Add status column to conversation table

main.wasp

@@ -129,6 +129,7 @@ entity Conversation {=psl
     conversation  Json
     createdAt                 DateTime        @default(now())
     updatedAt                 DateTime        @updatedAt
+    status String?
     chat        Chat?    @relation(fields: [chatId], references: [id])
     chatId      Int?
     user        User?    @relation(fields: [userId], references: [id])

migrations/20231123091639_add_status_to_conversation_model/migration.sql

@@ -0,0 +1,2 @@
+-- AlterTable
+ALTER TABLE "Conversation" ADD COLUMN     "status" TEXT;

src/client/components/ConversationWrapper.tsx

@@ -114,6 +114,7 @@ export default function ConversationWrapper() {
           // @ts-ignore
           ...[{ role: "assistant", content: response.content }],
         ],
+        status: response.team_status,
       };
       await updateConversation(openAIPayload);
       setIsLoading(false);

src/server/actions.ts

@@ -202,6 +202,7 @@ export const createChat: CreateChat<void, Conversation> = async (
 type UpdateConversationPayload = {
   conversation_id: number;
   conversations: any;
+  status: string;
 };
 
 export const updateConversation: UpdateConversation<
@@ -215,6 +216,7 @@ export const updateConversation: UpdateConversation<
     where: { id: args.conversation_id },
     data: {
       conversation: args.conversations,
+      status: args.status,
     },
   });
 };
@@ -248,7 +250,7 @@ export const getAgentResponse: GetAgentResponse<AgentPayload> = async (
       body: JSON.stringify(payload),
     });
 
-    const json = (await response.json()) as { detail?: string }; // Parse JSON once
+    const json: any = (await response.json()) as { detail?: string }; // Parse JSON once
 
     if (!response.ok) {
       const errorMsg =
@@ -257,7 +259,7 @@ export const getAgentResponse: GetAgentResponse<AgentPayload> = async (
       throw new Error(errorMsg);
     }
 
-    return { content: json };
+    return { content: json["content"], team_status: json["team_status"] };
   } catch (error: any) {
     throw new HttpError(500, "Something went wrong. Please try again later");
   }