apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-05-10 08:12:48 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Break apart metrics into their own module (#335)

Browse source 
Signed-off-by: José Ulises Niño Rivera <junr03@users.noreply.github.com>
This commit is contained in:
José Ulises Niño Rivera 2024-12-09 10:46:46 -08:00 • committed by GitHub
parent d002b2042a
commit cd1b561192
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
9 changed files with 59 additions and 61 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,5 +1,6 @@
use crate::embeddings::EmbeddingType;
use crate::filter_context::{EmbeddingsStore, WasmMetrics};
use crate::filter_context::EmbeddingsStore;
use crate::metrics::Metrics;
use acap::cos;
use common::api::hallucination::{
extract_messages_for_hallucination, HallucinationClassificationRequest,
@ -67,7 +68,7 @@ pub struct StreamContext {
pub prompt_targets: Rc<HashMap<String, PromptTarget>>,
pub embeddings_store: Option<Rc<EmbeddingsStore>>,
overrides: Rc<Option<Overrides>>,
pub metrics: Rc<WasmMetrics>,
pub metrics: Rc<Metrics>,
pub callouts: RefCell<HashMap<u32, StreamCallContext>>,
pub context_id: u32,
pub tool_calls: Option<Vec<ToolCall>>,
@ -90,7 +91,7 @@ impl StreamContext {
#[allow(clippy::too_many_arguments)]
pub fn new(
context_id: u32,
metrics: Rc<WasmMetrics>,
metrics: Rc<Metrics>,
system_prompt: Rc<Option<String>>,
prompt_targets: Rc<HashMap<String, PromptTarget>>,
prompt_guards: Rc<PromptGuards>,