mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 03:04:13 +00:00
053b0e741f
* test: fix import for test
* fix: fix bad error string
* docs: cleanup files docs
* fix(files/main.py): cleanup error string
* style: initial commit with a provider/config pattern for files api
google ai studio files api onboarding
* fix: test
* feat(gemini/files/transformation.py): support gemini files api response transformation
* fix(gemini/files/transformation.py): return file id as gemini uri
allows id to be passed in to chat completion request, just like openai
* feat(llm_http_handler.py): support async route for files api on llm_http_handler
* fix: fix linting errors
* fix: fix model info check
* fix: fix ruff errors
* fix: fix linting errors
* Revert "fix: fix linting errors"
This reverts commit 926a5a527f.
* fix: fix linting errors
* test: fix test
* test: fix tests
55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
"""
|
|
Translates from OpenAI's `/v1/chat/completions` to DeepSeek's `/v1/chat/completions`
|
|
"""
|
|
|
|
from typing import List, Optional, Tuple
|
|
|
|
from litellm.litellm_core_utils.prompt_templates.common_utils import (
|
|
handle_messages_with_content_list_to_str_conversion,
|
|
)
|
|
from litellm.secret_managers.main import get_secret_str
|
|
from litellm.types.llms.openai import AllMessageValues
|
|
|
|
from ...openai.chat.gpt_transformation import OpenAIGPTConfig
|
|
|
|
|
|
class DeepSeekChatConfig(OpenAIGPTConfig):
|
|
def _transform_messages(
|
|
self, messages: List[AllMessageValues], model: str
|
|
) -> List[AllMessageValues]:
|
|
"""
|
|
DeepSeek does not support content in list format.
|
|
"""
|
|
messages = handle_messages_with_content_list_to_str_conversion(messages)
|
|
return super()._transform_messages(messages=messages, model=model)
|
|
|
|
def _get_openai_compatible_provider_info(
|
|
self, api_base: Optional[str], api_key: Optional[str]
|
|
) -> Tuple[Optional[str], Optional[str]]:
|
|
api_base = (
|
|
api_base
|
|
or get_secret_str("DEEPSEEK_API_BASE")
|
|
or "https://api.deepseek.com/beta"
|
|
) # type: ignore
|
|
dynamic_api_key = api_key or get_secret_str("DEEPSEEK_API_KEY")
|
|
return api_base, dynamic_api_key
|
|
|
|
def get_complete_url(
|
|
self,
|
|
api_base: Optional[str],
|
|
api_key: Optional[str],
|
|
model: str,
|
|
optional_params: dict,
|
|
litellm_params: dict,
|
|
stream: Optional[bool] = None,
|
|
) -> str:
|
|
"""
|
|
If api_base is not provided, use the default DeepSeek /chat/completions endpoint.
|
|
"""
|
|
if not api_base:
|
|
api_base = "https://api.deepseek.com/beta"
|
|
|
|
if not api_base.endswith("/chat/completions"):
|
|
api_base = f"{api_base}/chat/completions"
|
|
|
|
return api_base
|