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

add support for jaeger tracing (#229)

Browse source
This commit is contained in:
Adil Hafeez 2024-11-07 22:11:00 -06:00 • committed by GitHub
parent fb67788be0
commit a72bb804eb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
64 changed files with 5032 additions and 1112 deletions
Download patch file Download diff file Expand all files Collapse all files

7
e2e_tests/test_prompt_gateway.py
View file

@ -147,6 +147,7 @@ def test_prompt_gateway_param_gathering(stream):
assert len(choices) > 0
tool_calls = choices[0].get("delta", {}).get("tool_calls", [])
assert len(tool_calls) == 0
# chunk would have "Could you provide the following details days"
else:
response_json = response.json()
assert response_json.get("model").startswith("Arch")
@ -261,9 +262,7 @@ def test_prompt_gateway_default_target(stream):
choices = response_json.get("choices", [])
assert len(choices) > 0
content = choices[0]["delta"]["content"]
assert (
content == "I can help you with weather forecast or insurance claim details"
)
assert content == "I can help you with weather forecast"
else:
response_json = response.json()
assert response_json.get("model").startswith("api_server")
@ -271,5 +270,5 @@ def test_prompt_gateway_default_target(stream):
assert response_json.get("choices")[0]["message"]["role"] == "assistant"
assert (
response_json.get("choices")[0]["message"]["content"]
== "I can help you with weather forecast or insurance claim details"
== "I can help you with weather forecast"
)