diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index 302e5ad71..0d0b18e9e 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -122,6 +122,9 @@ jobs: ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }} GROQ_API_KEY: ${{ secrets.GROQ_API_KEY }} ASSEMBLYAI_API_KEY: ${{ secrets.ASSEMBLYAI_API_KEY }} + AZURE_OPENAI_API_KEY: ${{ secrets.AZURE_OPENAI_API_KEY }} + AZURE_OPENAI_ENDPOINT: ${{ secrets.AZURE_OPENAI_ENDPOINT }} + OPENAI_API_VERSION: "2024-05-01-preview" # used by Azure FAL_KEY: ${{ secrets.FAL_KEY }} GOOGLE_APPLICATION_CREDENTIALS: google.json PYTEST_ADDOPTS: "--color=yes" diff --git a/tests/test_llm.py b/tests/test_llm.py index 4b71c0324..4343d69dd 100644 --- a/tests/test_llm.py +++ b/tests/test_llm.py @@ -81,6 +81,7 @@ def test_hashable_typeinfo(): LLMS: list[Callable[[], llm.LLM]] = [ pytest.param(lambda: openai.LLM(), id="openai"), + pytest.param(lambda: openai.LLM.with_azure(), id="openai.with_azure"), # lambda: openai.beta.AssistantLLM( # assistant_opts=openai.beta.AssistantOptions( # create_options=openai.beta.AssistantCreateOptions(