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

Include param default in parameters (#68)

Browse source 
* Include param default in parameters

* improve default param pr

* fix integration tests
This commit is contained in:
Adil Hafeez 2024-09-20 09:02:24 -07:00 • committed by GitHub
parent aa73a5aaae
commit 97b47c2ab4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 21 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

20
envoyfilter/src/stream_context.rs
View file

@ -343,6 +343,7 @@ impl StreamContext {
description: entity.description.clone(),
required: entity.required,
enum_values: entity.enum_values.clone(),
default: entity.default.clone(),
};
properties.insert(entity.name.clone(), param);
}
@ -425,7 +426,7 @@ impl StreamContext {
debug!("response received for function resolver");
let body_str = String::from_utf8(body).unwrap();
debug!("function_resolver response str: {:?}", body_str);
debug!("function_resolver response str: {}", body_str);
let boltfc_response: ChatCompletionsResponse = serde_json::from_str(&body_str).unwrap();
@ -459,8 +460,8 @@ impl StreamContext {
.get(callout_context.prompt_target_name.as_ref().unwrap())
.unwrap()
.clone();
// verify required parameters are present
// verify required parameters are present
prompt_target
.parameters
.as_ref()
@ -492,13 +493,20 @@ impl StreamContext {
tool_name, &prompt_target.name
);
}
let tool_params = &tools_call_response.tool_calls[0].arguments;
debug!("tool_name: {:?}", tool_name);
debug!("tool_params: {:?}", tool_params);
debug!("prompt_target: {:?}", prompt_target);
// extract all tool names
let tool_names: Vec<String> = tools_call_response
.tool_calls
.iter()
.map(|tool_call| tool_call.name.clone())
.collect();
let tool_params = &tools_call_response.tool_calls[0].arguments;
let tool_params_json_str = serde_json::to_string(&tool_params).unwrap();
debug!("tool_name(s): {:?}", tool_names);
debug!("tool_params: {}", tool_params_json_str);
debug!("prompt_target_name: {}", prompt_target.name);
let endpoint = prompt_target.endpoint.as_ref().unwrap();
let token_id = match self.dispatch_http_call(
&endpoint.cluster,