Update agent.py

agent.py

@@ -16,6 +16,18 @@ from langchain_core.tools import tool
 from langchain.tools.retriever import create_retriever_tool
 from supabase.client import Client, create_client
 
+class BasicAgent:
+    def __init__(self, provider="groq"):
+        self.graph = build_graph(provider=provider)
+    
+    def run(self, question: str) -> str:
+        messages = [HumanMessage(content=question)]
+        response = self.graph.invoke({"messages": messages})
+        # Return the last message from the assistant node (the answer)
+        return response["messages"][-1].content
+
+
+
 load_dotenv()
 
 @tool