mirror of https://github.com/hwchase17/langchain
openai[patch]: Upgrade @root_validators in preparation for pydantic 2 migration (#25491)
* Upgrade @root_validator in openai pkg * Ran notebooks for all but AzureAI embeddings --------- Co-authored-by: Bagatur <baskaryan@gmail.com>pull/25999/head
parent
0207dc1431
commit
bc3b851f08
@ -0,0 +1,38 @@
|
||||
from typing import Tuple, Type
|
||||
|
||||
from langchain_core.embeddings import Embeddings
|
||||
from langchain_standard_tests.unit_tests.embeddings import EmbeddingsUnitTests
|
||||
|
||||
from langchain_openai import AzureOpenAIEmbeddings
|
||||
|
||||
|
||||
class TestAzureOpenAIStandard(EmbeddingsUnitTests):
|
||||
@property
|
||||
def embeddings_class(self) -> Type[Embeddings]:
|
||||
return AzureOpenAIEmbeddings
|
||||
|
||||
@property
|
||||
def embedding_model_params(self) -> dict:
|
||||
return {"api_key": "api_key", "azure_endpoint": "https://endpoint.com"}
|
||||
|
||||
@property
|
||||
def init_from_env_params(self) -> Tuple[dict, dict, dict]:
|
||||
return (
|
||||
{
|
||||
"AZURE_OPENAI_API_KEY": "api_key",
|
||||
"AZURE_OPENAI_ENDPOINT": "https://endpoint.com",
|
||||
"AZURE_OPENAI_AD_TOKEN": "token",
|
||||
"OPENAI_ORG_ID": "org_id",
|
||||
"OPENAI_API_VERSION": "yyyy-mm-dd",
|
||||
"OPENAI_API_TYPE": "type",
|
||||
},
|
||||
{},
|
||||
{
|
||||
"openai_api_key": "api_key",
|
||||
"azure_endpoint": "https://endpoint.com",
|
||||
"azure_ad_token": "token",
|
||||
"openai_organization": "org_id",
|
||||
"openai_api_version": "yyyy-mm-dd",
|
||||
"openai_api_type": "type",
|
||||
},
|
||||
)
|
@ -0,0 +1,32 @@
|
||||
"""Standard LangChain interface tests"""
|
||||
|
||||
from typing import Tuple, Type
|
||||
|
||||
from langchain_core.embeddings import Embeddings
|
||||
from langchain_standard_tests.unit_tests.embeddings import EmbeddingsUnitTests
|
||||
|
||||
from langchain_openai import OpenAIEmbeddings
|
||||
|
||||
|
||||
class TestOpenAIStandard(EmbeddingsUnitTests):
|
||||
@property
|
||||
def embeddings_class(self) -> Type[Embeddings]:
|
||||
return OpenAIEmbeddings
|
||||
|
||||
@property
|
||||
def init_from_env_params(self) -> Tuple[dict, dict, dict]:
|
||||
return (
|
||||
{
|
||||
"OPENAI_API_KEY": "api_key",
|
||||
"OPENAI_ORG_ID": "org_id",
|
||||
"OPENAI_API_BASE": "api_base",
|
||||
"OPENAI_PROXY": "https://proxy.com",
|
||||
},
|
||||
{},
|
||||
{
|
||||
"openai_api_key": "api_key",
|
||||
"openai_organization": "org_id",
|
||||
"openai_api_base": "api_base",
|
||||
"openai_proxy": "https://proxy.com",
|
||||
},
|
||||
)
|
Loading…
Reference in New Issue