Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -222,12 +222,12 @@ Otherwise, answer directly."""
|
|
222 |
if "SEARCH_RESEARCH:" in response_text:
|
223 |
query = response_text.split("SEARCH_RESEARCH:")[1].strip()
|
224 |
results = chroma_manager.research_collection.as_retriever().invoke(query)
|
225 |
-
return {"messages": [AIMessage(content=f'Action: research_db_tool\n{{"query": "{query}"}}\n\nResults: {str(results)}')]
|
226 |
|
227 |
elif "SEARCH_DEV:" in response_text:
|
228 |
query = response_text.split("SEARCH_DEV:")[1].strip()
|
229 |
results = chroma_manager.dev_collection.as_retriever().invoke(query)
|
230 |
-
return {"messages": [AIMessage(content=f'Action: development_db_tool\n{{"query": "{query}"}}\n\nResults: {str(results)}')]
|
231 |
|
232 |
return {"messages": [AIMessage(content=response_text)]}
|
233 |
except Exception as e:
|
|
|
222 |
if "SEARCH_RESEARCH:" in response_text:
|
223 |
query = response_text.split("SEARCH_RESEARCH:")[1].strip()
|
224 |
results = chroma_manager.research_collection.as_retriever().invoke(query)
|
225 |
+
return {"messages": [AIMessage(content=f'Action: research_db_tool\n{{"query": "{query}"}}\n\nResults: {str(results)}')]}
|
226 |
|
227 |
elif "SEARCH_DEV:" in response_text:
|
228 |
query = response_text.split("SEARCH_DEV:")[1].strip()
|
229 |
results = chroma_manager.dev_collection.as_retriever().invoke(query)
|
230 |
+
return {"messages": [AIMessage(content=f'Action: development_db_tool\n{{"query": "{query}"}}\n\nResults: {str(results)}')]}
|
231 |
|
232 |
return {"messages": [AIMessage(content=response_text)]}
|
233 |
except Exception as e:
|