apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-06-17 15:25:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix tests

Browse source
This commit is contained in:
Adil Hafeez 2025-03-28 03:58:57 -07:00
parent e2edeedf55
commit e48918259e
No known key found for this signature in database
GPG key ID: 9B18EF7691369645
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/e2e/test_prompt_gateway.py
View file

@ -42,9 +42,11 @@ def test_prompt_gateway(stream):
assert "role" in choices[0]["delta"]
role = choices[0]["delta"]["role"]
assert role == "assistant"
tool_calls = choices[0].get("delta", {}).get("tool_calls", [])
print(f"choices: {choices}")
tool_call_str = choices[0].get("delta", {}).get("content", "")
tool_calls = json.loads(tool_call_str).get("tool_calls", [])
assert len(tool_calls) > 0
tool_call = tool_calls[0]["function"]
tool_call = tool_calls[0]
location = tool_call["arguments"]["location"]
assert expected_tool_call["arguments"]["location"] in location.lower()
del expected_tool_call["arguments"]["location"]