Don't set routing model, and always inject into run if unset

inngest · Dec 10, 2024 · 0a60c09 · 0a60c09
1 parent ee5d9d8
commit 0a60c09
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/src/network.ts b/src/network.ts
@@ -231,11 +231,10 @@ export class Network {
     routingAgent: RoutingAgent,
     input: string,
   ): Promise<Agent[] | undefined> {
-    if (routingAgent.model === undefined && this.defaultModel) {
-      routingAgent = routingAgent.withModel(this.defaultModel);
-    }
-
-    const result = await routingAgent.run(input, { network: this });
+    const result = await routingAgent.run(input, {
+      network: this,
+      model: routingAgent.model || this.defaultModel,
+    });
     const agentNames = routingAgent.lifecycles.onRoute({
       result,
       agent: routingAgent,