wt002 commited on
Commit
260b607
·
verified ·
1 Parent(s): a1fdb15

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -135,7 +135,7 @@ supabase: Client = create_client(
135
  vector_store = SupabaseVectorStore(
136
  client=supabase,
137
  embedding= embeddings,
138
- table_name="documents",
139
  query_name="match_documents_langchain",
140
  )
141
  create_retriever_tool = create_retriever_tool(
@@ -165,7 +165,7 @@ def build_graph(provider: str = "google"):
165
  llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
166
  elif provider == "groq":
167
  # Groq https://console.groq.com/docs/models
168
- llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
169
  elif provider == "huggingface":
170
  # TODO: Add huggingface endpoint
171
  llm = ChatHuggingFace(
 
135
  vector_store = SupabaseVectorStore(
136
  client=supabase,
137
  embedding= embeddings,
138
+ table_name="docs",
139
  query_name="match_documents_langchain",
140
  )
141
  create_retriever_tool = create_retriever_tool(
 
165
  llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
166
  elif provider == "groq":
167
  # Groq https://console.groq.com/docs/models
168
+ llm = ChatGroq(model="gemma2-9b-it", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
169
  elif provider == "huggingface":
170
  # TODO: Add huggingface endpoint
171
  llm = ChatHuggingFace(