move call to end of routine

1 files changed, 18 insertions, 1 deletions

diff --git a/src/main.rs b/src/main.rs
index 12cb491..a9ff484 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -54,6 +54,19 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
             parameters: json!({}),
         },
     });
+    tools.push(Tool {
+        tool_type: "function".into(),
+        function: Function {
+            name: "wikipedia_lookup".into(),
+            description: Some("Look up a wikipedia article and have its summary returned.".into()),
+            parameters: json!({
+                "title": {
+                    "type": "string",
+                    "description": "The title of the article to look up."
+                }
+            }),
+        },
+    });
     eprintln!("Tools Loaded");
     let date: String = Utc::now().date_naive().to_string();
     let system_prompt: String = format!("The current date is {date}.  {}", &config.system_prompt);
@@ -86,6 +99,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
             if let Some(reason) = choice.finish_reason {
                 match reason.as_str() {
                     "tool_calls" => {
+                        let mut need_call = false;
                         if let Some(calls) = choice.message.clone().tool_calls {
                             for call in calls {
                                 match call.function.name.as_str() {
@@ -95,7 +109,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
                                         let v = v["sign"].as_str().unwrap_or_default();
                                         let val: String = get_horoscope(v).await;
                                         req.messages.push(Message::tool(val, call.id));
-                                        response = client.chat(&req).await?;
+                                        need_call = true;
                                     },
                                     "stop" => {
                                         println!("Agent has stopped the conversation.");
@@ -109,6 +123,9 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
                                     _ => (),
                                 }
                             }
+                            if need_call {
+                                response = client.chat(&req).await?;
+                            }
                         }
                     },
                     _ => (),