handle no search return

agent.py

@@ -195,7 +195,15 @@ def build_graph(provider: str = "google"):
 
     def retriever(state: MessagesState):
         """Retriever node"""
-        similar_question = vector_store.similarity_search(state["messages"][0].content)
+        try:
+            # Use the vector store to find similar questions
+            similar_question = vector_store.similarity_search(state["messages"][0].content)
+            if not similar_question:
+                raise ValueError("No similar questions found.")
+        except Exception as e:
+            print(f"Error occurred while searching for similar questions: {e}")
+            return {"messages": [sys_msg] + state["messages"]}
+
         example_msg = HumanMessage(
             content=f"Here I provide a similar question and answer for reference: \n\n{similar_question[0].page_content}",
         )

app.py

@@ -211,7 +211,8 @@ def test_supabase_connection():
         
         # Test query
         result = supabase.table('documents').select("*").limit(1).execute()
-        print(f"Connection successful! Retrieved {len(result.data)} documents.")
+        print(f"✅ Supabase Connection successful!")
+        print(f"   Retrieved {len(result.data)} documents.")
         return True
         
     except Exception as e: