From 4961c13cc314b3be225d8d476a807d632f8faf12 Mon Sep 17 00:00:00 2001
From: Koudai Aono <koudai@mirascope.io>
Date: Thu, 12 Dec 2024 00:30:25 +0900
Subject: [PATCH] fix: fix stream with response_model

---
 mirascope/core/base/stream.py  | 2 +-
 tests/core/base/test_stream.py | 1 -
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/mirascope/core/base/stream.py b/mirascope/core/base/stream.py
index 46fc53155..bb20d70cb 100644
--- 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
index b8d8e23d3..00ff3c50d 100644
--- 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,
     }