phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fixing azure bug

Browse source
This commit is contained in:
Krrish Dholakia 2023-08-21 14:05:24 -07:00
parent 068f4ca866
commit 7f3b90db3c
2 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/main.py
View file

@ -164,7 +164,7 @@ def completion(
) )
else: else:
response = openai.ChatCompletion.create( response = openai.ChatCompletion.create(
model=model, messages=messages, **optional_params engine=model, messages=messages, **optional_params
) )
## LOGGING ## LOGGING
logging.post_call(input=messages, api_key=openai.api_key, original_response=response, additional_args={"headers": litellm.headers, "api_version": openai.api_version, "api_base": openai.api_base}) logging.post_call(input=messages, api_key=openai.api_key, original_response=response, additional_args={"headers": litellm.headers, "api_version": openai.api_version, "api_base": openai.api_base})

3
litellm/tests/test_completion.py
View file

@ -254,8 +254,7 @@ def test_completion_openai_with_functions():
def test_completion_azure(): def test_completion_azure():
try: try:
response = completion( response = completion(
model="gpt-3.5-turbo", model="chatgpt-test",
deployment_id="chatgpt-test",
messages=messages, messages=messages,
custom_llm_provider="azure", custom_llm_provider="azure",
) )