feat: Create projects inside a project group on bt functions push

scripts/functions-runner.py

@@ -98,6 +98,20 @@ def normalize_project_selector(project: Any) -> tuple[str | None, str | None]:
     return None, None
 
 
+def project_selector_fields(project: Any) -> dict[str, str]:
+    project_id, project_name = normalize_project_selector(project)
+    if project_id:
+        return {"project_id": project_id}
+    if not project_name:
+        return {}
+    fields = {"project_name": project_name}
+    # braintrust.framework2.Project exposes `.project_group_name` on SDKs that support project groups.
+    project_group_name = getattr(project, "project_group_name", None)
+    if isinstance(project_group_name, str) and project_group_name.strip():
+        fields["project_group_name"] = project_group_name.strip()
+    return fields
+
+
 def normalize_function_type(raw: Any) -> str | None:
     if isinstance(raw, str):
         value = raw.strip()
@@ -189,8 +203,6 @@ def collect_code_entries(functions_registry: Any) -> list[dict[str, Any]]:
         if not isinstance(name, str) or not isinstance(slug, str) or not name or not slug:
             continue
 
-        project_id, project_name = normalize_project_selector(getattr(item, "project", None))
-
         entry: dict[str, Any] = {
             "kind": "code",
             "name": name,
@@ -232,10 +244,7 @@ def collect_code_entries(functions_registry: Any) -> list[dict[str, Any]]:
             normalized_tags = [tag for tag in tags if isinstance(tag, str)]
             if normalized_tags:
                 entry["tags"] = normalized_tags
-        if project_id:
-            entry["project_id"] = project_id
-        if project_name:
-            entry["project_name"] = project_name
+        entry.update(project_selector_fields(getattr(item, "project", None)))
 
         preview = getattr(item, "preview", None)
         if isinstance(preview, str):
@@ -275,13 +284,13 @@ async def collect_function_event_entries(prompts_registry: Any) -> list[dict[str
             if isinstance(normalized, dict):
                 if normalized.get("if_exists") is None:
                     normalized.pop("if_exists", None)
-                project_id, project_name = normalize_project_selector(getattr(item, "project", None))
-                event_entry: dict[str, Any] = {"kind": "function_event", "event": normalized}
-                if project_id:
-                    event_entry["project_id"] = project_id
-                if project_name:
-                    event_entry["project_name"] = project_name
-                entries.append(event_entry)
+                entries.append(
+                    {
+                        "kind": "function_event",
+                        "event": normalized,
+                        **project_selector_fields(getattr(item, "project", None)),
+                    }
+                )
 
     return entries
 

scripts/functions-runner.ts

@@ -56,6 +56,7 @@ type CodeEntry = {
   kind: "code";
   project_id?: string;
   project_name?: string;
+  project_group_name?: string;
   name: string;
   slug: string;
   description?: string;
@@ -72,6 +73,7 @@ type FunctionEventEntry = {
   kind: "function_event";
   project_id?: string;
   project_name?: string;
+  project_group_name?: string;
   event: JsonValue;
 };
 
@@ -374,6 +376,7 @@ async function collectFunctionEvents(
       kind: "function_event",
       project_id: projectId,
       project_name: projectName,
+      project_group_name: selector.project_group_name,
       event: normalizedEvent,
     });
   }
@@ -439,6 +442,7 @@ async function collectLegacyPromptEvent(
     kind: "function_event",
     project_id: projectId,
     project_name: projectName,
+    project_group_name: selector.project_group_name,
     event,
   };
 }
@@ -526,6 +530,7 @@ function collectCodeEntries(items: CodeRegistryItem[]): CodeEntry[] {
         typeof selector.project_name === "string"
           ? selector.project_name
           : undefined,
+      project_group_name: selector.project_group_name,
       name: item.name,
       slug: item.slug,
       description:

scripts/runner-common.ts

@@ -6,11 +6,13 @@ export type JsonValue = JsonPrimitive | JsonArray | JsonObject;
 export type ProjectSelector = {
   project_id?: string;
   project_name?: string;
+  project_group_name?: string;
 };
 
 export type ProjectRef = {
   id?: string;
   name?: string;
+  projectGroupName?: string;
 };
 
 export function asProjectSelector(
@@ -25,7 +27,13 @@ export function asProjectSelector(
   }
 
   if (typeof project.name === "string" && project.name.trim().length > 0) {
-    return { project_name: project.name };
+    return typeof project.projectGroupName === "string" &&
+      project.projectGroupName.trim().length > 0
+      ? {
+          project_name: project.name,
+          project_group_name: project.projectGroupName,
+        }
+      : { project_name: project.name };
   }
 
   return {};

src/datasets/pipeline.rs

@@ -1808,7 +1808,7 @@ async fn resolve_target_project(
     if let Some(project) = get_project_by_name(client, project_name).await? {
         Ok(project)
     } else {
-        create_project(client, project_name)
+        create_project(client, project_name, None)
             .await
             .with_context(|| format!("project '{project_name}' not found, and creating it failed"))
     }