Skip to content

Commit fced37d

Browse files
committed
Updated pr.
1 parent 0c6d6ac commit fced37d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

ads/llm/langchain/plugins/embeddings/oci_data_science_model_deployment_endpoint.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,8 @@
88
import requests
99
from langchain_core.embeddings import Embeddings
1010
from langchain_core.language_models.llms import create_base_retry_decorator
11+
from langchain_core.pydantic_v1 import BaseModel, Field, root_validator
1112
from langchain_core.utils import get_from_dict_or_env
12-
from pydantic import BaseModel, Field, model_validator
1313

1414
DEFAULT_HEADER = {
1515
"Content-Type": "application/json",
@@ -64,7 +64,7 @@ class OCIModelDeploymentEndpointEmbeddings(BaseModel, Embeddings):
6464
max_retries: int = 1
6565
"""The maximum number of retries to make when generating."""
6666

67-
@model_validator(mode="before")
67+
@root_validator()
6868
def validate_environment( # pylint: disable=no-self-argument
6969
cls, values: Dict
7070
) -> Dict:
@@ -167,7 +167,7 @@ def _proceses_response(self, response: requests.Response) -> List[List[float]]:
167167
except Exception as e:
168168
raise ValueError(
169169
f"Error raised by inference API: {e}.\nResponse: {response.text}"
170-
)
170+
) from e
171171
return embeddings
172172

173173
def embed_documents(

0 commit comments

Comments
 (0)