update test

agent.py

@@ -107,6 +107,24 @@ def arvix_search(query: str) -> str:
     return {"arvix_results": formatted_search_docs}
 
 
+def test_supabase_connection():
+    load_dotenv()
+    
+    try:
+        supabase = create_client(
+            os.environ.get("SUPABASE_URL"),
+            os.environ.get("SUPABASE_SERVICE_KEY")
+        )
+        
+        # Test query
+        result = supabase.table('documents').select("*").limit(1).execute()
+        print("Connection successful!")
+        return True
+        
+    except Exception as e:
+        print(f"Connection failed: {e}")
+        return False
+
 
 # load the system prompt from the file
 with open("system_prompt.txt", "r", encoding="utf-8") as f:
@@ -132,7 +150,7 @@ create_retriever_tool = create_retriever_tool(
     description="A tool to retrieve similar questions from a vector store.",
 )
 
-
+test_supabase_connection()
 
 tools = [
     multiply,
@@ -196,4 +214,4 @@ def build_graph(provider: str = "google"):
     builder.add_edge("tools", "assistant")
 
     # Compile graph
-    return builder.compile()
+    return builder.compile()

app.py

@@ -25,28 +25,26 @@ class BasicAgent:
 
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
+        # Wrap the question in a HumanMessage from langchain_core
         messages = [HumanMessage(content=question)]
         print(f"messages: {messages}")
-        
         try:
             messages = self.graph.invoke({"messages": messages})
             print(f"Returned from graph.invoke(): {messages}")
-            
-            # Add validation
-            if not messages or 'messages' not in messages:
-                return "Error: No response received from the agent"
-                
-            message_list = messages['messages']
-            if not message_list:
-                return "Error: Empty response from the agent"
-                
-            answer = message_list[-1].content
-            # Only slice if the answer is long enough
-            return answer[14:] if len(answer) > 14 else answer
-            
         except Exception as e:
             print(f"Exception during graph.invoke: {e}")
-            return f"Error processing question: {str(e)}"
+            raise
+
+        # Add validation
+        if not messages or 'messages' not in messages:
+            print(f"Error: No response received from the agent")
+            
+        message_list = messages['messages']
+        if not message_list:
+            print(f"Error: Empty response from the agent")
+        answer = messages['messages'][-1].content
+        return answer[14:]
+
 
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):