diff --git a/environment.yaml b/environment.yaml index d66ab52..dc23260 100644 --- a/environment.yaml +++ b/environment.yaml @@ -11,7 +11,7 @@ dependencies: - pytest - typer - pip: - - litellm + - litellm>=1.40.6 - jupyter - numpydoc - semanticscholar>=0.8.1 diff --git a/just_agents/tools/search.py b/just_agents/tools/search.py index 5d6d46f..893c102 100644 --- a/just_agents/tools/search.py +++ b/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)