fix(init.py): expose complete client session

This commit is contained in:
Krrish Dholakia 2023-10-10 15:16:10 -07:00
parent d43160ab8a
commit 70bfa022a4
2 changed files with 4 additions and 9 deletions

View file

@ -4,13 +4,10 @@ import requests, certifi, ssl
class BaseLLM:
def create_client_session(self):
if litellm.verify_ssl is False:
if litellm.client_session:
session = litellm.client_session
else:
session = requests.Session()
session.verify = False
else:
ca_bundle_path = certifi.where() if litellm.ca_bundle_path is None else litellm.ca_bundle_path
session = requests.Session()
session.verify = ca_bundle_path
return session