From 23ed25342aa3b77f59ad44331fa1f9ebd9cd8403 Mon Sep 17 00:00:00 2001 From: Adil Hafeez Date: Wed, 6 Nov 2024 16:07:32 -0800 Subject: [PATCH] fix e2e --- crates/prompt_gateway/src/context.rs | 16 ++++++++-------- e2e_tests/test_prompt_gateway.py | 1 + 2 files changed, 9 insertions(+), 8 deletions(-) diff --git a/crates/prompt_gateway/src/context.rs b/crates/prompt_gateway/src/context.rs index 2df70374..567f41eb 100644 --- a/crates/prompt_gateway/src/context.rs +++ b/crates/prompt_gateway/src/context.rs @@ -34,14 +34,14 @@ impl Context for StreamContext { │ on_http_request_body ├──►│ prompt received ├──►│ get embeddings ├──►│ zeroshot intent │ │ │ │ │ │ │ │ │ └──────────────────────┘ └────────┬────────┘ └────────────────┘ └─────────────────┘ - │ ▲ - │ │ - │ │ - │ ┌────────┴───────┐ - │ │ │ - └───────────►│ arch guard │ - │ │ - └────────────────┘ + │ ▲ + │ │ + │ │ + │ ┌────────┴───────┐ + │ │ │ + └───────────►│ arch guard │ + │ │ + └────────────────┘ continue from zeroshot intent diff --git a/e2e_tests/test_prompt_gateway.py b/e2e_tests/test_prompt_gateway.py index 809fc710..33893c85 100644 --- a/e2e_tests/test_prompt_gateway.py +++ b/e2e_tests/test_prompt_gateway.py @@ -134,6 +134,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")