@townco/agent 0.1.124 → 0.1.126

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -0,0 +1,22 @@
1
+ import { z } from "zod";
2
+ export declare const CONVERSATION_SEARCH_TOOL_NAME = "conversation_search";
3
+ /**
4
+ * Creates a conversation search tool that searches past agent chat sessions.
5
+ * @param agentDir - The directory of the current agent (e.g., agents/researcher)
6
+ */
7
+ export declare function makeConversationSearchTool(agentDir: string): import("langchain").DynamicStructuredTool<z.ZodObject<{
8
+ query: z.ZodString;
9
+ date_from: z.ZodOptional<z.ZodString>;
10
+ date_to: z.ZodOptional<z.ZodString>;
11
+ max_results: z.ZodOptional<z.ZodNumber>;
12
+ }, z.core.$strip>, {
13
+ query: string;
14
+ date_from?: string | undefined;
15
+ date_to?: string | undefined;
16
+ max_results?: number | undefined;
17
+ }, {
18
+ query: string;
19
+ date_from?: string | undefined;
20
+ date_to?: string | undefined;
21
+ max_results?: number | undefined;
22
+ }, string>;
@@ -0,0 +1,137 @@
1
+ import { existsSync, readdirSync, readFileSync } from "node:fs";
2
+ import { join } from "node:path";
3
+ import { tool } from "langchain";
4
+ import { z } from "zod";
5
+ export const CONVERSATION_SEARCH_TOOL_NAME = "conversation_search";
6
+ const conversationSearchSchema = z.object({
7
+ query: z
8
+ .string()
9
+ .min(1)
10
+ .describe("The text to search for in conversation messages (case-insensitive)"),
11
+ date_from: z
12
+ .string()
13
+ .optional()
14
+ .describe("ISO date string to filter conversations from this date onwards"),
15
+ date_to: z
16
+ .string()
17
+ .optional()
18
+ .describe("ISO date string to filter conversations up to this date"),
19
+ max_results: z
20
+ .number()
21
+ .optional()
22
+ .describe("Maximum number of matching conversations to return (default: 10)"),
23
+ });
24
+ /**
25
+ * Creates a conversation search tool that searches past agent chat sessions.
26
+ * @param agentDir - The directory of the current agent (e.g., agents/researcher)
27
+ */
28
+ export function makeConversationSearchTool(agentDir) {
29
+ const conversationSearch = tool(async ({ query, date_from, date_to, max_results = 10 }) => {
30
+ const results = [];
31
+ const queryLower = query.toLowerCase();
32
+ const dateFromParsed = date_from ? new Date(date_from) : null;
33
+ const dateToParsed = date_to ? new Date(date_to) : null;
34
+ const sessionsDir = join(agentDir, ".sessions");
35
+ if (!existsSync(sessionsDir)) {
36
+ return `No conversations found matching "${query}".`;
37
+ }
38
+ let sessionFiles;
39
+ try {
40
+ sessionFiles = readdirSync(sessionsDir).filter((f) => f.endsWith(".json") && !f.endsWith(".tmp"));
41
+ }
42
+ catch {
43
+ return `No conversations found matching "${query}".`;
44
+ }
45
+ for (const file of sessionFiles) {
46
+ if (results.length >= max_results)
47
+ break;
48
+ try {
49
+ const content = readFileSync(join(sessionsDir, file), "utf-8");
50
+ const session = JSON.parse(content);
51
+ // Date filtering
52
+ const sessionDate = new Date(session.metadata.updatedAt);
53
+ if (dateFromParsed && sessionDate < dateFromParsed)
54
+ continue;
55
+ if (dateToParsed && sessionDate > dateToParsed)
56
+ continue;
57
+ // Search messages
58
+ const matches = [];
59
+ for (let i = 0; i < session.messages.length; i++) {
60
+ const msg = session.messages[i];
61
+ if (!msg)
62
+ continue;
63
+ const textBlocks = msg.content.filter((c) => c.type === "text");
64
+ for (const block of textBlocks) {
65
+ if (block.text.toLowerCase().includes(queryLower)) {
66
+ // Extract matched snippet with surrounding context
67
+ const matchIndex = block.text.toLowerCase().indexOf(queryLower);
68
+ const snippetStart = Math.max(0, matchIndex - 50);
69
+ const snippetEnd = Math.min(block.text.length, matchIndex + query.length + 50);
70
+ const matchedText = block.text.slice(snippetStart, snippetEnd);
71
+ matches.push({
72
+ messageIndex: i,
73
+ role: msg.role,
74
+ timestamp: msg.timestamp,
75
+ matchedText: (snippetStart > 0 ? "..." : "") +
76
+ matchedText +
77
+ (snippetEnd < block.text.length ? "..." : ""),
78
+ });
79
+ break; // One match per message is enough
80
+ }
81
+ }
82
+ }
83
+ if (matches.length > 0) {
84
+ results.push({
85
+ sessionId: session.sessionId,
86
+ agentName: session.metadata.agentName,
87
+ createdAt: session.metadata.createdAt,
88
+ updatedAt: session.metadata.updatedAt,
89
+ matches,
90
+ });
91
+ }
92
+ }
93
+ catch {
94
+ // Skip invalid session files
95
+ continue;
96
+ }
97
+ }
98
+ if (results.length === 0) {
99
+ return `No conversations found matching "${query}".`;
100
+ }
101
+ // Format results as readable string
102
+ let output = `Found ${results.length} conversation(s) matching "${query}":\n\n`;
103
+ for (const result of results) {
104
+ output += `=== Session: ${result.sessionId} (${result.agentName}) ===\n`;
105
+ output += `Created: ${result.createdAt} | Updated: ${result.updatedAt}\n\n`;
106
+ for (const match of result.matches) {
107
+ output += `[Message ${match.messageIndex + 1} - ${match.role} @ ${match.timestamp}]\n`;
108
+ output += `"${match.matchedText}"\n\n`;
109
+ }
110
+ output += "---\n\n";
111
+ }
112
+ return output;
113
+ }, {
114
+ name: CONVERSATION_SEARCH_TOOL_NAME,
115
+ description: `Search across past chat conversations to find previous discussions on specific topics.
116
+
117
+ Use this tool to:
118
+ - Find previous conversations that discussed specific topics
119
+ - Recall information from past sessions
120
+ - Search for patterns or recurring themes across conversations
121
+
122
+ Parameters:
123
+ - query: The text to search for (case-insensitive)
124
+ - date_from: (optional) Only search conversations from this date onwards (ISO format, e.g., "2025-01-01")
125
+ - date_to: (optional) Only search conversations up to this date (ISO format)
126
+ - max_results: (optional) Maximum results to return (default: 10)`,
127
+ schema: conversationSearchSchema,
128
+ });
129
+ conversationSearch.prettyName = "Conversation Search";
130
+ conversationSearch.icon = "MessageSquare";
131
+ conversationSearch.verbiage = {
132
+ active: "Searching conversations for {query}",
133
+ past: "Searched conversations for {query}",
134
+ paramKey: "query",
135
+ };
136
+ return conversationSearch;
137
+ }
@@ -1,6 +1,13 @@
1
1
  import type { z } from "zod";
