diff --git a/furumi-agent/src/ingest/normalize.rs b/furumi-agent/src/ingest/normalize.rs index 087bab1..cc19bce 100644 --- a/furumi-agent/src/ingest/normalize.rs +++ b/furumi-agent/src/ingest/normalize.rs @@ -33,6 +33,7 @@ pub async fn normalize( &user_message, state.config.ollama_auth.as_deref(), 0.5, + 512, Some(("normalized_metadata", schema.clone())), ) .await?; @@ -48,6 +49,7 @@ pub async fn normalize( &user_message, state.config.ollama_auth.as_deref(), 1.5, + 512, Some(("normalized_metadata", schema)), ) .await?; @@ -185,6 +187,7 @@ pub async fn call_ollama( user_message: &str, auth: Option<&str>, frequency_penalty: f64, + max_tokens: u32, schema: Option<(&str, serde_json::Value)>, ) -> anyhow::Result { let client = reqwest::Client::builder() @@ -211,7 +214,7 @@ pub async fn call_ollama( response_format, stream: false, temperature: 0.1, - max_tokens: 512, + max_tokens, frequency_penalty, }; diff --git a/furumi-agent/src/merge.rs b/furumi-agent/src/merge.rs index 3ab95a7..8c7f83c 100644 --- a/furumi-agent/src/merge.rs +++ b/furumi-agent/src/merge.rs @@ -66,6 +66,7 @@ pub async fn propose_merge(state: &Arc, merge_id: Uuid) -> anyhow::Res &user_message, state.config.ollama_auth.as_deref(), 0.5, + 4096, Some(("artist_merge", schema)), ).await?;