diff --git a/packages/gen/integration_tests/langchain_/test_openai.py b/packages/gen/integration_tests/langchain_/test_openai.py index 42b47faa..59d4c161 100644 --- a/packages/gen/integration_tests/langchain_/test_openai.py +++ b/packages/gen/integration_tests/langchain_/test_openai.py @@ -3,11 +3,11 @@ from parameterized import parameterized from pydantic import BaseModel -from integration_tests.constants import (OPENAI_GPT_O4_MINI_TEST_MODEL, +from ..constants import (OPENAI_GPT_O4_MINI_TEST_MODEL, OPENAI_GPT_4O_MINI_TEST_MODEL, OPENAI_GPT_5_TEST_MODEL_NANO, OPENAI_EMBEDDING_TEST_MODEL, OPENAI_GPT_O3_MINI_TEST_MODEL, MISTRAL_TEST_MODEL, NVIDIA_EMBEDDING_TEST_MODEL, PERPLEXITY_TEST_MODEL, - COHERE_COMMAND_A_TEST_MODEL, PERPLEXITY_SONAR_DEEP_RESEARCH_TEST_MODEL) -from integration_tests.setup_aicore import TestCaseAICoreSetupMixin + COHERE_COMMAND_A_TEST_MODEL) +from ..setup_aicore import TestCaseAICoreSetupMixin try: import openai @@ -45,8 +45,7 @@ class TestOpenAILLM(TestCaseAICoreSetupMixin, unittest.TestCase): @parameterized.expand( [ OPENAI_GPT_O4_MINI_TEST_MODEL, - PERPLEXITY_TEST_MODEL, - PERPLEXITY_SONAR_DEEP_RESEARCH_TEST_MODEL, + PERPLEXITY_TEST_MODEL ] ) def test_chat_model(self, model=OPENAI_GPT_4O_MINI_TEST_MODEL): @@ -121,8 +120,7 @@ class AsyncOpenAITests(TestCaseAICoreSetupMixin, unittest.IsolatedAsyncioTestCas @parameterized.expand( [ OPENAI_GPT_O3_MINI_TEST_MODEL, - MISTRAL_TEST_MODEL, - PERPLEXITY_SONAR_DEEP_RESEARCH_TEST_MODEL + MISTRAL_TEST_MODEL ] ) async def test_chat_model(self, model=OPENAI_GPT_4O_MINI_TEST_MODEL): diff --git a/packages/gen/integration_tests/native_clients/test_openai.py b/packages/gen/integration_tests/native_clients/test_openai.py index 94fcbec3..31476243 100644 --- a/packages/gen/integration_tests/native_clients/test_openai.py +++ b/packages/gen/integration_tests/native_clients/test_openai.py @@ -2,11 +2,10 @@ from parameterized import parameterized from pydantic import BaseModel -from integration_tests.constants import (MISTRAL_TEST_MODEL, OPENAI_EMBEDDING_TEST_MODEL, +from ..constants import (MISTRAL_TEST_MODEL, OPENAI_EMBEDDING_TEST_MODEL, OPENAI_GPT_5_MINI_TEST_MODEL, OPENAI_GPT_O4_MINI_TEST_MODEL, OPENAI_GPT_4O_MINI_TEST_MODEL, OPENAI_GPT_O3_MINI_TEST_MODEL, OPENAI_GPT_5_TEST_MODEL_NANO, NVIDIA_EMBEDDING_TEST_MODEL, PERPLEXITY_TEST_MODEL, - COHERE_COMMAND_A_TEST_MODEL, OPENAI_GPT_5_TEST_MODEL, - PERPLEXITY_SONAR_DEEP_RESEARCH_TEST_MODEL) + COHERE_COMMAND_A_TEST_MODEL, OPENAI_GPT_5_TEST_MODEL) try: import openai @@ -20,7 +19,7 @@ except ImportError: no_openai = True -from integration_tests.setup_aicore import TestCaseAICoreSetupMixin +from ..setup_aicore import TestCaseAICoreSetupMixin class Person(BaseModel): """ @@ -62,8 +61,7 @@ def test_nvidia_embedding(self, model=NVIDIA_EMBEDDING_TEST_MODEL): @parameterized.expand( [ - OPENAI_GPT_5_TEST_MODEL_NANO, - PERPLEXITY_SONAR_DEEP_RESEARCH_TEST_MODEL + OPENAI_GPT_5_TEST_MODEL_NANO ] ) def test_chat_completion(self, model=OPENAI_GPT_5_TEST_MODEL_NANO): @@ -310,8 +308,7 @@ async def test_async_nvidia_embedding(self, model=NVIDIA_EMBEDDING_TEST_MODEL): @parameterized.expand( [ - OPENAI_GPT_5_TEST_MODEL_NANO, - PERPLEXITY_SONAR_DEEP_RESEARCH_TEST_MODEL + OPENAI_GPT_5_TEST_MODEL_NANO ] ) async def test_async_chat_completion(self, model=OPENAI_GPT_5_TEST_MODEL_NANO):