fix tests

This commit is contained in:
Adil Hafeez 2025-03-14 17:02:09 -07:00
parent 2179b5a162
commit 898136470a
No known key found for this signature in database
GPG key ID: 9B18EF7691369645
7 changed files with 65 additions and 63 deletions

View file

@ -547,7 +547,7 @@ class ArchFunctionHandler(ArchBaseHandler):
messages=messages,
model=self.model_name,
stream=True,
extra_body={"temperature": 0.01, "logprobs": True},
extra_body=self.generation_params,
)
use_agent_orchestrator = req.metadata.get("use_agent_orchestrator", False)