diff --git a/litellm/main.py b/litellm/main.py index 2c906e990..bf4931168 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -11,6 +11,7 @@ import os, openai, sys, json, inspect, uuid, datetime, threading from typing import Any, Literal, Union, BinaryIO from typing_extensions import overload from functools import partial + import dotenv, traceback, random, asyncio, time, contextvars from copy import deepcopy import httpx diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index bb607e679..3b6b507d6 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -741,11 +741,13 @@ def test_completion_claude_3_function_plus_image(): print(response) + @pytest.mark.parametrize( - "provider", [ + "provider", + [ "azure", "azure_ai" - ] + ], ) def test_completion_azure_mistral_large_function_calling(provider): """