phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(fix) proxy cleanup

Browse source
This commit is contained in:
ishaan-jaff 2023-11-09 11:31:31 -08:00
parent 333268c8b7
commit 64a50d6b98
1 changed files with 14 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

21
litellm/proxy/proxy_server.py
View file

@ -79,7 +79,8 @@ def generate_feedback_box():
"\033[1;31mGive Feedback / Get Help: https://github.com/BerriAI/litellm/issues/new\033[0m"
)
print()
print("\033[1;34mDocs: https://docs.litellm.ai/docs/proxy_server\033[0m")
print("\033[1;34mDocs: https://docs.litellm.ai/docs/simple_proxy\033[0m\n")
print(f"\033[32mLiteLLM: Test your local endpoint with: \"litellm --test\" [In a new terminal tab]\033[0m\n")
print()
import litellm
@ -208,11 +209,7 @@ def load_router_config(router: Optional[litellm.Router], config_file_path: str):
except:
pass
## SERVER SETTINGS (e.g. default completion model = 'ollama/mistral')
_server_settings = config.get("server_settings", None)
if _server_settings:
server_settings = _server_settings
print_verbose(f"Configs passed in, loaded config YAML\n{config}")
## LITELLM MODULE SETTINGS (e.g. litellm.drop_params=True,..)
litellm_settings = config.get('litellm_settings', None)
if litellm_settings:
@ -223,6 +220,10 @@ def load_router_config(router: Optional[litellm.Router], config_file_path: str):
model_list = config.get('model_list', None)
if model_list:
router = litellm.Router(model_list=model_list)
print(f"\033[32mLiteLLM: Proxy initialized with Config, Set models:\033[0m")
for model in model_list:
print(f"\033[32m {model.get('model_name', '')}\033[0m")
print()
## ENVIRONMENT VARIABLES
environment_variables = config.get('environment_variables', None)
@ -392,7 +393,7 @@ def initialize(
if max_budget: # litellm-specific param
litellm.max_budget = max_budget
dynamic_config["general"]["max_budget"] = max_budget
if debug: # litellm-specific param
if debug==True: # litellm-specific param
litellm.set_verbose = True
if experimental:
pass
@ -415,7 +416,7 @@ def data_generator(response):
def litellm_completion(*args, **kwargs):
global user_temperature, user_request_timeout, user_max_tokens, user_api_base
call_type = kwargs.pop("call_type")
# override with user settings
# override with user settings, these are params passed via cli
if user_temperature:
kwargs["temperature"] = user_temperature
if user_request_timeout:
@ -427,7 +428,7 @@ def litellm_completion(*args, **kwargs):
## CHECK CONFIG ##
if llm_model_list and kwargs["model"] in [m["model_name"] for m in llm_model_list]:
for m in llm_model_list:
if kwargs["model"] == m["model_name"]:
if kwargs["model"] == m["model_name"]: # if user has specified a config, this will use the config
for key, value in m["litellm_params"].items():
kwargs[key] = value
break
@ -445,7 +446,7 @@ def startup_event():
import json
worker_config = json.loads(os.getenv("WORKER_CONFIG"))
initialize(**worker_config)
print(f"\033[32mWorker Initialized\033[0m\n")
# print(f"\033[32mWorker Initialized\033[0m\n")
#### API ENDPOINTS ####
@router.get("/v1/models")