zettcode-one-system 0.5.1__py3-none-any.whl

zettcode/one_system/__init__.py

@@ -0,0 +1,12 @@
+"""One system message: every instruction the agent gives the model, joined into the one message before the provider.
+
+A part of the program that lives in a distribution of its own, found through the ``zettcode.plugin`` entry point. It is
+on unless ``[plugins] disable = ["one-system"]`` names it: what it changes is the shape of the request, not what the
+model is told, so leaving it on is what an endpoint that accepts one system message expects.
+"""
+
+from __future__ import annotations
+
+from .plugin import OneSystem
+
+__all__ = ["OneSystem"]

zettcode/one_system/extension.py

@@ -0,0 +1,81 @@
+"""The hook that turns a request's instructions into one system message, and stops there.
+
+A request reaches the provider with several system messages: the agent's own prompt, the working directory, the
+project's instructions, skills, MCP notices, and whatever guidance an extension added for this request. Every one of
+them is an instruction, and every provider has its own way of taking them: the Anthropic-compatible one collects them
+and joins them with a blank line, an OpenAI-compatible one sends a message per entry, and an endpoint that accepts a
+single system message refuses the rest.
+
+Joining them here makes the shape the same everywhere, and does it on the text the model was going to read anyway: the
+parts are concatenated in the order they were already in, separated by a blank line, and the result sits where the first
+one was. Nothing is reordered past a conversation message, nothing is dropped, and a request that already has one
+system message is handed on untouched.
+"""
+
+from __future__ import annotations
+
+from collections.abc import AsyncIterator, Sequence
+from dataclasses import replace
+
+from zett_agent.agent import AgentRunContext
+from zett_agent.extensions.base import AgentExtension, ModelRequestNext
+from zett_agent.messages import AnyMessage, SystemMessage
+from zett_agent.model import ModelEvent, ModelRequest
+
+__all__ = ["OneSystemMessage", "merge_system"]
+
+
+def merge_system(messages: Sequence[AnyMessage]) -> Sequence[AnyMessage]:
+    """Return ``messages`` with every system message joined into the first one's place.
+
+    A request with fewer than two system messages, or with nothing but blank text between them, is returned as it was —
+    the same object, so a caller can tell the two cases apart without comparing contents.
+
+    Args:
+        messages: The request's messages, in the order the provider would see them.
+
+    Returns:
+        The messages with one :class:`~zett_agent.messages.SystemMessage` where the first one was, or ``messages``
+        itself when there was nothing to join.
+    """
+    found = [(index, message) for index, message in enumerate(messages) if isinstance(message, SystemMessage)]
+    if len(found) < 2:
+        return messages
+    joined = "\n\n".join(message.content.strip() for _index, message in found if message.content.strip())
+    if not joined:
+        return messages
+    # The first message's own flags and attributes are kept: only its text changes. `dataclasses.replace` gives a
+    # new message, so the one the runtime holds for later requests is left as it was.
+    where, first_message = found[0]
+    merged = replace(first_message, content=joined)
+    removed = {index for index, _message in found}
+    kept = [message for index, message in enumerate(messages) if index not in removed]
+    # Every removed place is at or after the first, so the first's index still counts among the ones that remain.
+    kept.insert(where, merged)
+    return tuple(kept)
+
+
+class OneSystemMessage(AgentExtension):
+    """Join a request's system messages into one, just before the provider is handed the request.
+
+    The priority is higher than any other extension's so that this runs innermost: a part that adds an instruction while
+    the request is being built — the harness's guidance, for one — has already been added by the time this sees the
+    request, and none is left behind to reach the provider on its own.
+    """
+
+    #: Innermost: the last extension to change the request, and the one closest to the provider. It has to clear the
+    #: two the framework itself runs late — the JIT harness's guidance at 900_000 and plan mode at 1_000_000 — because
+    #: both add a system message while the request is being built, and a tie is broken by registration order.
+    priority = 1_000_001
+
+    async def on_model_request(
+        self, context: AgentRunContext, request: ModelRequest, call_next: ModelRequestNext
+    ) -> AsyncIterator[ModelEvent]:
+        """Hand the provider one system message, or the very request it was given when there is nothing to join."""
+        merged = merge_system(request.messages)
+        if merged is request.messages:
+            async for event in call_next(request):
+                yield event
+            return
+        async for event in call_next(replace(request, messages=merged)):
+            yield event

