From 4554f5e4fbe6325b2b0b2f089239f7fd714e4609 Mon Sep 17 00:00:00 2001
From: Yuhong Sun <yuhongsun96@gmail.com>
Date: Sat, 14 Dec 2024 16:33:00 -0800
Subject: [PATCH] k

---
 .../versions/91a0a4d62b14_milestone.py        | 44 +++++++++++++++++++
 backend/requirements/ee.txt                   |  3 +-
 2 files changed, 46 insertions(+), 1 deletion(-)
 create mode 100644 backend/alembic/versions/91a0a4d62b14_milestone.py

diff --git a/backend/alembic/versions/91a0a4d62b14_milestone.py b/backend/alembic/versions/91a0a4d62b14_milestone.py
new file mode 100644
index 00000000000..a8b14bd2de5
--- /dev/null
+++ b/backend/alembic/versions/91a0a4d62b14_milestone.py
@@ -0,0 +1,44 @@
+"""Milestone
+
+Revision ID: 91a0a4d62b14
+Revises: dab04867cd88
+Create Date: 2024-12-13 19:03:30.947551
+
+"""
+from alembic import op
+import sqlalchemy as sa
+import fastapi_users_db_sqlalchemy
+from sqlalchemy.dialects import postgresql
+
+# revision identifiers, used by Alembic.
+revision = "91a0a4d62b14"
+down_revision = "dab04867cd88"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    op.create_table(
+        "milestone",
+        sa.Column("id", sa.UUID(), nullable=False),
+        sa.Column("tenant_id", sa.String(), nullable=True),
+        sa.Column(
+            "user_id",
+            fastapi_users_db_sqlalchemy.generics.GUID(),
+            nullable=True,
+        ),
+        sa.Column("event_type", sa.String(), nullable=False),
+        sa.Column(
+            "time_created",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.Column("event_tracker", postgresql.JSONB(), nullable=True),
+        sa.ForeignKeyConstraint(["user_id"], ["user.id"], ondelete="CASCADE"),
+        sa.PrimaryKeyConstraint("id"),
+    )
+
+
+def downgrade() -> None:
+    op.drop_table("milestone")
diff --git a/backend/requirements/ee.txt b/backend/requirements/ee.txt
index 18dc3200bdc..e587869fd75 100644
--- a/backend/requirements/ee.txt
+++ b/backend/requirements/ee.txt
@@ -1,2 +1,3 @@
+cohere==5.6.1
+posthog==3.7.4
 python3-saml==1.15.0
-cohere==5.6.1
\ No newline at end of file