community[patch]: Remove usage of @root_validator(allow_reuse=True) (#25235)

Remove usage of @root_validator(allow_reuse=True)
This commit is contained in:
Eugene Yurtsev
2024-08-09 10:57:42 -04:00
committed by GitHub
parent a2b4c33bd6
commit 6e57aa7c36
12 changed files with 76 additions and 74 deletions

View File

@@ -53,7 +53,7 @@ class GradientEmbeddings(BaseModel, Embeddings):
class Config:
extra = "forbid"
@root_validator(allow_reuse=True)
@root_validator(pre=True)
def validate_environment(cls, values: Dict) -> Dict:
"""Validate that api key and python package exists in environment."""
@@ -65,8 +65,15 @@ class GradientEmbeddings(BaseModel, Embeddings):
)
values["gradient_api_url"] = get_from_dict_or_env(
values, "gradient_api_url", "GRADIENT_API_URL"
values,
"gradient_api_url",
"GRADIENT_API_URL",
default="https://api.gradient.ai/api",
)
return values
@root_validator(pre=False, skip_on_failure=True)
def post_init(cls, values: Dict) -> Dict:
try:
import gradientai
except ImportError:
@@ -85,7 +92,6 @@ class GradientEmbeddings(BaseModel, Embeddings):
host=values["gradient_api_url"],
)
values["client"] = gradient.get_embeddings_model(slug=values["model"])
return values
def embed_documents(self, texts: List[str]) -> List[List[float]]: