Update main.py

main.py

@@ -30,5 +30,14 @@ async def generate_response(request: PromptRequest):
         temperature=request.temperature,
         base_url="http://localhost:11434"
     )
-    response = llm.invoke(request.prompt)
+    tool_prompt = f"System Role: You are an autonomous AI Agent with real-time internet access. Current Date: {{today_date}} TOOL_DEFINITION: - Name: Search_tool - Activation Command: Search [Your Query Here]"
+    response = llm.invoke(f'{{tool_prompt}}, User-Query:-{{request.prompt}}')
+
+    if "Search " in response:
+        new_query = response.removeprefix("Search ")
+        search_response = search_tool(query=new_query)
+        new_response = llm.invoke(f"Extra_information:- {{search_response}} User-Query:- {{request.prompt}}")
+        return {"response": new_response}
+    else:
+        return {"response": response}