feat(server): self-identification tools

2026-01-22 14:01:57 +07:00
parent cc2bdbf01f
commit ef9c538842
11 changed files with 557 additions and 175 deletions
@@ -1,76 +0,0 @@
-import { generateObject } from "ai";
-import { createOpenAI } from "@ai-sdk/openai";
-import { z } from "zod";
-import type { AgentTimelineItem } from "../server/agent/agent-sdk-types.js";
-import { curateAgentActivity } from "../server/agent/activity-curator.js";
-import type pino from "pino";
-
-let openai: ReturnType<typeof createOpenAI> | null = null;
-
-export function initializeTitleGenerator(logger: pino.Logger, apiKey: string): void {
-  openai = createOpenAI({ apiKey });
-  logger.child({ action: "initialize" }).info("Agent title generator initialized");
-}
-
-export function isTitleGeneratorInitialized(): boolean {
-  return openai !== null;
-}
-
-/**
- * Generate a concise title for an agent based on its activity
- * Returns a 3-5 word title similar to ChatGPT/Claude.ai
- */
-export async function generateAgentTitle(
-  logger: pino.Logger,
-  timeline: AgentTimelineItem[],
-  cwd: string
-): Promise<string> {
-  const titleLogger = logger.child({ action: "generate" });
-  if (!openai) {
-    throw new Error("Title generator not initialized");
-  }
-
-  if (timeline.length === 0) {
-    return "New Agent";
-  }
-
-  const activityContext = curateAgentActivity(timeline);
-
-  if (!activityContext.trim() || activityContext === "No activity to display.") {
-    return "New Agent";
-  }
-
-  try {
-    const titleSchema = z.object({
-      title: z.string().describe("A concise 3-5 word title describing what the agent is working on"),
-    });
-
-    const model = openai("gpt-4o-mini");
-    const prompt = `Generate a concise title for this agent session. The title should describe what the user asked the agent to do.
-
-IMPORTANT: Focus primarily on [User] messages to understand the task. User messages contain the actual request - use their words and intent to name the chat. Tool calls and assistant messages are just implementation details and should NOT drive the title.
-
-Be specific but brief (3-5 words). Examples: "Fix Authentication Bug", "Build Dashboard Component", "Refactor API Routes".
-
-Working directory: ${cwd}
-
-Activity:
-${activityContext}`;
-
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    const result = await (generateObject as any)({
-      model,
-      schema: titleSchema,
-      prompt,
-      temperature: 0.7,
-    });
-    const object = result.object as z.infer<typeof titleSchema>;
-
-    titleLogger.debug({ title: object.title }, "Generated agent title");
-
-    return object.title;
-  } catch (err) {
-    titleLogger.error({ err }, "Failed to generate agent title");
-    return "New Agent";
-  }
-}