Neda1 commited on
Commit
d484a74
Β·
verified Β·
1 Parent(s): 40bc56e

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +10 -3
agent.py CHANGED
@@ -122,9 +122,16 @@ sys_msg = SystemMessage(content=system_prompt)
122
 
123
  # build a retriever
124
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2") # dim=768
125
- supabase: Client = create_client(
126
- os.environ.get("SUPABASE_URL"),
127
- os.environ.get("SUPABASE_SERVICE_KEY"))
 
 
 
 
 
 
 
128
  vector_store = SupabaseVectorStore(
129
  client=supabase,
130
  embedding= embeddings,
 
122
 
123
  # build a retriever
124
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2") # dim=768
125
+ # supabase: Client = create_client(
126
+ # os.environ.get("SUPABASE_URL"),
127
+ # os.environ.get("SUPABASE_SERVICE_KEY"))
128
+ supabase_url = os.getenv("SUPABASE_URL")
129
+ supabase_key = os.getenv("SUPABASE_KEY")
130
+
131
+ if not supabase_url or not supabase_key:
132
+ raise ValueError("SUPABASE_URL and SUPABASE_KEY must be set in environment variables.")
133
+
134
+ supabase: Client = create_client(supabase_url, supabase_key)
135
  vector_store = SupabaseVectorStore(
136
  client=supabase,
137
  embedding= embeddings,