fix: add local env for connecting local server (#106)

<img width="334" alt="image"
src="https://github.com/landing-ai/vision-agent-ui/assets/132986242/e124adc4-acd1-409b-971f-914562d5ac00">

.github/workflows/cicd.yaml

@@ -142,3 +142,4 @@ jobs:
             --set env.LOKI_AUTH_USER_PASSWORD=${{ vars.LOKI_AUTH_USER_PASSWORD }} \
             --set env.OPENAI_API_KEY=${{ vars.OPENAI_API_KEY }} \
             --set env.POSTGRES_PRISMA_URL=${{ vars.POSTGRES_PRISMA_URL }}
+            --set env.AGENT_HOST=${{ vars.AGENT_HOST }}

.github/workflows/prod-deploy.yml

@@ -102,3 +102,4 @@ jobs:
             --set env.LOKI_AUTH_USER_PASSWORD=${{ vars.LOKI_AUTH_USER_PASSWORD }} \
             --set env.OPENAI_API_KEY=${{ vars.OPENAI_API_KEY }} \
             --set env.POSTGRES_PRISMA_URL=${{ vars.POSTGRES_PRISMA_URL }}
+            --set env.AGENT_HOST=${{ vars.AGENT_HOST }}

app/api/vision-agent/route.ts

@@ -121,14 +121,8 @@ export const POST = withLogging(
     formData.append('input', apiMessages);
     formData.append('image', encodeURI(mediaUrl));
 
-    const agentHost = process.env.LND_TIER
-      ? 'http://publicrestapi-app-lndsvc.publicrestapi.svc.cluster.local:5000'
-      : 'https://api.dev.landing.ai';
-
     const fetchResponse = await fetch(
-      // `${agentHost}/v1/agent/chat?agent_class=vision_agent&self_reflection=false`,
-      // `https://api.dev.landing.ai/v1/agent/chat?agent_class=vision_agent&self_reflection=false`,
-      `http://localhost:5001/v1/agent/chat?agent_class=vision_agent&self_reflection=false`,
+      `${process.env.AGENT_HOST}/v1/agent/chat?agent_class=vision_agent&self_reflection=false`,
       {
         method: 'POST',
         headers: {
@@ -183,6 +177,8 @@ export const POST = withLogging(
             message: 'Streaming cancelled',
             maxChunkSize,
             reason,
+            messageId,
+            chatId,
           },
           request,
           '__AGENT_STREAM_CANCELLED',
@@ -335,6 +331,8 @@ export const POST = withLogging(
               maxChunkSize,
               reason,
               error,
+              messageId,
+              chatId,
             },
             request,
             error ? '__AGENT_DONE_WITH_ERROR' : '__AGENT_DONE',

chart/values.yaml

@@ -98,6 +98,7 @@ env:
   POSTGRES_PRISMA_URL: ""
   NEXTAUTH_URL: ""
   LND_TIER: ""
+  AGENT_HOST: ""
 
 # Additional volumes on the output Deployment definition.
 volumes: []

lib/logger.ts

@@ -118,14 +118,14 @@ export const withLogging = (
   return async (req: Request) => {
     const session = await auth();
     const json = await req.json();
-    // logger.info(
-    //   session,
-    //   {
-    //     params: json,
-    //   },
-    //   req,
-    //   '_API_REQUEST',
-    // );
+    logger.info(
+      session,
+      {
+        params: json,
+      },
+      req,
+      '_API_REQUEST',
+    );
     return handler(session, json, req);
   };
 };