mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 19:24:27 +00:00
c7f14e936a
* remove unused imports * fix AmazonConverseConfig * fix test * fix import * ruff check fixes * test fixes * fix testing * fix imports
22 lines
772 B
Python
22 lines
772 B
Python
"""
|
|
Translate from OpenAI's `/v1/chat/completions` to Perplexity's `/v1/chat/completions`
|
|
"""
|
|
|
|
from typing import Optional, Tuple
|
|
|
|
from litellm.secret_managers.main import get_secret_str
|
|
|
|
from ...openai.chat.gpt_transformation import OpenAIGPTConfig
|
|
|
|
|
|
class PerplexityChatConfig(OpenAIGPTConfig):
|
|
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("PERPLEXITY_API_BASE") or "https://api.perplexity.ai" # type: ignore
|
|
dynamic_api_key = (
|
|
api_key
|
|
or get_secret_str("PERPLEXITYAI_API_KEY")
|
|
or get_secret_str("PERPLEXITY_API_KEY")
|
|
)
|
|
return api_base, dynamic_api_key
|