Use pydantic 2.6+

Author: cbornet

libs/knowledge-graph/pyproject.toml

@@ -36,7 +36,7 @@ pytest-dotenv = "^0.5.2"
 pytest-rerunfailures = "^14.0"
 mypy = "^1.10.1"
 types-pyyaml = "^6.0.1"
-pydantic = "<2" # for compatibility between LangChain and pydantic-yaml type checking
+pydantic = "^2.6.0"
 
 [build-system]
 requires = ["poetry-core"]

libs/knowledge-graph/ragstack_knowledge_graph/extraction.py

@@ -32,7 +32,8 @@
 def _format_example(idx: int, example: Example) -> str:
     from pydantic_yaml import to_yaml_str
 
-    return f"Example {idx}:\n```yaml\n{to_yaml_str(example)}\n```"
+    yaml_example = to_yaml_str(example)  # type: ignore[arg-type]
+    return f"Example {idx}:\n```yaml\n{yaml_example}\n```"
 
 
 class KnowledgeSchemaExtractor:

libs/knowledge-graph/ragstack_knowledge_graph/knowledge_schema.py

@@ -75,13 +75,13 @@ def from_file(cls, path: str | Path) -> Self:
         """
         from pydantic_yaml import parse_yaml_file_as
 
-        return parse_yaml_file_as(cls, path)
+        return parse_yaml_file_as(cls, path)  # type: ignore[type-var]
 
     def to_yaml_str(self) -> str:
         """Convert the schema to a YAML string."""
         from pydantic_yaml import to_yaml_str
 
-        return to_yaml_str(self)
+        return to_yaml_str(self)  # type: ignore[arg-type]
 
 
 class KnowledgeSchemaValidator: