rfnry-engine 0.1.0__py3-none-any.whl

rfnry_engine/__init__.py

@@ -0,0 +1,152 @@
+from rfnry_engine.engine import AgentInfo, Engine
+from rfnry_engine.errors import (
+    ConfigurationError,
+    DelegationDepthError,
+    DelegationLoopError,
+    EngineException,
+    InputError,
+    ModelError,
+    NotFoundError,
+    RunawayError,
+    StorageError,
+)
+from rfnry_engine.jail import Jail
+from rfnry_engine.models import (
+    Answer,
+    AnswerLike,
+    Audio,
+    Check,
+    CheckAnswer,
+    Choice,
+    ChoiceAnswer,
+    Content,
+    Decided,
+    DecidedLike,
+    DecisionModel,
+    Document,
+    GenerationModel,
+    HistoryLike,
+    Image,
+    Message,
+    Part,
+    Question,
+    QuestionLike,
+    Reply,
+    ReplyLike,
+    Score,
+    ScoreAnswer,
+    State,
+    StreamingGenerationModel,
+    Text,
+    ToolCall,
+    ToolSpec,
+    Usage,
+    confidence,
+)
+from rfnry_engine.records import (
+    Decision,
+    Event,
+    Fingerprint,
+    Generation,
+    Run,
+    Step,
+    TextDelta,
+    ToolCallStarted,
+    ToolResultReady,
+)
+from rfnry_engine.refinery.export import Bucket
+from rfnry_engine.refinery.holdout import is_holdout
+from rfnry_engine.refinery.labels import (
+    Conflict,
+    Exported,
+    Label,
+    Labeled,
+    Outcome,
+    Replay,
+    Review,
+)
+from rfnry_engine.refinery.postgres import PostgresStorage
+from rfnry_engine.refinery.refinery import Refinery
+from rfnry_engine.refinery.storage import LabelRow, RunRow, Storage
+from rfnry_engine.stop import Stop
+from rfnry_engine.tools.engine_tools import EngineTools
+from rfnry_engine.tools.tools import Tool, Tools
+from rfnry_engine.tools.workspace import Backend, Local, Process, Workspace
+
+__version__ = "0.1.0"
+
+__all__ = [
+    "AgentInfo",
+    "Answer",
+    "AnswerLike",
+    "Bucket",
+    "Check",
+    "CheckAnswer",
+    "Choice",
+    "ChoiceAnswer",
+    "ConfigurationError",
+    "Conflict",
+    "Decided",
+    "DecidedLike",
+    "Decision",
+    "DecisionModel",
+    "DelegationDepthError",
+    "DelegationLoopError",
+    "Engine",
+    "EngineException",
+    "EngineTools",
+    "Event",
+    "Exported",
+    "Fingerprint",
+    "Generation",
+    "GenerationModel",
+    "HistoryLike",
+    "InputError",
+    "Jail",
+    "Label",
+    "LabelRow",
+    "Labeled",
+    "Message",
+    "ModelError",
+    "NotFoundError",
+    "Outcome",
+    "PostgresStorage",
+    "Question",
+    "QuestionLike",
+    "Refinery",
+    "Replay",
+    "Reply",
+    "ReplyLike",
+    "Review",
+    "Run",
+    "RunRow",
+    "RunawayError",
+    "Score",
+    "ScoreAnswer",
+    "State",
+    "Step",
+    "Stop",
+    "Storage",
+    "StorageError",
+    "StreamingGenerationModel",
+    "TextDelta",
+    "Tool",
+    "ToolCall",
+    "ToolCallStarted",
+    "ToolResultReady",
+    "ToolSpec",
+    "Tools",
+    "Usage",
+    "Workspace",
+    "Backend",
+    "Local",
+    "Process",
+    "confidence",
+    "is_holdout",
+    "Text",
+    "Image",
+    "Audio",
+    "Document",
+    "Part",
+    "Content",
+]

rfnry_engine/bundle.py

@@ -0,0 +1,35 @@
+from __future__ import annotations
+
+from pathlib import Path
+
+from rfnry_engine.models import ToolSpec
+from rfnry_engine.tree import AgentSpec
+
+
+def _read(path: Path) -> str:
+    return path.read_text(encoding="utf-8").strip() if path.is_file() else ""
+
+
+def build_bundle(
+    spec: AgentSpec,
+    *,
+    task: str | None,
+    tools: list[ToolSpec],
+    agents: list[AgentSpec],
+    context: str | None,
+) -> str:
+    sections = [
+        ("AGENT", spec.body),
+        ("TASK", "" if task is None else _read(spec.root / "tasks" / f"{task}.md")),
+        ("FILES", spec.listing),
+        (
+            "TOOLS",
+            "\n".join(
+                f"- {t.name}: {t.description.splitlines()[0] if t.description else ''}"
+                for t in tools
+            ),
+        ),
+        ("AGENTS", "\n".join(f"- {a.name}: {a.description}" for a in agents)),
+        ("CONTEXT", (context or "").strip()),
+    ]
+    return "\n\n".join(f"<<<{label}>>>\n{body}" for label, body in sections if body)

rfnry_engine/chain.py

@@ -0,0 +1,37 @@
+from __future__ import annotations
+
+import hashlib
+import json
+
+from rfnry_engine.models import Message, Part, ToolSpec
+
+
+def opening(model: str | None, system: str, tools: list[ToolSpec], preload: list[Part]) -> bytes:
+    head = {
+        "model": model,
+        "system": system,
+        "tools": [tool.model_dump(mode="json") for tool in tools],
+        "preload": [part.model_dump(mode="json") for part in preload],
+    }
+    return hashlib.sha256(json.dumps(head, sort_keys=True, ensure_ascii=False).encode()).digest()
+
+
+def extend(digest: bytes, message: Message) -> bytes:
+    said = message.model_dump(mode="json", exclude={"prefix", "usage"}, fallback=repr)
+    return hashlib.sha256(
+        digest + json.dumps(said, sort_keys=True, ensure_ascii=False).encode()
+    ).digest()
+
+
+def mark(digest: bytes) -> str:
+    return digest.hex()[:16]
+
+
+def settle(history: list[Message], digest: bytes) -> tuple[list[Message], bytes]:
+    sent: list[Message] = []
+    for message in history:
+        if message.reasoning is not None and message.prefix != mark(digest):
+            message = message.model_copy(update={"reasoning": None})
+        sent.append(message)
+        digest = extend(digest, message)
+    return sent, digest