Feat: 调用本地 Ollama

2026-10-12 11:28:37 +08:00
parent c1a3461c49
commit 5d7228ebed
6 changed files with 98 additions and 43 deletions
@@ -36,12 +36,13 @@ load_dotenv()

 def llm_config() -> dict | None:
    key = os.getenv("OPENAI_API_KEY", "").strip()
-    if not key:
+    openai_base = os.getenv("OPENAI_BASE_URL", "").strip()
+    if key and openai_base:
+        model = os.getenv("OPENAI_MODEL", "gpt-4o-mini").strip() or "gpt-4o-mini"
+        return {"provider": "openai", "key": key, "base": openai_base.rstrip("/"), "model": model}
+
+    base = os.getenv("OLLAMA_BASE_URL", "http://10.7.9.3:11434").strip()
+    if not base:
        return None
-    base = os.getenv("OPENAI_BASE_URL", "https://api.openai.com/v1").strip()
-    model = os.getenv("OPENAI_MODEL", "gpt-4o-mini").strip()
-    return {
-        "key": key,
-        "base": (base or "https://api.openai.com/v1").rstrip("/"),
-        "model": model or "gpt-4o-mini",
-    }
+    model = os.getenv("OLLAMA_MODEL", "qwen3-vl:4b").strip() or "qwen3-vl:4b"
+    return {"provider": "ollama", "key": "ollama", "base": base.rstrip("/"), "model": model}