init

reelforge/llm_presets.py

@@ -0,0 +1,74 @@
+"""
+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
+