apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-06-20 15:28:07 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix for emtadata

Browse source
This commit is contained in:
cotran 2024-10-23 16:51:30 -07:00
parent 7a5852b401
commit 2910cc9169
1 changed files with 9 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -12,12 +12,7 @@ use common::common_types::{
}; };
use common::configuration::{Overrides, PromptGuards, PromptTarget}; use common::configuration::{Overrides, PromptGuards, PromptTarget};
use common::consts::{ use common::consts::{
ARCH_FC_INTERNAL_HOST, ARCH_FC_MODEL_NAME, ARCH_FC_REQUEST_TIMEOUT_MS, ARCH_FC_INTERNAL_HOST, ARCH_FC_MODEL_NAME, ARCH_FC_REQUEST_TIMEOUT_MS, ARCH_INTERNAL_CLUSTER_NAME, ARCH_MESSAGES_KEY, ARCH_MODEL_PREFIX, ARCH_STATE_HEADER, ARCH_UPSTREAM_HOST_HEADER, ASSISTANT_ROLE, DEFAULT_EMBEDDING_MODEL, DEFAULT_HALLUCINATED_THRESHOLD, DEFAULT_INTENT_MODEL, DEFAULT_PROMPT_TARGET_THRESHOLD, EMBEDDINGS_INTERNAL_HOST, HALLUCINATION_INTERNAL_HOST, REQUEST_ID_HEADER, SYSTEM_ROLE, TOOL_ROLE, USER_ROLE, ZEROSHOT_INTERNAL_HOST
ARCH_INTERNAL_CLUSTER_NAME, ARCH_MESSAGES_KEY, ARCH_MODEL_PREFIX, ARCH_STATE_HEADER,
ARCH_UPSTREAM_HOST_HEADER, DEFAULT_EMBEDDING_MODEL, DEFAULT_HALLUCINATED_THRESHOLD,
DEFAULT_INTENT_MODEL, DEFAULT_PROMPT_TARGET_THRESHOLD, EMBEDDINGS_INTERNAL_HOST,
HALLUCINATION_INTERNAL_HOST, REQUEST_ID_HEADER, SYSTEM_ROLE, TOOL_ROLE, USER_ROLE,
ZEROSHOT_INTERNAL_HOST,
}; };
use common::embeddings::{ use common::embeddings::{
CreateEmbeddingRequest, CreateEmbeddingRequestInput, CreateEmbeddingResponse, CreateEmbeddingRequest, CreateEmbeddingRequestInput, CreateEmbeddingResponse,
@ -340,6 +335,13 @@ impl StreamContext {
tool_call_id: None, tool_call_id: None,
}; };
let mut metadata: HashMap<String, String> = HashMap::new();
metadata.insert(
ARCH_STATE_HEADER.to_string(),
serde_json::to_string(&self.arch_state).unwrap(),
);
let chat_completion_response = ChatCompletionsResponse { let chat_completion_response = ChatCompletionsResponse {
choices: vec![Choice { choices: vec![Choice {
message, message,
@ -348,7 +350,7 @@ impl StreamContext {
}], }],
usage: None, usage: None,
model: ARCH_FC_MODEL_NAME.to_string(), model: ARCH_FC_MODEL_NAME.to_string(),
metadata: None, metadata: Some(metadata),
}; };
trace!("hallucination response: {:?}", chat_completion_response); trace!("hallucination response: {:?}", chat_completion_response);