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 rust tests

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

32
crates/prompt_gateway/tests/integration.rs
View file

@ -380,6 +380,7 @@ fn prompt_gateway_request_to_llm_gateway() {
.expect_log(Some(LogLevel::Warn), None) .expect_log(Some(LogLevel::Warn), None)
.expect_log(Some(LogLevel::Info), None) .expect_log(Some(LogLevel::Info), None)
.expect_log(Some(LogLevel::Debug), None) .expect_log(Some(LogLevel::Debug), None)
.expect_log(Some(LogLevel::Info), None)
.expect_log(Some(LogLevel::Debug), None) .expect_log(Some(LogLevel::Debug), None)
.expect_log(Some(LogLevel::Info), None) .expect_log(Some(LogLevel::Info), None)
.expect_log(Some(LogLevel::Debug), None) .expect_log(Some(LogLevel::Debug), None)
@ -453,6 +454,7 @@ fn prompt_gateway_request_to_llm_gateway() {
.expect_log(Some(LogLevel::Info), None) .expect_log(Some(LogLevel::Info), None)
.expect_set_buffer_bytes(Some(BufferType::HttpResponseBody), None) .expect_set_buffer_bytes(Some(BufferType::HttpResponseBody), None)
.expect_log(Some(LogLevel::Info), None) .expect_log(Some(LogLevel::Info), None)
.expect_log(Some(LogLevel::Info), None)
.expect_log(Some(LogLevel::Debug), None) .expect_log(Some(LogLevel::Debug), None)
.execute_and_expect(ReturnType::Action(Action::Continue)) .execute_and_expect(ReturnType::Action(Action::Continue))
.unwrap(); .unwrap();
@ -493,19 +495,9 @@ fn prompt_gateway_request_no_intent_match() {
finish_reason: Some("test".to_string()), finish_reason: Some("test".to_string()),
index: Some(0), index: Some(0),
message: Message { message: Message {
role: "system".to_string(), role: "assistant".to_string(),
content: None, content: None,
tool_calls: Some(vec![ToolCall { tool_calls: None,
id: String::from("test"),
tool_type: ToolType::Function,
function: FunctionCallDetail {
name: String::from("weather_forecast"),
arguments: Some(HashMap::from([(
String::from("city"),
Value::String(String::from("seattle")),
)])),
},
}]),
model: None, model: None,
tool_call_id: None, tool_call_id: None,
}, },
@ -523,7 +515,7 @@ fn prompt_gateway_request_no_intent_match() {
.expect_log(Some(LogLevel::Warn), None) .expect_log(Some(LogLevel::Warn), None)
.expect_log(Some(LogLevel::Info), None) .expect_log(Some(LogLevel::Info), None)
.expect_log(Some(LogLevel::Debug), None) .expect_log(Some(LogLevel::Debug), None)
.expect_log(Some(LogLevel::Info), Some("intent not matched")) .expect_log(Some(LogLevel::Info), Some("intent matched: false"))
.expect_log( .expect_log(
Some(LogLevel::Info), Some(LogLevel::Info),
Some("no default prompt target found, forwarding request to upstream llm"), Some("no default prompt target found, forwarding request to upstream llm"),
@ -651,17 +643,7 @@ fn prompt_gateway_request_no_intent_match_default_target() {
message: Message { message: Message {
role: "system".to_string(), role: "system".to_string(),
content: None, content: None,
tool_calls: Some(vec![ToolCall { tool_calls: None,
id: String::from("test"),
tool_type: ToolType::Function,
function: FunctionCallDetail {
name: String::from("weather_forecast"),
arguments: Some(HashMap::from([(
String::from("city"),
Value::String(String::from("seattle")),
)])),
},
}]),
model: None, model: None,
tool_call_id: None, tool_call_id: None,
}, },
@ -679,7 +661,7 @@ fn prompt_gateway_request_no_intent_match_default_target() {
.expect_log(Some(LogLevel::Warn), None) .expect_log(Some(LogLevel::Warn), None)
.expect_log(Some(LogLevel::Info), None) .expect_log(Some(LogLevel::Info), None)
.expect_log(Some(LogLevel::Debug), None) .expect_log(Some(LogLevel::Debug), None)
.expect_log(Some(LogLevel::Info), Some("intent not matched")) .expect_log(Some(LogLevel::Info), Some("intent matched: false"))
.expect_log( .expect_log(
Some(LogLevel::Info), Some(LogLevel::Info),
Some("default prompt target found, forwarding request to default prompt target"), Some("default prompt target found, forwarding request to default prompt target"),