Skip to content

Commit 0f9147c

Browse files
committed
fix tests
Signed-off-by: Akihiko Kuroda <akihikokuroda2020@gmail.com>
1 parent 092b02b commit 0f9147c

3 files changed

Lines changed: 3 additions & 3 deletions

File tree

docs/examples/agents/react/react_using_mellea.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
# Simple tool for searching. Requires the langchain-community package.
1818
# Mellea allows you to interop with langchain defined tools.
1919
lc_ddg_search = DuckDuckGoSearchResults(output_format="list")
20-
search_tool = MelleaTool.from_langchain(lc_ddg_search)
20+
search_tool: MelleaTool = MelleaTool.from_langchain(lc_ddg_search)
2121

2222

2323
class Email(pydantic.BaseModel):

docs/examples/tools/smolagents_example.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
python_tool_hf = PythonInterpreterTool()
2424

2525
# Convert to Mellea tool - now you can use it with Mellea!
26-
python_tool = MelleaTool.from_smolagents(python_tool_hf)
26+
python_tool: MelleaTool = MelleaTool.from_smolagents(python_tool_hf)
2727

2828
# Use with Mellea session
2929
m = start_session()

test/backends/test_mellea_tool.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,7 @@ def test_from_langchain_args_handling(caplog):
127127
@pytest.mark.ollama
128128
@pytest.mark.e2e
129129
def test_from_langchain_generation(session: MelleaSession):
130-
t = MelleaTool.from_langchain(langchain_tool)
130+
t: MelleaTool = MelleaTool.from_langchain(langchain_tool)
131131

132132
out = session.instruct(
133133
"Call the langchain_tool.",

0 commit comments

Comments
 (0)