phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 03:34:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

expose api_version

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

1
litellm/__init__.py
View file

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

2
litellm/main.py
View file

@ -67,7 +67,7 @@ def completion(
# azure configs
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_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:
openai.api_key = api_key
elif litellm.azure_key:

2
pyproject.toml
View file

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