diff --git a/agentic_reasoning/deep_research.py b/agentic_reasoning/deep_research.py index f61c3a881..cc5fdb91d 100644 --- a/agentic_reasoning/deep_research.py +++ b/agentic_reasoning/deep_research.py @@ -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: diff --git a/rag/utils/tavily_conn.py b/rag/utils/tavily_conn.py index c5320a9fb..7d78636e0 100644 --- a/rag/utils/tavily_conn.py +++ b/rag/utils/tavily_conn.py @@ -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} \ No newline at end of file