Update agent.py

agent.py

@@ -1,58 +1,48 @@
 import os
-from langchain.agents import initialize_agent, AgentType, Tool
-from langchain_huggingface import HuggingFaceEndpoint
+from langchain_google_genai import ChatGoogleGenerativeAI
+from langchain.agents import initialize_agent, Tool, AgentType
 from langchain_community.tools import DuckDuckGoSearchResults
-from langchain_experimental.tools import PythonREPLTool
-from huggingface_hub import login
-
-# LLM: Mistral-7B-Instruct über Hugging Face Inference API
-llm = HuggingFaceEndpoint(
-    repo_id="mistralai/Mistral-7B-Instruct-v0.2",
-    temperature=0.2,
-    max_new_tokens=512,
+from langchain.memory import ConversationBufferMemory
+from langchain_core.messages import SystemMessage
+
+# API Key automatisch aus Environment ziehen
+google_api_key = os.getenv("GOOGLE_API_KEY")
+
+# LLM: Gemini 2.0 Flash
+llm = ChatGoogleGenerativeAI(
+    model="gemini-2.0-flash",
+    google_api_key=google_api_key,
+    temperature=0,
+    max_output_tokens=2048,
+    system_message=SystemMessage(content=(
+        "You are a highly accurate AI assistant. "
+        "You must answer precisely, concisely, and only if you are confident. "
+        "Use the available tools like Web Search if needed. "
+        "Always prefer exact information over assumptions."
+    ))
 )
 
-# Tools definieren
+# Tools: DuckDuckGo Web Search
 search_tool = DuckDuckGoSearchResults()
-python_tool = PythonREPLTool()
 
 tools = [
     Tool(
-        name="Search",
+        name="WebSearch",
         func=search_tool.run,
-        description="Useful for when you need to answer questions about current events or look up information online."
-    ),
-    Tool(
-        name="Python_REPL",
-        func=python_tool.run,
-        description="Useful for math, calculations, or running simple python code."
+        description="Use this to search the internet for up-to-date or unknown information."
     ),
 ]
 
-from langchain_core.prompts import SystemMessagePromptTemplate
-
-system_prompt = """You are an expert AI assistant specialized in answering exam-style factual questions. 
-Follow these guidelines:
-
-- Use the Search tool when external knowledge is needed (especially about recent events or niche topics).
-- Use the Python_REPL tool for any math calculations, even if simple.
-- Always attempt to provide a direct and concise answer without extra commentary.
-- Do not apologize or state limitations.
-- If a file is attached, explain how you would process it or the key steps to extract an answer.
-- When dates are mentioned, be very precise and double-check calculations using the appropriate tools.
-- If unsure, use the Search tool before responding.
-
-Respond directly to the user’s question based solely on facts and without unnecessary elaboration.
-Only provide what is explicitly asked for.
-"""
+# Memory (optional, kann auch weggelassen werden, falls nicht gebraucht)
+memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
 
-# Agent initialisieren
+# Agent
 agent_executor = initialize_agent(
-    tools,
-    llm,
+    tools=tools,
+    llm=llm,
     agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
     verbose=True,
+    memory=memory,
     handle_parsing_errors=True,
-    system_message=SystemMessagePromptTemplate.from_template(system_prompt),
 )