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-05 22:39:00 -07:00
parent fab537fee9
commit c8884e21e7
7 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/main.py
View file

@ -75,7 +75,7 @@ def completion(
else:
openai.api_key = get_secret("AZURE_API_KEY")
## LOGGING
logging(model=model, input=messages, azure=azure, logger_fn=logger_fn)
logging(model=model, input=messages, additional_args=optional_params, azure=azure, logger_fn=logger_fn)
## COMPLETION CALL
if litellm.headers:
response = openai.ChatCompletion.create(
@ -86,7 +86,7 @@ def completion(
)
else:
response = openai.ChatCompletion.create(
engine=model,
model=model,
messages = messages,
**optional_params
)
@ -103,7 +103,7 @@ def completion(
else:
openai.api_key = get_secret("OPENAI_API_KEY")
## LOGGING
logging(model=model, input=messages, azure=azure, logger_fn=logger_fn)
logging(model=model, input=messages, additional_args=optional_params, azure=azure, logger_fn=logger_fn)
## COMPLETION CALL
if litellm.headers:
response = openai.ChatCompletion.create(
@ -132,7 +132,7 @@ def completion(
openai.organization = litellm.organization
prompt = " ".join([message["content"] for message in messages])
## LOGGING
logging(model=model, input=prompt, azure=azure, logger_fn=logger_fn)
logging(model=model, input=prompt, additional_args=optional_params, azure=azure, logger_fn=logger_fn)
## COMPLETION CALL
if litellm.headers:
response = openai.Completion.create(