Merge pull request #707 from Mirascope/fix-dynamic-config-message-typing-issue

Fix dynamic config message typing issue

Author: willbakst

mirascope/core/base/_utils/_setup_call.py

@@ -4,13 +4,9 @@
 from collections.abc import (
     Awaitable,
     Callable,
+    Sequence,
 )
-from typing import (
-    Any,
-    Protocol,
-    TypeVar,
-    cast,
-)
+from typing import Any, Protocol, TypeVar, cast
 
 from ..call_kwargs import BaseCallKwargs
 from ..call_params import BaseCallParams, CommonCallParams
@@ -33,17 +29,17 @@ def __call__(self, common_params: CommonCallParams) -> _BaseCallParamsT: ...
 
 def setup_call(
     fn: Callable[..., _BaseDynamicConfigT | Awaitable[_BaseDynamicConfigT]]
-    | Callable[..., list[BaseMessageParam]]
-    | Callable[..., Awaitable[list[BaseMessageParam]]],
+    | Callable[..., Sequence[BaseMessageParam]]
+    | Callable[..., Awaitable[Sequence[BaseMessageParam]]],
     fn_args: dict[str, Any],
     dynamic_config: _BaseDynamicConfigT,
-    tools: list[type[BaseTool] | Callable] | None,
+    tools: Sequence[type[BaseTool] | Callable] | None,
     tool_type: type[_BaseToolT],
     call_params: _BaseCallParamsT | CommonCallParams,
     convert_common_call_params: ConvertCommonParamsFunc[_BaseCallParamsT],
 ) -> tuple[
     str | None,
-    list[BaseMessageParam | Any],
+    Sequence[BaseMessageParam | Any],
     list[type[_BaseToolT]] | None,
     BaseCallKwargs,
 ]:

mirascope/core/base/dynamic_config.py

@@ -1,6 +1,6 @@
 """The base type in a function as an LLM call to return for dynamic configuration."""
 
-from collections.abc import Callable
+from collections.abc import Callable, Sequence
 from typing import Any, Generic, TypeVar
 
 from typing_extensions import NotRequired, TypedDict
@@ -17,11 +17,11 @@
 class DynamicConfigBase(TypedDict):
     metadata: NotRequired[Metadata]
     computed_fields: NotRequired[dict[str, Any | list[Any] | list[list[Any]]]]
-    tools: NotRequired[list[type[BaseTool] | Callable]]
+    tools: NotRequired[Sequence[type[BaseTool] | Callable]]
 
 
 class DynamicConfigMessages(DynamicConfigBase, Generic[_MessageParamT]):
-    messages: NotRequired[list[_MessageParamT]]
+    messages: NotRequired[Sequence[_MessageParamT]]
 
 
 class DynamicConfigCallParams(DynamicConfigBase, Generic[_CallParamsT]):
@@ -35,12 +35,12 @@ class DynamicConfigClient(DynamicConfigBase, Generic[_ClientT]):
 class DynamicConfigMessagesCallParams(
     DynamicConfigBase, Generic[_MessageParamT, _CallParamsT]
 ):
-    messages: NotRequired[list[_MessageParamT]]
+    messages: NotRequired[Sequence[_MessageParamT]]
     call_params: NotRequired[_CallParamsT]
 
 
 class DynamicConfigMessagesClient(DynamicConfigBase, Generic[_MessageParamT, _ClientT]):
-    messages: NotRequired[list[_MessageParamT]]
+    messages: NotRequired[Sequence[_MessageParamT]]
     client: NotRequired[_ClientT | None]
 
 
@@ -52,7 +52,7 @@ class DynamicConfigCallParamsClient(DynamicConfigBase, Generic[_CallParamsT, _Cl
 class DynamicConfigFull(
     DynamicConfigBase, Generic[_MessageParamT, _CallParamsT, _ClientT]
 ):
-    messages: NotRequired[list[_MessageParamT]]
+    messages: NotRequired[Sequence[_MessageParamT]]
     call_params: NotRequired[_CallParamsT]
     client: NotRequired[_ClientT | None]
 

mirascope/integrations/_middleware_factory.py

@@ -47,6 +47,7 @@ def default_context_manager(
 
 
 def middleware_factory(
+    *,
     custom_context_manager: Callable[
         [SyncFunc | AsyncFunc], AbstractContextManager[_T]
     ] = default_context_manager,

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mirascope"
-version = "1.9.4"
+version = "1.9.5"
 description = "LLM abstractions that aren't obstructions"
 readme = "README.md"
 license = { file = "LICENSE" }