From 5049082a4f1d08509ca4d11a44ec9f3dc01c4541 Mon Sep 17 00:00:00 2001 From: cotran Date: Tue, 15 Oct 2024 11:23:30 -0700 Subject: [PATCH] correct init of user_messages --- arch/src/stream_context.rs | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/arch/src/stream_context.rs b/arch/src/stream_context.rs index 50ac0462..b72e4057 100644 --- a/arch/src/stream_context.rs +++ b/arch/src/stream_context.rs @@ -728,9 +728,10 @@ impl StreamContext { None => HashMap::new(), // Return an empty HashMap if v is not an object }; - let mut user_messages = Vec::new(); + let messages = &callout_context.request_body.messages; let mut arch_assistant = false; + let mut user_messages = Vec::new(); if messages.len() >= 2 { let latest_assistant_message = &messages[messages.len() - 2]; @@ -739,11 +740,8 @@ impl StreamContext { arch_assistant = true; } } - } else { - info!("no assistant message found, probably first interaction"); - } - - if arch_assistant { + } + if arch_assistant { for message in messages.iter() { if let Some(model) = message.model.as_ref() { if !model.starts_with(ARCH_MODEL_PREFIX) {