from datetime import UTC, datetime from sqlalchemy import select from sqlalchemy.ext.asyncio import AsyncSession from app.infrastructure.db import SessionFactory from app.model.configuration import ( AgentIntentConfig, ConfigRelease, PlatformConfigItem, PromptTemplateVersion, ) from app.service.intent_classifier import IntentConfigEntry class RuntimeConfigService: """Reads only configuration rows bound to the caller's released version.""" def __init__(self, session: AsyncSession) -> None: self.session = session async def prompt( self, release_id: int, prompt_code: str, task_type: str, agent_type: str | None ) -> PromptTemplateVersion | None: result = await self.session.scalar( select(PromptTemplateVersion).where( PromptTemplateVersion.release_id == release_id, PromptTemplateVersion.prompt_code == prompt_code, PromptTemplateVersion.task_type == task_type, (PromptTemplateVersion.agent_type == agent_type) | PromptTemplateVersion.agent_type.is_(None), ) ) return result async def intent(self, agent_type: str, intent_code: str) -> AgentIntentConfig | None: result = await self.session.scalar( select(AgentIntentConfig).where( AgentIntentConfig.agent_type == agent_type, AgentIntentConfig.intent_code == intent_code, AgentIntentConfig.status == "active", ) ) return result async def active_intents(self, agent_type: str) -> tuple[IntentConfigEntry, ...]: """读取某 Agent 当前生效的意图配置,供意图分类链路使用。 生效判定与 `intent()` 一致:`status='active'`;另外尊重 `effective_at/expire_at` 的有效期窗口(未填写视为立即生效、不过期)。该表通过生成列唯一键 `uk_intent_config_active_one` 保证同一 `agent_type:intent_code` 最多一个 active 版本,因此按 intent_code 匹配不会歧义;仍按 `priority` 排序以保证结果稳定。 """ now = datetime.now(UTC).replace(tzinfo=None) rows = await self.session.scalars( select(AgentIntentConfig) .where( AgentIntentConfig.agent_type == agent_type, AgentIntentConfig.status == "active", (AgentIntentConfig.effective_at.is_(None)) | (AgentIntentConfig.effective_at <= now), (AgentIntentConfig.expire_at.is_(None)) | (AgentIntentConfig.expire_at > now), ) .order_by(AgentIntentConfig.priority, AgentIntentConfig.intent_code) ) return tuple(self._entry(row) for row in rows) @staticmethod def _entry(row: AgentIntentConfig) -> IntentConfigEntry: return IntentConfigEntry( intent_code=row.intent_code, intent_name=row.intent_name or "", description=row.description, examples=tuple(str(item) for item in (row.examples or ())), classifier_instruction=row.classifier_instruction, confidence_threshold=float(row.confidence_threshold), ) async def allowed_tools( self, release_id: int, agent_type: str, intent_code: str ) -> tuple[str, ...]: config_key = f"{agent_type}:{intent_code}" item = await self.session.scalar( select(PlatformConfigItem).where( PlatformConfigItem.release_id == release_id, PlatformConfigItem.namespace == "agent_tools", PlatformConfigItem.config_key == config_key, ) ) if item is None: return () raw_tools = item.value_json.get("allowed_tools", []) if not isinstance(raw_tools, list) or not all(isinstance(tool, str) for tool in raw_tools): raise ValueError("invalid tool whitelist configuration") return tuple(raw_tools) async def fund_quote(self, release_id: int) -> dict[str, object]: """读取已发布行情配置;缺失时返回空映射,由 Service 使用安全默认值。""" item = await self.session.scalar( select(PlatformConfigItem).where( PlatformConfigItem.release_id == release_id, PlatformConfigItem.namespace == "fund_market", PlatformConfigItem.config_key == "default", ) ) return item.value_json if item is not None else {} async def load_fund_quote_config() -> dict[str, object]: """读取当前激活行情配置;配置中心不可用时交给调用方使用默认值。""" async with SessionFactory() as session: release = await session.scalar( select(ConfigRelease).where(ConfigRelease.status == "active") ) if release is None: return {} return await RuntimeConfigService(session).fund_quote(release.id) async def load_active_intent_configs(agent_type: str) -> tuple[IntentConfigEntry, ...]: """意图分类链路的运行期配置装载器:只读 `agent_intent_config` 的 active 行。 该表不绑定 `config_release`(没有 `release_id` 外键),因此这里按 Agent 直接读取; 没有配置时返回空元组,分类退化为代码声明的意图清单——**行为与接入前一致**。 """ async with SessionFactory() as session: return await RuntimeConfigService(session).active_intents(agent_type)