Integrate Captn agent (#25)

airtai · Nov 14, 2023 · 9fca95e · 9fca95e
2 parents 1d16006 + b19fbe0
commit 9fca95e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/src/client/ChatPage.jsx b/src/client/ChatPage.jsx
@@ -117,7 +117,10 @@ export default function ChatPage(props) {
           }
           await updateConversation(payload)
           // 2. call backend python server to get agent response
-          const response = await getAgentResponse({conversation: payload.conversations})
+          const response = await getAgentResponse({
+              message: userQuery,
+              conv_id: payload.conversation_id,
+            })
           // 3. add agent response as new conversation in the table
           const openAIPayload = {
               conversation_id: conversations.id,

diff --git a/src/server/actions.ts b/src/server/actions.ts
@@ -202,18 +202,19 @@ export const updateConversation: UpdateConversation<UpdateConversationPayload, C
 }
 
 type AgentPayload = {
-  conversation: any;
+  message: string;
+  conv_id: number;
 };
 
 export const getAgentResponse: GetAgentResponse<AgentPayload> = async (
-  { conversation },
+  { message, conv_id },
   context
 ) => {
   if (!context.user) {
     throw new HttpError(401);
   }
 
-  const payload = { conversation: conversation };
+  const payload = { message: message, conv_id: conv_id, user_id: context.user.id };
   try {
     const response = await fetch(`${ADS_SERVER_URL}/chat`, {
       method: 'POST',