@townco/agent 0.1.134 → 0.1.136

package/dist/runner/langchain/tools/conversation_search.d.ts

@@ -0,0 +1,22 @@
+import { z } from "zod";
+export declare const CONVERSATION_SEARCH_TOOL_NAME = "conversation_search";
+/**
+ * Creates a conversation search tool that searches past agent chat sessions.
+ * @param agentDir - The directory of the current agent (e.g., agents/researcher)
+ */
+export declare function makeConversationSearchTool(agentDir: string): import("langchain").DynamicStructuredTool<z.ZodObject<{
+    query: z.ZodString;
+    date_from: z.ZodOptional<z.ZodString>;
+    date_to: z.ZodOptional<z.ZodString>;
+    max_results: z.ZodOptional<z.ZodNumber>;
+}, z.core.$strip>, {
+    query: string;
+    date_from?: string | undefined;
+    date_to?: string | undefined;
+    max_results?: number | undefined;
+}, {
+    query: string;
+    date_from?: string | undefined;
+    date_to?: string | undefined;
+    max_results?: number | undefined;
+}, string>;

package/dist/runner/langchain/tools/conversation_search.js

@@ -0,0 +1,137 @@
+import { existsSync, readdirSync, readFileSync } from "node:fs";
+import { join } from "node:path";
+import { tool } from "langchain";
+import { z } from "zod";
+export const CONVERSATION_SEARCH_TOOL_NAME = "conversation_search";
+const conversationSearchSchema = z.object({
+    query: z
+        .string()
+        .min(1)
+        .describe("The text to search for in conversation messages (case-insensitive)"),
+    date_from: z
+        .string()
+        .optional()
+        .describe("ISO date string to filter conversations from this date onwards"),
+    date_to: z
+        .string()
+        .optional()
+        .describe("ISO date string to filter conversations up to this date"),
+    max_results: z
+        .number()
+        .optional()
+        .describe("Maximum number of matching conversations to return (default: 10)"),
+});
+/**
+ * Creates a conversation search tool that searches past agent chat sessions.
+ * @param agentDir - The directory of the current agent (e.g., agents/researcher)
+ */
+export function makeConversationSearchTool(agentDir) {
+    const conversationSearch = tool(async ({ query, date_from, date_to, max_results = 10 }) => {
+        const results = [];
+        const queryLower = query.toLowerCase();
+        const dateFromParsed = date_from ? new Date(date_from) : null;
+        const dateToParsed = date_to ? new Date(date_to) : null;
+        const sessionsDir = join(agentDir, ".sessions");
+        if (!existsSync(sessionsDir)) {
+            return `No conversations found matching "${query}".`;
+        }
+        let sessionFiles;
+        try {
+            sessionFiles = readdirSync(sessionsDir).filter((f) => f.endsWith(".json") && !f.endsWith(".tmp"));
+        }
+        catch {
+            return `No conversations found matching "${query}".`;
+        }
+        for (const file of sessionFiles) {
+            if (results.length >= max_results)
+                break;
+            try {
+                const content = readFileSync(join(sessionsDir, file), "utf-8");
+                const session = JSON.parse(content);
+                // Date filtering
+                const sessionDate = new Date(session.metadata.updatedAt);
+                if (dateFromParsed && sessionDate < dateFromParsed)
+                    continue;
+                if (dateToParsed && sessionDate > dateToParsed)
+                    continue;
+                // Search messages
+                const matches = [];
+                for (let i = 0; i < session.messages.length; i++) {
+                    const msg = session.messages[i];
+                    if (!msg)
+                        continue;
+                    const textBlocks = msg.content.filter((c) => c.type === "text");
+                    for (const block of textBlocks) {
+                        if (block.text.toLowerCase().includes(queryLower)) {
+                            // Extract matched snippet with surrounding context
+                            const matchIndex = block.text.toLowerCase().indexOf(queryLower);
+                            const snippetStart = Math.max(0, matchIndex - 50);
+                            const snippetEnd = Math.min(block.text.length, matchIndex + query.length + 50);
+                            const matchedText = block.text.slice(snippetStart, snippetEnd);
+                            matches.push({
+                                messageIndex: i,
+                                role: msg.role,
+                                timestamp: msg.timestamp,
+                                matchedText: (snippetStart > 0 ? "..." : "") +
+                                    matchedText +
+                                    (snippetEnd < block.text.length ? "..." : ""),
+                            });
+                            break; // One match per message is enough
+                        }
+                    }
+                }
+                if (matches.length > 0) {
+                    results.push({
+                        sessionId: session.sessionId,
+                        agentName: session.metadata.agentName,
+                        createdAt: session.metadata.createdAt,
+                        updatedAt: session.metadata.updatedAt,
+                        matches,
+                    });
+                }
+            }
+            catch {
+                // Skip invalid session files
+                continue;
+            }
+        }
+        if (results.length === 0) {
+            return `No conversations found matching "${query}".`;
+        }
+        // Format results as readable string
+        let output = `Found ${results.length} conversation(s) matching "${query}":\n\n`;
+        for (const result of results) {
+            output += `=== Session: ${result.sessionId} (${result.agentName}) ===\n`;
+            output += `Created: ${result.createdAt} | Updated: ${result.updatedAt}\n\n`;
+            for (const match of result.matches) {
+                output += `[Message ${match.messageIndex + 1} - ${match.role} @ ${match.timestamp}]\n`;
+                output += `"${match.matchedText}"\n\n`;
+            }
+            output += "---\n\n";
+        }
+        return output;
+    }, {
+        name: CONVERSATION_SEARCH_TOOL_NAME,
+        description: `Search across past chat conversations to find previous discussions on specific topics.
+
+Use this tool to:
+- Find previous conversations that discussed specific topics
+- Recall information from past sessions
+- Search for patterns or recurring themes across conversations
+
+Parameters:
+- query: The text to search for (case-insensitive)
+- date_from: (optional) Only search conversations from this date onwards (ISO format, e.g., "2025-01-01")
+- date_to: (optional) Only search conversations up to this date (ISO format)
+- max_results: (optional) Maximum results to return (default: 10)`,
+        schema: conversationSearchSchema,
+    });
+    conversationSearch.prettyName = "Conversation Search";
+    conversationSearch.icon = "MessageSquare";
+    conversationSearch.verbiage = {
+        active: "Searching conversations for {query}",
+        past: "Searched conversations for {query}",
+        paramKey: "query",
+    };
+    return conversationSearch;
+}