@convex-dev/agent 0.6.3 → 0.7.0-alpha.0

package/src/{client → vercel/client}/utils.ts

@@ -1,4 +1,5 @@
-import type { StepResult, StopCondition } from "ai";
+import type { Context } from "@ai-sdk/provider-utils";
+import type { StepResult, StopCondition, ToolSet } from "ai";
 
 /**
  * A stop condition that only matches tool calls which completed
@@ -14,18 +15,23 @@ export function hasSuccessfulToolCall(toolName: string): StopCondition<any> {
     ) ?? false;
 }
 
-export async function willContinue(
-  steps: StepResult<any>[],
-
-  stopWhen: StopCondition<any> | Array<StopCondition<any>> | undefined,
+export async function willContinue<
+  TOOLS extends ToolSet,
+  RUNTIME_CONTEXT extends Context,
+>(
+  steps: StepResult<TOOLS, RUNTIME_CONTEXT>[],
+  stopWhen:
+    | StopCondition<TOOLS, RUNTIME_CONTEXT>
+    | Array<StopCondition<TOOLS, RUNTIME_CONTEXT>>
+    | undefined,
 ): Promise<boolean> {
   const step = steps.at(-1)!;
   // we aren't doing another round after a tool result
   // TODO: whether to handle continuing after too much context used..
   if (step.finishReason !== "tool-calls") return false;
   // Count both successful results and errors as completed outputs.
-  // In AI SDK v6, failed tool calls produce tool-error content parts
-  // instead of tool-result, so only checking toolResults misses them.
+  // Failed tool calls are represented as tool-error content parts, so only
+  // checking toolResults misses them.
   const completedOutputs =
     step.content?.filter(
       (p) => p.type === "tool-result" || p.type === "tool-error",