feat(vision): il runner, unica implementazione dell'esecuzione
Aggiunge il sottomodulo vendor/visionsuite (pin f095fcc), src/vision/runner.py (run_graph/engine_version, pura libreria: niente HTTP, niente DB) e i test. Deviazione dal brief: vs-task/vs-measure/vs-pm2d dichiarano requires-python>=3.13 a monte, mentre il monorepo resta a 3.11. L'extra `vision` porta marker `python_version>='3.13'` cosi' la sync di base (server+client+dev, 3.11) resta intatta; per lavorare sulla vision serve `uv sync --extra vision --extra dev --python 3.13`. Ricade sul Task 3: il worker (Dockerfile.vision) non puo' partire da python:3.11-slim come previsto dal suo brief, deve usare 3.13.
This commit is contained in:
@@ -0,0 +1,91 @@
|
||||
"""Graph plus image in, numbers out.
|
||||
|
||||
The one implementation of vision execution: the worker beside the server and
|
||||
the station agent both call this, and they must agree to the digit. No HTTP, no
|
||||
database, no filesystem beyond reading the pinned submodule's commit.
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
import subprocess
|
||||
import time
|
||||
from dataclasses import dataclass, field
|
||||
from functools import lru_cache
|
||||
from pathlib import Path
|
||||
from typing import Any
|
||||
|
||||
import numpy as np
|
||||
from visionsuite.core.image import Image
|
||||
from visionsuite.task import Task
|
||||
|
||||
_SUBMODULE = Path(__file__).resolve().parents[2] / "vendor" / "visionsuite"
|
||||
|
||||
|
||||
@dataclass
|
||||
class ToolFailure:
|
||||
tool_id: str
|
||||
tool_name: str
|
||||
error: str
|
||||
|
||||
|
||||
@dataclass
|
||||
class RunOutcome:
|
||||
outputs: dict[str, float] = field(default_factory=dict)
|
||||
failures: list[ToolFailure] = field(default_factory=list)
|
||||
engine_version: str = ""
|
||||
duration_ms: int = 0
|
||||
|
||||
|
||||
@lru_cache(maxsize=1)
|
||||
def engine_version() -> str:
|
||||
"""The pinned VisionSuite commit: stamped on every number produced."""
|
||||
return subprocess.check_output(
|
||||
["git", "-C", str(_SUBMODULE), "rev-parse", "HEAD"],
|
||||
text=True,
|
||||
).strip()
|
||||
|
||||
|
||||
def run_graph(
|
||||
image: Any,
|
||||
graph: dict,
|
||||
calibration: dict | None = None,
|
||||
) -> RunOutcome:
|
||||
"""Run a serialised vs-task graph against one image.
|
||||
|
||||
A tool that fails does not stop the others - on a crooked piece an edge may
|
||||
simply not be there, and the other quotes are still wanted. Failures come
|
||||
back named, they are not swallowed.
|
||||
|
||||
`image` is whatever the caller already has in hand: a plain numpy array is
|
||||
wrapped into the `visionsuite.core.image.Image` that vs-measure's tools
|
||||
actually read via `.pixels`; an `Image` passed in is used as-is.
|
||||
"""
|
||||
task = Task.from_dict(graph)
|
||||
wrapped_image = image if isinstance(image, Image) else Image(pixels=np.asarray(image))
|
||||
context: dict[str, Any] = {"image": wrapped_image}
|
||||
if calibration is not None:
|
||||
context["calibration"] = calibration
|
||||
|
||||
started = time.monotonic()
|
||||
report = task.run(context=context)
|
||||
elapsed_ms = int((time.monotonic() - started) * 1000)
|
||||
|
||||
outputs: dict[str, float] = {}
|
||||
failures: list[ToolFailure] = []
|
||||
for result in report:
|
||||
if not result.ok:
|
||||
failures.append(ToolFailure(
|
||||
tool_id=result.tool_id,
|
||||
tool_name=result.tool_name,
|
||||
error=result.error or "unknown failure",
|
||||
))
|
||||
continue
|
||||
for name, value in result.outputs.items():
|
||||
if isinstance(value, (int, float)):
|
||||
outputs[f"{result.tool_id}.{name}"] = float(value)
|
||||
|
||||
return RunOutcome(
|
||||
outputs=outputs,
|
||||
failures=failures,
|
||||
engine_version=engine_version(),
|
||||
duration_ms=elapsed_ms,
|
||||
)
|
||||
Reference in New Issue
Block a user