composer-agent 0.1.0__py3-none-any.whl

chat/apps.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class ChatConfig(AppConfig):
+    default_auto_field = "django.db.models.BigAutoField"
+    name = "chat"

chat/migrations/0001_initial.py

@@ -0,0 +1,83 @@
+# Generated by Django 6.0.6 on 2026-06-07 17:56
+
+import django.db.models.deletion
+import uuid
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='ChatSession',
+            fields=[
+                ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('name', models.CharField(blank=True, default='', max_length=255)),
+                ('config', models.JSONField(default=dict)),
+                ('active_branch_id', models.UUIDField()),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('updated_at', models.DateTimeField(auto_now=True)),
+            ],
+            options={
+                'ordering': ['-updated_at'],
+            },
+        ),
+        migrations.CreateModel(
+            name='Project',
+            fields=[
+                ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('name', models.CharField(max_length=255, unique=True)),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('updated_at', models.DateTimeField(auto_now=True)),
+            ],
+            options={
+                'ordering': ['name'],
+            },
+        ),
+        migrations.CreateModel(
+            name='BranchNodeRecord',
+            fields=[
+                ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('compressed_payload', models.JSONField(blank=True, null=True)),
+                ('compressed_through', models.PositiveIntegerField()),
+                ('visible_end', models.PositiveIntegerField(blank=True, null=True)),
+                ('child_order', models.JSONField(default=list)),
+                ('parent', models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.CASCADE, related_name='children_nodes', to='chat.branchnoderecord')),
+                ('session', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='branch_nodes', to='chat.chatsession')),
+            ],
+            options={
+                'ordering': ['compressed_through'],
+            },
+        ),
+        migrations.AddField(
+            model_name='chatsession',
+            name='project',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='sessions', to='chat.project'),
+        ),
+        migrations.CreateModel(
+            name='StoredMessage',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('position', models.PositiveIntegerField()),
+                ('message_type', models.CharField(max_length=32)),
+                ('payload', models.JSONField()),
+                ('session', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='messages', to='chat.chatsession')),
+            ],
+            options={
+                'ordering': ['position'],
+            },
+        ),
+        migrations.AddConstraint(
+            model_name='chatsession',
+            constraint=models.UniqueConstraint(condition=models.Q(('name', ''), _negated=True), fields=('project', 'name'), name='uniq_named_session_per_project'),
+        ),
+        migrations.AddConstraint(
+            model_name='storedmessage',
+            constraint=models.UniqueConstraint(fields=('session', 'position'), name='uniq_message_position_per_session'),
+        ),
+    ]

chat/models.py

@@ -0,0 +1,88 @@
+import uuid
+
+from django.db import models
+from django.db.models import Q
+
+
+class Project(models.Model):
+    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
+    name = models.CharField(max_length=255, unique=True)
+    created_at = models.DateTimeField(auto_now_add=True)
+    updated_at = models.DateTimeField(auto_now=True)
+
+    class Meta:
+        ordering = ["name"]
+
+    def __str__(self) -> str:
+        return self.name
+
+
+class ChatSession(models.Model):
+    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
+    project = models.ForeignKey(
+        Project,
+        on_delete=models.CASCADE,
+        related_name="sessions",
+    )
+    name = models.CharField(max_length=255, blank=True, default="")
+    config = models.JSONField(default=dict)
+    active_branch_id = models.UUIDField()
+    created_at = models.DateTimeField(auto_now_add=True)
+    updated_at = models.DateTimeField(auto_now=True)
+
+    class Meta:
+        ordering = ["-updated_at"]
+        constraints = [
+            models.UniqueConstraint(
+                fields=["project", "name"],
+                condition=~Q(name=""),
+                name="uniq_named_session_per_project",
+            ),
+        ]
+
+    def __str__(self) -> str:
+        label = self.name or str(self.id)
+        return f"{self.project.name}:{label}"
+
+
+class StoredMessage(models.Model):
+    session = models.ForeignKey(
+        ChatSession,
+        on_delete=models.CASCADE,
+        related_name="messages",
+    )
+    position = models.PositiveIntegerField()
+    message_type = models.CharField(max_length=32)
+    payload = models.JSONField()
+
+    class Meta:
+        ordering = ["position"]
+        constraints = [
+            models.UniqueConstraint(
+                fields=["session", "position"],
+                name="uniq_message_position_per_session",
+            ),
+        ]
+
+
+class BranchNodeRecord(models.Model):
+    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
+    session = models.ForeignKey(
+        ChatSession,
+        on_delete=models.CASCADE,
+        related_name="branch_nodes",
+    )
+    parent = models.ForeignKey(
+        "self",
+        on_delete=models.CASCADE,
+        null=True,
+        blank=True,
+        related_name="children_nodes",
+    )
+    compressed_payload = models.JSONField(null=True, blank=True)
+    compressed_through = models.PositiveIntegerField()
+    visible_end = models.PositiveIntegerField(null=True, blank=True)
+    child_order = models.JSONField(default=list)
+
+    class Meta:
+        ordering = ["compressed_through"]

composer/__init__.py

@@ -0,0 +1,97 @@
+"""Composer — thread-based agents with typed streaming."""
+
+__version__ = "0.1.0"
+
+from .agent import (
+    Agent,
+    MCPClient,
+    MCPPromptInfo,
+    MCPResourceInfo,
+    StreamEvent,
+    StreamEventKind,
+    StreamEventProcessor,
+    ThinkingEvent,
+    AssistantEvent,
+    ToolCallEvent,
+    ToolResultEvent,
+    ToolCall,
+    ToolResult,
+    combine_tools,
+    extract_thinking,
+    extract_assistant_text,
+)
+from .tools import run_tool_call
+from .vector import Vector
+from .image import ImageAttach
+from .thread import (
+    Thread,
+    HumanMessage,
+    ImageMessage,
+    AIMessage,
+    SystemMessage,
+    ToolMessage,
+    CompressedMessage,
+    Message,
+    EncoderType,
+    TokenCalculator,
+)
+from .tool_hide import (
+    ToolResultHideRule,
+    HideMode,
+    get_original_content,
+    is_hidden_for_model,
+    message_matches_rule,
+    resolve_full_tool_name,
+    restore_hidden_tool_messages,
+)
+
+__all__ = [
+    "Agent",
+    "Vector",
+    "MCPClient",
+    "MCPPromptInfo",
+    "MCPResourceInfo",
+    "Thread",
+    "HumanMessage",
+    "ImageMessage",
+    "ImageAttach",
+    "AIMessage",
+    "SystemMessage",
+    "ToolMessage",
+    "CompressedMessage",
+    "Message",
+    "EncoderType",
+    "TokenCalculator",
+    "HideMode",
+    "ToolResultHideRule",
+    "get_original_content",
+    "is_hidden_for_model",
+    "message_matches_rule",
+    "resolve_full_tool_name",
+    "restore_hidden_tool_messages",
+    "ThinkingEvent",
+    "AssistantEvent",
+    "ToolCallEvent",
+    "ToolResultEvent",
+    "StreamEvent",
+    "StreamEventKind",
+    "StreamEventProcessor",
+    "ToolCall",
+    "ToolResult",
+    "combine_tools",
+    "run_tool_call",
+    "extract_thinking",
+    "extract_assistant_text",
+    "ChatProject",
+    "ChatSession",
+]
+
+_LAZY_IMPORTS = {"ChatProject", "ChatSession"}
+
+
+def __getattr__(name: str):
+    if name in _LAZY_IMPORTS:
+        from .persistence import ChatProject, ChatSession
+
+        return {"ChatProject": ChatProject, "ChatSession": ChatSession}[name]
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")