@mastra/factory 0.10.0-alpha.2 → 0.10.0-alpha.8

package/dist/integrations/github/routes.js

@@ -3,12 +3,15 @@ import { listPullRequestSubscriptionsForThread, subscribeToPullRequest } from ".
 import { SandboxBudgetError } from "../../sandbox/fleet.js";
 import { MaterializeError, WorktreeError, commitAll, ensureProjectSandbox, isValidGitRef as isValidGitRef$1, materializeRepo, pushBranch, teardownProjectSandbox } from "./sandbox.js";
 import { baseCheckpointIsStale } from "../../sandbox/base-checkpoint-triggers.js";
+import { normalizeSessionTitle } from "../../session/session-title.js";
 import { getGithubFeatureDiagnostics, isGithubFeatureEnabled } from "./config.js";
 import { reclaimDeletedSessionSandbox } from "./sandbox-release.js";
 import { handleGithubWebhook } from "./webhook.js";
 import { registerApiRoute } from "@mastra/core/server";
 import { randomUUID } from "crypto";
 import { UniqueViolationError } from "@mastra/core/storage";
+import { resolveModel } from "@mastra/code-sdk/agents/model";
+import { RequestContext } from "@mastra/core/request-context";
 import { streamSSE } from "hono/streaming";
 //#region src/integrations/github/routes.ts
 /**
@@ -24,7 +27,6 @@ import { streamSSE } from "hono/streaming";
 * so the SPA can cleanly hide all GitHub UI.
 */
 const sessionOperationLocks = /* @__PURE__ */ new Map();
-const MAX_SESSION_TITLE_LENGTH = 80;
 const USER_SESSION_BRANCH_PREFIX = "user/session-";
 const UUID_PATTERN = /^[0-9a-f]{8}-[0-9a-f]{4}-[1-8][0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/;
 /**
@@ -69,9 +71,6 @@ function isValidGitRef(value) {
 function isJsonObject(value) {
 	return typeof value === "object" && value !== null && !Array.isArray(value);
 }
-function normalizeSessionTitle(title) {
-	return [...title.replace(/\s+/g, " ").trim()].slice(0, MAX_SESSION_TITLE_LENGTH).join("").trimEnd() || null;
-}
 /**
 * Resolve the org-scoped tenant for a GitHub request. GitHub project features
 * are org-owned, so they require both a signed-in user and a WorkOS
@@ -208,7 +207,7 @@ async function ingestPolledEvents(events, ingestFactoryEvent) {
 */
 function buildGithubRoutes(options) {
 	const routes = [];
-	const { auth, fleet, storage, github, stateSigner, controller, emitAudit, sessionRetirement } = options;
+	const { auth, fleet, storage, github, stateSigner, controller, memorySettings, emitAudit, sessionRetirement } = options;
 	const diagnostics = () => getGithubFeatureDiagnostics({
 		github,
 		auth,
@@ -688,6 +687,7 @@ function buildGithubRoutes(options) {
 		auth,
 		fleet,
 		controller,
+		memorySettings,
 		emitAudit,
 		sessionRetirement
 	}));
@@ -894,7 +894,33 @@ async function loadOwnedProject(options) {
 		sandboxRow: await loadOrCreateSandboxRow(github, project, userId)
 	};
 }
-function buildProjectGitRoutes({ github, auth, fleet, controller, emitAudit, sessionRetirement }) {
+/**
+* One naming per session at a time: a second caller joins the run already in
+* flight instead of paying for another model call and racing its rename.
+*/
+function createSessionNaming() {
+	const inFlight = /* @__PURE__ */ new Map();
+	return (sessionId, run) => {
+		const pending = inFlight.get(sessionId);
+		if (pending) return pending;
+		const started = run().finally(() => inFlight.delete(sessionId));
+		inFlight.set(sessionId, started);
+		return started;
+	};
+}
+/**
+* Name a thread with a stored model id. Resolution has to go through
+* mastracode's gateway: a bare id handed to core's model router looks for a
+* process env key instead of the caller's stored provider credentials.
+*/
+function titleModel(modelId) {
+	return ({ requestContext }) => resolveModel(modelId, {
+		remapForCodexOAuth: true,
+		requestContext
+	});
+}
+function buildProjectGitRoutes({ github, auth, fleet, controller, memorySettings, emitAudit, sessionRetirement }) {
+	const nameSession = createSessionNaming();
 	return [
 		registerApiRoute("/web/github/projects/:id/sessions", {
 			method: "GET",
@@ -1029,6 +1055,49 @@ function buildProjectGitRoutes({ github, auth, fleet, controller, emitAudit, ses
 				return c.json({ removed: true });
 			}
 		}),
+		registerApiRoute("/web/user-sessions/:sessionId/title", {
+			method: "POST",
+			requiresAuth: false,
+			handler: async (c) => {
+				const resolved = await resolveOrgTenant(loose(c), auth);
+				if ("response" in resolved) return resolved.response;
+				const sessionId = c.req.param("sessionId");
+				const row = await github.sourceControlStorage.sessions.getBySessionId(sessionId);
+				if (!row || row.orgId !== resolved.tenant.orgId || row.userId !== resolved.tenant.userId) return c.json({ error: "Session not found" }, 404);
+				if (!controller) return c.json({ error: "Sessions are not available on this server." }, 503);
+				const thread = (await controller.queryThreads({ resourceId: sessionId })).sort((a, b) => b.updatedAt.getTime() - a.updatedAt.getTime())[0];
+				if (!thread) return c.json({ error: "This session has no conversation to name yet." }, 409);
+				const stored = await memorySettings.get({
+					orgId: row.orgId,
+					userId: row.userId
+				});
+				const requestContext = new RequestContext();
+				requestContext.set("user", {
+					workosId: row.userId,
+					organizationId: row.orgId
+				});
+				try {
+					const title = await nameSession(sessionId, async () => {
+						const generated = await controller.generateThreadTitle({
+							threadId: thread.id,
+							resourceId: sessionId,
+							requestContext,
+							...stored?.observerModelId ? { model: titleModel(stored.observerModelId) } : {}
+						});
+						const named = generated ? normalizeSessionTitle(generated) : null;
+						if (named) await github.sourceControlStorage.sessions.rename({
+							sessionId,
+							title: named
+						});
+						return named;
+					});
+					if (!title) return c.json({ error: "The model returned an empty title. Try again." }, 502);
+					return c.json({ title });
+				} catch (error) {
+					return c.json({ error: error instanceof Error ? error.message : String(error) }, 500);
+				}
+			}
+		}),
 		registerApiRoute("/web/github/projects/:id/commit", {
 			method: "POST",
 			requiresAuth: false,