mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 19:24:27 +00:00
29 lines
830 B
Python
29 lines
830 B
Python
from typing import Optional, Tuple
|
|
|
|
|
|
from litellm.llms.openai.openai import OpenAIConfig
|
|
|
|
from ..authenticator import Authenticator
|
|
|
|
class GithubCopilotConfig(OpenAIConfig):
|
|
def __init__(
|
|
self,
|
|
api_key: Optional[str] = None,
|
|
api_base: Optional[str] = None,
|
|
custom_llm_provider: str = "openai",
|
|
) -> None:
|
|
super().__init__()
|
|
|
|
self.authenticator = Authenticator()
|
|
|
|
def _get_openai_compatible_provider_info(
|
|
self,
|
|
model: str,
|
|
api_base: Optional[str],
|
|
api_key: Optional[str],
|
|
custom_llm_provider: str,
|
|
) -> Tuple[Optional[str], Optional[str], str]:
|
|
api_base = "https://api.githubcopilot.com"
|
|
dynamic_api_key = self.authenticator.get_api_key()
|
|
|
|
return api_base, dynamic_api_key, custom_llm_provider
|