diff --git a/src/funcchain/parser/primitive_types.py b/src/funcchain/parser/primitive_types.py index 58b8729..7757b2f 100644 --- a/src/funcchain/parser/primitive_types.py +++ b/src/funcchain/parser/primitive_types.py @@ -24,7 +24,7 @@ def __init__( retry_llm: BaseChatModel | str | None = None, ) -> None: super().__init__( - pydantic_object=create_model("ExtractPrimitiveType", value=(primitive_type, ...)), + pydantic_object=create_model("Extract", value=(primitive_type, ...)), retry=retry, retry_llm=retry_llm, )