Changeset 38764

Timestamp:

2024-02-21T18:23:46+13:00 (4 months ago)

Author:

davidb

Message:

Stricter (more strongly type) code, which turned out to be needed to run as 'npm run build'

File:

• gs3-installations/thewillow/trunk/sites/thewillow/dlcol-chatgpt/react-gui/pages/api/chat.ts (modified) (8 diffs)

gs3-installations/thewillow/trunk/sites/thewillow/dlcol-chatgpt/react-gui/pages/api/chat.ts

@@ -8 +8 @@
 
 
+// For an (alternative) approach that looks like it retrieve the already created Assistant directly, see:
+//    https://gist.github.com/drifterz13/0cbe93ced5dc7958d7841a29c1721d1c
+
 import {
-  //Configuration,
-  OpenAI,
-  //ChatCompletionRequestMessageRoleEnum
+    //Configuration,
+    OpenAI,
+    //ChatCompletionRequestMessageRoleEnum
 } from "openai";
+
+
+import { MessageContentText } from 'openai/resources/beta/threads/messages/messages'
+
 
 import type { NextApiRequest, NextApiResponse } from "next";
@@ -21 +28 @@
 });
 
-const threadByUser = {}; // Store thread IDs by user
+//const threadByUser = {}; // Store thread IDs by user
 
-async function assistantHandler(res,req) {
-    
-    const assistantIdToUse = process.env.ASSISTANT_ID;
+const threadByUser = new Map<string, any>();
+//const threadByUser = {};
+
+async function assistantHandler(
+    req: NextApiRequest,
+    res: NextApiResponse
+) {    
+    const assistantIdToUse = process.env.ASSISTANT_ID || "";
     const modelToUse = "gpt-4-turbo-preview"; // Specify the model you want to use
     
     //const userId = req.body.userId; // You should include the user ID in the request
-    const userId = "a-username";
+    const userId = "test_username";
     
     // Create a new thread if it's the user's first message
-    if (!threadByUser[userId]) {
+    //if (!threadByUser[userId]) {
+    if (!threadByUser.has(userId)) {
     try {
         const myThread = await openai.beta.threads.create();
         console.log("New thread created with ID: ", myThread.id, "\n");
-        threadByUser[userId] = myThread.id; // Store the thread ID for this user
+        //threadByUser[userId] = myThread.id; // Store the thread ID for this user
+        threadByUser.set(userId,myThread.id); // Store the thread ID for this user
     } catch (error) {
         console.error("Error creating thread:", error);
@@ -56 +70 @@
     try {
     const myThreadMessage = await openai.beta.threads.messages.create(
-        threadByUser[userId], // Use the stored thread ID for this user
+        threadByUser.get(userId), // Use the stored thread ID for this user
         {
         role: "user",
@@ -66 +80 @@
     // Run the Assistant
     const myRun = await openai.beta.threads.runs.create(
-        threadByUser[userId], // Use the stored thread ID for this user
+        threadByUser.get(userId), // Use the stored thread ID for this user
         {
         assistant_id: assistantIdToUse,
@@ -84 +98 @@
         while (myRun.status !== "completed") {
         keepRetrievingRun = await openai.beta.threads.runs.retrieve(
-            threadByUser[userId], // Use the stored thread ID for this user
+            threadByUser.get(userId), // Use the stored thread ID for this user
             myRun.id
         );
@@ -103 +117 @@
         
         const allMessages = await openai.beta.threads.messages.list(
-        threadByUser[userId] // Use the stored thread ID for this user
+        threadByUser.get(userId) // Use the stored thread ID for this user
         );
         
@@ -115 +129 @@
         "------------------------------------------------------------ \n"
         );
-        
-        console.log("User: ", myThreadMessage.content[0].text.value);
-        console.log("Assistant: ", allMessages.data[0].content[0].text.value);
+
+        const user_request_message: MessageContentText = (myThreadMessage.content[0]) as MessageContentText;
+        const assistant_response_message = allMessages.data[0].content[0] as MessageContentText;
+    
+        console.log("User: ", user_request_message.text.value);
+        console.log("Assistant: ", assistant_response_message.text.value);
 
         var returned_message = {
         role: "assistant",
-        content: allMessages.data[0].content[0].text.value
+        content: assistant_response_message.text.value
         };
         
@@ -170 +187 @@
     */
     
-    const assistant_message = await assistantHandler(res,req);
+    const assistant_message = await assistantHandler(req,res);
 
     console.log("**** (assistant) completion message: ");