From 48c473de5678dc4d508dc754e0d557b9b3cc96f7 Mon Sep 17 00:00:00 2001
From: AB-UK <ab@hexor.cy>
Date: Tue, 7 Apr 2026 22:52:23 +0100
Subject: [PATCH] fix(agent): increase max_tokens for merge requests to avoid
 truncated responses

normalize: 512 tokens (sufficient for single track metadata)
merge: 4096 tokens (needed for artists with many albums)

Co-Authored-By: Claude Sonnet 4.6 (1M context) <noreply@anthropic.com>
---
 furumi-agent/src/ingest/normalize.rs | 5 ++++-
 furumi-agent/src/merge.rs            | 1 +
 2 files changed, 5 insertions(+), 1 deletion(-)

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<String> {
     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<AppState>, merge_id: Uuid) -> anyhow::Res
         &user_message,
         state.config.ollama_auth.as_deref(),
         0.5,
+        4096,
         Some(("artist_merge", schema)),
     ).await?;