expose api_version

This commit is contained in:
Krrish Dholakia 2023-08-05 21:57:12 -07:00
parent 0a1525a670
commit 8c5118cae6
3 changed files with 3 additions and 2 deletions

View file

@ -23,6 +23,7 @@ def identify(event_details):
####### PROXY PARAMS ################### configurable params if you use proxy models like Helicone ####### PROXY PARAMS ################### configurable params if you use proxy models like Helicone
api_base = None api_base = None
headers = None headers = None
api_version = None
####### Secret Manager ##################### ####### Secret Manager #####################
secret_manager_client = None secret_manager_client = None
####### COMPLETION MODELS ################### ####### COMPLETION MODELS ###################

View file

@ -67,7 +67,7 @@ def completion(
# azure configs # azure configs
openai.api_type = "azure" openai.api_type = "azure"
openai.api_base = litellm.api_base if litellm.api_base is not None else get_secret("AZURE_API_BASE") openai.api_base = litellm.api_base if litellm.api_base is not None else get_secret("AZURE_API_BASE")
openai.api_version = get_secret("AZURE_API_VERSION") openai.api_version = litellm.api_version if litellm.api_version is not None else get_secret("AZURE_API_VERSION")
if api_key: if api_key:
openai.api_key = api_key openai.api_key = api_key
elif litellm.azure_key: elif litellm.azure_key:

View file

@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "litellm" name = "litellm"
version = "0.1.347" version = "0.1.348"
description = "Library to easily interface with LLM API providers" description = "Library to easily interface with LLM API providers"
authors = ["BerriAI"] authors = ["BerriAI"]
license = "MIT License" license = "MIT License"