diff --git a/app/agent/__init__.py b/app/agent/__init__.py index a92a7392..96ef40f7 100644 --- a/app/agent/__init__.py +++ b/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, - max_retries=3, - temperature=settings.LLM_TEMPERATURE, - streaming=True, - callbacks=[self.callback_handler] - ) + + # 使用线程锁保护的临时环境变量配置 + @contextmanager + def _temp_proxy_env(): + """线程安全的临时设置代理环境变量的上下文管理器""" + with _env_lock: + 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: