From 93fc44aaa214b5eac33b263edba8db9237a2b6a0 Mon Sep 17 00:00:00 2001
From: Mikita Kandratsyeu <mikita_kandratsyeu1@epam.com>
Date: Fri, 18 Apr 2025 14:41:42 +0300
Subject: [PATCH 1/4] updated ai libs

---
 .../chat-main/chat-input-footer.tsx           |   6 +-
 .../_components/chat-main/chat.tsx            |   2 +-
 app/api/ai/completions/route.ts               |  13 +-
 components/ai/generate-text-response-ai.tsx   |   9 +-
 package.json                                  |   3 +-
 yarn.lock                                     | 181 ++----------------
 6 files changed, 27 insertions(+), 187 deletions(-)

diff --git a/app/(chat)/(routes)/chat/[[...slug]]/_components/chat-main/chat-input-footer.tsx b/app/(chat)/(routes)/chat/[[...slug]]/_components/chat-main/chat-input-footer.tsx
index e3eef37a..599b50e1 100644
--- a/app/(chat)/(routes)/chat/[[...slug]]/_components/chat-main/chat-input-footer.tsx
+++ b/app/(chat)/(routes)/chat/[[...slug]]/_components/chat-main/chat-input-footer.tsx
@@ -1,6 +1,6 @@
 'use client';
 
-import { ImageIcon, Paperclip, SendHorizonal, StopCircle } from 'lucide-react';
+import { ImageIcon, SendHorizonal, StopCircle } from 'lucide-react';
 import { useTranslations } from 'next-intl';
 
 import { Badge, Button, Separator } from '@/components/ui';
@@ -54,9 +54,9 @@ export const ChatInputFooter = ({
             )}
           />
         </button>
