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

raise exception when prisma init fails

Browse source
This commit is contained in:
ishaan-jaff 2024-01-09 13:36:02 +05:30
parent 8f8237a1a0
commit 599318b40e
1 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/proxy/proxy_server.py
View file

@ -311,6 +311,7 @@ def prisma_setup(database_url: Optional[str]):
database_url=database_url, proxy_logging_obj=proxy_logging_obj database_url=database_url, proxy_logging_obj=proxy_logging_obj
) )
except Exception as e: except Exception as e:
raise e
verbose_proxy_logger.debug( verbose_proxy_logger.debug(
f"Error when initializing prisma, Ensure you run pip install prisma {str(e)}" f"Error when initializing prisma, Ensure you run pip install prisma {str(e)}"
) )
@ -510,10 +511,9 @@ class ProxyConfig:
def is_yaml(self, config_file_path: str) -> bool: def is_yaml(self, config_file_path: str) -> bool:
if not os.path.isfile(config_file_path): if not os.path.isfile(config_file_path):
return False return False
_, file_extension = os.path.splitext(config_file_path) _, file_extension = os.path.splitext(config_file_path)
return file_extension.lower() == '.yaml' or file_extension.lower() == '.yml' return file_extension.lower() == ".yaml" or file_extension.lower() == ".yml"
async def get_config(self, config_file_path: Optional[str] = None) -> dict: async def get_config(self, config_file_path: Optional[str] = None) -> dict:
global prisma_client, user_config_file_path global prisma_client, user_config_file_path
@ -1169,7 +1169,9 @@ async def startup_event():
llm_router, llm_router,
llm_model_list, llm_model_list,
general_settings, general_settings,
) = await proxy_config.load_config(router=llm_router, config_file_path=worker_config) ) = await proxy_config.load_config(
router=llm_router, config_file_path=worker_config
)
else: else:
await initialize(**worker_config) await initialize(**worker_config)
else: else: