Spaces:
Running
Running
Update agent.py
Browse files
agent.py
CHANGED
@@ -135,7 +135,7 @@ supabase: Client = create_client(
|
|
135 |
vector_store = SupabaseVectorStore(
|
136 |
client=supabase,
|
137 |
embedding= embeddings,
|
138 |
-
table_name="
|
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="
|
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(
|