From 3e419fd906a670311b132785d98829221b01c487 Mon Sep 17 00:00:00 2001
From: evelynmitchell <efmphone@gmail.com>
Date: Sun, 3 Dec 2023 17:41:16 -0700
Subject: [PATCH] openai_models remove pydantic
---
swarms/models/openai_models.py | 11 ++++-------
1 file changed, 4 insertions(+), 7 deletions(-)
diff --git a/swarms/models/openai_models.py b/swarms/models/openai_models.py
index 12830cec..233b99c3 100644
--- a/swarms/models/openai_models.py
+++ b/swarms/models/openai_models.py
@@ -37,7 +37,7 @@ from langchain.utils.utils import build_extra_kwargs
from importlib.metadata import version
from packaging.version import parse
-from pydantic import model_validator, ConfigDict
+
logger = logging.getLogger(__name__)
@@ -248,9 +248,8 @@ class BaseOpenAI(BaseLLM):
"""Initialize the OpenAI object."""
data.get("model_name", "")
return super().__new__(cls)
- model_config = ConfigDict(populate_by_name=True)
- @model_validator(mode="before")
+
@classmethod
def build_extra(cls, values: Dict[str, Any]) -> Dict[str, Any]:
"""Build extra kwargs from additional params that were passed in."""
@@ -261,7 +260,7 @@ class BaseOpenAI(BaseLLM):
)
return values
- @model_validator()
+
@classmethod
def validate_environment(cls, values: Dict) -> Dict:
"""Validate that api key and python package exists in environment."""
@@ -757,7 +756,6 @@ class AzureOpenAI(BaseOpenAI):
openai_api_type: str = ""
openai_api_version: str = ""
- @model_validator()
@classmethod
def validate_azure_settings(cls, values: Dict) -> Dict:
values["openai_api_version"] = get_from_dict_or_env(
@@ -847,7 +845,6 @@ class OpenAIChat(BaseLLM):
disallowed_special: Union[Literal["all"], Collection[str]] = "all"
"""Set of special tokens that are not allowed。"""
- @model_validator(mode="before")
@classmethod
def build_extra(cls, values: Dict[str, Any]) -> Dict[str, Any]:
"""Build extra kwargs from additional params that were passed in."""
@@ -866,7 +863,7 @@ class OpenAIChat(BaseLLM):
values["model_kwargs"] = extra
return values
- @model_validator()
+
@classmethod
def validate_environment(cls, values: Dict) -> Dict:
"""Validate that api key and python package exists in environment."""