feat: add openai_base_url for DeepSeek/Ollama/custom endpoint support

Generated with [Claude Code](https://claude.ai/code)
via [Happy](https://happy.engineering)

Co-Authored-By: Claude <noreply@anthropic.com>
Co-Authored-By: Happy <yesreply@happy.engineering>

Author: 3 people

.env.example

@@ -9,6 +9,7 @@ MINIFLUX_API_KEY=
 AI_PROVIDER=claude
 ANTHROPIC_API_KEY=
 OPENAI_API_KEY=
+OPENAI_BASE_URL=              # custom endpoint: https://api.deepseek.com/v1, etc.
 
 # Telegram
 TELEGRAM_BOT_TOKEN=

src/muse/analyzer/ai_client.py

@@ -11,7 +11,7 @@
 logger = structlog.get_logger()
 
 CLAUDE_API_URL = "https://api.anthropic.com/v1/messages"
-OPENAI_API_URL = "https://api.openai.com/v1/chat/completions"
+DEFAULT_OPENAI_BASE_URL = "https://api.openai.com/v1"
 
 
 class AIRequestError(Exception):
@@ -23,6 +23,7 @@ class AIClient:
     provider: str  # "claude" | "openai"
     api_key: str
     model: str = ""
+    base_url: str = ""
     max_retries: int = 3
     base_delay: float = 2.0
 
@@ -109,9 +110,14 @@ async def _call_claude(
     async def _call_openai(
         self, system_prompt: str, user_prompt: str
     ) -> tuple[str, dict]:
+        url = (
+            f"{self.base_url.rstrip('/')}/chat/completions"
+            if self.base_url
+            else f"{DEFAULT_OPENAI_BASE_URL}/chat/completions"
+        )
         async with httpx.AsyncClient(timeout=60.0) as client:
             resp = await client.post(
-                OPENAI_API_URL,
+                url,
                 headers={
                     "Authorization": f"Bearer {self.api_key}",
                     "Content-Type": "application/json",

src/muse/config.py

@@ -20,6 +20,7 @@ class Settings(BaseSettings):
     ai_provider: str = "claude"  # claude | openai
     anthropic_api_key: str = ""
     openai_api_key: str = ""
+    openai_base_url: str = ""  # custom endpoint for DeepSeek, Ollama, etc.
 
     # Telegram
     telegram_bot_token: str

src/muse/scheduler.py

@@ -90,7 +90,9 @@ async def collect_signals_job(
         if settings.ai_provider == "claude"
         else settings.openai_api_key
     )
-    ai_client = AIClient(provider=settings.ai_provider, api_key=api_key)
+    ai_client = AIClient(
+        provider=settings.ai_provider, api_key=api_key, base_url=settings.openai_base_url
+    )
     detector = SignalDetector(
         ai_client=ai_client,
         system_prompt_path=str(PROMPTS_DIR / "signal_detection_system.txt"),
@@ -201,7 +203,9 @@ async def extract_opportunities_job(
         if settings.ai_provider == "claude"
         else settings.openai_api_key
     )
-    ai_client = AIClient(provider=settings.ai_provider, api_key=api_key)
+    ai_client = AIClient(
+        provider=settings.ai_provider, api_key=api_key, base_url=settings.openai_base_url
+    )
     extractor = OpportunityExtractor(
         ai_client=ai_client,
         system_prompt_path=str(PROMPTS_DIR / "opportunity_extraction_system.txt"),
@@ -331,7 +335,9 @@ async def generate_ideas_job(
         if settings.ai_provider == "claude"
         else settings.openai_api_key
     )
-    ai_client = AIClient(provider=settings.ai_provider, api_key=api_key)
+    ai_client = AIClient(
+        provider=settings.ai_provider, api_key=api_key, base_url=settings.openai_base_url
+    )
     generator = IdeaGenerator(
         ai_client=ai_client,
         system_prompt_path=str(PROMPTS_DIR / "idea_generation_system.txt"),

tests/conftest.py

@@ -10,6 +10,7 @@
 os.environ.setdefault("SMTP_USER", "")
 os.environ.setdefault("SMTP_PASSWORD", "")
 os.environ.setdefault("EMAIL_RECIPIENTS", "")
+os.environ.setdefault("OPENAI_BASE_URL", "")
 os.environ.setdefault("NOTION_API_KEY", "")
 os.environ.setdefault("NOTION_IDEAS_DATABASE_ID", "")
 os.environ.setdefault("MONTHLY_SCHEDULE_DAY", "1")