feat: project archiving and deletion (#2585)

Adds deleteProject and archiveProject mutations. No deletion currently happens, we simply mark the project as archived and don't return it from the projects field on Query.

app/schema.graphql

@@ -485,6 +485,8 @@ type Mutation {
   Given a list of clusters, export the corresponding data subset in Parquet format. File name is optional, but if specified, should be without file extension. By default the exported file name is current timestamp.
   """
   exportClusters(clusters: [ClusterInput!]!, fileName: String): ExportedFile!
+  deleteProject(id: GlobalID!): Query!
+  archiveProject(id: GlobalID!): Query!
 }
 
 """A node in the graph with a globally unique ID"""

src/phoenix/core/project.py

@@ -91,6 +91,7 @@ class Project:
     def __init__(self) -> None:
         self._spans = _Spans()
         self._evals = _Evals()
+        self._is_archived = False
 
     @property
     def last_updated_at(self) -> Optional[datetime]:
@@ -192,6 +193,13 @@ def get_document_evaluation_scores(
     def export_evaluations(self) -> List[Evaluations]:
         return self._evals.export_evaluations()
 
+    def archive(self) -> None:
+        self._is_archived = True
+
+    @property
+    def is_archived(self) -> bool:
+        return self._is_archived
+
 
 class _Spans:
     def __init__(self) -> None:

src/phoenix/core/traces.py

@@ -38,10 +38,26 @@ def get_project(self, project_name: str) -> Optional["Project"]:
         with self._lock:
             return self._projects.get(project_name)
 
-    def get_projects(self) -> Iterator[Tuple[str, "Project"]]:
+    def get_projects(self) -> Iterator[Tuple[int, str, "Project"]]:
         with self._lock:
-            projects = tuple(self._projects.items())
-        yield from projects
+            for project_id, (project_name, project) in enumerate(self._projects.items()):
+                if project.is_archived:
+                    continue
+                yield project_id, project_name, project
+
+    def archive_project(self, id: int) -> Optional["Project"]:
+        with self._lock:
+            active_projects = {
+                project_id: project
+                for project_id, _, project in self.get_projects()
+                if not project.is_archived
+            }
+            if len(active_projects) <= 1:
+                return None
+            if project := active_projects.get(id):
+                project.archive()
+                return project
+        return None
 
     def put(
         self,

src/phoenix/server/api/schema.py

@@ -56,8 +56,8 @@ def projects(
             []
             if (traces := info.context.traces) is None
             else [
-                Project(id_attr=i, name=name, project=project)
-                for i, (name, project) in enumerate(traces.get_projects())
+                Project(id_attr=project_id, name=project_name, project=project)
+                for project_id, project_name, project in traces.get_projects()
             ]
         )
         return connection_from_list(data=data, args=args)
@@ -86,8 +86,11 @@ def node(self, id: GlobalID, info: Info[Context, None]) -> Node:
             return to_gql_embedding_dimension(node_id, embedding_dimension)
         elif type_name == "Project":
             if (traces := info.context.traces) is not None:
-                projects = list(traces.get_projects())
-                if node_id < len(projects):
+                projects = {
+                    project_id: (project_name, project)
+                    for project_id, project_name, project in traces.get_projects()
+                }
+                if node_id in projects:
                     name, project = projects[node_id]
                     return Project(id_attr=node_id, name=name, project=project)
             raise Exception(f"Unknown project: {id}")
@@ -224,7 +227,26 @@ def hdbscan_clustering(
 
 
 @strawberry.type
-class Mutation(ExportEventsMutation): ...
+class Mutation(ExportEventsMutation):
+    @strawberry.mutation
+    def delete_project(self, info: Info[Context, None], id: GlobalID) -> Query:
+        if (traces := info.context.traces) is None:
+            return Query()
+        type_name, node_id = from_global_id(str(id))
+        if type_name != "Project":
+            return Query()
+        traces.archive_project(node_id)
+        return Query()
+
+    @strawberry.mutation
+    def archive_project(self, info: Info[Context, None], id: GlobalID) -> Query:
+        if (traces := info.context.traces) is None:
+            return Query()
+        type_name, node_id = from_global_id(str(id))
+        if type_name != "Project":
+            return Query()
+        traces.archive_project(node_id)
+        return Query()
 
 
 schema = strawberry.Schema(query=Query, mutation=Mutation)