mgbam commited on
Commit
9742255
·
verified ·
1 Parent(s): d94f105

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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: