@corbat-tech/coco 2.8.1 → 2.8.2

package/dist/index.d.ts

@@ -1032,6 +1032,7 @@ interface StreamChunk {
     type: "text" | "tool_use_start" | "tool_use_delta" | "tool_use_end" | "done";
     text?: string;
     toolCall?: Partial<ToolCall>;
+    stopReason?: "end_turn" | "max_tokens" | "stop_sequence" | "tool_use";
 }
 /**
  * LLM Provider interface

package/dist/index.js

@@ -11803,6 +11803,7 @@ var AnthropicProvider = class {
       };
       const timeoutInterval = setInterval(checkTimeout, 5e3);
       try {
+        let streamStopReason;
         for await (const event of stream) {
           lastActivityTime = Date.now();
           if (event.type === "content_block_delta") {
@@ -11810,9 +11811,14 @@ var AnthropicProvider = class {
             if (delta.type === "text_delta" && delta.text) {
               yield { type: "text", text: delta.text };
             }
+          } else if (event.type === "message_delta") {
+            const delta = event.delta;
+            if (delta.stop_reason) {
+              streamStopReason = this.mapStopReason(delta.stop_reason);
+            }
           }
         }
-        yield { type: "done" };
+        yield { type: "done", stopReason: streamStopReason };
       } finally {
         clearInterval(timeoutInterval);
       }
@@ -11849,9 +11855,15 @@ var AnthropicProvider = class {
       };
       const timeoutInterval = setInterval(checkTimeout, 5e3);
       try {
+        let streamStopReason;
         for await (const event of stream) {
           lastActivityTime = Date.now();
-          if (event.type === "content_block_start") {
+          if (event.type === "message_delta") {
+            const delta = event.delta;
+            if (delta.stop_reason) {
+              streamStopReason = this.mapStopReason(delta.stop_reason);
+            }
+          } else if (event.type === "content_block_start") {
             const contentBlock = event.content_block;
             if (contentBlock.type === "tool_use") {
               if (currentToolCall) {
@@ -11922,7 +11934,7 @@ var AnthropicProvider = class {
             }
           }
         }
-        yield { type: "done" };
+        yield { type: "done", stopReason: streamStopReason };
       } finally {
         clearInterval(timeoutInterval);
       }
@@ -12418,13 +12430,18 @@ var OpenAIProvider = class {
         stream: true,
         ...supportsTemp && { temperature: options?.temperature ?? this.config.temperature ?? 0 }
       });
+      let streamStopReason;
       for await (const chunk of stream) {
         const delta = chunk.choices[0]?.delta;
         if (delta?.content) {
           yield { type: "text", text: delta.content };
         }
+        const finishReason = chunk.choices[0]?.finish_reason;
+        if (finishReason) {
+          streamStopReason = this.mapFinishReason(finishReason);
+        }
       }
-      yield { type: "done" };
+      yield { type: "done", stopReason: streamStopReason };
     } catch (error) {
       throw this.handleError(error);
     }
@@ -12489,6 +12506,7 @@ var OpenAIProvider = class {
         return input;
       };
       try {
+        let streamStopReason;
         for await (const chunk of stream) {
           const delta = chunk.choices[0]?.delta;
           if (delta?.content || delta?.tool_calls) {
@@ -12535,6 +12553,9 @@ var OpenAIProvider = class {
             }
           }
           const finishReason = chunk.choices[0]?.finish_reason;
+          if (finishReason) {
+            streamStopReason = this.mapFinishReason(finishReason);
+          }
           if (finishReason && toolCallBuilders.size > 0) {
             for (const [, builder] of toolCallBuilders) {
               yield {
@@ -12559,7 +12580,7 @@ var OpenAIProvider = class {
             }
           };
         }
-        yield { type: "done" };
+        yield { type: "done", stopReason: streamStopReason };
       } finally {
         clearInterval(timeoutInterval);
       }
@@ -13322,7 +13343,7 @@ var CodexProvider = class {
         }
       }
     }
-    yield { type: "done" };
+    yield { type: "done", stopReason: response.stopReason };
   }
   /**
    * Stream a chat response with tool use
@@ -13475,13 +13496,18 @@ var GeminiProvider = class {
       const { history, lastMessage } = this.convertMessages(messages);
       const chat = model.startChat({ history });
       const result = await chat.sendMessageStream(lastMessage);
+      let streamStopReason;
       for await (const chunk of result.stream) {
         const text = chunk.text();
         if (text) {
           yield { type: "text", text };
         }
+        const finishReason = chunk.candidates?.[0]?.finishReason;
+        if (finishReason) {
+          streamStopReason = this.mapFinishReason(finishReason);
+        }
       }
-      yield { type: "done" };
+      yield { type: "done", stopReason: streamStopReason };
     } catch (error) {
       throw this.handleError(error);
     }
@@ -13516,11 +13542,16 @@ var GeminiProvider = class {
       const chat = model.startChat({ history });
       const result = await chat.sendMessageStream(lastMessage);
       const emittedToolCalls = /* @__PURE__ */ new Set();
+      let streamStopReason;
       for await (const chunk of result.stream) {
         const text = chunk.text();
         if (text) {
           yield { type: "text", text };
         }
+        const finishReason = chunk.candidates?.[0]?.finishReason;
+        if (finishReason) {
+          streamStopReason = this.mapFinishReason(finishReason);
+        }
         const candidate = chunk.candidates?.[0];
         if (candidate?.content?.parts) {
           for (const part of candidate.content.parts) {
@@ -13554,7 +13585,7 @@ var GeminiProvider = class {
           }
         }
       }
-      yield { type: "done" };
+      yield { type: "done", stopReason: streamStopReason };
     } catch (error) {
       throw this.handleError(error);
     }