fix: do not consider empty message.tool_calls as valid

evilsocket · Dec 10, 2024 · 6fbe6b5 · 6fbe6b5
1 parent c8ed591
commit 6fbe6b5
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/src/agent/generator/openai.rs b/src/agent/generator/openai.rs
@@ -178,11 +178,13 @@ impl Client for OpenAIClient {
             if !comp.choices.is_empty() {
                 let first = comp.choices.first().unwrap();
                 if let Some(m) = first.message.as_ref() {
-                    if m.tool_calls.is_some() {
-                        return Ok(SupportedFeatures {
-                            system_prompt: true,
-                            tools: true,
-                        });
+                    if let Some(calls) = m.tool_calls.as_ref() {
+                        if !calls.is_empty() {
+                            return Ok(SupportedFeatures {
+                                system_prompt: true,
+                                tools: true,
+                            });
+                        }
                     }
                 }
             }