From dc4a09dc89d61dbcfb38bf9eba2d1f5ef18f92cb Mon Sep 17 00:00:00 2001
From: vaibhavsingh-materialplus <Vaibhav.Singh@materialplus.io>
Date: Tue, 3 Mar 2026 09:07:15 +0530
Subject: [PATCH 1/4] type(model).model_fields accesses the attribute on the
 class, which is the correct way per Pydantic v2.11+.

---
 burr/integrations/pydantic.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/burr/integrations/pydantic.py b/burr/integrations/pydantic.py
index dd1f95a5..348a7de7 100644
--- a/burr/integrations/pydantic.py
+++ b/burr/integrations/pydantic.py
@@ -51,7 +51,7 @@
 
 def model_to_dict(model: pydantic.BaseModel, include: Optional[List[str]] = None) -> dict:
     """Utility function to convert a pydantic model to a dictionary."""
-    keys = model.model_fields.keys()
+    keys = type(model).model_fields.keys()
     keys = keys if include is None else [item for item in include if item in model.model_fields]
     return {key: getattr(model, key) for key in keys}
 

From 347c14856a712b892ce7b26e349bf9a442100781 Mon Sep 17 00:00:00 2001
From: vaibhavsingh-materialplus <Vaibhav.Singh@materialplus.io>
Date: Tue, 3 Mar 2026 09:14:21 +0530
Subject: [PATCH 2/4] all instances use type(model).model_fields correctly.

---
 burr/integrations/pydantic.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/burr/integrations/pydantic.py b/burr/integrations/pydantic.py
index 348a7de7..77f0f6ca 100644
--- a/burr/integrations/pydantic.py
+++ b/burr/integrations/pydantic.py
@@ -52,7 +52,7 @@
 def model_to_dict(model: pydantic.BaseModel, include: Optional[List[str]] = None) -> dict:
     """Utility function to convert a pydantic model to a dictionary."""
     keys = type(model).model_fields.keys()
-    keys = keys if include is None else [item for item in include if item in model.model_fields]
+    keys = keys if include is None else [item for item in include if item in type(model).model_fields]
     return {key: getattr(model, key) for key in keys}
 
 
@@ -76,7 +76,7 @@ def subset_model(
     """
     new_fields = {}
 
-    for name, field_info in model.model_fields.items():
+    for name, field_info in type(model).model_fields.items():
         if name in fields:
             # copy directly
             # TODO -- handle cross-field validation
@@ -114,7 +114,7 @@ def model_from_state(model: Type[ModelType], state: State) -> ModelType:
     :param state: state object to create from
     :return: model object
     """
-    keys = [item for item in model.model_fields.keys() if item in state]
+    keys = [item for item in type(model).model_fields.keys() if item in state]
     return model(**{key: state[key] for key in keys})
 
 
@@ -153,7 +153,7 @@ def _validate_and_extract_signature_types(
 
 
 def _validate_keys(model: Type[pydantic.BaseModel], keys: List[str], fn: Callable) -> None:
-    missing_keys = [key for key in keys if key not in model.model_fields]
+    missing_keys = [key for key in keys if key not in type(model).model_fields]
     if missing_keys:
         raise ValueError(
             f"Function fn: {fn.__qualname__} is not a valid pydantic action. "

From 342ff7769b1a819aaccda8c3b3bbc251a3e4354c Mon Sep 17 00:00:00 2001
From: Vaibhav Singh <95634443+vbhavh@users.noreply.github.com>
Date: Fri, 6 Mar 2026 14:13:39 +0530
Subject: [PATCH 3/4] Update pydantic version constraint to >=2.11

---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index e88dce53..76c6b890 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -120,7 +120,7 @@ documentation = [
 ]
 
 tracking-client = [
-  "pydantic>1"
+  "pydantic>=2.11"
 ]
 
 tracking-client-s3 = [

From cfaf738aaf31dd78d75b306dce7fb5b2ae3b980e Mon Sep 17 00:00:00 2001
From: Vaibhav Singh <95634443+vbhavh@users.noreply.github.com>
Date: Sat, 7 Mar 2026 00:28:32 +0530
Subject: [PATCH 4/4] Update pydantic version to >=2.11 in pyproject.toml

---
 pyproject.toml | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 76c6b890..7cdd092d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -89,7 +89,7 @@ tests = [
   "langchain_core",
   "langchain_community",
   "pandas",
-  "pydantic[email]",
+  "pydantic[email]>=2.11",
   "pyarrow",
   "apache-burr[aiosqlite]",
   "apache-burr[asyncpg]",
@@ -141,7 +141,7 @@ tracking-server = [
   "click",
   "fastapi",
   "uvicorn",
-  "pydantic",
+  "pydantic>=2.11",
   "pydantic-settings",
   "fastapi-pagination",
   "fastapi-utils",
@@ -153,7 +153,7 @@ tracking-server = [
 ]
 
 pydantic = [
-  "pydantic"
+  "pydantic>=2.11"
 ]
 
 haystack = [