Merge pull request #5152 from Pollo3470/feat/ai-proxy

app/agent/__init__.py

@@ -1,6 +1,7 @@
 """MoviePilot AI智能体实现"""
 
 import asyncio
+import threading
 from typing import Dict, List, Any
 
 from langchain.agents import AgentExecutor, create_openai_tools_agent
@@ -20,6 +21,9 @@ from app.helper.message import MessageHelper
 from app.log import logger
 from app.schemas import Notification
 
+# 用于保护环境变量修改的线程锁
+_env_lock = threading.Lock()
+
 
 class AgentChain(ChainBase):
     pass
@@ -73,15 +77,44 @@ class MoviePilotAgent:
             raise ValueError("未配置 LLM_API_KEY")
 
         if provider == "google":
+            import os
+            from contextlib import contextmanager
             from langchain_google_genai import ChatGoogleGenerativeAI
-            return ChatGoogleGenerativeAI(
+
-                model=settings.LLM_MODEL,
+            # 使用线程锁保护的临时环境变量配置
-                google_api_key=api_key,
+            @contextmanager
-                max_retries=3,
+            def _temp_proxy_env():
-                temperature=settings.LLM_TEMPERATURE,
+                """线程安全的临时设置代理环境变量的上下文管理器"""
-                streaming=True,
+                with _env_lock:
-                callbacks=[self.callback_handler]
+                    old_http = os.environ.get("HTTP_PROXY")
-            )
+                    old_https = os.environ.get("HTTPS_PROXY")
+                    try:
+                        if settings.PROXY_HOST:
+                            os.environ["HTTP_PROXY"] = settings.PROXY_HOST
+                            os.environ["HTTPS_PROXY"] = settings.PROXY_HOST
+                        yield
+                    finally:
+                        # 恢复原始环境变量
+                        if old_http is not None:
+                            os.environ["HTTP_PROXY"] = old_http
+                        elif "HTTP_PROXY" in os.environ:
+                            del os.environ["HTTP_PROXY"]
+
+                        if old_https is not None:
+                            os.environ["HTTPS_PROXY"] = old_https
+                        elif "HTTPS_PROXY" in os.environ:
+                            del os.environ["HTTPS_PROXY"]
+
+            # 在临时环境变量中初始化 ChatGoogleGenerativeAI
+            with _temp_proxy_env():
+                return ChatGoogleGenerativeAI(
+                    model=settings.LLM_MODEL,
+                    google_api_key=api_key,
+                    max_retries=3,
+                    temperature=settings.LLM_TEMPERATURE,
+                    streaming=True,
+                    callbacks=[self.callback_handler]
+                )
         elif provider == "deepseek":
             from langchain_deepseek import ChatDeepSeek
             return ChatDeepSeek(
@@ -103,7 +136,8 @@ class MoviePilotAgent:
                 temperature=settings.LLM_TEMPERATURE,
                 streaming=True,
                 callbacks=[self.callback_handler],
-                stream_usage=True
+                stream_usage=True,
+                openai_proxy=settings.PROXY_HOST
             )
 
     def _initialize_tools(self) -> List: