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

refactor cohere to be in a folder

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-26 14:16:25 -07:00
parent 174b1c43e3
commit f9ea0d8fa9
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/cohere_chat.py → litellm/llms/cohere/chat.py
View file

@ -13,7 +13,7 @@ import litellm
from litellm.types.llms.cohere import ToolResultObject from litellm.types.llms.cohere import ToolResultObject
from litellm.utils import Choices, Message, ModelResponse, Usage from litellm.utils import Choices, Message, ModelResponse, Usage
from .prompt_templates.factory import cohere_message_pt, cohere_messages_pt_v2 from ..prompt_templates.factory import cohere_message_pt, cohere_messages_pt_v2
class CohereError(Exception): class CohereError(Exception):

2
litellm/main.py
View file

@ -83,7 +83,6 @@ from .llms import (
clarifai, clarifai,
cloudflare, cloudflare,
cohere, cohere,
cohere_chat,
gemini, gemini,
huggingface_restapi, huggingface_restapi,
maritalk, maritalk,
@ -107,6 +106,7 @@ from .llms.anthropic_text import AnthropicTextCompletion
from .llms.azure import AzureChatCompletion, _check_dynamic_azure_params 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.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
from .llms.huggingface_restapi import Huggingface from .llms.huggingface_restapi import Huggingface