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

use common folder for cohere

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-26 14:28:50 -07:00
parent 2481d6007b
commit 5c1ebb6ac2
5 changed files with 459 additions and 417 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/main.py
View file

@ -107,6 +107,8 @@ from .llms.azure import AzureChatCompletion, _check_dynamic_azure_params
from .llms.azure_text import AzureTextCompletion
from .llms.bedrock_httpx import BedrockConverseLLM, BedrockLLM
from .llms.cohere import chat as cohere_chat
from .llms.cohere import completion as cohere_completion
from .llms.cohere import embed as cohere_embed
from .llms.custom_llm import CustomLLM, custom_chat_llm_router
from .llms.databricks import DatabricksChatCompletion
from .llms.huggingface_restapi import Huggingface
@ -1645,7 +1647,7 @@ def completion(
if extra_headers is not None:
headers.update(extra_headers)
model_response = cohere.completion(
model_response = cohere_completion.completion(
model=model,
messages=messages,
api_base=api_base,
@ -3457,7 +3459,7 @@ def embedding(
headers = extra_headers
else:
headers = {}
response = cohere.embedding(
response = cohere_embed.embedding(
model=model,
input=input,
optional_params=optional_params,