summitsdk 0.1.0__py3-none-any.whl

marlsdk/__init__.py

@@ -0,0 +1,5 @@
+from marlsdk.tracer import Tracer
+from marlsdk.schema import InteractionTrace
+from marlsdk.exporters.local import LocalExporter
+
+__all__ = ["Tracer", "InteractionTrace", "LocalExporter"]

marlsdk/exporters/__init__.py

@@ -0,0 +1,3 @@
+from marlsdk.exporters.local import LocalExporter
+
+__all__ = ["LocalExporter"]

marlsdk/exporters/local.py

@@ -0,0 +1,24 @@
+import json
+import os
+from marlsdk.schema import InteractionTrace
+
+
+class LocalExporter:
+    def __init__(self, output_dir: str = "data/traces"):
+        self.output_dir = output_dir
+        os.makedirs(output_dir, exist_ok=True)
+
+    def export(self, trace: InteractionTrace) -> str:
+        filename = (
+            f"{trace.task_id}_{trace.from_agent}_to_{trace.to_agent}"
+            f"_r{trace.round_trip}_{trace.trace_id[:6]}.json"
+        )
+        filepath = os.path.join(self.output_dir, filename)
+        with open(filepath, "w") as f:
+            json.dump(trace.to_dict(), f, indent=2)
+        return filepath
+
+    def list_traces(self) -> list:
+        if not os.path.exists(self.output_dir):
+            return []
+        return sorted(f for f in os.listdir(self.output_dir) if f.endswith(".json"))

marlsdk/schema.py

@@ -0,0 +1,37 @@
+from dataclasses import dataclass, field
+from datetime import datetime, timezone
+import uuid
+
+
+@dataclass
+class InteractionTrace:
+    task_id: str
+    from_agent: str
+    to_agent: str
+    content: str
+    model: str
+    input_tokens: int
+    output_tokens: int
+    round_trip: int
+    framework: str = "unknown"
+    trace_id: str = field(default_factory=lambda: str(uuid.uuid4()))
+    timestamp: str = field(
+        default_factory=lambda: datetime.now(timezone.utc).isoformat()
+    )
+    metadata: dict = field(default_factory=dict)
+
+    def to_dict(self) -> dict:
+        return {
+            "trace_id": self.trace_id,
+            "task_id": self.task_id,
+            "from_agent": self.from_agent,
+            "to_agent": self.to_agent,
+            "content": self.content,
+            "model": self.model,
+            "input_tokens": self.input_tokens,
+            "output_tokens": self.output_tokens,
+            "round_trip": self.round_trip,
+            "timestamp": self.timestamp,
+            "framework": self.framework,
+            "metadata": self.metadata,
+        }

marlsdk/tracer.py

@@ -0,0 +1,36 @@
+from marlsdk.schema import InteractionTrace
+from marlsdk.exporters.local import LocalExporter
+
+
+class Tracer:
+    def __init__(self, exporter=None):
+        self.exporter = exporter or LocalExporter()
+
+    def record(self, trace: InteractionTrace) -> None:
+        try:
+            self.exporter.export(trace)
+        except Exception as e:
+            import sys
+            print(
+                f"[summitsdk] Warning: export failed"
+                f" for trace {trace.trace_id}:"
+                f" {type(e).__name__}: {e}",
+                file=sys.stderr,
+            )
+
+    def wrap_anthropic(self, client, from_agent, to_agent, task_id, round_trip):
+        from marlsdk.wrappers.anthropic_wrapper import WrappedAnthropicMessages
+        client.messages = WrappedAnthropicMessages(
+            client.messages, self, from_agent, to_agent, task_id, round_trip
+        )
+        return client
+
+    def wrap_openai(self, client, from_agent, to_agent, task_id, round_trip):
+        try:
+            from marlsdk.wrappers.openai_wrapper import WrappedOpenAICompletions
+            client.chat.completions = WrappedOpenAICompletions(
+                client.chat.completions, self, from_agent, to_agent, task_id, round_trip
+            )
+        except (ImportError, AttributeError):
+            pass
+        return client

