Update agent.py

agent.py

@@ -122,17 +122,9 @@ sys_msg = SystemMessage(content=system_prompt)
 
 # build a retriever
 embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2") #  dim=768
-# supabase: Client = create_client(
-#     os.environ.get("SUPABASE_URL"), 
-#     os.environ.get("SUPABASE_SERVICE_KEY"))
-
-supabase_url = os.getenv("SUPABASE_URL")
-supabase_key = os.getenv("SUPABASE_KEY")
-
-if not supabase_url or not supabase_key:
-    raise ValueError("SUPABASE_URL and SUPABASE_KEY must be set in environment variables.")
-
-supabase: Client = create_client(supabase_url, supabase_key)
+supabase: Client = create_client(
+    os.environ.get("SUPABASE_URL"), 
+    os.environ.get("SUPABASE_SERVICE_KEY"))
 vector_store = SupabaseVectorStore(
     client=supabase,
     embedding= embeddings,
@@ -218,4 +210,4 @@ if __name__ == "__main__":
     messages = [HumanMessage(content=question)]
     messages = graph.invoke({"messages": messages})
     for m in messages["messages"]:
-        m.pretty_print()
+        m.pretty_print()