Update app.py

app.py

@@ -11,6 +11,10 @@ from llama_index.llms.gemini import Gemini
 from llama_index.core.memory import ChatMemoryBuffer
 from llama_index.readers.web import FireCrawlWebReader
 from fastapi.middleware.cors import CORSMiddleware
+from llama_index.core.agent import ReActAgent
+from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
+from llama_index.core.llms import ChatMessage
+
 
 dotenv.load_dotenv()
 
@@ -125,6 +129,28 @@ async def query(request: QueryRequest):
 
     return {"response": response.response}
 
+
+@app.post("/agent-search/")
+async def agent_search(request: QueryRequest):
+    code_tools=DuckDuckGoSearchToolSpec()
+    tool = code_tools.to_tool_list()
+    llm = Gemini(api_key=os.environ["GOOGLE_API_KEY"], model="models/gemini-1.5-pro")
+    prefix_messages = [
+        ChatMessage(
+            role="system",
+            content=(
+                "You are now a integration agent, and what  ever you are requested, you will try to execute utilizing your toools."
+            ),
+        )
+    ]
+    agent = ReActAgent.from_tools(tool, llm=llm, verbose=True)
+    response = agent.chat(request.query)
+    state['chat_history'].append(("User", request.query))
+    state['chat_history'].append(("Assistant", str(response.response)))
+
+    return {"response": response.response}
+
+
 @app.get("/chat-history/")
 async def get_chat_history():
     return {"chat_history": state['chat_history']}