lam-ho commited on
Commit
250ec74
·
verified ·
1 Parent(s): 6906d29

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -204,9 +204,9 @@ class ArvixSearchTool(Tool):
204
  condensed_docs = "\n\n---\n\n".join(
205
  [
206
  f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content[:1000]}\n</Document>'
207
- for doc in search_docs
208
  ])
209
- return {"arvix_search_results": formatted_search_docs}
210
 
211
 
212
  tools=[
 
204
  condensed_docs = "\n\n---\n\n".join(
205
  [
206
  f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content[:1000]}\n</Document>'
207
+ for doc in documents
208
  ])
209
+ return {"arvix_search_results": condensed_docs}
210
 
211
 
212
  tools=[