diff --git a/knowledge_storm/rm.py b/knowledge_storm/rm.py index 563116fe..effbe9ff 100644 --- a/knowledge_storm/rm.py +++ b/knowledge_storm/rm.py @@ -938,11 +938,12 @@ def forward( for query in queries: args = { + "query": query, "max_results": self.k, "include_raw_contents": self.include_raw_content, } # list of dicts that will be parsed to return - responseData = self.tavily_client.search(query) + responseData = self.tavily_client.search(**args) results = responseData.get("results") for d in results: # assert d is dict