Update LG_agent.py

LG_agent.py

@@ -16,7 +16,21 @@ chat = ChatOpenAI(
     model="gpt-3.5-turbo",
     openai_api_key=OPENAI_API_KEY,
     temperature=0,
+    model_kwargs={
+        "messages": [
+            {
+                "role": "system",
+                "content": (
+                    "You are a step-by-step reasoning assistant. "
+                    "Always think out loud, use available tools when appropriate, and aim for factual, complete answers. "
+                    "If you use a tool like search_wikipedia and get a long text response that might contain numbers, "
+                    "use extract_number_from_text immediately after to summarize the relevant count or value."
+                )
+            }
+        ]
+    }
 )
+
 chat_with_tools = chat.bind_tools(all_tools)
 
 # 2. Define the agent state