@townco/agent 0.1.42 → 0.1.43

package/dist/runner/langchain/index.js

@@ -17,6 +17,8 @@ const getWeather = tool(({ city }) => `It's always sunny in ${city}!`, {
         city: z.string(),
     }),
 });
+getWeather.prettyName = "Get Weather";
+getWeather.icon = "Cloud";
 export const TOOL_REGISTRY = {
     todo_write: todoWrite,
     get_weather: getWeather,
@@ -27,11 +29,14 @@ export const TOOL_REGISTRY = {
 // Custom tool loading
 // ============================================================================
 function toLangchainTool(resolved) {
-    return tool(resolved.fn, {
+    const t = tool(resolved.fn, {
         name: resolved.name,
         description: resolved.description,
         schema: resolved.schema,
     });
+    t.prettyName = resolved.prettyName;
+    t.icon = resolved.icon;
+    return t;
 }
 async function loadCustomTool(modulePath) {
     const resolved = await loadCustomToolModule(modulePath);
@@ -84,6 +89,8 @@ export class LangchainAgent {
                         description: t.description,
                         schema: t.schema,
                     });
+                    addedTool.prettyName = t.prettyName;
+                    addedTool.icon = t.icon;
                     enabledTools.push(addedTool);
                 }
             }
@@ -229,6 +236,9 @@ export class LangchainAgent {
                         //  todoWriteToolCallIds.add(toolCall.id);
                         //  continue;
                         //}
+                        const matchingTool = finalTools.find((t) => t.name === toolCall.name);
+                        const prettyName = matchingTool?.prettyName;
+                        const icon = matchingTool?.icon;
                         yield {
                             sessionUpdate: "tool_call",
                             toolCallId: toolCall.id,
@@ -237,7 +247,11 @@ export class LangchainAgent {
                             status: "pending",
                             rawInput: toolCall.args,
                             ...(tokenUsage ? { tokenUsage } : {}),
-                            _meta: { messageId: req.messageId },
+                            _meta: {
+                                messageId: req.messageId,
+                                ...(prettyName ? { prettyName } : {}),
+                                ...(icon ? { icon } : {}),
+                            },
                         };
                         yield {
                             sessionUpdate: "tool_call_update",

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

@@ -188,6 +188,8 @@ export function makeFilesystemTools(workingDirectory) {
                 .describe("Enable multiline mode where . matches newlines and patterns can span lines (rg -U --multiline-dotall). Default: false."),
         }),
     });
+    grep.prettyName = "Codebase Search";
+    grep.icon = "Search";
     const read = tool(async ({ file_path, offset, limit }) => {
         await ensureSandbox(resolvedWd);
         assertAbsolutePath(file_path, "file_path");
@@ -230,6 +232,8 @@ export function makeFilesystemTools(workingDirectory) {
                 .describe("The number of lines to read. Only provide if the file is too large to read at once."),
         }),
     });
+    read.prettyName = "Read File";
+    read.icon = "FileText";
     const write = tool(async ({ file_path, content }) => {
         await ensureSandbox(resolvedWd);
         assertAbsolutePath(file_path, "file_path");
@@ -257,5 +261,7 @@ export function makeFilesystemTools(workingDirectory) {
             content: z.string().describe("The content to write to the file"),
         }),
     });
+    write.prettyName = "Write File";
+    write.icon = "Edit";
     return [grep, read, write];
 }

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

@@ -76,3 +76,5 @@ When in doubt, use this tool. Being proactive with task management demonstrates
         todos: z.array(todoItemSchema),
     }),
 });
+todoWrite.prettyName = "Todo List";
+todoWrite.icon = "CheckSquare";

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

@@ -46,6 +46,8 @@ export function makeWebSearchTools() {
             query: z.string().describe("The search query to use"),
         }),
     });
+    webSearch.prettyName = "Web Search";
+    webSearch.icon = "Globe";
     // WebFetch tool - get contents of specific URLs
     const webFetch = tool(async ({ url, prompt }) => {
         const client = getExaClient();
@@ -118,6 +120,8 @@ export function makeWebSearchTools() {
             prompt: z.string().describe("The prompt to run on the fetched content"),
         }),
     });
+    webFetch.prettyName = "Web Fetch";
+    webFetch.icon = "Link";
     return [webSearch, webFetch];
 }
 function buildWebFetchUserMessage(pageContent, prompt) {

package/dist/runner/tool-loader.d.ts

@@ -4,11 +4,15 @@ export type CustomToolModule = {
     schema: z.ZodTypeAny | ((z: typeof import("zod").z) => z.ZodTypeAny);
     name: string;
     description: string;
+    prettyName?: string;
+    icon?: string;
 };
 export type ResolvedCustomTool = {
     fn: (input: unknown) => unknown | Promise<unknown>;
     schema: z.ZodTypeAny;
     name: string;
     description: string;
+    prettyName?: string;
+    icon?: string;
 };
 export declare function loadCustomToolModule(modulePath: string): Promise<ResolvedCustomTool>;

package/dist/runner/tool-loader.js

@@ -35,6 +35,8 @@ export async function loadCustomToolModule(modulePath) {
         schema: resolvedSchema,
         name: mod.name,
         description: mod.description,
+        ...(mod.prettyName && { prettyName: mod.prettyName }),
+        ...(mod.icon && { icon: mod.icon }),
     };
     // Cache the resolved tool
     customToolCache.set(modulePath, resolved);

package/dist/runner/tools.d.ts

@@ -8,6 +8,8 @@ declare const zDirectTool: z.ZodObject<{
     description: z.ZodString;
     fn: z.ZodFunction<z.core.$ZodFunctionArgs, z.core.$ZodFunctionOut>;
     schema: z.ZodAny;
+    prettyName: z.ZodOptional<z.ZodString>;
+    icon: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>;
 /** Tool type - can be a built-in tool string or custom tool object. */
 export declare const zToolType: z.ZodUnion<readonly [z.ZodUnion<readonly [z.ZodLiteral<"todo_write">, z.ZodLiteral<"get_weather">, z.ZodLiteral<"web_search">, z.ZodLiteral<"filesystem">]>, z.ZodObject<{
@@ -22,6 +24,8 @@ export declare const zToolType: z.ZodUnion<readonly [z.ZodUnion<readonly [z.ZodL
     description: z.ZodString;
     fn: z.ZodFunction<z.core.$ZodFunctionArgs, z.core.$ZodFunctionOut>;
     schema: z.ZodAny;
+    prettyName: z.ZodOptional<z.ZodString>;
+    icon: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>]>;
 export type ToolType = z.infer<typeof zToolType>;
 export type BuiltInToolType = z.infer<typeof zBuiltInToolType>;

package/dist/runner/tools.js

@@ -23,6 +23,8 @@ const zDirectTool = z.object({
     description: z.string(),
     fn: z.function(),
     schema: z.any(), // Accept any Zod schema
+    prettyName: z.string().optional(),
+    icon: z.string().optional(),
 });
 /** Tool type - can be a built-in tool string or custom tool object. */
 export const zToolType = z.union([