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

Merge branch 'main' into integrate-Arch-Function-v1.1

Browse source
This commit is contained in:
Adil Hafeez 2025-04-14 09:16:27 -07:00
commit 0ee92798b0
No known key found for this signature in database
GPG key ID: 9B18EF7691369645
22 changed files with 73 additions and 54 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/e2e/test_prompt_gateway.py
View file

@ -64,7 +64,7 @@ def test_prompt_gateway(stream):
# third..end chunk is summarization (role = assistant)
response_json = json.loads(chunks[2])
assert response_json.get("model").startswith("gpt-4o-mini")
assert response_json.get("model").startswith("llama-3.2-3b-preview")
choices = response_json.get("choices", [])
assert len(choices) > 0
assert "role" in choices[0]["delta"]
@ -73,7 +73,7 @@ def test_prompt_gateway(stream):
else:
response_json = response.json()
assert response_json.get("model").startswith("gpt-4o-mini")
assert response_json.get("model").startswith("llama-3.2-3b-preview")
choices = response_json.get("choices", [])
assert len(choices) > 0
assert "role" in choices[0]["message"]
@ -233,7 +233,7 @@ def test_prompt_gateway_param_tool_call(stream):
# third..end chunk is summarization (role = assistant)
response_json = json.loads(chunks[2])
assert response_json.get("model").startswith("gpt-4o-mini")
assert response_json.get("model").startswith("llama-3.2-3b-preview")
choices = response_json.get("choices", [])
assert len(choices) > 0
assert "role" in choices[0]["delta"]
@ -242,7 +242,7 @@ def test_prompt_gateway_param_tool_call(stream):
else:
response_json = response.json()
assert response_json.get("model").startswith("gpt-4o-mini")
assert response_json.get("model").startswith("llama-3.2-3b-preview")
choices = response_json.get("choices", [])
assert len(choices) > 0
assert "role" in choices[0]["message"]