refactor: unify host threadpool boundary

app/agent/callback/__init__.py

@@ -3,8 +3,7 @@ import re
 import threading
 from typing import Any, Optional, Tuple
 
-from fastapi.concurrency import run_in_threadpool
-
+from app.runtime.execution import run_in_threadpool
 from app.agent.policy.sanitizer import sanitize_for_host
 from app.chain import ChainBase
 from app.runtime.log import logger

app/agent/orchestrator.py

@@ -11,7 +11,6 @@ from dataclasses import dataclass
 from datetime import datetime, timedelta
 from typing import Any, Awaitable, Callable, Dict, List, Optional
 
-from fastapi.concurrency import run_in_threadpool
 from langchain.agents import create_agent
 from langchain_core.messages import (  # noqa: F401
     HumanMessage,
@@ -21,6 +20,7 @@ from langchain_core.messages import (  # noqa: F401
 
 from langgraph.checkpoint.memory import InMemorySaver
 
+from app.runtime.execution import run_in_threadpool
 from app.agent.callback import StreamingHandler
 from app.agent.contracts import ReplyMode, build_display_message
 from app.agent.llm.helper import LLMHelper

app/api/endpoints/agent.py

@@ -14,9 +14,9 @@ from typing import Any, AsyncIterator, Awaitable, Callable, Optional, Union
 
 import aiofiles
 from fastapi import Depends, File, Form, HTTPException, Request, UploadFile, status
-from fastapi.concurrency import run_in_threadpool
 from fastapi.responses import FileResponse, StreamingResponse
 
+from app.runtime.execution import run_in_threadpool
 from app.schemas.agent import AgentChatDisplaySaveRequest as _SchemaAgentChatDisplaySaveRequest
 from app.schemas.agent import AgentChatSessionDetail as _SchemaAgentChatSessionDetail
 from app.schemas.agent import AgentChatSessionSummary as _SchemaAgentChatSessionSummary

app/api/endpoints/dashboard.py

@@ -2,8 +2,8 @@ from pathlib import Path
 from typing import Any, List, Optional, Annotated
 
 from fastapi import Depends
-from fastapi.concurrency import run_in_threadpool
 
+from app.runtime.execution import run_in_threadpool
 from app.schemas.dashboard import DashboardMemoryInfo as _SchemaDashboardMemoryInfo
 from app.schemas.dashboard import DashboardSystemInfo as _SchemaDashboardSystemInfo
 from app.schemas.dashboard import DownloaderInfo as _SchemaDownloaderInfo

app/api/endpoints/plugin.py

@@ -6,10 +6,10 @@ from typing import Annotated, Any, Dict, List, Optional
 import aiofiles
 from anyio import Path as AsyncPath
 from fastapi import Depends, Header, HTTPException, Security
-from fastapi.concurrency import run_in_threadpool
 from starlette import status
 from starlette.responses import StreamingResponse
 
+from app.runtime.execution import run_in_threadpool
 from app.schemas.common import JsonObject as _SchemaJsonObject
 from app.schemas.plugin import Plugin as _SchemaPlugin
 from app.schemas.plugin import PluginDashboard as _SchemaPluginDashboard

app/chain/_recognition.py

@@ -7,8 +7,7 @@
 import copy
 from typing import Optional
 
-from fastapi.concurrency import run_in_threadpool
-
+from app.runtime.execution import run_in_threadpool
 from app.adapters.external.server import MoviePilotServerHelper
 from app.application.configuration import get_configured_system_config
 from app.domain.context import MediaInfo, MusicInfo

app/chain/media.py

@@ -3,8 +3,7 @@ from pathlib import Path
 from threading import Lock
 from typing import Any, Iterable, List, Optional, Tuple, Union
 
-from fastapi.concurrency import run_in_threadpool
-
+from app.runtime.execution import run_in_threadpool
 from app.schemas.event import MediaRecognizeConvertEventData as _SchemaMediaRecognizeConvertEventData
 from app.chain import ChainBase
 from app.chain.acoustid import AcoustIdChain

app/chain/search.py

@@ -10,8 +10,7 @@ from typing import AsyncIterator, Any, Dict, Iterable, Tuple
 from typing import List, Optional
 from unicodedata import normalize
 
-from fastapi.concurrency import run_in_threadpool
-
+from app.runtime.execution import run_in_threadpool
 from app.chain import ChainBase
 from app.chain.media import MediaChain
 from app.runtime.config import global_vars

app/modules/acoustid/__init__.py

@@ -9,8 +9,7 @@ from pathlib import Path
 from typing import Any, Optional, Tuple, Union
 from uuid import UUID
 
-from fastapi.concurrency import run_in_threadpool
-
+from app.runtime.execution import run_in_threadpool
 from app.runtime.settings import RuntimeSettingsCompat
 
 settings = RuntimeSettingsCompat()

app/modules/discord/discord.py

@@ -6,8 +6,8 @@ from typing import Optional, List, Dict, Any, Tuple, Union
 
 import discord
 from discord import app_commands
-from fastapi.concurrency import run_in_threadpool
 
+from app.runtime.execution import run_in_threadpool
 from app.runtime.settings import RuntimeSettingsCompat
 
 settings = RuntimeSettingsCompat()

app/modules/indexer/spider/__init__.py

@@ -5,10 +5,10 @@ from typing import Any, Optional
 from typing import List
 from urllib.parse import quote, urlparse, parse_qs
 
-from fastapi.concurrency import run_in_threadpool
 from jinja2 import Template
 from pyquery import PyQuery
 
+from app.runtime.execution import run_in_threadpool
 from app.runtime.settings import RuntimeSettingsCompat
 
 settings = RuntimeSettingsCompat()