Skip to content

Commit 30ca963

Browse files
fix integration tests (#77)
1 parent 888f209 commit 30ca963

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

tests_integration/simple_llm_assistant/openrouter_tool_example.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ async def test_openrouter_tool_stream() -> None:
5151
# invoke with custom chat params #
5252
# --------------------------------------------------------------------------- #
5353
async def test_openrouter_tool_with_chat_param() -> None:
54-
chat_param = {"temperature": 0.1, "max_tokens": 15}
54+
chat_param = {"temperature": 0.1, "max_tokens": 120}
5555

5656
await event_store.clear_events()
5757
or_tool = OpenRouterTool.builder().api_key(api_key).chat_params(chat_param).build()
@@ -61,11 +61,11 @@ async def test_openrouter_tool_with_chat_param() -> None:
6161
[Message(role="user", content="Hello, my name is Grafi, how are you doing?")],
6262
):
6363
for m in msgs:
64+
print(m)
6465
assert m.role == "assistant"
6566
assert m.content and "Grafi" in m.content
66-
print(m.content)
6767
if isinstance(m.content, str):
68-
assert len(m.content) < 70 # 15 tokens ≈ < 70 chars
68+
assert len(m.content) < 300
6969

7070
assert len(await event_store.get_events()) == 2
7171

0 commit comments

Comments
 (0)