HideyoshiNakazone/jambo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 12 Wiki Activity

Merge pull request #15 from HideyoshiNakazone/fix/better-typing-output

Browse Source 
Fixes Typing Output
This commit was merged in pull request #15.
This commit is contained in:
Vitor Hideyoshi
2025-06-02 20:43:13 -03:00
committed by GitHub
parent 393eaa5e0a 9e52783b22
commit 894969332d
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
jambo/schema_converter.py
View File

@@ -18,7 +18,7 @@ class SchemaConverter:
""" """
@staticmethod @staticmethod
def build(schema: JSONSchema) -> ModelT: def build(schema: JSONSchema) -> type[ModelT]:
""" """
Converts a JSON Schema to a Pydantic model. Converts a JSON Schema to a Pydantic model.
:param schema: The JSON Schema to convert. :param schema: The JSON Schema to convert.
@@ -33,7 +33,7 @@ class SchemaConverter:
def build_object( def build_object(
name: str, name: str,
schema: JSONSchema, schema: JSONSchema,
) -> ModelT: ) -> type[ModelT]:
""" """
Converts a JSON Schema object to a Pydantic model given a name. Converts a JSON Schema object to a Pydantic model given a name.
:param name: :param name:
@@ -59,7 +59,7 @@ class SchemaConverter:
@staticmethod @staticmethod
def _build_model_from_properties( def _build_model_from_properties(
model_name: str, model_properties: dict, required_keys: list[str] model_name: str, model_properties: dict, required_keys: list[str]
) -> ModelT: ) -> type[ModelT]:
properties = SchemaConverter._parse_properties(model_properties, required_keys) properties = SchemaConverter._parse_properties(model_properties, required_keys)
return create_model(model_name, **properties) return create_model(model_name, **properties)