diff --git a/Cargo.lock b/Cargo.lock
index 7328eff..25acc82 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -104,7 +104,7 @@ dependencies = [
 
 [[package]]
 name = "baked-potato"
-version = "0.10.0"
+version = "0.11.0"
 dependencies = [
  "mockito",
  "potato-agent",
@@ -1343,7 +1343,7 @@ checksum = "f89776e4d69bb58bc6993e99ffa1d11f228b839984854c7daeb5d37f87cbe950"
 
 [[package]]
 name = "potato-agent"
-version = "0.10.0"
+version = "0.11.0"
 dependencies = [
  "base64",
  "gcloud-auth",
@@ -1365,7 +1365,7 @@ dependencies = [
 
 [[package]]
 name = "potato-head"
-version = "0.10.0"
+version = "0.11.0"
 dependencies = [
  "baked-potato",
  "potato-agent",
@@ -1377,7 +1377,7 @@ dependencies = [
 
 [[package]]
 name = "potato-provider"
-version = "0.10.0"
+version = "0.11.0"
 dependencies = [
  "base64",
  "gcloud-auth",
@@ -1398,7 +1398,7 @@ dependencies = [
 
 [[package]]
 name = "potato-state"
-version = "0.10.0"
+version = "0.11.0"
 dependencies = [
  "thiserror 2.0.17",
  "tokio",
@@ -1407,7 +1407,7 @@ dependencies = [
 
 [[package]]
 name = "potato-type"
-version = "0.10.0"
+version = "0.11.0"
 dependencies = [
  "potato-util",
  "potatohead-macro",
@@ -1423,7 +1423,7 @@ dependencies = [
 
 [[package]]
 name = "potato-util"
-version = "0.10.0"
+version = "0.11.0"
 dependencies = [
  "colored_json",
  "pyo3",
@@ -1438,7 +1438,7 @@ dependencies = [
 
 [[package]]
 name = "potato-workflow"
-version = "0.10.0"
+version = "0.11.0"
 dependencies = [
  "chrono",
  "potato-agent",
@@ -1457,7 +1457,7 @@ dependencies = [
 
 [[package]]
 name = "potatohead-macro"
-version = "0.10.0"
+version = "0.11.0"
 dependencies = [
  "thiserror 2.0.17",
 ]
@@ -1497,7 +1497,7 @@ dependencies = [
 
 [[package]]
 name = "py-potatohead"
-version = "0.10.0"
+version = "0.11.0"
 dependencies = [
  "potato-head",
  "pyo3",
diff --git a/Cargo.toml b/Cargo.toml
index cb8f028..53219f5 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -9,22 +9,22 @@ default-members = [
 ]
 
 [workspace.package]
-version = "0.10.0"
+version = "0.11.0"
 authors = ["demml <support@demmlai.com>"]
 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.*"
diff --git a/py-potato/pyproject.toml b/py-potato/pyproject.toml
index 32b4553..5b432d0 100644
--- a/py-potato/pyproject.toml
+++ b/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 = "<support@demmlai.com>"},
diff --git a/py-potato/python/potato_head/_potato_head.pyi b/py-potato/python/potato_head/_potato_head.pyi
index 634c792..8971ccb 100644
--- a/py-potato/python/potato_head/_potato_head.pyi
+++ b/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.
 
diff --git a/py-potato/uv.lock b/py-potato/uv.lock
index 0c12ce8..0561a76 100644
--- a/py-potato/uv.lock
+++ b/py-potato/uv.lock
@@ -2180,7 +2180,7 @@ wheels = [
 
 [[package]]
 name = "potato-head"
-version = "0.10.0"
+version = "0.11.0"
 source = { editable = "." }
 
 [package.dev-dependencies]