Update agent.py

agent.py

@@ -9,12 +9,12 @@ from langchain_core.messages import AnyMessage, HumanMessage
 from langchain_groq import ChatGroq
 from typing import TypedDict, Annotated
 
-# Load environment variables
+# Load API keys
 load_dotenv()
 groq_api_key = os.getenv("GROQ_API_KEY")
 serpapi_api_key = os.getenv("SERPAPI_API_KEY")
 
-# --- أدوات مساعدة ---
+# --- أدوات ---
 
 def calculator_tool_func(query: str):
     try:
@@ -45,33 +45,31 @@ SerpAPI_tool = Tool(
     description="Searches the web for recent information."
 )
 
-# --- بناء الوكيل ---
+# --- تعريف موديل الذكاء الاصطناعي والوكيل ---
 
-def create_search_only_agent():
-    tools = [SerpAPI_tool, calculator_tool, weather_tool]
-    llm = ChatGroq(model="deepseek-r1-distill-llama-70b", groq_api_key=groq_api_key)
-    llm_with_tools = llm.bind_tools(tools)
+tools = [SerpAPI_tool, calculator_tool, weather_tool]
+llm = ChatGroq(model="deepseek-r1-distill-llama-70b", groq_api_key=groq_api_key)
+llm_with_tools = llm.bind_tools(tools)
 
-    class AgentState(TypedDict):
-        messages: Annotated[list[AnyMessage], add_messages]
+class AgentState(TypedDict):
+    messages: Annotated[list[AnyMessage], add_messages]
 
-    def assistant(state: AgentState):
-        return {"messages": [llm_with_tools.invoke(state["messages"])]}
+def assistant(state: AgentState):
+    return {"messages": [llm_with_tools.invoke(state["messages"])]}
 
-    builder = StateGraph(AgentState)
-    builder.add_node("assistant", assistant)
-    builder.add_node("tools", ToolNode(tools))
-    builder.add_edge(START, "assistant")
-    builder.add_conditional_edges("assistant", tools_condition)
-    builder.add_edge("tools", "assistant")
+builder = StateGraph(AgentState)
+builder.add_node("assistant", assistant)
+builder.add_node("tools", ToolNode(tools))
+builder.add_edge(START, "assistant")
+builder.add_conditional_edges("assistant", tools_condition)
+builder.add_edge("tools", "assistant")
 
-    return builder.compile()
+# 👇 وكيل باسم `ninu` يمكن استيراده من أي ملف آخر
+ninu = builder.compile()
 
-# --- دالة التفاعل ---
-
-def run_search_agent(query):
-    agent = create_search_only_agent()
+# --- دالة تشغيل مستقلة (اختياري) ---
 
+def run_ninu(query: str):
     conversation = []
 
     intro_prompt = """
@@ -87,11 +85,5 @@ Only include what the user asked for in the final answer.
     conversation.append(HumanMessage(content=intro_prompt))
     conversation.append(HumanMessage(content=query))
 
-    response = agent.invoke({"messages": conversation})
+    response = ninu.invoke({"messages": conversation})
     return response["messages"][-1].content
-
-# يمكن استخدام الدالة التالية لاختبار الوكيل
-if __name__ == "__main__":
-    question = "ما هي عاصمة اليابان وكم عدد سكانها؟"
-    answer = run_search_agent(question)
-    print(answer)