Typing

Cargo.toml

@@ -9,22 +9,22 @@ default-members = [
 ]
 
 [workspace.package]
-version = "0.10.0"
+version = "0.11.0"
 authors = ["demml <[email protected]>"]
 edition = "2021"
 license = "MIT"
 repository = "https://github.com/demml/potatohead"
 
 [workspace.dependencies]
-baked-potato = { path = "crates/baked_potato", version = "0.10.0" }
-potato-agent = { path = "crates/potato_agent", version = "0.10.0" }
-potato-provider = { path = "crates/potato_provider", version = "0.10.0" }
-potatohead-macro = { path = "crates/potato_macro", version = "0.10.0" }
-potato-type = { path = "crates/potato_type", version = "0.10.0" }
-potato-workflow = { path = "crates/potato_workflow", version = "0.10.0" }
-potato-util = { path = "crates/potato_util", version = "0.10.0" }
-potato-head = { path = "crates/potato_head", version = "0.10.0" }
-potato-state = { path = "crates/potato_state", version = "0.10.0" }
+baked-potato = { path = "crates/baked_potato", version = "0.11.0" }
+potato-agent = { path = "crates/potato_agent", version = "0.11.0" }
+potato-provider = { path = "crates/potato_provider", version = "0.11.0" }
+potatohead-macro = { path = "crates/potato_macro", version = "0.11.0" }
+potato-type = { path = "crates/potato_type", version = "0.11.0" }
+potato-workflow = { path = "crates/potato_workflow", version = "0.11.0" }
+potato-util = { path = "crates/potato_util", version = "0.11.0" }
+potato-head = { path = "crates/potato_head", version = "0.11.0" }
+potato-state = { path = "crates/potato_state", version = "0.11.0" }
 
 anyhow = "1.0.93"
 base64 = "0.*"

py-potato/pyproject.toml

@@ -6,7 +6,7 @@ classifiers = [
     "Programming Language :: Python :: Implementation :: CPython",
     "Programming Language :: Python :: Implementation :: PyPy",
 ]
-version = "0.10.0"
+version = "0.11.0"
 description = ""
 authors = [
     {name = "Thorrester", email = "<[email protected]>"},

py-potato/python/potato_head/_potato_head.pyi

@@ -2,7 +2,7 @@
 
 import datetime
 from pathlib import Path
-from typing import Any, Dict, Generic, List, Optional, TypeAlias, Union, overload
+from typing import Any, Dict, Generic, List, Optional, Type, TypeAlias, Union, overload
 
 from typing_extensions import TypeVar
 
@@ -227,7 +227,7 @@ class Prompt(Generic[OutputType]):
         provider: Provider | str,
         system_instructions: Optional[PromptMessage] = None,
         model_settings: Optional[ModelSettings | OpenAIChatSettings | GeminiSettings | AnthropicSettings] = None,
-        output_type: Optional[OutputType] = None,
+        output_type: Optional[Type[OutputType]] = None,
     ) -> None:
         """Initialize a Prompt object.