Update agent.py

agent.py

@@ -14,15 +14,26 @@ from langchain_community.vectorstores import SupabaseVectorStore
 from langchain_core.messages import SystemMessage, HumanMessage
 from langchain_core.tools import tool
 from langchain.tools.retriever import create_retriever_tool
-from supabase.client import Client, create_client
+from supabase import create_client, Client
 
-from dotenv import load_dotenv
-
-load_dotenv()  # Load environment variables from .env file
 
 supabase_url = os.getenv("SUPABASE_URL")
 supabase_key = os.getenv("SUPABASE_KEY")
 
+# Obtener credenciales de entorno (funcionará con ambos métodos)
+supabase_url = os.environ.get('SUPABASE_URL')
+supabase_key = os.environ.get('SUPABASE_KEY')
+
+if not supabase_url or not supabase_key:
+    raise ValueError("""
+    Missing Supabase credentials. Please set:
+    1. SUPABASE_URL and SUPABASE_KEY as Secrets in HF Space settings
+    OR
+    2. As environment variables if running locally
+    """)
+
+supabase: Client = create_client(supabase_url, supabase_key)
+
 @tool
 def multiply(a: int, b: int) -> int:
     """Multiply two numbers.