2
2
  import type { AgentDefinition, McpConfigSchema } from "../definition";
3
3
  type McpConfig = z.infer<typeof McpConfigSchema>;
4
+ /** Config for a subagent to wire up via makeSubagentsTool */
5
+ export interface SubagentConfig {
6
+ agentName: string;
7
+ toolDescription: string;
8
+ displayName?: string;
9
+ path: string;
10
+ }
4
11
  export interface TemplateVars {
5
12
  name: string;
6
13
  model: string;
@@ -21,6 +28,7 @@ export interface TemplateVars {
21
28
  schema: unknown;
22
29
  }>;
23
30
  mcps?: McpConfig[];
31
+ subagents?: SubagentConfig[];
24
32
  systemPrompt: string | null;
25
33
  hasWebSearch: boolean;
26
34
  hasGenerateImage: boolean;
@@ -23,6 +23,11 @@ export function getTemplateVars(name, definition) {
23
23
  if (definition.mcps && definition.mcps.length > 0) {
24
24
  result.mcps = definition.mcps;
25
25
  }
26
+ // Handle subagents if defined (using type assertion since AgentDefinition may not have subagents yet)
27
+ const defWithSubagents = definition;
28
+ if (defWithSubagents.subagents && defWithSubagents.subagents.length > 0) {
29
+ result.subagents = defWithSubagents.subagents;
30
+ }
26
31
  return result;
27
32
  }
28
33
  export function generatePackageJson(vars) {
@@ -59,33 +64,57 @@ export function generatePackageJson(vars) {
59
64
  return JSON.stringify(pkg, null, 2);
60
65
  }
61
66
  export async function generateIndexTs(vars) {
62
- // Build agent definition with fields in a logical order
63
- const agentDef = {
64
- model: vars.model,
65
- };
67
+ const hasSubagents = vars.subagents && vars.subagents.length > 0;
68
+ let imports = 'import type { AgentDefinition } from "@townco/agent/definition";';
69
+ if (hasSubagents) {
70
+ imports += '\nimport { makeSubagentsTool } from "@townco/agent/utils";';
71
+ }
72
+ // Build tools code - either plain JSON or with makeSubagentsTool call
73
+ const toolsCode = (() => {
74
+ if (!hasSubagents) {
75
+ return JSON.stringify(vars.tools);
76
+ }
77
+ // makeSubagentsTool expects 'description' not 'toolDescription'
78
+ const subagentConfigs = vars.subagents?.map((s) => ({
79
+ agentName: s.agentName,
80
+ description: s.toolDescription,
81
+ ...(s.displayName && { displayName: s.displayName }),
82
+ path: s.path,
83
+ })) ?? [];
84
+ const subagentToolCall = `makeSubagentsTool(${JSON.stringify(subagentConfigs, null, 2)})`;
85
+ const toolItems = vars.tools.map((t) => JSON.stringify(t));
86
+ toolItems.push(subagentToolCall);
87
+ return `[${toolItems.join(", ")}]`;
88
+ })();
89
+ // Build optional fields
90
+ const optionalFields = [];
66
91
  if (vars.displayName) {
67
- agentDef.displayName = vars.displayName;
92
+ optionalFields.push(`displayName: ${JSON.stringify(vars.displayName)},`);
68
93
  }
69
94
  if (vars.description) {
70
- agentDef.description = vars.description;
95
+ optionalFields.push(`description: ${JSON.stringify(vars.description)},`);
71
96
  }
72
97
  if (vars.suggestedPrompts) {
73
- agentDef.suggestedPrompts = vars.suggestedPrompts;
98
+ optionalFields.push(`suggestedPrompts: ${JSON.stringify(vars.suggestedPrompts)},`);
74
99
  }
75
- agentDef.systemPrompt = vars.systemPrompt;
76
- agentDef.tools = vars.tools;
77
100
  if (vars.mcps && vars.mcps.length > 0) {
78
- agentDef.mcps = vars.mcps;
101
+ optionalFields.push(`mcps: ${JSON.stringify(vars.mcps)},`);
79
102
  }
80
103
  if (vars.hooks) {
81
- agentDef.hooks = vars.hooks;
104
+ optionalFields.push(`hooks: ${JSON.stringify(vars.hooks)},`);
82
105
  }
83
- return prettier.format(`import type { AgentDefinition } from "@townco/agent/definition";
106
+ const optionalFieldsStr = optionalFields.length > 0 ? `\n ${optionalFields.join("\n ")}` : "";
107
+ const code = `${imports}
84
108
 
85
- const agent: AgentDefinition = ${JSON.stringify(agentDef)};
109
+ const agent: AgentDefinition = {
110
+ model: ${JSON.stringify(vars.model)},${optionalFieldsStr}
111
+ systemPrompt: ${JSON.stringify(vars.systemPrompt)},
112
+ tools: ${toolsCode},
113
+ };
86
114
 
87
115
  export default agent;
88
- `, { parser: "typescript" });
116
+ `;
117
+ return prettier.format(code, { parser: "typescript" });
89
118
  }
90
119
  export function generateBinTs() {
91
120
  return `#!/usr/bin/env bun