feat: wip

latitude-dev · Jan 13, 2025 · 962fe63 · 962fe63
1 parent 2899bd9
commit 962fe63
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 55 deletions.
diff --git a/packages/constants/src/ai.ts b/packages/constants/src/ai.ts
@@ -69,7 +69,6 @@ export enum ChainEventTypes {
   Step = 'chain-step',
   Complete = 'chain-complete',
   StepComplete = 'chain-step-complete',
-  ToolsCalled = 'chain-tool-call',
 }
 
 export type ProviderData =

diff --git a/packages/core/src/services/chains/ChainStreamConsumer/index.ts b/packages/core/src/services/chains/ChainStreamConsumer/index.ts
@@ -125,39 +125,6 @@ export class ChainStreamConsumer {
     controller.close()
   }
 
-  static toolsCalled({
-    response,
-    config,
-    controller,
-  }: {
-    controller: ReadableStreamDefaultController
-    response: ChainStepResponse<StreamType>
-    config: Config
-  }) {
-    const content = parseResponseText(response)
-
-    enqueueChainEvent(controller, {
-      event: StreamEventTypes.Latitude,
-      data: {
-        type: ChainEventTypes.ToolsCalled,
-        config,
-        documentLogUuid: response.documentLogUuid,
-        response,
-        messages: [
-          {
-            role: MessageRole.assistant,
-            toolCalls:
-              response.streamType === 'text' ? response.toolCalls || [] : [],
-            // Review after rebase with alex's fixes
-            content,
-          },
-        ],
-      },
-    })
-
-    controller.close()
-  }
-
   static chainError({
     controller,
     e,
@@ -240,20 +207,6 @@ export class ChainStreamConsumer {
     })
   }
 
-  toolsCalled({
-    step,
-    response,
-  }: {
-    step: ValidatedStep
-    response: ChainStepResponse<StreamType>
-  }) {
-    ChainStreamConsumer.toolsCalled({
-      controller: this.controller,
-      response,
-      config: step.conversation.config as Config,
-    })
-  }
-
   chainError(e: unknown) {
     return ChainStreamConsumer.chainError({
       controller: this.controller,

diff --git a/packages/core/src/services/chains/buildStep/index.ts b/packages/core/src/services/chains/buildStep/index.ts
@@ -70,21 +70,17 @@ export async function buildStepExecution({
     const toolCalls = getToolCalls({ response: finalResponse })
     const hasTools = isPromptl && toolCalls.length > 0
 
-    if (isPromptl && toolCalls.length > 0) {
+    if (hasTools) {
       await cacheChain({ workspace, chain, documentLogUuid, toolCalls })
-      streamConsumer.toolsCalled({
-        step,
-        response: finalResponse,
-      })
-
-      return finalResponse
     }
 
     // Instead of ChainCompleted emit chain completed with stop reason tool-call
     if (step.chainCompleted || hasTools) {
       streamConsumer.chainCompleted({
         step,
         response: finalResponse,
+        // TODO: Add this
+        finishReason
       })
 
       return finalResponse