mirror of
https://github.com/HKUDS/nanobot.git
synced 2026-09-04 02:01:48 +03:00
refactor(providers): define typed usage contract
This commit is contained in:
@@ -412,7 +412,7 @@ class TestEphemeralDirect:
|
||||
provider.supports_tools = True
|
||||
provider.generation = MagicMock(max_tokens=4096)
|
||||
provider.chat_with_retry = AsyncMock(
|
||||
return_value=LLMResponse(content="done", tool_calls=[], finish_reason="stop", usage={})
|
||||
return_value=LLMResponse(content="done", tool_calls=[], finish_reason="stop", usage=None)
|
||||
)
|
||||
|
||||
with (
|
||||
@@ -556,9 +556,9 @@ class TestEphemeralDirect:
|
||||
"new_text": "replacement",
|
||||
},
|
||||
)],
|
||||
usage={},
|
||||
usage=None,
|
||||
),
|
||||
LLMResponse(content="done", finish_reason="stop", tool_calls=[], usage={}),
|
||||
LLMResponse(content="done", finish_reason="stop", tool_calls=[], usage=None),
|
||||
])
|
||||
|
||||
resp = await loop.process_direct(
|
||||
@@ -646,7 +646,7 @@ class TestEphemeralHooks:
|
||||
provider.generation = MagicMock(max_tokens=4096)
|
||||
provider.chat_with_retry = AsyncMock(
|
||||
return_value=LLMResponse(
|
||||
content="done", finish_reason="stop", tool_calls=[], usage={},
|
||||
content="done", finish_reason="stop", tool_calls=[], usage=None,
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user