Update agent.py

agent.py

@@ -122,9 +122,17 @@ 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: 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)
 vector_store = SupabaseVectorStore(
     client=supabase,
     embedding= embeddings,