feat(ph-ai): agent artifacts API endpoint

Author: kappa90

ee/migrations/0031_agentartifact.py

@@ -0,0 +1,67 @@
+# Generated by Django 4.2.26 on 2025-11-21 16:00
+
+import django.db.models.deletion
+from django.conf import settings
+from django.db import migrations, models
+
+import posthog.models.utils
+
+import ee.models.assistant
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("posthog", "0914_alter_userproductlist_reason"),
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+        ("ee", "0030_singlesessionsummary_distinct_id_and_more"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="AgentArtifact",
+            fields=[
+                ("created_at", models.DateTimeField(auto_now_add=True)),
+                ("updated_at", models.DateTimeField(auto_now=True, null=True)),
+                ("deleted", models.BooleanField(blank=True, default=False, null=True)),
+                ("deleted_at", models.DateTimeField(blank=True, null=True)),
+                (
+                    "id",
+                    models.UUIDField(
+                        default=posthog.models.utils.uuid7, editable=False, primary_key=True, serialize=False
+                    ),
+                ),
+                ("short_id", models.CharField(default=ee.models.assistant.generate_short_id, max_length=4)),
+                ("name", models.CharField(max_length=400)),
+                (
+                    "type",
+                    models.CharField(
+                        choices=[("visualization", "Visualization"), ("notebook", "Notebook")], max_length=50
+                    ),
+                ),
+                ("data", models.JSONField(help_text="Artifact content. Structure depends on artifact type.")),
+                (
+                    "conversation",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.CASCADE, related_name="artifacts", to="ee.conversation"
+                    ),
+                ),
+                (
+                    "created_by",
+                    models.ForeignKey(
+                        blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, to=settings.AUTH_USER_MODEL
+                    ),
+                ),
+                ("team", models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to="posthog.team")),
+            ],
+            options={
+                "indexes": [
+                    models.Index(fields=["team", "short_id"], name="ee_agentart_team_id_d05587_idx"),
+                    models.Index(fields=["team", "conversation", "created_at"], name="ee_agentart_team_id_e402e1_idx"),
+                ],
+            },
+        ),
+        migrations.AddConstraint(
+            model_name="agentartifact",
+            constraint=models.UniqueConstraint(fields=("team", "short_id"), name="unique_team_short_id"),
+        ),
+    ]

ee/migrations/max_migration.txt

@@ -1 +1 @@
-0030_singlesessionsummary_distinct_id_and_more
+0031_agentartifact

ee/models/__init__.py

@@ -1,4 +1,5 @@
 from .assistant import (
+    AgentArtifact,
     Conversation,
     ConversationCheckpoint,
     ConversationCheckpointBlob,
@@ -16,6 +17,7 @@
 
 __all__ = [
     "AccessControl",
+    "AgentArtifact",
     "ConversationCheckpoint",
     "ConversationCheckpointBlob",
     "ConversationCheckpointWrite",

ee/models/assistant.py

@@ -1,11 +1,22 @@
+import string
+import secrets
 from datetime import timedelta
 
-from django.db import models
+from django.db import IntegrityError, models
 from django.utils import timezone
 
 from posthog.models.team.team import Team
 from posthog.models.user import User
-from posthog.models.utils import UUIDTModel
+from posthog.models.utils import CreatedMetaFields, DeletedMetaFields, UpdatedMetaFields, UUIDModel, UUIDTModel
+
+
+def generate_short_id():
+    """Generate securely random 4 characters long alphanumeric ID.
+
+    With team-scoped uniqueness, 4 characters (62^4 = 14.7M combinations)
+    is sufficient to avoid collisions within a single team.
+    """
+    return "".join(secrets.choice(string.ascii_letters + string.digits) for _ in range(4))
 
 
 class Conversation(UUIDTModel):
@@ -183,3 +194,36 @@ def answers_left(self) -> int:
         if self.initial_text.endswith("\nAnswer:"):
             answers_given -= 1
         return MAX_ONBOARDING_QUESTIONS - answers_given
+
+
+class AgentArtifact(UUIDModel, CreatedMetaFields, UpdatedMetaFields, DeletedMetaFields):
+    class Type(models.TextChoices):
+        VISUALIZATION = "visualization", "Visualization"
+        NOTEBOOK = "notebook", "Notebook"
+
+    short_id = models.CharField(max_length=4, default=generate_short_id)
+    name = models.CharField(max_length=400)
+    type = models.CharField(max_length=50, choices=Type.choices)
+    data = models.JSONField(help_text="Artifact content. Structure depends on artifact type.")
+    conversation = models.ForeignKey(Conversation, on_delete=models.CASCADE, related_name="artifacts")
+    team = models.ForeignKey(Team, on_delete=models.CASCADE)
+
+    class Meta:
+        indexes = [
+            models.Index(fields=["team", "short_id"]),
+            models.Index(fields=["team", "conversation", "created_at"]),
+        ]
+        constraints = [
+            models.UniqueConstraint(fields=["team", "short_id"], name="unique_team_short_id"),
+        ]
+
+    def save(self, *args, **kwargs):
+        max_retries = 5
+        for attempt in range(max_retries):
+            try:
+                return super().save(*args, **kwargs)
+            except IntegrityError as e:
+                if "short_id" in str(e) and attempt < max_retries - 1:
+                    self.short_id = generate_short_id()
+                else:
+                    raise