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

fix e2e

Browse source
This commit is contained in:
Adil Hafeez 2024-11-06 16:07:32 -08:00
parent 60e442b168
commit 23ed25342a
2 changed files with 9 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

16
crates/prompt_gateway/src/context.rs
View file

@ -34,14 +34,14 @@ impl Context for StreamContext {
│ on_http_request_body ├──►│ prompt received ├──►│ get embeddings ├──►│ zeroshot intent │ │ on_http_request_body ├──►│ prompt received ├──►│ get embeddings ├──►│ zeroshot intent │
│ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │
└──────────────────────┘ └────────┬────────┘ └────────────────┘ └─────────────────┘ └──────────────────────┘ └────────┬────────┘ └────────────────┘ └─────────────────┘
│ ▲ │ ▲
│ │ │ │
│ │ │ │
│ ┌────────┴───────┐ │ ┌────────┴───────┐
│ │ │ │ │ │
└───────────►│ arch guard │ └───────────►│ arch guard │
│ │ │ │
└────────────────┘ └────────────────┘
continue from zeroshot intent continue from zeroshot intent

1
e2e_tests/test_prompt_gateway.py
View file

@ -134,6 +134,7 @@ def test_prompt_gateway_param_gathering(stream):
assert len(choices) > 0 assert len(choices) > 0
tool_calls = choices[0].get("delta", {}).get("tool_calls", []) tool_calls = choices[0].get("delta", {}).get("tool_calls", [])
assert len(tool_calls) == 0 assert len(tool_calls) == 0
# chunk would have "Could you provide the following details days"
else: else:
response_json = response.json() response_json = response.json()
assert response_json.get("model").startswith("Arch") assert response_json.get("model").startswith("Arch")