Browse Source

Fix: tavily search error. (#5653)

### What problem does this PR solve?

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
tags/v0.17.1
Kevin Hu 8 months ago
parent
commit
e5041749a2
No account linked to committer's email address
2 changed files with 3 additions and 2 deletions
  1. 1
    1
      agentic_reasoning/deep_research.py
  2. 2
    1
      rag/utils/tavily_conn.py

+ 1
- 1
agentic_reasoning/deep_research.py View File

@@ -118,7 +118,7 @@ class DeepResearcher:

if self.prompt_config.get("tavily_api_key"):
tav = Tavily(self.prompt_config["tavily_api_key"])
tav_res = tav.retrieve_chunks(" ".join(search_query))
tav_res = tav.retrieve_chunks(search_query)
kbinfos["chunks"].extend(tav_res["chunks"])
kbinfos["doc_aggs"].extend(tav_res["doc_aggs"])
if self.prompt_config.get("use_kg") and self._kg_retrieve:

+ 2
- 1
rag/utils/tavily_conn.py View File

@@ -38,6 +38,7 @@ class Tavily:
def retrieve_chunks(self, question):
chunks = []
aggs = []
logging.info("[Tavily]Q: " + question)
for r in self.search(question):
id = get_uuid()
chunks.append({
@@ -62,5 +63,5 @@ class Tavily:
"count": 1,
"url": r["url"]
})
logging.info("[Tavily]: "+r["content"][:128]+"...")
logging.info("[Tavily]R: "+r["content"][:128]+"...")
return {"chunks": chunks, "doc_aggs": aggs}

Loading…
Cancel
Save