marlsdk/wrappers/anthropic_wrapper.py

@@ -0,0 +1,55 @@
+from marlsdk.schema import InteractionTrace
+
+
+class WrappedAnthropicMessages:
+    def __init__(self, original, tracer, from_agent, to_agent, task_id, round_trip):
+        self._original = original
+        self._tracer = tracer
+        self._from_agent = from_agent
+        self._to_agent = to_agent
+        self._task_id = task_id
+        self._round_trip = round_trip
+
+    def create(self, *args, **kwargs):
+        response = self._original.create(*args, **kwargs)
+        try:
+            content = ""
+            if response.content:
+                for block in response.content:
+                    if hasattr(block, "text"):
+                        content = block.text
+                        break
+            input_tokens = getattr(response.usage, "input_tokens", 0) or 0
+            output_tokens = getattr(response.usage, "output_tokens", 0) or 0
+            model = getattr(response, "model", "unknown")
+            trace = InteractionTrace(
+                task_id=self._task_id,
+                from_agent=self._from_agent,
+                to_agent=self._to_agent,
+                content=content,
+                model=model,
+                input_tokens=input_tokens,
+                output_tokens=output_tokens,
+                round_trip=self._round_trip,
+                framework="anthropic",
+            )
+            self._tracer.record(trace)
+        except Exception as e:
+            import sys
+            print(
+                f"[summitsdk] Warning: trace recording failed"
+                f" for {self._from_agent}->{self._to_agent}:"
+                f" {type(e).__name__}: {e}",
+                file=sys.stderr,
+            )
+        return response
+
+    def stream(self, *args, **kwargs):
+        raise NotImplementedError(
+            "summitsdk does not support streaming. "
+            "Use client.messages.create() instead. "
+            "Streaming calls will not produce traces."
+        )
+
+    def __getattr__(self, name):
+        return getattr(self._original, name)

marlsdk/wrappers/openai_wrapper.py

@@ -0,0 +1,58 @@
+try:
+    from marlsdk.schema import InteractionTrace
+
+    class WrappedOpenAICompletions:
+        def __init__(self, original, tracer, from_agent, to_agent, task_id, round_trip):
+            self._original = original
+            self._tracer = tracer
+            self._from_agent = from_agent
+            self._to_agent = to_agent
+            self._task_id = task_id
+            self._round_trip = round_trip
+
+        def create(self, *args, **kwargs):
+            response = self._original.create(*args, **kwargs)
+            try:
+                choices = getattr(response, "choices", None) or []
+                content = ""
+                if choices:
+                    msg = getattr(choices[0], "message", None)
+                    content = getattr(msg, "content", "") or ""
+                usage = getattr(response, "usage", None)
+                input_tokens = getattr(usage, "prompt_tokens", 0) or 0
+                output_tokens = getattr(usage, "completion_tokens", 0) or 0
+                model = getattr(response, "model", "unknown")
+                trace = InteractionTrace(
+                    task_id=self._task_id,
+                    from_agent=self._from_agent,
+                    to_agent=self._to_agent,
+                    content=content,
+                    model=model,
+                    input_tokens=input_tokens,
+                    output_tokens=output_tokens,
+                    round_trip=self._round_trip,
+                    framework="openai",
+                )
+                self._tracer.record(trace)
+            except Exception as e:
+                import sys
+                print(
+                    f"[summitsdk] Warning: trace recording failed"
+                    f" for {self._from_agent}->{self._to_agent}:"
+                    f" {type(e).__name__}: {e}",
+                    file=sys.stderr,
+                )
+            return response
+
+        def stream(self, *args, **kwargs):
+            raise NotImplementedError(
+                "summitsdk does not support streaming. "
+                "Use client.chat.completions.create() instead. "
+                "Streaming calls will not produce traces."
+            )
+
+        def __getattr__(self, name):
+            return getattr(self._original, name)
+
+except ImportError:
+    pass

summitsdk-0.1.0.dist-info/METADATA