zettcode/one_system/plugin.py

@@ -0,0 +1,38 @@
+"""The plugin that hands the agent the one-system-message hook."""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from zettcode.engine.plugins.context import PluginContext
+from zettcode.engine.plugins.plugin import Plugin
+
+if TYPE_CHECKING:  # pragma: no cover - annotations only
+    from zett_agent.extensions.base import AgentExtension
+
+    from zettcode.engine.plugins.state import AgentBuild
+
+
+class OneSystem(Plugin[PluginContext]):
+    """Every instruction in one system message, for the endpoints that take one and for the ones that take many.
+
+    The merge runs once per request, at the point the provider would see the messages, so it covers the instructions a
+    part added for that request as well as the ones the agent carries everywhere. It changes nothing a model reads — the
+    same text, in the same order, separated by the blank line the Anthropic-compatible provider already used between
+    them — and it is off for `[plugins] disable = ["one-system"]` like any other part.
+    """
+
+    name = "one-system"
+    description = "All instructions the model is given in one system message"
+
+    def activate(self, ctx: PluginContext) -> None:
+        """Register the extension as a factory: it is built with the agent, not while the first frame is drawn."""
+
+        # Imported in the factory: the extension imports the framework's message and hook types, which a launch that
+        # never builds an agent does not need.
+        def build(_: AgentBuild) -> AgentExtension:
+            from .extension import OneSystemMessage
+
+            return OneSystemMessage()
+
+        ctx.agent.register_extension_factory("OneSystemMessage", build)

zettcode_one_system-0.5.1.dist-info/METADATA

@@ -0,0 +1,19 @@
+Metadata-Version: 2.5
+Name: zettcode-one-system
+Version: 0.5.1
+Summary: Every instruction ZettCode gives the model in one system message, as a part of the agent that lives apart from it
+Project-URL: Repository, https://github.com/Chang-LeHung/zettcode
+Author: Chang-LeHung
+License-Expression: MIT
+Keywords: agent,prompt,system-message,zettcode
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Programming Language :: Python :: 3.15
+Classifier: Typing :: Typed
+Requires-Python: >=3.10
+Requires-Dist: zett-agent<0.3,>=0.2.10
+Requires-Dist: zettcode-engine<0.6,>=0.5.1

zettcode_one_system-0.5.1.dist-info/RECORD

@@ -0,0 +1,8 @@
+zettcode/one_system/__init__.py,sha256=2OsABcOsB5tyjXWdYpESd8vkvHZiB3_eWDNQIObJktY,544
+zettcode/one_system/extension.py,sha256=RbmJ0i4kNdAqIoSozIoVcK--wKQyYkp-Y2inWwB6_-I,4344
+zettcode/one_system/plugin.py,sha256=7zDk5xNrK9phckqaejeVqZJ1y-SCljoUMxNhkCppoMM,1684
+zettcode/one_system/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+zettcode_one_system-0.5.1.dist-info/METADATA,sha256=Zurg4BsYInnnPNvYtBx3I_vlPRgu1Fu3AqPvEP0l6jM,840
+zettcode_one_system-0.5.1.dist-info/WHEEL,sha256=2iZrJkDIkeDh0k9SikKMqFuRP-ua6mb4Ir1AsER4BEw,87
+zettcode_one_system-0.5.1.dist-info/entry_points.txt,sha256=N_kR-dJzRKscTVr6x1tqrUmxXy0CFV0OQtwnAJyvqJE,68
+zettcode_one_system-0.5.1.dist-info/RECORD,,

zettcode_one_system-0.5.1.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.33.0
+Root-Is-Purelib: true
+Tag: py3-none-any

zettcode_one_system-0.5.1.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[zettcode.plugin]
+one-system = zettcode.one_system.plugin:OneSystem