Merge pull request #739 from Mirascope/fix-stream-response_model

fix: fix stream with response_model
Mirascope · Dec 16, 2024 · 02895ae · 02895ae
2 parents df4e732 + d051993
commit 02895ae
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/mirascope/core/base/stream.py b/mirascope/core/base/stream.py
@@ -337,7 +337,7 @@ def decorator(
         json_mode: bool,
         client: _SameSyncAndAsyncClientT | _SyncBaseClientT | _AsyncBaseClientT | None,
         call_params: _BaseCallParamsT,
-        partial_tools: bool,
+        partial_tools: bool = False,
     ) -> Callable[_P, BaseStream] | Callable[_P, Awaitable[BaseStream]]:
         if not is_prompt_template(fn):
             fn = cast(

diff --git a/tests/core/base/test_stream.py b/tests/core/base/test_stream.py
@@ -18,7 +18,6 @@ def mock_stream_decorator_kwargs() -> dict:
         "json_mode": True,
         "client": MagicMock(),
         "call_params": MagicMock(),
-        "partial_tools": False,
     }