zeitlich 0.2.2 → 0.2.3

package/src/tools/task-list/handler.ts

@@ -2,27 +2,22 @@ import type {
   AgentStateManager,
   JsonSerializable,
 } from "../../lib/state-manager";
-import type { ToolHandlerResponse } from "../../lib/tool-router";
+import type { ToolHandler } from "../../lib/tool-router";
 import type { WorkflowTask } from "../../lib/types";
-import type { TaskListToolSchemaType } from "./tool";
+import type { TaskListArgs } from "./tool";
 
 /**
  * Creates a TaskList handler that returns all tasks.
  *
  * @param stateManager - State manager containing tasks state
- * @returns A tool handler function
- *
- * @example
- * const handler = createTaskListHandler(stateManager);
+ * @returns A ToolHandler for TaskList tool calls
  */
 export function createTaskListHandler<
   TCustom extends JsonSerializable<TCustom>,
 >(
   stateManager: AgentStateManager<TCustom>
-): (args: TaskListToolSchemaType) => ToolHandlerResponse<WorkflowTask[]> {
-  return (
-    _args: TaskListToolSchemaType
-  ): ToolHandlerResponse<WorkflowTask[]> => {
+): ToolHandler<TaskListArgs, WorkflowTask[]> {
+  return () => {
     const taskList = stateManager.getTasks();
 
     return {

package/src/tools/task-list/tool.ts

@@ -1,9 +1,10 @@
 import z from "zod";
+import type { ToolDefinition } from "../../lib/tool-router";
 
 export const taskListTool = {
   name: "TaskList" as const,
   description: `List all tasks with current state.`,
   schema: z.object({}),
-};
+} satisfies ToolDefinition;
 
-export type TaskListToolSchemaType = z.infer<typeof taskListTool.schema>;
+export type TaskListArgs = z.infer<typeof taskListTool.schema>;

package/src/tools/task-update/handler.ts

@@ -2,29 +2,22 @@ import type {
   AgentStateManager,
   JsonSerializable,
 } from "../../lib/state-manager";
-import type { ToolHandlerResponse } from "../../lib/tool-router";
+import type { ToolHandler } from "../../lib/tool-router";
 import type { WorkflowTask } from "../../lib/types";
-import type { TaskUpdateToolSchemaType } from "./tool";
+import type { TaskUpdateArgs } from "./tool";
 
 /**
  * Creates a TaskUpdate handler that modifies task status and dependencies.
  *
  * @param stateManager - State manager containing tasks state
- * @returns A tool handler function
- *
- * @example
- * const handler = createTaskUpdateHandler(stateManager);
+ * @returns A ToolHandler for TaskUpdate tool calls
  */
 export function createTaskUpdateHandler<
   TCustom extends JsonSerializable<TCustom>,
 >(
   stateManager: AgentStateManager<TCustom>
-): (
-  args: TaskUpdateToolSchemaType
-) => ToolHandlerResponse<WorkflowTask | null> {
-  return (
-    args: TaskUpdateToolSchemaType
-  ): ToolHandlerResponse<WorkflowTask | null> => {
+): ToolHandler<TaskUpdateArgs, WorkflowTask | null> {
+  return (args) => {
     const task = stateManager.getTask(args.taskId);
 
     if (!task) {

package/src/tools/task-update/tool.ts

@@ -1,4 +1,5 @@
 import z from "zod";
+import type { ToolDefinition } from "../../lib/tool-router";
 
 export const taskUpdateTool = {
   name: "TaskUpdate" as const,
@@ -15,6 +16,6 @@ export const taskUpdateTool = {
       .array(z.string())
       .describe("The IDs of the tasks that this task is blocking"),
   }),
-};
+} satisfies ToolDefinition;
 
-export type TaskUpdateToolSchemaType = z.infer<typeof taskUpdateTool.schema>;
+export type TaskUpdateArgs = z.infer<typeof taskUpdateTool.schema>;

package/src/tools/{write → write-file}/tool.ts

@@ -1,11 +1,12 @@
 import { z } from "zod";
+import type { ToolDefinition } from "../../lib/tool-router";
 
 export const writeTool = {
   name: "FileWrite" as const,
   description: `Create or overwrite a file with new content.
 
 Usage:
-- Provide the absolute virtual path to the file
+- Provide the absolute path to the file
 - The file will be created if it doesn't exist
 - If the file exists, it will be completely overwritten
 
@@ -15,12 +16,10 @@ IMPORTANT:
 - Path must be absolute (e.g., "/docs/readme.md", not "docs/readme.md")
 `,
   schema: z.object({
-    file_path: z
-      .string()
-      .describe("The absolute virtual path to the file to write"),
+    file_path: z.string().describe("The absolute path to the file to write"),
     content: z.string().describe("The content to write to the file"),
   }),
   strict: true,
-};
+} satisfies ToolDefinition;
 
-export type WriteToolSchemaType = z.infer<typeof writeTool.schema>;
+export type FileWriteArgs = z.infer<typeof writeTool.schema>;

package/src/workflow.ts

@@ -16,7 +16,7 @@
  */
 
 // Session
-export { createSession } from "./lib/session";
+export { createSession, proxyDefaultThreadOps } from "./lib/session";
 export type { ZeitlichSession, SessionLifecycleHooks } from "./lib/session";
 
 // State management
@@ -33,7 +33,12 @@ export type {
 } from "./lib/state-manager";
 
 // Tool router (includes registry functionality)
-export { createToolRouter, hasNoOtherToolCalls, defineTool, defineSubagent } from "./lib/tool-router";
+export {
+  createToolRouter,
+  hasNoOtherToolCalls,
+  defineTool,
+  defineSubagent,
+} from "./lib/tool-router";
 export type {
   // Tool definition types
   ToolDefinition,
@@ -68,6 +73,7 @@ export type {
   BaseAgentState,
   AgentFile,
   AgentResponse,
+  ThreadOps,
   ZeitlichAgentConfig,
   RunAgentConfig,
   RunAgentActivity,
@@ -96,10 +102,7 @@ export { isTerminalStatus } from "./lib/types";
 
 // Subagent support
 export { createTaskTool } from "./tools/task/tool";
-export type {
-  TaskToolSchemaType,
-  GenericTaskToolSchemaType,
-} from "./tools/task/tool";
+export type { TaskArgs } from "./tools/task/tool";
 export type { TaskHandlerResult } from "./tools/task/handler";
 
 // Activity type interfaces (types only, no runtime code)
@@ -108,33 +111,34 @@ export type { ZeitlichSharedActivities } from "./activities";
 
 // Tool definitions (schemas only - no handlers)
 export { askUserQuestionTool } from "./tools/ask-user-question/tool";
-export type { AskUserQuestionToolSchemaType } from "./tools/ask-user-question/tool";
+export type { AskUserQuestionArgs } from "./tools/ask-user-question/tool";
 export { globTool } from "./tools/glob/tool";
-export type { GlobToolSchemaType } from "./tools/glob/tool";
+export type { GlobArgs } from "./tools/glob/tool";
 export { grepTool } from "./tools/grep/tool";
-export type { GrepToolSchemaType } from "./tools/grep/tool";
-export { readTool } from "./tools/read/tool";
-export type { ReadToolSchemaType } from "./tools/read/tool";
-export { writeTool } from "./tools/write/tool";
-export type { WriteToolSchemaType } from "./tools/write/tool";
+export type { GrepArgs } from "./tools/grep/tool";
+export { readTool } from "./tools/read-file/tool";
+export type { FileReadArgs } from "./tools/read-file/tool";
+export { writeTool } from "./tools/write-file/tool";
+export type { FileWriteArgs } from "./tools/write-file/tool";
 export { editTool } from "./tools/edit/tool";
-export type { EditToolSchemaType } from "./tools/edit/tool";
+export type { FileEditArgs } from "./tools/edit/tool";
 
 // Workflow task tools (state-only, no activities needed)
 export { taskCreateTool } from "./tools/task-create/tool";
-export type { TaskCreateToolSchemaType } from "./tools/task-create/tool";
+export type { TaskCreateArgs } from "./tools/task-create/tool";
 export { createTaskCreateHandler } from "./tools/task-create/handler";
 
 export { taskGetTool } from "./tools/task-get/tool";
-export type { TaskGetToolSchemaType } from "./tools/task-get/tool";
+export type { TaskGetArgs } from "./tools/task-get/tool";
 export { createTaskGetHandler } from "./tools/task-get/handler";
 
 export { taskListTool } from "./tools/task-list/tool";
+export type { TaskListArgs } from "./tools/task-list/tool";
 export { createTaskListHandler } from "./tools/task-list/handler";
 
 export { taskUpdateTool } from "./tools/task-update/tool";
-export type { TaskUpdateToolSchemaType } from "./tools/task-update/tool";
+export type { TaskUpdateArgs } from "./tools/task-update/tool";
 export { createTaskUpdateHandler } from "./tools/task-update/handler";
 
-export { bashTool } from "./tools/bash/tool";
-export type { bashToolSchemaType } from "./tools/bash/tool";
+export { bashTool, createBashToolDescription } from "./tools/bash/tool";
+export type { BashArgs } from "./tools/bash/tool";