ktluege commited on
Commit
3269d51
Β·
verified Β·
1 Parent(s): 41cd066

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +13 -4
agent.py CHANGED
@@ -162,13 +162,22 @@ tools = [
162
  with open("system_prompt.txt", "r", encoding="utf-8") as f:
163
  sys_msg = SystemMessage(content=f.read())
164
 
 
 
 
 
 
 
 
 
 
165
  # vectorstore setup (Supabase)
166
- emb = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
167
- sup = create_client(os.getenv("SUPABASE_URL"), os.getenv("SUPABASE_SERVICE_ROLE_KEY"))
168
  vector_store = SupabaseVectorStore(
169
- client=sup, embedding=emb,
170
- table_name=os.getenv("VECTORTABLE_NAME", "documents2"),
 
171
  query_name=os.getenv("VECTOR_QUERY_NAME","match_documents_langchain")
 
172
 
173
  def build_graph():
174
  """Build the LangGraph agent using OpenAI ChatGPT only."""
 
162
  with open("system_prompt.txt", "r", encoding="utf-8") as f:
163
  sys_msg = SystemMessage(content=f.read())
164
 
165
+
166
+ emb = HuggingFaceEmbeddings(
167
+ model_name="sentence-transformers/all-mpnet-base-v2"
168
+ )
169
+
170
+ sup = create_client(
171
+ os.getenv("SUPABASE_URL"),
172
+ os.getenv("SUPABASE_SERVICE_ROLE_KEY")
173
+ )
174
  # vectorstore setup (Supabase)
 
 
175
  vector_store = SupabaseVectorStore(
176
+ client=sup,
177
+ embedding=emb,
178
+ table_name=os.getenv("VECTORTABLE_NAME","documents2"),
179
  query_name=os.getenv("VECTOR_QUERY_NAME","match_documents_langchain")
180
+ )
181
 
182
  def build_graph():
183
  """Build the LangGraph agent using OpenAI ChatGPT only."""