Include manual testing

.env.sample

@@ -0,0 +1,7 @@
+# Required API Keys
+GROQ_API_KEY=your-groq-api-key-here
+TAVILY_API_KEY=your-tavily-api-key-here
+
+# Optional Environment Variables
+PYTHONPATH=.
+DEBUG=false

agent.py

@@ -1,6 +1,7 @@
 import os
-from typing import Dict
+from typing import Dict, List, Sequence, TypedDict, cast
 
+from dotenv import load_dotenv
 from langchain_community.document_loaders import WikipediaLoader
 from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
 from langchain_core.tools import tool
@@ -128,9 +129,22 @@ def build_agent_graph(provider: str = "groq"):
     llm_with_tools = llm.bind_tools(tools)
 
     # Create nodes
-    def assistant(state: MessagesState):
+    def assistant(state: MessagesState) -> Dict[str, List[AIMessage]]:
         """Assistant node"""
-        return {"messages": [llm_with_tools.invoke(state["messages"])]}
+        try:
+            # Get last message
+            messages = state.get("messages", [])
+            if not messages:
+                return {"messages": [AIMessage(content="Error: No messages found")]}
+
+            # Run LLM and ensure AIMessage response
+            response = llm_with_tools.invoke(messages)
+            if isinstance(response, AIMessage):
+                return {"messages": [response]}
+            return {"messages": [AIMessage(content=str(response))]}
+
+        except Exception as e:
+            return {"messages": [AIMessage(content=f"Error: {str(e)}")]}
 
     # Build graph
     builder = StateGraph(MessagesState)
@@ -143,3 +157,63 @@ def build_agent_graph(provider: str = "groq"):
     builder.add_edge("assistant", END)
 
     return builder.compile()
+
+
+# Manual test function
+def test_agent():
+    """Run a manual test of the agent"""
+    # Load environment variables from .env file
+    load_dotenv()
+    print("\n" + "=" * 50)
+    print("Starting Agent Test")
+    print("=" * 50)
+
+    # Check environment variables
+    if not os.getenv("GROQ_API_KEY"):
+        print("\nError: GROQ_API_KEY not set")
+        return
+    if not os.getenv("TAVILY_API_KEY"):
+        print("\nWarning: TAVILY_API_KEY not set - web search will be unavailable")
+
+    print("\nInitializing agent...")
+    try:
+        graph = build_agent_graph()
+        print("Agent initialized successfully")
+    except Exception as e:
+        print(f"Failed to initialize agent: {str(e)}")
+        return
+
+    # Test a single question
+    question = "What is the first name of the only Malko Competition recipient from the 20th Century (after 1977) whose nationality on record is a country that no longer exists?"
+    print("\nTesting question:", question)
+    print("-" * 50)
+
+    try:
+        # Create messages state
+        messages = [system_prompt, HumanMessage(content=question)]
+
+        # Run agent
+        print("\nWaiting for response...")
+        result = graph.invoke({"messages": messages})
+
+        # Get answer
+        if result and "messages" in result and result["messages"]:
+            answer = result["messages"][-1].content
+            print("\nResponse received:")
+            print("-" * 20)
+            print(answer)
+            print("-" * 20)
+        else:
+            print("\nError: No response from agent")
+
+    except Exception as e:
+        print(f"\nError processing question: {str(e)}")
+
+    print("\n" + "=" * 50)
+    print("Test Complete")
+    print("=" * 50 + "\n")
+
+
+# Run test if script is run directly
+if __name__ == "__main__":
+    test_agent()

requirements.txt

@@ -1,15 +1,13 @@
-gradio>=4.0.0
-requests>=2.31.0
-langchain>=0.1.0
-langchain-core>=0.1.0
-langchain-community>=0.0.10
-langchain_huggingface>=0.0.10
-langchain-groq>=0.0.5
-langchain-experimental>=0.0.40
-langchain-tavily>=0.0.5
-langgraph>=0.0.15
-tavily-python>=0.3.0
-wikipedia>=1.4.0
+langchain
+langchain-core
+langchain-community
+langchain-groq
+langchain-tavily
+langgraph
+python-dotenv
+wikipedia
+requests
+pydantic
+lxml
 youtube-transcript-api==0.6.3
-pytube>=15.0.0
-pydantic>=2.0.0
+pytube>=15.0.0