We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0d7e419 commit fd6656dCopy full SHA for fd6656d
pydantic_ai_slim/pydantic_ai/durable_exec/restate/_model.py
@@ -7,11 +7,13 @@
7
from pydantic_ai.models import Model
8
from pydantic_ai.models.wrapper import WrapperModel
9
10
+MODEL_RESPONSE_SERDE = PydanticTypeAdapter(ModelResponse)
11
+
12
13
class RestateModelWrapper(WrapperModel):
14
def __init__(self, wrapped: Model, context: Context, max_attempts: int | None = None):
15
super().__init__(wrapped)
- self.options = RunOptions[ModelResponse](serde=PydanticTypeAdapter(ModelResponse), max_attempts=max_attempts)
16
+ self.options = RunOptions(serde=MODEL_RESPONSE_SERDE, max_attempts=max_attempts)
17
self.context = context
18
19
async def request(self, *args: Any, **kwargs: Any) -> ModelResponse:
0 commit comments