@vellumai/assistant 0.8.8 → 0.8.9-staging.2

package/src/__tests__/agent-loop-callsite-precedence.test.ts

@@ -100,11 +100,19 @@ describe("AgentLoop — call-site precedence", () => {
     });
 
     const { provider, lastConfig } = makePipeline("anthropic");
-    const loop = new AgentLoop(provider, "system", {
+    const loop = new AgentLoop({
+      provider: provider,
+      systemPrompt: "system",
+      conversationId: "test-conversation",
       config: { maxTokens: 64000 },
     });
 
-    await loop.run([userMessage], () => {}, { callSite: "mainAgent" });
+    await loop.run({
+      messages: [userMessage],
+      onEvent: () => {},
+      trust: { sourceChannel: "vellum", trustClass: "unknown" },
+      callSite: "mainAgent",
+    });
 
     expect(lastConfig()!.max_tokens).toBe(4096);
   });
@@ -120,14 +128,22 @@ describe("AgentLoop — call-site precedence", () => {
     });
 
     const { provider, lastConfig } = makePipeline("anthropic");
-    const loop = new AgentLoop(provider, "system", {
+    const loop = new AgentLoop({
+      provider: provider,
+      systemPrompt: "system",
+      conversationId: "test-conversation",
       config: {
         maxTokens: 64000,
         effort: "high",
       },
     });
 
-    await loop.run([userMessage], () => {}, { callSite: "mainAgent" });
+    await loop.run({
+      messages: [userMessage],
+      onEvent: () => {},
+      trust: { sourceChannel: "vellum", trustClass: "unknown" },
+      callSite: "mainAgent",
+    });
 
     expect(lastConfig()!.effort).toBe("low");
   });
@@ -143,7 +159,10 @@ describe("AgentLoop — call-site precedence", () => {
     });
 
     const { provider, lastConfig } = makePipeline("anthropic");
-    const loop = new AgentLoop(provider, "system", {
+    const loop = new AgentLoop({
+      provider: provider,
+      systemPrompt: "system",
+      conversationId: "test-conversation",
       config: {
         maxTokens: 64000,
         effort: "high",
@@ -153,7 +172,12 @@ describe("AgentLoop — call-site precedence", () => {
       },
     });
 
-    await loop.run([userMessage], () => {}, { callSite: "mainAgent" });
+    await loop.run({
+      messages: [userMessage],
+      onEvent: () => {},
+      trust: { sourceChannel: "vellum", trustClass: "unknown" },
+      callSite: "mainAgent",
+    });
 
     expect(lastConfig()!.speed).toBe("fast");
   });
@@ -174,7 +198,10 @@ describe("AgentLoop — call-site precedence", () => {
     });
 
     const { provider, lastConfig } = makePipeline("anthropic");
-    const loop = new AgentLoop(provider, "system", {
+    const loop = new AgentLoop({
+      provider: provider,
+      systemPrompt: "system",
+      conversationId: "test-conversation",
       config: {
         maxTokens: 64000,
         // Conversation default also has thinking on — without the fix, this
@@ -184,7 +211,12 @@ describe("AgentLoop — call-site precedence", () => {
       },
     });
 
-    await loop.run([userMessage], () => {}, { callSite: "mainAgent" });
+    await loop.run({
+      messages: [userMessage],
+      onEvent: () => {},
+      trust: { sourceChannel: "vellum", trustClass: "unknown" },
+      callSite: "mainAgent",
+    });
 
     // Call-site override resolves `thinking.enabled: false`, so the
     // RetryProvider normalizer must send Anthropic's explicit disabled shape.
@@ -202,11 +234,19 @@ describe("AgentLoop — call-site precedence", () => {
     });
 
     const { provider, lastConfig } = makePipeline("anthropic");
-    const loop = new AgentLoop(provider, "system", {
+    const loop = new AgentLoop({
+      provider: provider,
+      systemPrompt: "system",
+      conversationId: "test-conversation",
       config: { maxTokens: 64000 },
     });
 
-    await loop.run([userMessage], () => {}, { callSite: "mainAgent" });
+    await loop.run({
+      messages: [userMessage],
+      onEvent: () => {},
+      trust: { sourceChannel: "vellum", trustClass: "unknown" },
+      callSite: "mainAgent",
+    });
 
     // Must be wire-format `{ type: "adaptive" }` so the Anthropic SDK's
     // `ThinkingConfigParam` accepts it. The schema-shape `{ enabled,
@@ -227,7 +267,10 @@ describe("AgentLoop — call-site precedence", () => {
     });
 
     const { provider, lastConfig } = makePipeline("anthropic");
-    const loop = new AgentLoop(provider, "system", {
+    const loop = new AgentLoop({
+      provider: provider,
+      systemPrompt: "system",
+      conversationId: "test-conversation",
       config: {
         maxTokens: 64000,
         effort: "high",
@@ -236,7 +279,11 @@ describe("AgentLoop — call-site precedence", () => {
       },
     });
 
-    await loop.run([userMessage], () => {});
+    await loop.run({
+      messages: [userMessage],
+      onEvent: () => {},
+      trust: { sourceChannel: "vellum", trustClass: "unknown" },
+    });
 
     const config = lastConfig()!;
     expect(config.max_tokens).toBe(64000);
@@ -262,12 +309,20 @@ describe("AgentLoop — call-site precedence", () => {
       maxTokens: 8192,
     });
 
-    const loop = new AgentLoop(provider, "system", {
+    const loop = new AgentLoop({
+      provider: provider,
+      systemPrompt: "system",
+      conversationId: "test-conversation",
       config: { maxTokens: 64000 },
       resolveSystemPrompt: resolveSystemPrompt,
     });
 
-    await loop.run([userMessage], () => {}, { callSite: "mainAgent" });
+    await loop.run({
+      messages: [userMessage],
+      onEvent: () => {},
+      trust: { sourceChannel: "vellum", trustClass: "unknown" },
+      callSite: "mainAgent",
+    });
 
     // Per-turn explicit value beats both the call-site (4096) and the
     // default (64000).