75 lines
2.0 KiB
Python
75 lines
2.0 KiB
Python
"""
|
|
LLM Presets - Predefined configurations for popular LLM providers
|
|
|
|
All providers support OpenAI SDK protocol.
|
|
"""
|
|
|
|
from typing import Dict, Any, List
|
|
|
|
|
|
LLM_PRESETS: List[Dict[str, Any]] = [
|
|
{
|
|
"name": "Qwen",
|
|
"base_url": "https://dashscope.aliyuncs.com/compatible-mode/v1",
|
|
"model": "qwen-max",
|
|
"api_key_url": "https://dashscope.console.aliyun.com/apiKey",
|
|
},
|
|
{
|
|
"name": "OpenAI",
|
|
"base_url": "https://api.openai.com/v1",
|
|
"model": "gpt-4o",
|
|
"api_key_url": "https://platform.openai.com/api-keys",
|
|
},
|
|
{
|
|
"name": "Claude",
|
|
"base_url": "https://api.anthropic.com/v1/",
|
|
"model": "claude-sonnet-4-5",
|
|
"api_key_url": "https://console.anthropic.com/settings/keys",
|
|
},
|
|
{
|
|
"name": "DeepSeek",
|
|
"base_url": "https://api.deepseek.com",
|
|
"model": "deepseek-chat",
|
|
"api_key_url": "https://platform.deepseek.com/api_keys",
|
|
},
|
|
{
|
|
"name": "Ollama",
|
|
"base_url": "http://localhost:11434/v1",
|
|
"model": "llama3.2",
|
|
"api_key_url": "https://ollama.com/download",
|
|
},
|
|
{
|
|
"name": "Moonshot",
|
|
"base_url": "https://api.moonshot.cn/v1",
|
|
"model": "moonshot-v1-8k",
|
|
"api_key_url": "https://platform.moonshot.cn/console/api-keys",
|
|
},
|
|
]
|
|
|
|
|
|
def get_preset_names() -> List[str]:
|
|
"""Get list of preset names"""
|
|
return [preset["name"] for preset in LLM_PRESETS]
|
|
|
|
|
|
def get_preset(name: str) -> Dict[str, Any]:
|
|
"""Get preset configuration by name"""
|
|
for preset in LLM_PRESETS:
|
|
if preset["name"] == name:
|
|
return preset
|
|
return {}
|
|
|
|
|
|
def find_preset_by_base_url_and_model(base_url: str, model: str) -> str | None:
|
|
"""
|
|
Find preset name by base_url and model
|
|
|
|
Returns:
|
|
Preset name if found, None otherwise
|
|
"""
|
|
for preset in LLM_PRESETS:
|
|
if preset["base_url"] == base_url and preset["model"] == model:
|
|
return preset["name"]
|
|
return None
|
|
|