zettcode-one-system 0.5.1__tar.gz

zettcode_one_system-0.5.1/.gitignore

@@ -0,0 +1,40 @@
+**/.venv/
+**/__pycache__/
+*.py[cod]
+*.egg-info/
+build/
+dist/
+.smoke/
+.coverage
+**/.pytest_cache/
+**/.ruff_cache/
+.zett-tool-output/
+.env
+.DS_Store
+
+# viztracer writes its trace here by default.
+/result.json
+
+# The docs site: dependencies, build output, and VitePress' cache.
+node_modules/
+docs/.vitepress/dist/
+docs/.vitepress/cache/
+docs/.vitepress/.temp/
+
+# The TypeScript client is built from its sources; only dist/ is published.
+clients/typescript/dist/
+
+# The gateway's page is built from clients/gateway into the package when a distribution is made (`make gateway-ui`);
+# the build output is not kept in the repository. pyproject's `artifacts` puts it in the wheel all the same.
+packages/zettcode-gateway/src/zettcode/gateway/static/
+
+# The palette gallery is generated by `make palettes`: every widget painted in
+# every palette, written by docs/scripts/palette_gallery.py. Regenerate it
+# rather than committing a 2 MB copy that goes stale the moment a colour moves.
+docs/internal/palettes.html
+
+# The release order is generated from the pyproject.toml files (`make order`); it is not kept in git.
+packages/order.txt
+
+# electron-builder writes the desktop installers under release/.
+clients/desktop/release/

zettcode_one_system-0.5.1/PKG-INFO

@@ -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/pyproject.toml

@@ -0,0 +1,51 @@
+[project]
+name = "zettcode-one-system"
+description = "Every instruction ZettCode gives the model in one system message, as a part of the agent that lives apart from it"
+requires-python = ">=3.10"
+dynamic = ["version"]
+license = "MIT"
+authors = [{name = "Chang-LeHung"}]
+keywords = ["zettcode", "prompt", "system-message", "agent"]
+classifiers = [
+  "Development Status :: 5 - Production/Stable",
+  "Programming Language :: Python :: 3.10",
+  "Programming Language :: Python :: 3.11",
+  "Programming Language :: Python :: 3.12",
+  "Programming Language :: Python :: 3.13",
+  "Programming Language :: Python :: 3.14",
+  "Programming Language :: Python :: 3.15",
+  "Typing :: Typed",
+]
+dependencies = [
+  # The plugin base classes.
+  "zettcode-engine>=0.5.1,<0.6",
+  # The message types and the extension hook the merge runs on.
+  "zett-agent>=0.2.10,<0.3",
+]
+
+# A part of the program, not a separately released plugin: the `[tool.zettcode] plugin = true` marker is what the
+# plugin release line looks for, and this ships with the program like the tool guidelines and the long-output cut. It is
+# on unless `[plugins] disable = ["one-system"]` names it.
+[project.entry-points."zettcode.plugin"]
+one-system = "zettcode.one_system.plugin:OneSystem"
+
+[project.urls]
+Repository = "https://github.com/Chang-LeHung/zettcode"
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.hatch.version]
+# One version for the whole repository: the engine ships the file.
+path = "../zettcode-engine/src/zettcode/version.py"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/zettcode"]
+
+[tool.hatch.build.targets.sdist]
+include = ["/src", "/pyproject.toml"]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+asyncio_mode = "auto"

zettcode_one_system-0.5.1/src/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-0.5.1/src/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-0.5.1/src/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)