diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index 40bc14fab8..f23e1cbc10 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -372,6 +372,38 @@ def test_completion_openai_with_functions(): # test_completion_openai_with_functions() +# def test_completion_openai_azure_with_functions(): +# function1 = [ +# { +# "name": "get_current_weather", +# "description": "Get the current weather in a given location", +# "parameters": { +# "type": "object", +# "properties": { +# "location": { +# "type": "string", +# "description": "The city and state, e.g. San Francisco, CA", +# }, +# "unit": {"type": "string", "enum": ["celsius", "fahrenheit"]}, +# }, +# "required": ["location"], +# }, +# } +# ] +# try: +# response = completion( +# model="azure/chatgpt-functioncalling", messages=messages, stream=True +# ) +# # Add any assertions here to check the response +# print(response) +# for chunk in response: +# print(chunk) +# print(chunk["choices"][0]["finish_reason"]) +# except Exception as e: +# pytest.fail(f"Error occurred: {e}") +# test_completion_openai_azure_with_functions() + + def test_completion_azure(): try: print("azure gpt-3.5 test\n\n")