github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-12 12:59:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

openai[patch]: fix azure integration tests (#26791)

Browse Source
This commit is contained in:
ccurme
2024-09-23 17:49:15 -04:00
committed by GitHub
parent 1ce056d1b2
commit 2a4c5713cd
3 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
libs/partners/openai/tests/integration_tests/chat_models/test_azure_standard.py
View File

@@ -9,6 +9,9 @@ from langchain_standard_tests.integration_tests import ChatModelIntegrationTests
from langchain_openai import AzureChatOpenAI
OPENAI_API_VERSION = os.environ.get("AZURE_OPENAI_API_VERSION", "")
OPENAI_API_BASE = os.environ.get("AZURE_OPENAI_API_BASE", "")
class TestAzureOpenAIStandard(ChatModelIntegrationTests):
@property
@@ -20,6 +23,8 @@ class TestAzureOpenAIStandard(ChatModelIntegrationTests):
return {
"deployment_name": os.environ["AZURE_OPENAI_CHAT_DEPLOYMENT_NAME"],
"model": "gpt-4o",
"openai_api_version": OPENAI_API_VERSION,
"azure_endpoint": OPENAI_API_BASE,
}
@property
@@ -41,7 +46,9 @@ class TestAzureOpenAIStandardLegacy(ChatModelIntegrationTests):
@property
def chat_model_params(self) -> dict:
return {
"deployment_name": os.environ["AZURE_OPENAI_LEGACY_CHAT_DEPLOYMENT_NAME"]
"deployment_name": os.environ["AZURE_OPENAI_LEGACY_CHAT_DEPLOYMENT_NAME"],
"openai_api_version": OPENAI_API_VERSION,
"azure_endpoint": OPENAI_API_BASE,
}
@pytest.mark.xfail(reason="Not yet supported.")