-        <button type="button" disabled={isSubmitting}>
+        {/* <button type="button" disabled={isSubmitting}>
           <Paperclip className="w-4 h-4 text-muted-foreground" />
-        </button>
+        </button> */}
         <Separator orientation="vertical" className="mr-4 ml-2 h-6" />
         <Button
           className={cn(
diff --git a/app/(chat)/(routes)/chat/[[...slug]]/_components/chat-main/chat.tsx b/app/(chat)/(routes)/chat/[[...slug]]/_components/chat-main/chat.tsx
index 4d37f246..a354a676 100644
--- a/app/(chat)/(routes)/chat/[[...slug]]/_components/chat-main/chat.tsx
+++ b/app/(chat)/(routes)/chat/[[...slug]]/_components/chat-main/chat.tsx
@@ -138,7 +138,7 @@ export const Chat = ({ conversations = [], initialData, isEmbed, isShared }: Cha
 
         const completionStream = await fetcher.post('/api/ai/completions', {
           body: {
-            messages: [...messages, ...(options?.regenerate ? [] : messagesForApi)].map(
+            input: [...messages, ...(options?.regenerate ? [] : messagesForApi)].map(
               ({ content, role }) => ({
                 content,
                 role,
diff --git a/app/api/ai/completions/route.ts b/app/api/ai/completions/route.ts
index 87879b3d..c69e5d1c 100644
--- a/app/api/ai/completions/route.ts
+++ b/app/api/ai/completions/route.ts
@@ -1,4 +1,3 @@
-import { OpenAIStream, StreamingTextResponse } from 'ai';
 import { ReasonPhrases, StatusCodes } from 'http-status-codes';
 import { NextRequest, NextResponse } from 'next/server';
 
@@ -16,7 +15,7 @@ export const POST = async (req: NextRequest) => {
   const provider = AIProvider(config?.ai?.provider as string);
 
   try {
-    const { messages, model, system } = await req.json();
+    const { input, model, instructions } = await req.json();
 
     if (!user) {
       return new NextResponse(ReasonPhrases.UNAUTHORIZED, { status: StatusCodes.UNAUTHORIZED });
@@ -35,16 +34,14 @@ export const POST = async (req: NextRequest) => {
       });
     }
 
-    const completion = await provider.chat.completions.create({
-      messages: [...(system ? [system] : []), ...messages],
+    const completion = await provider.responses.create({
+      input,
+      instructions,
       model,
-      top_p: 0.5,
       stream: true,
     });
 
-    const stream = OpenAIStream(completion);
-
-    return new StreamingTextResponse(stream);
+    return completion;
   } catch (error) {
     console.error('[OPEN_AI_COMPLETIONS]', error);
 
diff --git a/components/ai/generate-text-response-ai.tsx b/components/ai/generate-text-response-ai.tsx
index 8025b5e0..02e3b462 100644
--- a/components/ai/generate-text-response-ai.tsx
+++ b/components/ai/generate-text-response-ai.tsx
@@ -6,7 +6,7 @@ import { Dispatch, SetStateAction, useRef, useState } from 'react';
 import { BsStars } from 'react-icons/bs';
 
 import { useToast } from '@/components/ui/use-toast';
-import { ChatCompletionRole, SYSTEM_COURSE_PROMPT, SYSTEM_TRANSLATE_PROMPT } from '@/constants/ai';
+import { SYSTEM_COURSE_PROMPT, SYSTEM_TRANSLATE_PROMPT } from '@/constants/ai';
 import { TEN_MINUTE_SEC } from '@/constants/common';
 import { useAppConfigStore } from '@/hooks/store/use-app-config-store';
 import { getValueFromMemoryCache, setValueToMemoryCache } from '@/lib/cache';
@@ -65,11 +65,8 @@ export const GenerateTextResponseAi = ({
 
       const completionStream = await fetcher.post('/api/ai/completions', {
         body: {
-          messages,
-          system: {
-            role: ChatCompletionRole.SYSTEM,
-            content: isTranslateButton ? SYSTEM_TRANSLATE_PROMPT : SYSTEM_COURSE_PROMPT,
-          },
+          input: messages,
+          instructions: isTranslateButton ? SYSTEM_TRANSLATE_PROMPT : SYSTEM_COURSE_PROMPT,
           model: DEFAULT_MODEL,
         },
         cache: 'no-cache',
diff --git a/package.json b/package.json
index 58d252e9..417aa004 100644
--- a/package.json
+++ b/package.json
@@ -50,7 +50,6 @@
     "@vercel/kv": "^1.0.1",
     "@vercel/speed-insights": "^1.0.10",
     "@yandex/smart-captcha": "^2.6.0",
-    "ai": "^2.2.34",
     "bcrypt": "^5.1.1",
     "class-variance-authority": "^0.7.0",
     "clsx": "^2.1.0",
@@ -69,7 +68,7 @@
     "next-intl": "^3.17.6",
     "next-themes": "^0.2.1",
     "node-cache": "^5.1.2",
-    "openai": "^4.27.0",
+    "openai": "^4.95.0",
     "otplib": "^12.0.1",
     "plaiceholder": "^3.0.0",
     "pusher": "^5.2.0",
diff --git a/yarn.lock b/yarn.lock
index d0bed166..85c60e25 100644
--- a/yarn.lock
+++ b/yarn.lock
@@ -3195,35 +3195,6 @@ __metadata:
   languageName: node
   linkType: hard
 
-"ai@npm:^2.2.34":
-  version: 2.2.35
-  resolution: "ai@npm:2.2.35"
-  dependencies:
-    eventsource-parser: "npm:1.0.0"
-    nanoid: "npm:3.3.6"
-    solid-swr-store: "npm:0.10.7"
-    sswr: "npm:2.0.0"
-    swr: "npm:2.2.0"
-    swr-store: "npm:0.10.6"
-    swrv: "npm:1.0.4"
-  peerDependencies:
-    react: ^18.2.0
-    solid-js: ^1.7.7
-    svelte: ^3.0.0 || ^4.0.0
-    vue: ^3.3.4
-  peerDependenciesMeta:
-    react:
-      optional: true
-    solid-js:
-      optional: true
-    svelte:
-      optional: true
-    vue:
-      optional: true
-  checksum: 10c0/f1fdaf48974abd0c91d36e2b7bc5c35191a83bb1ad349b6d2db6f5b0383a60aaf4ef13bdd1eb966ad04abf667546fc214a51139563d2e4c0834a47f17e981873
-  languageName: node
-  linkType: hard
-
 "ajv@npm:^6.12.4":
   version: 6.12.6
   resolution: "ajv@npm:6.12.6"
@@ -3596,13 +3567,6 @@ __metadata:
   languageName: node
   linkType: hard
 
-"base-64@npm:^0.1.0":
-  version: 0.1.0
-  resolution: "base-64@npm:0.1.0"
-  checksum: 10c0/fe0dcf076e823f04db7ee9b02495be08a91c445fbc6db03cb9913be9680e2fcc0af8b74459041fe08ad16800b1f65a549501d8f08696a8a6d32880789b7de69d
-  languageName: node
-  linkType: hard
-
 "base64-js@npm:^1.3.1":
   version: 1.5.1
   resolution: "base64-js@npm:1.5.1"
@@ -3844,13 +3808,6 @@ __metadata:
   languageName: node
   linkType: hard
 
-"charenc@npm:0.0.2":
-  version: 0.0.2
-  resolution: "charenc@npm:0.0.2"
-  checksum: 10c0/a45ec39363a16799d0f9365c8dd0c78e711415113c6f14787a22462ef451f5013efae8a28f1c058f81fc01f2a6a16955f7a5fd0cd56247ce94a45349c89877d8
-  languageName: node
-  linkType: hard
-
 "chokidar@npm:^3.5.3":
   version: 3.6.0
   resolution: "chokidar@npm:3.6.0"
@@ -4173,13 +4130,6 @@ __metadata:
   languageName: node
   linkType: hard
 
-"crypt@npm:0.0.2":
-  version: 0.0.2
-  resolution: "crypt@npm:0.0.2"
-  checksum: 10c0/adbf263441dd801665d5425f044647533f39f4612544071b1471962209d235042fb703c27eea2795c7c53e1dfc242405173003f83cf4f4761a633d11f9653f18
-  languageName: node
-  linkType: hard
-
 "crypto-js@npm:^4.2.0":
   version: 4.2.0
   resolution: "crypto-js@npm:4.2.0"
@@ -4495,16 +4445,6 @@ __metadata:
   languageName: node
   linkType: hard
 
-"digest-fetch@npm:^1.3.0":
-  version: 1.3.0
-  resolution: "digest-fetch@npm:1.3.0"
-  dependencies:
-    base-64: "npm:^0.1.0"
-    md5: "npm:^2.3.0"
-  checksum: 10c0/0fb389e33b9c6baf5e6a9ed287aa9d0d8b373d59b49d49c62c261e1ab24eaaf1d5aea3a105c1b31ba4a23e29e129365d839ce4c5974fa004a85d1a4568bc3585
-  languageName: node
-  linkType: hard
-
 "dijkstrajs@npm:^1.0.1":
   version: 1.0.3
   resolution: "dijkstrajs@npm:1.0.3"
@@ -5242,13 +5182,6 @@ __metadata:
   languageName: node
   linkType: hard
 
-"eventsource-parser@npm:1.0.0":
-  version: 1.0.0
-  resolution: "eventsource-parser@npm:1.0.0"
-  checksum: 10c0/eb6b66231bd2b383f7a5fa1c5ca17d61cd8145afc560de8cc9f11ab60a05a22afd760de594886241378b831314c3b875c64ae3f596e9ce90cded99cbeef0ef9a
-  languageName: node
-  linkType: hard
-
 "execa@npm:8.0.1":
   version: 8.0.1
   resolution: "execa@npm:8.0.1"
@@ -6281,13 +6214,6 @@ __metadata:
   languageName: node
   linkType: hard
 
-"is-buffer@npm:~1.1.6":
-  version: 1.1.6
-  resolution: "is-buffer@npm:1.1.6"
-  checksum: 10c0/ae18aa0b6e113d6c490ad1db5e8df9bdb57758382b313f5a22c9c61084875c6396d50bbf49315f5b1926d142d74dfb8d31b40d993a383e0a158b15fea7a82234
-  languageName: node
-  linkType: hard
-
 "is-callable@npm:^1.1.3, is-callable@npm:^1.1.4, is-callable@npm:^1.2.7":
   version: 1.2.7
   resolution: "is-callable@npm:1.2.7"
@@ -6823,7 +6749,6 @@ __metadata:
     "@vercel/kv": "npm:^1.0.1"
     "@vercel/speed-insights": "npm:^1.0.10"
     "@yandex/smart-captcha": "npm:^2.6.0"
-    ai: "npm:^2.2.34"
     autoprefixer: "npm:^10.0.1"
     bcrypt: "npm:^5.1.1"
     class-variance-authority: "npm:^0.7.0"
@@ -6850,7 +6775,7 @@ __metadata:
     next-intl: "npm:^3.17.6"
     next-themes: "npm:^0.2.1"
     node-cache: "npm:^5.1.2"
-    openai: "npm:^4.27.0"
+    openai: "npm:^4.95.0"
     otplib: "npm:^12.0.1"
     plaiceholder: "npm:^3.0.0"
     postcss: "npm:^8"
@@ -7079,17 +7004,6 @@ __metadata:
   languageName: node
   linkType: hard
 
-"md5@npm:^2.3.0":
-  version: 2.3.0
-  resolution: "md5@npm:2.3.0"
-  dependencies:
-    charenc: "npm:0.0.2"
-    crypt: "npm:0.0.2"
-    is-buffer: "npm:~1.1.6"
-  checksum: 10c0/14a21d597d92e5b738255fbe7fe379905b8cb97e0a49d44a20b58526a646ec5518c337b817ce0094ca94d3e81a3313879c4c7b510d250c282d53afbbdede9110
-  languageName: node
-  linkType: hard
-
 "mdast-util-find-and-replace@npm:^3.0.0":
   version: 3.0.1
   resolution: "mdast-util-find-and-replace@npm:3.0.1"
@@ -7868,15 +7782,6 @@ __metadata:
   languageName: node
   linkType: hard
 
-"nanoid@npm:3.3.6":
-  version: 3.3.6
-  resolution: "nanoid@npm:3.3.6"
-  bin:
-    nanoid: bin/nanoid.cjs
-  checksum: 10c0/606b355960d0fcbe3d27924c4c52ef7d47d3b57208808ece73279420d91469b01ec1dce10fae512b6d4a8c5a5432b352b228336a8b2202a6ea68e67fa348e2ee
-  languageName: node
-  linkType: hard
-
 "nanoid@npm:^3.3.6, nanoid@npm:^3.3.7":
   version: 3.3.7
   resolution: "nanoid@npm:3.3.7"
@@ -8310,22 +8215,28 @@ __metadata:
   languageName: node
   linkType: hard
 
-"openai@npm:^4.27.0":
-  version: 4.28.0
-  resolution: "openai@npm:4.28.0"
+"openai@npm:^4.95.0":
+  version: 4.95.0
+  resolution: "openai@npm:4.95.0"
   dependencies:
     "@types/node": "npm:^18.11.18"
     "@types/node-fetch": "npm:^2.6.4"
     abort-controller: "npm:^3.0.0"
     agentkeepalive: "npm:^4.2.1"
-    digest-fetch: "npm:^1.3.0"
     form-data-encoder: "npm:1.7.2"
     formdata-node: "npm:^4.3.2"
     node-fetch: "npm:^2.6.7"
-    web-streams-polyfill: "npm:^3.2.1"
+  peerDependencies:
+    ws: ^8.18.0
+    zod: ^3.23.8
+  peerDependenciesMeta:
+    ws:
+      optional: true
+    zod:
+      optional: true
   bin:
     openai: bin/cli
-  checksum: 10c0/b40cd3cf553ea947dbc916f6416c10528116eb523cd5453e3c442f11d9514de228e54b90b5c9cca1af34cc296baa0bd29ccbc8078bf9bf0b74bb18860468509a
+  checksum: 10c0/cb5d560539834da5da49abc54c09fcd6cd09d8d8d24ebdaf86549859098dbbdbfd62d43d0477aee4f58f7bb36006e8598b2f2281ecda443ae9d6c1a3b8120253
   languageName: node
   linkType: hard
 
@@ -9973,16 +9884,6 @@ __metadata:
   languageName: node
   linkType: hard
 
-"solid-swr-store@npm:0.10.7":
-  version: 0.10.7
-  resolution: "solid-swr-store@npm:0.10.7"
-  peerDependencies:
-    solid-js: ^1.2
-    swr-store: ^0.10
-  checksum: 10c0/da45f4eeb85323f8686b38dad09dbaf6aac2552191a7314be74bdc088080a11386d25ff7ba63754a880456b2647035f72640450ae07ec237bba4de2d884c2eb2
-  languageName: node
-  linkType: hard
-
 "source-map-js@npm:^1.0.2":
   version: 1.0.2
   resolution: "source-map-js@npm:1.0.2"
@@ -10041,17 +9942,6 @@ __metadata:
   languageName: node
   linkType: hard
 
-"sswr@npm:2.0.0":
-  version: 2.0.0
-  resolution: "sswr@npm:2.0.0"
-  dependencies:
-    swrev: "npm:^4.0.0"
-  peerDependencies:
-    svelte: ^4.0.0
-  checksum: 10c0/3eea2df761c58a11959bccceed5333100cfff6b1d4449c703a59e90c9f2662af00c476311ef1300bca24cca3dc6535b27caa9fa1a12e691318915fad01ed6cd5
-  languageName: node
-  linkType: hard
-
 "streamsearch@npm:^1.1.0":
   version: 1.1.0
   resolution: "streamsearch@npm:1.1.0"
@@ -10313,42 +10203,6 @@ __metadata:
   languageName: node
   linkType: hard
 
-"swr-store@npm:0.10.6":
-  version: 0.10.6
-  resolution: "swr-store@npm:0.10.6"
-  dependencies:
-    dequal: "npm:^2.0.3"
-  checksum: 10c0/4d26084ec1aa83ef2e1c27be39e75a7f1e8e5b222dd5e0003c31495ed1827443647460b9de21879bd125f686fedc28928d7a067f03d9dde9a9298289cc217540
-  languageName: node
-  linkType: hard
-
-"swr@npm:2.2.0":
-  version: 2.2.0
-  resolution: "swr@npm:2.2.0"
-  dependencies:
-    use-sync-external-store: "npm:^1.2.0"
-  peerDependencies:
-    react: ^16.11.0 || ^17.0.0 || ^18.0.0
-  checksum: 10c0/f631dfd206a989313cd82e7d9477b11af08496000c339752c745e8b42ba58667528ee0c70ea529910a7b6dee9b4085e8a2cd622efb70b9709883211111e0d756
-  languageName: node
-  linkType: hard
-
-"swrev@npm:^4.0.0":
-  version: 4.0.0
-  resolution: "swrev@npm:4.0.0"
-  checksum: 10c0/c2d2328f3f29d85af3b8fd9d449195f7be737d2efa465aeeaf1bf9b64473b610f0db82114261f9a11380bbe30c4a09fb702da24753144789de4073ede1d10824
-  languageName: node
-  linkType: hard
-
-"swrv@npm:1.0.4":
-  version: 1.0.4
-  resolution: "swrv@npm:1.0.4"
-  peerDependencies:
-    vue: ">=3.2.26 < 4"
-  checksum: 10c0/2bdf8bf3e461c21f9030ad47f1996e2429ca99cb31d0bb91ae5b04ba9b48e8fce93c356491cc69a3a586575a6ad0f5b29812b2a3a3e7ee28c1f2086c4f72f8b9
-  languageName: node
-  linkType: hard
-
 "synckit@npm:^0.8.6":
   version: 0.8.8
   resolution: "synckit@npm:0.8.8"
@@ -10922,7 +10776,7 @@ __metadata:
   languageName: node
   linkType: hard
 
-"use-sync-external-store@npm:1.2.0, use-sync-external-store@npm:^1.0.0, use-sync-external-store@npm:^1.2.0":
+"use-sync-external-store@npm:1.2.0, use-sync-external-store@npm:^1.0.0":
   version: 1.2.0
   resolution: "use-sync-external-store@npm:1.2.0"
   peerDependencies:
@@ -11018,13 +10872,6 @@ __metadata:
   languageName: node
   linkType: hard
 
-"web-streams-polyfill@npm:^3.2.1":
-  version: 3.3.2
-  resolution: "web-streams-polyfill@npm:3.3.2"
-  checksum: 10c0/623c2fced2ef77d5afdbc43acef64b8af609a32125b691eae286d534a36004c8a71030f0e78068516774a97fd90dbfb3726b10fd569a2d158e60c83a539c489e
-  languageName: node
-  linkType: hard
-
 "webidl-conversions@npm:^3.0.0":
   version: 3.0.1
   resolution: "webidl-conversions@npm:3.0.1"

From 0ddbb5587c5c84155c66882018d0f40209c86df6 Mon Sep 17 00:00:00 2001
From: Mikita Kandratsyeu <mikita_kandratsyeu1@epam.com>
Date: Fri, 18 Apr 2025 16:09:44 +0300
Subject: [PATCH 2/4] updated deps

---
 actions/chat/get-chat-initial.ts              | 12 +--
 .../_components/chat-main/chat.tsx            | 12 ++-
 app/api/ai/completions/route.ts               | 44 +++++++++-
 components/ai/generate-text-response-ai.tsx   | 12 ++-
 package.json                                  |  4 +-
 yarn.lock                                     | 88 +++++++++----------
 6 files changed, 111 insertions(+), 61 deletions(-)

diff --git a/actions/chat/get-chat-initial.ts b/actions/chat/get-chat-initial.ts
index ad4f67df..be4a1094 100644
--- a/actions/chat/get-chat-initial.ts
+++ b/actions/chat/get-chat-initial.ts
@@ -22,19 +22,15 @@ export const getChatInitial = async () => {
     const introMessages = await fetchCachedData(
       `chat-initial-[${locale}]`,
       async () => {
-        const response = await provider.chat.completions.create({
-          messages: [
-            {
-              role: 'system',
-              content: 'You are a machine that only returns array format.',
-            },
+        const response = await provider.responses.create({
+          instructions: 'You are a machine that only returns array format.',
+          input: [
             {
               content: `Generate 4 questions ranging from 120 to 150 characters long for an intelligent chat on the topic of programming. Language code is ${locale}. Write the result to an array.`,
               role: ChatCompletionRole.USER as unknown as ChatCompletionUserMessageParam['role'],
             },
           ],
           model: DEFAULT_MODEL,
-          temperature: 0.8,
         });
 
         return response;
@@ -43,7 +39,7 @@ export const getChatInitial = async () => {
     );
 
     return {
-      introMessages: JSON.parse(introMessages.choices[0].message.content || '[]'),
+      introMessages: JSON.parse(introMessages.output_text || '[]'),
     };
   } catch (error) {
     console.error('[GET_CHAT_INITIAL_ACTION]', error);
diff --git a/app/(chat)/(routes)/chat/[[...slug]]/_components/chat-main/chat.tsx b/app/(chat)/(routes)/chat/[[...slug]]/_components/chat-main/chat.tsx
index a354a676..77d8f5a8 100644
--- a/app/(chat)/(routes)/chat/[[...slug]]/_components/chat-main/chat.tsx
+++ b/app/(chat)/(routes)/chat/[[...slug]]/_components/chat-main/chat.tsx
@@ -145,6 +145,7 @@ export const Chat = ({ conversations = [], initialData, isEmbed, isShared }: Cha
               }),
             ),
             model: currentModel,
+            stream: true,
           },
           cache: 'no-cache',
           headers: {
@@ -170,9 +171,16 @@ export const Chat = ({ conversations = [], initialData, isEmbed, isShared }: Cha
           }
 
           const chunk = decoder.decode(value);
-          streamAssistMessage += chunk;
+          const lines = chunk.split('\n').filter((line) => line.trim());
 
-          setAssistantMessage((prev) => prev + chunk);
+          for (const line of lines) {
+            if (line.startsWith('data: ')) {
+              const data = JSON.parse(line.slice(6));
+
+              streamAssistMessage += data.delta;
+              setAssistantMessage((prev) => prev + data.delta);
+            }
+          }
         }
       }
     } catch (error: any) {
diff --git a/app/api/ai/completions/route.ts b/app/api/ai/completions/route.ts
index c69e5d1c..457b17d5 100644
--- a/app/api/ai/completions/route.ts
+++ b/app/api/ai/completions/route.ts
@@ -15,7 +15,7 @@ export const POST = async (req: NextRequest) => {
   const provider = AIProvider(config?.ai?.provider as string);
 
   try {
-    const { input, model, instructions } = await req.json();
+    const { input, instructions, model, stream } = await req.json();
 
     if (!user) {
       return new NextResponse(ReasonPhrases.UNAUTHORIZED, { status: StatusCodes.UNAUTHORIZED });
@@ -38,10 +38,48 @@ export const POST = async (req: NextRequest) => {
       input,
       instructions,
       model,
-      stream: true,
+      stream,
     });
 
-    return completion;
+    if (stream) {
+      const encoder = new TextEncoder();
+      const stream_response = new TransformStream();
+
+      (async () => {
+        const writer = stream_response.writable.getWriter();
+
+        try {
+          for await (const event of completion as any) {
+            if (event.type === 'response.output_text.delta') {
+              const data = {
+                item_id: event.item_id,
+                output_index: event.output_index,
+                content_index: event.content_index,
+                delta: event.delta,
+              };
+              await writer.write(encoder.encode(`data: ${JSON.stringify(data)}\n\n`));
+            }
+          }
+        } catch (error) {
+          console.error('Stream processing error:', error);
+          await writer.write(
+            encoder.encode(`data: ${JSON.stringify({ error: 'Stream processing error' })}\n\n`),
+          );
+        } finally {
+          await writer.close();
+        }
+      })();
+
+      return new NextResponse(stream_response.readable, {
+        headers: {
+          'Content-Type': 'text/event-stream',
+          'Cache-Control': 'no-cache',
+          Connection: 'keep-alive',
+        },
+      });
+    }
+
+    return NextResponse.json({ completion });
   } catch (error) {
     console.error('[OPEN_AI_COMPLETIONS]', error);
 
diff --git a/components/ai/generate-text-response-ai.tsx b/components/ai/generate-text-response-ai.tsx
index 02e3b462..e37c8cc2 100644
--- a/components/ai/generate-text-response-ai.tsx
+++ b/components/ai/generate-text-response-ai.tsx
@@ -68,6 +68,7 @@ export const GenerateTextResponseAi = ({
           input: messages,
           instructions: isTranslateButton ? SYSTEM_TRANSLATE_PROMPT : SYSTEM_COURSE_PROMPT,
           model: DEFAULT_MODEL,
+          stream: true,
         },
         cache: 'no-cache',
         headers: {
@@ -96,9 +97,16 @@ export const GenerateTextResponseAi = ({
         }
 
         const chunk = decoder.decode(value);
+        const lines = chunk.split('\n').filter((line) => line.trim());
 
-        text += chunk;
-        callback((prev) => prev + chunk);
+        for (const line of lines) {
+          if (line.startsWith('data: ')) {
+            const data = JSON.parse(line.slice(6));
+
+            text += data.delta;
+            callback((prev) => prev + data.delta);
+          }
+        }
       }
 
       if (shouldCacheResponse) {
diff --git a/package.json b/package.json
index 417aa004..ec0d5bd6 100644
--- a/package.json
+++ b/package.json
@@ -26,7 +26,7 @@
     "@hello-pangea/dnd": "^16.5.0",
     "@hookform/resolvers": "^3.3.4",
     "@plaiceholder/next": "^3.0.0",
-    "@prisma/client": "^6.5.0",
+    "@prisma/client": "^6.6.0",
     "@radix-ui/react-accordion": "^1.1.2",
     "@radix-ui/react-alert-dialog": "^1.0.5",
     "@radix-ui/react-avatar": "^1.0.4",
@@ -130,7 +130,7 @@
     "lint-staged": "^15.2.2",
     "postcss": "^8",
     "prettier": "^3.2.4",
-    "prisma": "^6.5.0",
+    "prisma": "^6.6.0",
     "tailwindcss": "^3.3.0",
     "typescript": "^5"
   },
diff --git a/yarn.lock b/yarn.lock
index 85c60e25..49358697 100644
--- a/yarn.lock
+++ b/yarn.lock
@@ -849,9 +849,9 @@ __metadata:
   languageName: node
   linkType: hard
 
-"@prisma/client@npm:^6.5.0":
-  version: 6.5.0
-  resolution: "@prisma/client@npm:6.5.0"
+"@prisma/client@npm:^6.6.0":
+  version: 6.6.0
+  resolution: "@prisma/client@npm:6.6.0"
   peerDependencies:
     prisma: "*"
     typescript: ">=5.1.0"
@@ -860,63 +860,63 @@ __metadata:
       optional: true
     typescript:
       optional: true
-  checksum: 10c0/2bf6a213bd65de20092b2773b9bb105f023558da2214e92542fe745255778dee8386068ec6648d0d651e0af7443ca69ed9278aa505c0342d39a30098fd0090d2
+  checksum: 10c0/174e4f445ae671dc16af272743b33312f36be34d8d5c0ab92eba8c81d74dc86967fa316bd4db082f1aa0970abf31899a8305e09a6717cb285bdc700aa9273f70
   languageName: node
   linkType: hard
 
-"@prisma/config@npm:6.5.0":
-  version: 6.5.0
-  resolution: "@prisma/config@npm:6.5.0"
+"@prisma/config@npm:6.6.0":
+  version: 6.6.0
+  resolution: "@prisma/config@npm:6.6.0"
   dependencies:
     esbuild: "npm:>=0.12 <1"
     esbuild-register: "npm:3.6.0"
-  checksum: 10c0/6096e946aa8787cd216a9661ab12641776cfc21c2142f25b2fa3a199042a761de3fbfeca0bf0e8acd77f4913e8ecb1cea37a04ecf51a4a110f06c17963665b7c
+  checksum: 10c0/a301daed825783c76a7e7fc01f26c90f2e9f8b3a7d0a00a83422e1dcb3b3aa2a30fa4548a23ec0d2ca418c103d48dbc87578a8717d18586ff76992096eaf2edd
   languageName: node
   linkType: hard
 
-"@prisma/debug@npm:6.5.0":
-  version: 6.5.0
-  resolution: "@prisma/debug@npm:6.5.0"
-  checksum: 10c0/e33ba5750642c3ecf73844df7b8d7a1b35bd45601e6449bd6fdd56df7bf5a1c09f3b034594b2cd150aae86e822b6de5f29fae5681a0c94c38b1e393ce439e923
+"@prisma/debug@npm:6.6.0":
+  version: 6.6.0
+  resolution: "@prisma/debug@npm:6.6.0"
+  checksum: 10c0/49bed1c7f562b2a565637c5ba435ce81c337256791287c4c8923aa8ddb5724e625213614dc9b0ae5564a28018051e91f7426bca2182018d903c6579dec702a40
   languageName: node
   linkType: hard
 
-"@prisma/engines-version@npm:6.5.0-73.173f8d54f8d52e692c7e27e72a88314ec7aeff60":
-  version: 6.5.0-73.173f8d54f8d52e692c7e27e72a88314ec7aeff60
-  resolution: "@prisma/engines-version@npm:6.5.0-73.173f8d54f8d52e692c7e27e72a88314ec7aeff60"
-  checksum: 10c0/92792c7f4ece28bfb3f6be14c8d3c625bc201a802ab9c71cf5b04e7e1fef827c75e73f63258c3ae5e63658eec97306d172b8104e5ee4e73583e7cb586479d724
+"@prisma/engines-version@npm:6.6.0-53.f676762280b54cd07c770017ed3711ddde35f37a":
+  version: 6.6.0-53.f676762280b54cd07c770017ed3711ddde35f37a
+  resolution: "@prisma/engines-version@npm:6.6.0-53.f676762280b54cd07c770017ed3711ddde35f37a"
+  checksum: 10c0/7f1ed0143cdde34d768846a24f6bacac8e4826dc0ed0fd9e47066d81e7876d92c51dda6491144d734fa97b573ed9a500238b544c78516443bfc353c362c42e15
   languageName: node
   linkType: hard
 
-"@prisma/engines@npm:6.5.0":
-  version: 6.5.0
-  resolution: "@prisma/engines@npm:6.5.0"
+"@prisma/engines@npm:6.6.0":
+  version: 6.6.0
+  resolution: "@prisma/engines@npm:6.6.0"
   dependencies:
-    "@prisma/debug": "npm:6.5.0"
-    "@prisma/engines-version": "npm:6.5.0-73.173f8d54f8d52e692c7e27e72a88314ec7aeff60"
-    "@prisma/fetch-engine": "npm:6.5.0"
-    "@prisma/get-platform": "npm:6.5.0"
-  checksum: 10c0/61cc51e28f403a9540183d7276e3f4e191daa854005642f80747f122f4d99e322d036b112da5ab0be77938198b5298138243c72d9d56e6841066b808a6c73f93
+    "@prisma/debug": "npm:6.6.0"
+    "@prisma/engines-version": "npm:6.6.0-53.f676762280b54cd07c770017ed3711ddde35f37a"
+    "@prisma/fetch-engine": "npm:6.6.0"
+    "@prisma/get-platform": "npm:6.6.0"
+  checksum: 10c0/e470d5e5c18694ee9f0a67dfa15d2eb0dbe98fe05d3aee9ad8777ecb5e3b8b48e53eacd1ae3b8b02764327d992d8d6f13d26ae895a391d09032a21b084905be6
   languageName: node
   linkType: hard
 
-"@prisma/fetch-engine@npm:6.5.0":
-  version: 6.5.0
-  resolution: "@prisma/fetch-engine@npm:6.5.0"
+"@prisma/fetch-engine@npm:6.6.0":
+  version: 6.6.0
+  resolution: "@prisma/fetch-engine@npm:6.6.0"
   dependencies:
-    "@prisma/debug": "npm:6.5.0"
-    "@prisma/engines-version": "npm:6.5.0-73.173f8d54f8d52e692c7e27e72a88314ec7aeff60"
-    "@prisma/get-platform": "npm:6.5.0"
-  checksum: 10c0/4f26fb07fd183f9a975674b29f976d94fe0301a4fd0571b85b3f6995127ce70d0a65e84a821daef95f4053ae3a02b0cc9535e1e113d363f70d3f606eca587df5
+    "@prisma/debug": "npm:6.6.0"
+    "@prisma/engines-version": "npm:6.6.0-53.f676762280b54cd07c770017ed3711ddde35f37a"
+    "@prisma/get-platform": "npm:6.6.0"
+  checksum: 10c0/08dca7f1bf54b57fee350d4461a45c0dd2a6ca364e9c9839fff9c41fadacc1a2d98273ee994fe86b67508672a3d739b8db33d5963a9db38216b2ca7730654795
   languageName: node
   linkType: hard
 
-"@prisma/get-platform@npm:6.5.0":
-  version: 6.5.0
-  resolution: "@prisma/get-platform@npm:6.5.0"
+"@prisma/get-platform@npm:6.6.0":
+  version: 6.6.0
+  resolution: "@prisma/get-platform@npm:6.6.0"
   dependencies:
-    "@prisma/debug": "npm:6.5.0"
-  checksum: 10c0/45033d8c20a19d6e0329ff15901606a53be9f43c843ddfd99848c8ac388b12e1b65c202e95a1d5cdc7da4bfee77ed3795ba3c7e0b1d1fa6380c9a253e1bedfd2
+    "@prisma/debug": "npm:6.6.0"
+  checksum: 10c0/28faef47f95256f180a9a417d36339a85062a6fa1fc00bb633b4f3c3c6cf933fcb7e73e89cac0f17807ad144e53381fd71dd0974ea58418ef3ccd08e610cc069
   languageName: node
   linkType: hard
 
@@ -6711,7 +6711,7 @@ __metadata:
     "@hello-pangea/dnd": "npm:^16.5.0"
     "@hookform/resolvers": "npm:^3.3.4"
     "@plaiceholder/next": "npm:^3.0.0"
-    "@prisma/client": "npm:^6.5.0"
+    "@prisma/client": "npm:^6.6.0"
     "@radix-ui/react-accordion": "npm:^1.1.2"
     "@radix-ui/react-alert-dialog": "npm:^1.0.5"
     "@radix-ui/react-avatar": "npm:^1.0.4"
@@ -6780,7 +6780,7 @@ __metadata:
     plaiceholder: "npm:^3.0.0"
     postcss: "npm:^8"
     prettier: "npm:^3.2.4"
-    prisma: "npm:^6.5.0"
+    prisma: "npm:^6.6.0"
     pusher: "npm:^5.2.0"
     pusher-js: "npm:^8.4.0-rc2"
     qrcode: "npm:^1.5.3"
@@ -8676,12 +8676,12 @@ __metadata:
   languageName: node
   linkType: hard
 
-"prisma@npm:^6.5.0":
-  version: 6.5.0
-  resolution: "prisma@npm:6.5.0"
+"prisma@npm:^6.6.0":
+  version: 6.6.0
+  resolution: "prisma@npm:6.6.0"
   dependencies:
-    "@prisma/config": "npm:6.5.0"
-    "@prisma/engines": "npm:6.5.0"
+    "@prisma/config": "npm:6.6.0"
+    "@prisma/engines": "npm:6.6.0"
     fsevents: "npm:2.3.3"
   peerDependencies:
     typescript: ">=5.1.0"
@@ -8693,7 +8693,7 @@ __metadata:
       optional: true
   bin:
     prisma: build/index.js
-  checksum: 10c0/d859336d9b121987723e775d48a69e78b7eed7d6c56ccc040d2d53f3e6d999dc04b23021320bc4ee2c6db968929c2a863ae747c870f359b1faf831ced6f8132c
+  checksum: 10c0/ab8932e58864f8a46916491530003220202e7489667b3f72934b8fb6dd0ab3bb955505bb99750987a6ec672d2c791f8bdeb94b30f3ce91d964e28215fc4280da
   languageName: node
   linkType: hard
 

From ff2d29de8591d4abe336a3a2bcc1d73ced27275c Mon Sep 17 00:00:00 2001
From: Mikita Kandratsyeu <mikita_kandratsyeu1@epam.com>
Date: Fri, 18 Apr 2025 16:52:20 +0300
Subject: [PATCH 3/4] code refactor

---
 actions/ai/generate-completion.ts | 76 +++++++++++++++++++++++++++++++
 actions/ai/generate-image.ts      | 42 +++++++++++++++++
 actions/auth/get-login-quote.ts   | 28 ++++--------
 actions/chat/get-chat-initial.ts  | 13 ++----
 app/api/ai/completions/route.ts   | 58 ++++-------------------
 app/api/ai/image/route.ts         | 31 +++++--------
 6 files changed, 152 insertions(+), 96 deletions(-)
 create mode 100644 actions/ai/generate-completion.ts
 create mode 100644 actions/ai/generate-image.ts

diff --git a/actions/ai/generate-completion.ts b/actions/ai/generate-completion.ts
new file mode 100644
index 00000000..ddbeeec9
--- /dev/null
+++ b/actions/ai/generate-completion.ts
@@ -0,0 +1,76 @@
+'use server';
+
+import { ResponseCreateParamsBase } from 'openai/resources/responses/responses.mjs';
+
+import { isOwner } from '@/lib/owner';
+import { AIProvider } from '@/server/ai-provider';
+
+import { getCurrentUser } from '../auth/get-current-user';
+import { getAppConfig } from '../configs/get-app-config';
+
+type GenerateCompletion = Omit<ResponseCreateParamsBase, 'model'> & {
+  model?: string;
+};
+
+export const generateCompletion = async ({
+  input,
+  instructions,
+  model,
+  stream = false,
+}: GenerateCompletion) => {
+  const user = await getCurrentUser();
+  const config = await getAppConfig();
+
+  const aiModel = model ?? config?.ai?.['text-models']?.[0].value ?? '';
+  const provider = AIProvider(config?.ai?.provider as string);
+
+  const TEXT_MODELS = config?.ai?.['text-models'] ?? [];
+  const models = (isOwner(user?.userId) ? TEXT_MODELS : TEXT_MODELS.slice(0, 2)).map(
+    ({ value }) => value,
+  );
+
+  if (!models.includes(aiModel)) {
+    return { completion: null, model: aiModel };
+  }
+
+  const completion = await provider.responses.create({
+    input,
+    instructions,
+    model: aiModel,
+    stream,
+  });
+
+  if (stream) {
+    const encoder = new TextEncoder();
+    const stream_response = new TransformStream();
+
+    (async () => {
+      const writer = stream_response.writable.getWriter();
+
+      try {
+        for await (const event of completion as any) {
+          if (event.type === 'response.output_text.delta') {
+            const data = {
+              item_id: event.item_id,
+              output_index: event.output_index,
+              content_index: event.content_index,
+              delta: event.delta,
+            };
+            await writer.write(encoder.encode(`data: ${JSON.stringify(data)}\n\n`));
+          }
+        }
+      } catch (error) {
+        console.error('Stream processing error:', error);
+        await writer.write(
+          encoder.encode(`data: ${JSON.stringify({ error: 'Stream processing error' })}\n\n`),
+        );
+      } finally {
+        await writer.close();
+      }
+    })();
+
+    return { completion: stream_response.readable, model: aiModel };
+  }
+
+  return { completion, model: aiModel };
+};
diff --git a/actions/ai/generate-image.ts b/actions/ai/generate-image.ts
new file mode 100644
index 00000000..93d59202
--- /dev/null
+++ b/actions/ai/generate-image.ts
@@ -0,0 +1,42 @@
+'use server';
+
+import { ImageGenerateParams } from 'openai/resources/images.mjs';
+
+import { AIProvider } from '@/server/ai-provider';
+
+import { getAppConfig } from '../configs/get-app-config';
+
+type GenerateImage = Omit<ImageGenerateParams, 'model'> & {
+  model?: string;
+};
+
+export const generateImage = async ({ model, prompt }: GenerateImage) => {
+  const config = await getAppConfig();
+
+  const aiModel = model ?? config?.ai?.['image-models']?.[0].value ?? '';
+  const provider = AIProvider(config?.ai?.provider as string);
+
+  const IMAGE_MODELS = config?.ai?.['image-models'] ?? [];
+  const models = IMAGE_MODELS.map(({ value }) => value);
+
+  if (!models.includes(aiModel)) {
+    return {
+      image: null,
+      model: aiModel,
+    };
+  }
+
+  const response = await provider.images.generate({
+    model: aiModel,
+    n: 1,
+    prompt,
+    quality: 'hd',
+    response_format: 'b64_json',
+    size: '1024x1024',
+  });
+
+  return {
+    image: response,
+    model: null,
+  };
+};
diff --git a/actions/auth/get-login-quote.ts b/actions/auth/get-login-quote.ts
index f0fc7451..984d9d84 100644
--- a/actions/auth/get-login-quote.ts
+++ b/actions/auth/get-login-quote.ts
@@ -6,36 +6,25 @@ import { ChatCompletionUserMessageParam } from 'openai/resources/index.mjs';
 import { ChatCompletionRole } from '@/constants/ai';
 import { TEN_MINUTE_SEC } from '@/constants/common';
 import { fetchCachedData } from '@/lib/cache';
-import { AIProvider } from '@/server/ai-provider';
 
-import { getAppConfig } from '../configs/get-app-config';
+import { generateCompletion } from '../ai/generate-completion';
 
 export const getLoginQuote = async () => {
   const locale = await getLocale();
-  const config = await getAppConfig();
-
-  const provider = AIProvider(config?.ai?.provider as string);
-
-  const DEFAULT_MODEL = config?.ai?.['text-models']?.[0].value;
 
   try {
     const response = await fetchCachedData(
       `login-quote-[${locale}]`,
       async () => {
-        const response = await provider.chat.completions.create({
-          messages: [
-            {
-              role: 'system',
-              content:
-                'You are a machine that only returns JSON object format without unnecessary symbols.',
-            },
+        const response = await generateCompletion({
+          instructions:
+            'You are a machine that only returns JSON object format without unnecessary symbols.',
+          input: [
             {
               content: `Generate a quote from a famous philosopher. Language code is ${locale}. Write it down in JSON format - {"quote": "Quote", "author": "Quote the author"}`,
               role: ChatCompletionRole.USER as unknown as ChatCompletionUserMessageParam['role'],
             },
           ],
-          model: DEFAULT_MODEL,
-          temperature: 0.8,
         });
 
         return response;
@@ -43,11 +32,12 @@ export const getLoginQuote = async () => {
       TEN_MINUTE_SEC,
     );
 
-    const generatedQuote = JSON.parse(response.choices[0].message.content || '{}');
+    const generatedQuote = JSON.parse(response.completion.output_text || '{}');
+    const model = response.model ?? '';
 
     return {
       author: generatedQuote?.author ?? '',
-      model: DEFAULT_MODEL,
+      model,
       quote: generatedQuote?.quote ?? '',
     };
   } catch (error) {
@@ -55,7 +45,7 @@ export const getLoginQuote = async () => {
 
     return {
       author: '',
-      model: DEFAULT_MODEL,
+      model: '',
       quote: '',
     };
   }
diff --git a/actions/chat/get-chat-initial.ts b/actions/chat/get-chat-initial.ts
index be4a1094..79b43dd4 100644
--- a/actions/chat/get-chat-initial.ts
+++ b/actions/chat/get-chat-initial.ts
@@ -6,23 +6,17 @@ import { ChatCompletionUserMessageParam } from 'openai/resources/index.mjs';
 import { ChatCompletionRole } from '@/constants/ai';
 import { ONE_DAY_SEC } from '@/constants/common';
 import { fetchCachedData } from '@/lib/cache';
-import { AIProvider } from '@/server/ai-provider';
 
-import { getAppConfig } from '../configs/get-app-config';
+import { generateCompletion } from '../ai/generate-completion';
 
 export const getChatInitial = async () => {
   const locale = await getLocale();
-  const config = await getAppConfig();
-
-  const provider = AIProvider(config?.ai?.provider as string);
-
-  const DEFAULT_MODEL = config?.ai?.['text-models']?.[0].value;
 
   try {
     const introMessages = await fetchCachedData(
       `chat-initial-[${locale}]`,
       async () => {
-        const response = await provider.responses.create({
+        const response = await generateCompletion({
           instructions: 'You are a machine that only returns array format.',
           input: [
             {
@@ -30,10 +24,9 @@ export const getChatInitial = async () => {
               role: ChatCompletionRole.USER as unknown as ChatCompletionUserMessageParam['role'],
             },
           ],
-          model: DEFAULT_MODEL,
         });
 
-        return response;
+        return response.completion;
       },
       ONE_DAY_SEC,
     );
diff --git a/app/api/ai/completions/route.ts b/app/api/ai/completions/route.ts
index 457b17d5..c61bf315 100644
--- a/app/api/ai/completions/route.ts
+++ b/app/api/ai/completions/route.ts
@@ -1,18 +1,13 @@
 import { ReasonPhrases, StatusCodes } from 'http-status-codes';
 import { NextRequest, NextResponse } from 'next/server';
 
+import { generateCompletion } from '@/actions/ai/generate-completion';
 import { getCurrentUser } from '@/actions/auth/get-current-user';
-import { getAppConfig } from '@/actions/configs/get-app-config';
-import { isOwner } from '@/lib/owner';
-import { AIProvider } from '@/server/ai-provider';
 
 export const maxDuration = 60;
 
 export const POST = async (req: NextRequest) => {
   const user = await getCurrentUser();
-  const config = await getAppConfig();
-
-  const provider = AIProvider(config?.ai?.provider as string);
 
   try {
     const { input, instructions, model, stream } = await req.json();
@@ -21,12 +16,14 @@ export const POST = async (req: NextRequest) => {
       return new NextResponse(ReasonPhrases.UNAUTHORIZED, { status: StatusCodes.UNAUTHORIZED });
     }
 
-    const TEXT_MODELS = config?.ai?.['text-models'] ?? [];
-    const models = (isOwner(user?.userId) ? TEXT_MODELS : TEXT_MODELS.slice(0, 2)).map(
-      ({ value }) => value,
-    );
+    const response = await generateCompletion({
+      input,
+      instructions,
+      model,
+      stream,
+    });
 
-    if (!models.includes(model)) {
+    if (!response.completion) {
       console.error('[OPEN_AI_FORBIDDEN_MODEL]', user);
 
       return new NextResponse(ReasonPhrases.FORBIDDEN, {
@@ -34,43 +31,8 @@ export const POST = async (req: NextRequest) => {
       });
     }
 
-    const completion = await provider.responses.create({
-      input,
-      instructions,
-      model,
-      stream,
-    });
-
     if (stream) {
-      const encoder = new TextEncoder();
-      const stream_response = new TransformStream();
-
-      (async () => {
-        const writer = stream_response.writable.getWriter();
-
-        try {
-          for await (const event of completion as any) {
-            if (event.type === 'response.output_text.delta') {
-              const data = {
-                item_id: event.item_id,
-                output_index: event.output_index,
-                content_index: event.content_index,
-                delta: event.delta,
-              };
-              await writer.write(encoder.encode(`data: ${JSON.stringify(data)}\n\n`));
-            }
-          }
-        } catch (error) {
-          console.error('Stream processing error:', error);
-          await writer.write(
-            encoder.encode(`data: ${JSON.stringify({ error: 'Stream processing error' })}\n\n`),
-          );
-        } finally {
-          await writer.close();
-        }
-      })();
-
-      return new NextResponse(stream_response.readable, {
+      return new NextResponse(response.completion as any, {
         headers: {
           'Content-Type': 'text/event-stream',
           'Cache-Control': 'no-cache',
@@ -79,7 +41,7 @@ export const POST = async (req: NextRequest) => {
       });
     }
 
-    return NextResponse.json({ completion });
+    return NextResponse.json({ completion: response.completion });
   } catch (error) {
     console.error('[OPEN_AI_COMPLETIONS]', error);
 
diff --git a/app/api/ai/image/route.ts b/app/api/ai/image/route.ts
index a5404c1e..01136d51 100644
--- a/app/api/ai/image/route.ts
+++ b/app/api/ai/image/route.ts
@@ -2,18 +2,14 @@ import { ReasonPhrases, StatusCodes } from 'http-status-codes';
 import { NextRequest, NextResponse } from 'next/server';
 import { v4 as uuidv4 } from 'uuid';
 
+import { generateImage } from '@/actions/ai/generate-image';
 import { getCurrentUser } from '@/actions/auth/get-current-user';
-import { getAppConfig } from '@/actions/configs/get-app-config';
 import { uploadFiles } from '@/actions/uploadthing/upload-files';
-import { AIProvider } from '@/server/ai-provider';
 
 export const maxDuration = 60;
 
 export const POST = async (req: NextRequest) => {
   const user = await getCurrentUser();
-  const config = await getAppConfig();
-
-  const provider = AIProvider(config?.ai?.provider as string);
 
   try {
     const { model, prompt } = await req.json();
@@ -22,18 +18,7 @@ export const POST = async (req: NextRequest) => {
       return new NextResponse(ReasonPhrases.UNAUTHORIZED, { status: StatusCodes.UNAUTHORIZED });
     }
 
-    const IMAGE_MODELS = config?.ai?.['image-models'] ?? [];
-    const models = IMAGE_MODELS.map(({ value }) => value);
-
-    if (!models.includes(model)) {
-      console.error('[OPEN_AI_FORBIDDEN_IMAGE_MODEL]', user);
-
-      return new NextResponse(ReasonPhrases.FORBIDDEN, {
-        status: StatusCodes.FORBIDDEN,
-      });
-    }
-
-    const response = await provider.images.generate({
+    const response = await generateImage({
       model,
       n: 1,
       prompt,
@@ -42,16 +27,24 @@ export const POST = async (req: NextRequest) => {
       size: '1024x1024',
     });
 
+    if (!response.image) {
+      console.error('[OPEN_AI_FORBIDDEN_IMAGE_MODEL]', user);
+
+      return new NextResponse(ReasonPhrases.FORBIDDEN, {
+        status: StatusCodes.FORBIDDEN,
+      });
+    }
+
     const files = await uploadFiles([
       {
         name: `${uuidv4()}.png`,
-        base64: response.data[0].b64_json ?? '',
+        base64: response.image.data[0].b64_json ?? '',
         contentType: 'image/png',
       },
     ]);
 
     return NextResponse.json({
-      revisedPrompt: response.data[0].revised_prompt,
+      revisedPrompt: response.image.data[0].revised_prompt,
       url: files[0].data?.url,
     });
   } catch (error) {

From 9f492e30b0c58f7a552cd3b227d3e4cd16d6af4a Mon Sep 17 00:00:00 2001
From: Mikita Kandratsyeu <mikita_kandratsyeu1@epam.com>
Date: Fri, 18 Apr 2025 17:01:49 +0300
Subject: [PATCH 4/4] sonar

---
 actions/ai/generate-completion.ts | 2 +-
 actions/ai/generate-image.ts      | 2 +-
 actions/auth/get-login-quote.ts   | 2 +-
 actions/chat/get-chat-initial.ts  | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/actions/ai/generate-completion.ts b/actions/ai/generate-completion.ts
index ddbeeec9..45ba0158 100644
--- a/actions/ai/generate-completion.ts
+++ b/actions/ai/generate-completion.ts
@@ -22,7 +22,7 @@ export const generateCompletion = async ({
   const config = await getAppConfig();
 
   const aiModel = model ?? config?.ai?.['text-models']?.[0].value ?? '';
-  const provider = AIProvider(config?.ai?.provider as string);
+  const provider = AIProvider(config?.ai?.provider);
 
   const TEXT_MODELS = config?.ai?.['text-models'] ?? [];
   const models = (isOwner(user?.userId) ? TEXT_MODELS : TEXT_MODELS.slice(0, 2)).map(
diff --git a/actions/ai/generate-image.ts b/actions/ai/generate-image.ts
index 93d59202..e2382c93 100644
--- a/actions/ai/generate-image.ts
+++ b/actions/ai/generate-image.ts
@@ -14,7 +14,7 @@ export const generateImage = async ({ model, prompt }: GenerateImage) => {
   const config = await getAppConfig();
 
   const aiModel = model ?? config?.ai?.['image-models']?.[0].value ?? '';
-  const provider = AIProvider(config?.ai?.provider as string);
+  const provider = AIProvider(config?.ai?.provider);
 
   const IMAGE_MODELS = config?.ai?.['image-models'] ?? [];
   const models = IMAGE_MODELS.map(({ value }) => value);
diff --git a/actions/auth/get-login-quote.ts b/actions/auth/get-login-quote.ts
index 984d9d84..f45f5e3a 100644
--- a/actions/auth/get-login-quote.ts
+++ b/actions/auth/get-login-quote.ts
@@ -32,7 +32,7 @@ export const getLoginQuote = async () => {
       TEN_MINUTE_SEC,
     );
 
-    const generatedQuote = JSON.parse(response.completion.output_text || '{}');
+    const generatedQuote = JSON.parse(response.completion.output_text ?? '{}');
     const model = response.model ?? '';
 
     return {
diff --git a/actions/chat/get-chat-initial.ts b/actions/chat/get-chat-initial.ts
index 79b43dd4..497ac40d 100644
--- a/actions/chat/get-chat-initial.ts
+++ b/actions/chat/get-chat-initial.ts
@@ -32,7 +32,7 @@ export const getChatInitial = async () => {
     );
 
     return {
-      introMessages: JSON.parse(introMessages.output_text || '[]'),
+      introMessages: JSON.parse(introMessages.output_text ?? '[]'),
     };
   } catch (error) {
     console.error('[GET_CHAT_INITIAL_ACTION]', error);