104 lines
4.6 KiB
Python
104 lines
4.6 KiB
Python
# ============================================================
|
||
# 外部算法引擎适配器（moduleId: engines-external, 可重生 ✅）
|
||
# ISchedulingEngine：组中性 DTO → HTTP/local skill → 写回 world
|
||
# ============================================================
|
||
from __future__ import annotations
|
||
|
||
from typing import Any
|
||
|
||
import httpx
|
||
|
||
from server.aps_domain.scheduling_dto import (
|
||
SchedulingSolution, apply_flex_solution, world_to_flex_problem,
|
||
)
|
||
from server.contracts import ScheduleResult
|
||
from server.engines.base import EngineParams, ISchedulingEngine
|
||
|
||
|
||
class ExternalEngine(ISchedulingEngine):
|
||
"""外部算法排产：通过 SkillRegistry 选定 skill，HTTP 或 local://stub。"""
|
||
|
||
name = "EXTERNAL"
|
||
supports_anytime = False
|
||
|
||
def __init__(self, skill_id: str | None = None) -> None:
|
||
self.skill_id = skill_id
|
||
|
||
def solve(self, world: dict[str, Any], params: EngineParams, next_id) -> ScheduleResult:
|
||
from server.agent_core.skills import get_skills
|
||
|
||
skill_id = self.skill_id or (params.meta_skill_id if hasattr(params, "meta_skill_id") else None)
|
||
# EngineParams 无自定义字段时从 constraints 或 name 透传；workflow 会设 world 临时键
|
||
skill_id = skill_id or world.pop("_externalSkillId", None)
|
||
track = "flex" # 本切片外部 skill 主路径走柔性轨
|
||
reg = get_skills()
|
||
skill = reg.get_enabled(skill_id, track=track) or reg.get_enabled(skill_id) or reg.get_enabled()
|
||
if not skill:
|
||
raise ValueError("没有可用的外部算法 skill（请先登记并启用）")
|
||
|
||
order_ids = list(params.orderIds or []) or None
|
||
problem = world_to_flex_problem(
|
||
world, order_ids=order_ids,
|
||
start_time=params.startDate + " 08:00" if params.startDate and " " not in params.startDate
|
||
else params.startDate,
|
||
)
|
||
solution = self._call_skill(skill, problem.model_dump())
|
||
summary = apply_flex_solution(world, solution, next_id, skill_id=skill["skill_id"])
|
||
|
||
return ScheduleResult(
|
||
versionId=summary["versionId"],
|
||
versionNo=summary["versionNo"],
|
||
engineType="EXTERNAL",
|
||
strategy=f"EXTERNAL:{skill['skill_id']}",
|
||
status="DRAFT",
|
||
orderCount=len(problem.orders),
|
||
poCount=summary["vlCount"],
|
||
woCount=summary["woCount"],
|
||
conflictCount=summary["conflictCount"],
|
||
totalTardiness=float((solution.kpi or {}).get("totalTardiness") or 0),
|
||
avgUtilization=float(summary.get("avgUtilization") or 0),
|
||
evidenceRefs=[f"skill:{skill['skill_id']}", f"run:{solution.runId}"],
|
||
solveStatus=solution.status,
|
||
solveTimeSec=float((solution.solverMeta or {}).get("timeSec") or 0) or None,
|
||
)
|
||
|
||
def _call_skill(self, skill: dict[str, Any], problem: dict) -> SchedulingSolution:
|
||
ep = skill.get("endpoint") or ""
|
||
if ep.startswith("local://"):
|
||
from server.integrations.algo_skill_stub import solve_problem
|
||
return solve_problem(problem)
|
||
url = ep.rstrip("/") + "/schedule"
|
||
headers = {"Content-Type": "application/json"}
|
||
if skill.get("auth"):
|
||
headers["Authorization"] = f"Bearer {skill['auth']}"
|
||
timeout = float(skill.get("timeout_sec") or 60)
|
||
try:
|
||
with httpx.Client(timeout=timeout) as client:
|
||
resp = client.post(url, json=problem, headers=headers)
|
||
resp.raise_for_status()
|
||
return SchedulingSolution(**resp.json())
|
||
except Exception as exc:
|
||
raise ValueError(f"外部算法 skill「{skill.get('name')}」调用失败：{exc}") from exc
|
||
|
||
|
||
def run_external_flex(store, *, skill_id: str | None = None,
|
||
order_ids: list[int] | None = None, actor: str = "planner") -> dict[str, Any]:
|
||
"""供 workflow 直接调用的柔性外部排产（不经固定轨 ScheduleResult 路径）。"""
|
||
from server.agent_core.skills import get_skills
|
||
from server.timeutil import add_minutes, fmt_dt, today0
|
||
|
||
reg = get_skills()
|
||
skill = reg.get_enabled(skill_id, track="flex") or reg.get_enabled(skill_id) or reg.get_enabled()
|
||
if not skill:
|
||
raise ValueError("没有可用的外部算法 skill")
|
||
|
||
problem = world_to_flex_problem(
|
||
store.data, order_ids=order_ids,
|
||
start_time=fmt_dt(add_minutes(today0(), 24 * 60)),
|
||
)
|
||
eng = ExternalEngine(skill_id=skill["skill_id"])
|
||
solution = eng._call_skill(skill, problem.model_dump())
|
||
summary = apply_flex_solution(store.data, solution, store.next_id, skill_id=skill["skill_id"])
|
||
store.save()
|
||
return summary
|