fix literature search

environment.yaml

@@ -11,7 +11,7 @@ dependencies:
   - pytest
   - typer
   - pip:
-      - litellm
+      - litellm>=1.40.6
       - jupyter
       - numpydoc
       - semanticscholar>=0.8.1

just_agents/llm_session.py

@@ -208,7 +208,6 @@ def _process_function_calls(self, response: ModelResponse) -> Optional[ModelResp
             message = self.message_from_response(response)
             self.memory.add_message(message)
             for tool_call in tool_calls:
-                print(f"Calling function {function_name}({function_args})")
                 function_name = tool_call.function.name
                 function_to_call = self.available_tools[function_name]
                 function_args = json.loads(tool_call.function.arguments)

just_agents/tools/search.py

@@ -119,4 +119,4 @@ def literature_search(query: str, limit: int = 20):
     :param limit: limit number of results
     :return: extracts from scientific papers where SOURCE field is the id of the paper
     """
-    hybrid_search(text=query, limit=limit)
+    return hybrid_search(text=query, limit=limit)