@@ -0,0 +1,71 @@
+Metadata-Version: 2.4
+Name: summitsdk
+Version: 0.1.0
+Summary: Instrumentation layer for multi-agent AI pipelines
+Home-page: https://cronys.xyz/summit
+Author: Shaz Hussain
+Author-email: shaz@cronys.xyz
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: anthropic
+Provides-Extra: openai
+Requires-Dist: openai; extra == "openai"
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# MARL
+
+Simulation and evaluation environment for
+Summit by Cronys.
+
+## Structure
+
+sim/        Planner, Worker, Judge agents
+eval/       Reward model evaluator
+improve.py  Single-pass improvement loop
+cycle.py    Automated multi-cycle runner
+marlsdk/    summitsdk package source
+data/       Traces and evaluation results
+
+## Usage
+
+```python
+from anthropic import Anthropic
+from marlsdk.tracer import Tracer
+from marlsdk.exporters.local import LocalExporter
+
+exporter = LocalExporter(output_dir="traces/")
+tracer = Tracer(exporter=exporter)
+
+client = Anthropic()
+wrapped = tracer.wrap_anthropic(
+    client,
+    from_agent="planner",
+    to_agent="worker",
+    task_id="task-001",
+    round_trip=1
+)
+
+response = wrapped.messages.create(
+    model="claude-sonnet-4-20250514",
+    max_tokens=1024,
+    messages=[{"role": "user", "content": "your prompt"}]
+)
+# trace written automatically to traces/
+```
+
+OpenAI: use tracer.wrap_openai(client, ...)
+with the same arguments.

summitsdk-0.1.0.dist-info/RECORD

@@ -0,0 +1,13 @@
+marlsdk/__init__.py,sha256=MieX0EwpdtKC7LwbKbLznpdFMYYf64dA0DCxObWIQb8,187
+marlsdk/schema.py,sha256=JV30B0i8XU69zq3uWIEyjp3ySdOY-eBcb17a9lmSaOA,1083
+marlsdk/tracer.py,sha256=UMQRgRnzSTSNOrTaidOh_tYJ9RyI0ivVU4x39XS3h5s,1325
+marlsdk/exporters/__init__.py,sha256=MexA6fnAt9kKWiyajkRxnKAD2Qlevk1SIhZoprQ6joM,79
+marlsdk/exporters/local.py,sha256=rtQ_NQT-HuobGlHvYbXQ56h80MM4jYFPhcKdvsm6ga4,813
+marlsdk/wrappers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+marlsdk/wrappers/anthropic_wrapper.py,sha256=lx9AIAbTcuxw6MimLk6TvOMw8M6GQZNNQqSXaGU76lE,2006
+marlsdk/wrappers/openai_wrapper.py,sha256=Do59EVQkPCo9h6Y72pRtaLFZNgvdclaarsAnqJHa80I,2289
+summitsdk-0.1.0.dist-info/licenses/LICENSE,sha256=-piU4bFNOw_DasMxGO6Xz2ciRrf6SCLUkEm7sKiI7UU,1069
+summitsdk-0.1.0.dist-info/METADATA,sha256=TlCynx4g-79MlkuRi5rBVZsN-No8IGrsrur5GN5E8uY,1807
+summitsdk-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+summitsdk-0.1.0.dist-info/top_level.txt,sha256=IxMmo2HJH49y0hDeIDvcMIvUhcFdDBBAVan0WbH4qoI,8
+summitsdk-0.1.0.dist-info/RECORD,,

summitsdk-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

summitsdk-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,30 @@
+MIT License
+
+Copyright (c) 2026 Shaz Hussain
+
+Permission is hereby granted, free of charge,
+to any person obtaining a copy of this software
+and associated documentation files (the
+"Software"), to deal in the Software without
+restriction, including without limitation the
+rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of
+the Software, and to permit persons to whom the
+Software is furnished to do so, subject to the
+following conditions:
+
+The above copyright notice and this permission
+notice shall be included in all copies or
+substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT
+WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
+INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR
+PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
+THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR
+ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
+ARISING FROM, OUT OF OR IN CONNECTION WITH THE
+SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

summitsdk-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+marlsdk