1 line
1.0 KiB
JSON
1 line
1.0 KiB
JSON
{"repo": "belingud/yaicli", "files": {"yaicli/llms/providers/infiniai_provider.py": {"530c77440d4723337c11d33d4765e63259804e94": {"t": 1785595460.5309365, "n": 791, "h": "adb8913fc1a262ec", "c": "from ...config import cfg\nfrom .openai_provider import OpenAIProvider\n\n\nclass InfiniAIProvider(OpenAIProvider):\n \"\"\"InfiniAI provider implementation based on openai-compatible API\"\"\"\n\n DEFAULT_BASE_URL = \"https://cloud.infini-ai.com/maas/v1\"\n COMPLETION_PARAMS_KEYS = {\n \"model\": \"MODEL\",\n \"temperature\": \"TEMPERATURE\",\n \"top_p\": \"TOP_P\",\n \"max_tokens\": \"MAX_TOKENS\",\n \"timeout\": \"TIMEOUT\",\n \"extra_body\": \"EXTRA_BODY\",\n \"frequency_penalty\": \"FREQUENCY_PENALTY\",\n }\n\n def __init__(self, config: dict = cfg, **kwargs):\n super().__init__(config, **kwargs)\n if self.enable_function:\n self.console.print(\"InfiniAI does not support functions, disabled\", style=\"yellow\")\n self.enable_function = False\n"}}}} |