feat: wip

latitude-dev · Jan 15, 2025 · 5f622ed · 5f622ed
1 parent 915f766
commit 5f622ed
Show file tree

Hide file tree

Showing 6 changed files with 16 additions and 16 deletions.
diff --git a/...ommitUuid]/documents/[documentUuid]/_components/DocumentEditor/Editor/Playground/Chat.tsx b/...ommitUuid]/documents/[documentUuid]/_components/DocumentEditor/Editor/Playground/Chat.tsx
@@ -166,6 +166,7 @@ export default function Chat({
             }
             break
           }
+
           default:
             break
         }
@@ -221,6 +222,8 @@ export default function Chat({
 
           switch (event) {
             case StreamEventTypes.Latitude: {
+              console.log('DATA', data)
+
               if (data.type === ChainEventTypes.Complete) {
                 setUsage(data.response.usage)
               } else if (data.type === ChainEventTypes.Error) {
@@ -237,6 +240,7 @@ export default function Chat({
               }
               break
             }
+
             default:
               break
           }

diff --git a/packages/core/src/constants.ts b/packages/core/src/constants.ts
@@ -261,7 +261,7 @@ const toolResultContentSchema = z.object({
   type: z.literal('tool-result'),
   toolCallId: z.string(),
   toolName: z.string(),
-  result: z.string(),
+  result: z.any(),
   isError: z.boolean().optional(),
 })
 

diff --git a/packages/core/src/services/chains/ChainStreamConsumer/index.ts b/packages/core/src/services/chains/ChainStreamConsumer/index.ts
@@ -12,6 +12,7 @@ import { Config } from '../../ai'
 import { ChainError } from '../ChainErrors'
 import { ValidatedStep } from '../ChainValidator'
 import { FinishReason } from 'ai'
+import { buildMessagesFromResponse } from '../../../helpers'
 
 export function enqueueChainEvent(
   controller: ReadableStreamDefaultController,
@@ -141,13 +142,15 @@ export class ChainStreamConsumer {
     step,
     response,
     finishReason,
-    responseMessages,
+    responseMessages: defaultResponseMessages,
   }: {
     step: ValidatedStep
     response: ChainStepResponse<StreamType>
     finishReason: FinishReason
-    responseMessages: Message[]
+    responseMessages?: Message[]
   }) {
+    const responseMessages =
+      defaultResponseMessages ?? buildMessagesFromResponse({ response })
     return ChainStreamConsumer.chainCompleted({
       controller: this.controller,
       response,

diff --git a/packages/core/src/services/chains/buildStep/index.ts b/packages/core/src/services/chains/buildStep/index.ts
@@ -11,7 +11,6 @@ import {
 } from '../ProviderProcessor/saveOrPublishProviderLogs'
 import { cacheChain } from '../chainCache'
 import { buildMessagesFromResponse } from '../../../helpers'
-import { Message } from '@latitude-data/compiler'
 
 function getToolCalls({
   response,
@@ -72,11 +71,9 @@ export async function buildStepExecution({
     const isPromptl = chain instanceof PromptlChain
     const toolCalls = getToolCalls({ response: finalResponse })
     const hasTools = isPromptl && toolCalls.length > 0
-    let responseMessages: Message[] = []
-
-    if (hasTools || step.chainCompleted) {
-      responseMessages = buildMessagesFromResponse({ response: finalResponse })
-    }
+    const responseMessages = buildMessagesFromResponse({
+      response: finalResponse,
+    })
 
     if (hasTools) {
       await cacheChain({

diff --git a/packages/core/src/services/chains/run.ts b/packages/core/src/services/chains/run.ts
@@ -192,10 +192,10 @@ export async function runStep({
       streamConsumer.chainCompleted({
         step,
         response: previousResponse!,
-        finishReason: 'stop',
-        responseMessages: [],
+        finishReason: previousResponse?.finishReason ?? 'stop',
       })
 
+      previousResponse!.chainCompleted = true
       return previousResponse!
     }
 
@@ -242,7 +242,6 @@ export async function runStep({
       return executeStep({ finalResponse })
     }
 
-    messages.forEach((m) => console.log('Message TO AI:', m))
     const aiResult = await ai({
       messages,
       config: step.config,
@@ -261,7 +260,6 @@ export async function runStep({
 
     if (consumedStream.error) throw consumedStream.error
 
-    step.chainCompleted
     const _response = await processResponse({
       aiResult,
       apiProvider: step.provider,

diff --git a/packages/web-ui/src/ds/molecules/Chat/Message/index.tsx b/packages/web-ui/src/ds/molecules/Chat/Message/index.tsx
@@ -165,9 +165,7 @@ function JSONContent({ content }: { content: unknown }) {
     return (
       <div className='py-2 max-w-full'>
         <div className='overflow-hidden rounded-xl w-full'>
-          <CodeBlock language='json'>
-            {JSON.stringify(parsedValue, null, 2)}
-          </CodeBlock>
+          <CodeBlock language='json'>{parsedValue}</CodeBlock>
         </div>
       </div>
     )