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

fix entrypoint

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-26 20:32:23 -07:00
parent a49a7f4458
commit ffb752d878
2 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/main.py
View file

@ -82,7 +82,6 @@ from .llms import (
bedrock, bedrock,
clarifai, clarifai,
cloudflare, cloudflare,
cohere,
gemini, gemini,
huggingface_restapi, huggingface_restapi,
maritalk, maritalk,
@ -107,7 +106,7 @@ from .llms.azure import AzureChatCompletion, _check_dynamic_azure_params
from .llms.azure_text import AzureTextCompletion from .llms.azure_text import AzureTextCompletion
from .llms.bedrock_httpx import BedrockConverseLLM, BedrockLLM from .llms.bedrock_httpx import BedrockConverseLLM, BedrockLLM
from .llms.cohere import chat as cohere_chat from .llms.cohere import chat as cohere_chat
from .llms.cohere import completion as cohere_completion from .llms.cohere import completion as cohere_completion # type: ignore
from .llms.cohere import embed as cohere_embed from .llms.cohere import embed as cohere_embed
from .llms.custom_llm import CustomLLM, custom_chat_llm_router from .llms.custom_llm import CustomLLM, custom_chat_llm_router
from .llms.databricks import DatabricksChatCompletion from .llms.databricks import DatabricksChatCompletion

4
litellm/proxy/prisma_migration.py
View file

@ -51,6 +51,10 @@ while retry_count < max_retries and exit_code != 0:
retry_count += 1 retry_count += 1
print(f"Attempt {retry_count}...") # noqa print(f"Attempt {retry_count}...") # noqa
# run prisma generate
result = subprocess.run(["prisma", "generate"], capture_output=True)
exit_code = result.returncode
# Run the Prisma db push command # Run the Prisma db push command
result = subprocess.run( result = subprocess.run(
["prisma", "db", "push", "--accept-data-loss"], capture_output=True ["prisma", "db", "push", "--accept-data-loss"], capture_output=True