From 2d8e4ddfa02e081ef7807fe37b419895cf48c855 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 11 Jun 2024 15:13:50 -0700 Subject: [PATCH] fix(main.py): trigger new build --- litellm/main.py | 1 + litellm/tests/test_completion.py | 6 ++++-- 2 files changed, 5 insertions(+), 2 deletions(-) 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): """