zeitlich 0.1.0

package/dist/workflow.js

@@ -0,0 +1,762 @@
+'use strict';
+
+var workflow = require('@temporalio/workflow');
+var z2 = require('zod');
+var minimatch = require('minimatch');
+
+function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
+
+var z2__default = /*#__PURE__*/_interopDefault(z2);
+
+// src/lib/session.ts
+var TASK_TOOL = "Task";
+function buildTaskDescription(subagents) {
+  const subagentList = subagents.map((s) => `- **${s.name}**: ${s.description}`).join("\n");
+  return `Launch a new agent to handle complex, multi-step tasks autonomously.
+
+The ${TASK_TOOL} tool launches specialized agents (subprocesses) that autonomously handle complex tasks. Each agent type has specific capabilities and tools available to it.
+
+Available agent types:
+
+${subagentList}
+
+When using the ${TASK_TOOL} tool, you must specify a subagent parameter to select which agent type to use.
+
+Usage notes:
+
+- Always include a short description (3-5 words) summarizing what the agent will do
+- Launch multiple agents concurrently whenever possible, to maximize performance; to do that, use a single message with multiple tool uses
+- When the agent is done, it will return a single message back to you. The result returned by the agent is not visible to the user. To show the user the result, you should send a text message back to the user with a concise summary of the result.
+- Each invocation starts fresh - provide a detailed task description with all necessary context.
+- Provide clear, detailed prompts so the agent can work autonomously and return exactly the information you need.
+- The agent's outputs should generally be trusted
+- Clearly tell the agent what type of work you expect since it is not aware of the user's intent
+- If the agent description mentions that it should be used proactively, then you should try your best to use it without the user having to ask for it first. Use your judgement.`;
+}
+function createTaskTool(subagents) {
+  if (subagents.length === 0) {
+    throw new Error("createTaskTool requires at least one subagent");
+  }
+  const names = subagents.map((s) => s.name);
+  return {
+    name: TASK_TOOL,
+    description: buildTaskDescription(subagents),
+    schema: z2__default.default.object({
+      subagent: z2__default.default.enum(names).describe("The type of subagent to launch"),
+      description: z2__default.default.string().describe("A short (3-5 word) description of the task"),
+      prompt: z2__default.default.string().describe("The task for the agent to perform")
+    })
+  };
+}
+function createTaskHandler(subagents) {
+  const { workflowId: parentWorkflowId, taskQueue: parentTaskQueue } = workflow.workflowInfo();
+  return async (args, _toolCallId) => {
+    const config = subagents.find((s) => s.name === args.subagent);
+    if (!config) {
+      throw new Error(
+        `Unknown subagent: ${args.subagent}. Available: ${subagents.map((s) => s.name).join(", ")}`
+      );
+    }
+    const childWorkflowId = `${parentWorkflowId}-${args.subagent}-${workflow.uuid4()}`;
+    const childResult = await workflow.executeChild(config.workflowType, {
+      workflowId: childWorkflowId,
+      args: [{ prompt: args.prompt }],
+      taskQueue: config.taskQueue ?? parentTaskQueue
+    });
+    const validated = config.resultSchema ? config.resultSchema.parse(childResult) : childResult;
+    const content = typeof validated === "string" ? validated : JSON.stringify(validated, null, 2);
+    return {
+      content,
+      result: {
+        result: validated,
+        childWorkflowId
+      }
+    };
+  };
+}
+
+// src/lib/subagent-support.ts
+function withSubagentSupport(userTools, config) {
+  if (config.subagents.length === 0) {
+    throw new Error("withSubagentSupport requires at least one subagent");
+  }
+  const taskTool = createTaskTool(config.subagents);
+  const taskHandler = createTaskHandler(config.subagents);
+  return {
+    tools: {
+      ...userTools,
+      Task: taskTool
+    },
+    taskHandler
+  };
+}
+function hasTaskTool(tools) {
+  return "Task" in tools;
+}
+
+// src/lib/session.ts
+var createSession = async ({ threadId, agentName, maxTurns = 50, metadata = {} }, {
+  runAgent,
+  promptManager,
+  toolRouter,
+  toolRegistry,
+  hooks = {}
+}) => {
+  const { initializeThread, appendHumanMessage, parseToolCalls } = workflow.proxyActivities({
+    startToCloseTimeout: "30m",
+    retry: {
+      maximumAttempts: 6,
+      initialInterval: "5s",
+      maximumInterval: "15m",
+      backoffCoefficient: 4
+    },
+    heartbeatTimeout: "5m"
+  });
+  const callSessionEnd = async (exitReason, turns) => {
+    if (hooks.onSessionEnd) {
+      await hooks.onSessionEnd({
+        threadId,
+        agentName,
+        exitReason,
+        turns,
+        metadata
+      });
+    }
+  };
+  return {
+    runSession: async (prompt, stateManager) => {
+      if (hooks.onSessionStart) {
+        await hooks.onSessionStart({
+          threadId,
+          agentName,
+          metadata
+        });
+      }
+      await initializeThread(threadId);
+      await appendHumanMessage(
+        threadId,
+        await promptManager.buildContextMessage(prompt)
+      );
+      let exitReason = "completed";
+      try {
+        while (stateManager.isRunning() && !stateManager.isTerminal() && stateManager.getTurns() < maxTurns) {
+          stateManager.incrementTurns();
+          const currentTurn = stateManager.getTurns();
+          const { message, stopReason } = await runAgent(
+            {
+              threadId,
+              agentName,
+              metadata
+            },
+            {
+              systemPrompt: await promptManager.getSystemPrompt()
+            }
+          );
+          if (stopReason === "end_turn") {
+            stateManager.complete();
+            exitReason = "completed";
+            return message;
+          }
+          const rawToolCalls = await parseToolCalls(message);
+          const parsedToolCalls = rawToolCalls.map(
+            (tc) => toolRegistry.parseToolCall(tc)
+          );
+          await toolRouter.processToolCalls(parsedToolCalls, {
+            turn: currentTurn
+          });
+          if (stateManager.getStatus() === "WAITING_FOR_INPUT") {
+            exitReason = "waiting_for_input";
+            break;
+          }
+        }
+        if (stateManager.getTurns() >= maxTurns && stateManager.isRunning()) {
+          exitReason = "max_turns";
+        }
+      } catch (error) {
+        exitReason = "failed";
+        throw error;
+      } finally {
+        await callSessionEnd(exitReason, stateManager.getTurns());
+      }
+      return null;
+    }
+  };
+};
+
+// src/lib/types.ts
+function isTerminalStatus(status) {
+  return status === "COMPLETED" || status === "FAILED" || status === "CANCELLED";
+}
+
+// src/lib/state-manager.ts
+function createAgentStateManager(config) {
+  let status = "RUNNING";
+  let version = 0;
+  let turns = 0;
+  const customState = { ...config?.initialState ?? {} };
+  function buildState() {
+    return {
+      status,
+      version,
+      turns,
+      ...customState
+    };
+  }
+  return {
+    getStatus() {
+      return status;
+    },
+    isRunning() {
+      return status === "RUNNING";
+    },
+    isTerminal() {
+      return isTerminalStatus(status);
+    },
+    getTurns() {
+      return turns;
+    },
+    getVersion() {
+      return version;
+    },
+    run() {
+      status = "RUNNING";
+      version++;
+    },
+    waitForInput() {
+      status = "WAITING_FOR_INPUT";
+      version++;
+    },
+    complete() {
+      status = "COMPLETED";
+      version++;
+    },
+    fail() {
+      status = "FAILED";
+      version++;
+    },
+    cancel() {
+      status = "CANCELLED";
+      version++;
+    },
+    incrementVersion() {
+      version++;
+    },
+    incrementTurns() {
+      turns++;
+    },
+    get(key) {
+      return customState[key];
+    },
+    set(key, value) {
+      customState[key] = value;
+      version++;
+    },
+    getCurrentState() {
+      return buildState();
+    },
+    shouldReturnFromWait(lastKnownVersion) {
+      return version > lastKnownVersion || isTerminalStatus(status);
+    }
+  };
+}
+var AGENT_HANDLER_NAMES = {
+  getAgentState: "getAgentState",
+  waitForStateChange: "waitForStateChange",
+  addMessage: "addMessage"
+};
+
+// src/lib/prompt-manager.ts
+function createPromptManager(config) {
+  const { baseSystemPrompt, instructionsPrompt, buildContextMessage } = config;
+  async function resolvePrompt(prompt) {
+    if (typeof prompt === "function") {
+      return prompt();
+    }
+    return prompt;
+  }
+  return {
+    async getSystemPrompt() {
+      const base = await resolvePrompt(baseSystemPrompt);
+      const instructions = await resolvePrompt(instructionsPrompt);
+      return [base, instructions].join("\n");
+    },
+    async buildContextMessage(context) {
+      return buildContextMessage(context);
+    }
+  };
+}
+
+// src/lib/tool-registry.ts
+function createToolRegistry(tools) {
+  const toolMap = /* @__PURE__ */ new Map();
+  for (const [_key, tool] of Object.entries(tools)) {
+    toolMap.set(tool.name, tool);
+  }
+  return {
+    parseToolCall(toolCall) {
+      const tool = toolMap.get(toolCall.name);
+      if (!tool) {
+        throw new Error(`Tool ${toolCall.name} not found`);
+      }
+      const parsedArgs = tool.schema.parse(toolCall.args);
+      return {
+        id: toolCall.id ?? "",
+        name: toolCall.name,
+        args: parsedArgs
+      };
+    },
+    getToolList() {
+      return Object.values(tools);
+    },
+    getTool(name) {
+      return tools[name];
+    },
+    hasTool(name) {
+      return toolMap.has(name);
+    },
+    getToolNames() {
+      return Array.from(toolMap.keys());
+    }
+  };
+}
+
+// src/lib/tool-router.ts
+function createToolRouter(options, handlers) {
+  const { parallel = true, threadId, appendToolResult, hooks } = options;
+  async function processToolCall(toolCall, turn) {
+    const startTime = Date.now();
+    let effectiveArgs = toolCall.args;
+    if (hooks?.onPreToolUse) {
+      const preResult = await hooks.onPreToolUse({
+        toolCall,
+        threadId,
+        turn
+      });
+      if (preResult?.skip) {
+        await appendToolResult({
+          threadId,
+          toolCallId: toolCall.id,
+          content: JSON.stringify({
+            skipped: true,
+            reason: "Skipped by PreToolUse hook"
+          })
+        });
+        return null;
+      }
+      if (preResult?.modifiedArgs !== void 0) {
+        effectiveArgs = preResult.modifiedArgs;
+      }
+    }
+    const handler = handlers[toolCall.name];
+    let result;
+    let content;
+    try {
+      if (handler) {
+        const response = await handler(
+          effectiveArgs,
+          toolCall.id
+        );
+        result = response.result;
+        content = response.content;
+      } else {
+        result = { error: `Unknown tool: ${toolCall.name}` };
+        content = JSON.stringify(result, null, 2);
+      }
+    } catch (error) {
+      if (hooks?.onPostToolUseFailure) {
+        const failureResult = await hooks.onPostToolUseFailure({
+          toolCall,
+          error: error instanceof Error ? error : new Error(String(error)),
+          threadId,
+          turn
+        });
+        if (failureResult?.fallbackContent !== void 0) {
+          content = failureResult.fallbackContent;
+          result = { error: String(error), recovered: true };
+        } else if (failureResult?.suppress) {
+          content = JSON.stringify({ error: String(error), suppressed: true });
+          result = { error: String(error), suppressed: true };
+        } else {
+          throw error;
+        }
+      } else {
+        throw error;
+      }
+    }
+    await appendToolResult({ threadId, toolCallId: toolCall.id, content });
+    const toolResult = {
+      toolCallId: toolCall.id,
+      name: toolCall.name,
+      result
+    };
+    if (hooks?.onPostToolUse) {
+      const durationMs = Date.now() - startTime;
+      await hooks.onPostToolUse({
+        toolCall,
+        result: toolResult,
+        threadId,
+        turn,
+        durationMs
+      });
+    }
+    return toolResult;
+  }
+  return {
+    async processToolCalls(toolCalls, context) {
+      if (toolCalls.length === 0) {
+        return [];
+      }
+      const turn = context?.turn ?? 0;
+      if (parallel) {
+        const results2 = await Promise.all(
+          toolCalls.map((tc) => processToolCall(tc, turn))
+        );
+        return results2.filter(
+          (r) => r !== null
+        );
+      }
+      const results = [];
+      for (const toolCall of toolCalls) {
+        const result = await processToolCall(toolCall, turn);
+        if (result !== null) {
+          results.push(result);
+        }
+      }
+      return results;
+    },
+    async processToolCallsByName(toolCalls, toolName, handler) {
+      const matchingCalls = toolCalls.filter((tc) => tc.name === toolName);
+      if (matchingCalls.length === 0) {
+        return [];
+      }
+      const processOne = async (toolCall) => {
+        const response = await handler(
+          toolCall.args,
+          toolCall.id
+        );
+        await appendToolResult({
+          threadId,
+          toolCallId: toolCall.id,
+          content: response.content
+        });
+        return {
+          toolCallId: toolCall.id,
+          name: toolCall.name,
+          result: response.result
+        };
+      };
+      if (parallel) {
+        return Promise.all(matchingCalls.map(processOne));
+      }
+      const results = [];
+      for (const toolCall of matchingCalls) {
+        results.push(await processOne(toolCall));
+      }
+      return results;
+    },
+    filterByName(toolCalls, name) {
+      return toolCalls.filter(
+        (tc) => tc.name === name
+      );
+    },
+    hasToolCall(toolCalls, name) {
+      return toolCalls.some((tc) => tc.name === name);
+    },
+    getResultsByName(results, name) {
+      return results.filter(
+        (r) => r.name === name
+      );
+    }
+  };
+}
+function hasNoOtherToolCalls(toolCalls, excludeName) {
+  return toolCalls.filter((tc) => tc.name !== excludeName).length === 0;
+}
+var askUserQuestionTool = {
+  name: "AskUserQuestion",
+  description: `Use this tool when you need to ask the user questions during execution. This allows you to:
+
+1. Gather user preferences or requirements
+2. Clarify ambiguous instructions
+3. Get decisions on implementation choices as you work
+4. Offer choices to the user about what direction to take.
+
+Usage notes:
+
+* Users will always be able to select "Other" to provide custom text input
+* Use multiSelect: true to allow multiple answers to be selected for a question
+* If you recommend a specific option, make that the first option in the list and add "(Recommended)" at the end of the label
+`,
+  schema: z2__default.default.object({
+    questions: z2__default.default.array(
+      z2__default.default.object({
+        question: z2__default.default.string().describe("The full question text to display"),
+        header: z2__default.default.string().describe("Short label for the question (max 12 characters)"),
+        options: z2__default.default.array(
+          z2__default.default.object({
+            label: z2__default.default.string(),
+            description: z2__default.default.string()
+          })
+        ).min(0).max(4).describe("Array of 0-4 choices, each with label and description"),
+        multiSelect: z2__default.default.boolean().describe("If true, users can select multiple options")
+      })
+    )
+  }),
+  strict: true
+};
+var globTool = {
+  name: "Glob",
+  description: `Search for files matching a glob pattern within the available file system.
+
+Usage:
+- Use glob patterns like "**/*.ts" to find all TypeScript files
+- Use "docs/**" to find all files in the docs directory
+- Patterns are matched against virtual paths in the file system
+
+Examples:
+- "*.md" - Find all markdown files in the root
+- "**/*.test.ts" - Find all test files recursively
+- "src/**/*.ts" - Find all TypeScript files in src directory
+`,
+  schema: z2.z.object({
+    pattern: z2.z.string().describe("Glob pattern to match files against"),
+    root: z2.z.string().optional().describe("Optional root directory to search from")
+  }),
+  strict: true
+};
+var grepTool = {
+  name: "Grep",
+  description: `Search file contents for a pattern within the available file system.
+
+Usage:
+- Searches for a regex pattern across file contents
+- Returns matching lines with file paths and line numbers
+- Can filter by file patterns and limit results
+
+Examples:
+- Search for "TODO" in all files
+- Search for function definitions with "function.*handleClick"
+- Search case-insensitively with ignoreCase: true
+`,
+  schema: z2.z.object({
+    pattern: z2.z.string().describe("Regex pattern to search for in file contents"),
+    ignoreCase: z2.z.boolean().optional().describe("Case-insensitive search (default: false)"),
+    maxMatches: z2.z.number().optional().describe("Maximum number of matches to return (default: 50)"),
+    includePatterns: z2.z.array(z2.z.string()).optional().describe("Glob patterns to include (e.g., ['*.ts', '*.js'])"),
+    excludePatterns: z2.z.array(z2.z.string()).optional().describe("Glob patterns to exclude (e.g., ['*.test.ts'])"),
+    contextLines: z2.z.number().optional().describe("Number of context lines to show around matches")
+  }),
+  strict: true
+};
+var readTool = {
+  name: "FileRead",
+  description: `Read file contents with optional pagination.
+
+Usage:
+- Provide the virtual path to the file you want to read
+- Supports text files, images, and PDFs
+- For large files, use offset and limit to read specific portions
+
+The tool returns the file content in an appropriate format:
+- Text files: Plain text content
+- Images: Base64-encoded image data
+- PDFs: Extracted text content
+`,
+  schema: z2.z.object({
+    path: z2.z.string().describe("Virtual path to the file to read"),
+    offset: z2.z.number().optional().describe(
+      "Line number to start reading from (1-indexed, for text files)"
+    ),
+    limit: z2.z.number().optional().describe("Maximum number of lines to read (for text files)")
+  }),
+  strict: true
+};
+var writeTool = {
+  name: "FileWrite",
+  description: `Create or overwrite a file with new content.
+
+Usage:
+- Provide the absolute virtual path to the file
+- The file will be created if it doesn't exist
+- If the file exists, it will be completely overwritten
+
+IMPORTANT:
+- You must read the file first (in this session) before writing to it
+- This is an atomic write operation - the entire file is replaced
+- Path must be absolute (e.g., "/docs/readme.md", not "docs/readme.md")
+`,
+  schema: z2.z.object({
+    file_path: z2.z.string().describe("The absolute virtual path to the file to write"),
+    content: z2.z.string().describe("The content to write to the file")
+  }),
+  strict: true
+};
+var editTool = {
+  name: "FileEdit",
+  description: `Edit specific sections of a file by replacing text.
+
+Usage:
+- Provide the exact text to find and replace
+- The old_string must match exactly (whitespace-sensitive)
+- By default, only replaces the first occurrence
+- Use replace_all: true to replace all occurrences
+
+IMPORTANT:
+- You must read the file first (in this session) before editing it
+- old_string must be unique in the file (unless using replace_all)
+- The operation fails if old_string is not found
+- old_string and new_string must be different
+`,
+  schema: z2.z.object({
+    file_path: z2.z.string().describe("The absolute virtual path to the file to modify"),
+    old_string: z2.z.string().describe("The exact text to replace"),
+    new_string: z2.z.string().describe(
+      "The text to replace it with (must be different from old_string)"
+    ),
+    replace_all: z2.z.boolean().optional().describe(
+      "If true, replace all occurrences of old_string (default: false)"
+    )
+  }),
+  strict: true
+};
+
+// src/lib/filesystem/types.ts
+function fileContentToMessageContent(content) {
+  switch (content.type) {
+    case "text":
+      return [{ type: "text", text: content.content }];
+    case "image":
+      return [
+        {
+          type: "image_url",
+          image_url: {
+            url: `data:${content.mimeType};base64,${content.data}`
+          }
+        }
+      ];
+    case "pdf":
+      return [{ type: "text", text: content.content }];
+  }
+}
+var DEFAULT_HEADER = "Available files and directories:";
+var DEFAULT_DESCRIPTION = "You have access to the following files. Use the Read, Glob, and Grep tools to explore them.";
+function isExcluded(path, excludePatterns) {
+  if (!excludePatterns || excludePatterns.length === 0) {
+    return false;
+  }
+  return excludePatterns.some((pattern) => minimatch.minimatch(path, pattern));
+}
+function renderNode(node, options, depth, indent) {
+  const lines = [];
+  if (options.maxDepth !== void 0 && depth > options.maxDepth) {
+    return lines;
+  }
+  if (isExcluded(node.path, options.excludePatterns)) {
+    return lines;
+  }
+  const parts = node.path.split("/");
+  const name = parts[parts.length - 1] || node.path;
+  let line = indent;
+  if (node.type === "directory") {
+    line += `${name}/`;
+  } else {
+    line += name;
+  }
+  if (options.showMimeTypes && node.mimeType) {
+    line += ` [${node.mimeType}]`;
+  }
+  if (options.showDescriptions !== false && node.description) {
+    line += ` - ${node.description}`;
+  }
+  lines.push(line);
+  if (node.type === "directory" && node.children) {
+    const childIndent = indent + "  ";
+    for (const child of node.children) {
+      lines.push(...renderNode(child, options, depth + 1, childIndent));
+    }
+  }
+  return lines;
+}
+function buildFileTreePrompt(nodes, options = {}) {
+  const header = options.headerText ?? DEFAULT_HEADER;
+  const description = options.descriptionText ?? DEFAULT_DESCRIPTION;
+  const lines = [];
+  for (const node of nodes) {
+    lines.push(...renderNode(node, options, 0, ""));
+  }
+  if (lines.length === 0) {
+    return `<file_system>
+${header}
+
+${description}
+
+(no files available)
+</file_system>`;
+  }
+  return `<file_system>
+${header}
+
+${lines.join("\n")}
+</file_system>`;
+}
+function flattenFileTree(nodes) {
+  const paths = [];
+  function traverse(node) {
+    if (node.type === "file") {
+      paths.push(node.path);
+    }
+    if (node.children) {
+      for (const child of node.children) {
+        traverse(child);
+      }
+    }
+  }
+  for (const node of nodes) {
+    traverse(node);
+  }
+  return paths;
+}
+function isPathInScope(path, scopedNodes) {
+  const allowedPaths = flattenFileTree(scopedNodes);
+  return allowedPaths.includes(path);
+}
+function findNodeByPath(path, nodes) {
+  for (const node of nodes) {
+    if (node.path === path) {
+      return node;
+    }
+    if (node.children) {
+      const found = findNodeByPath(path, node.children);
+      if (found) {
+        return found;
+      }
+    }
+  }
+  return void 0;
+}
+
+exports.AGENT_HANDLER_NAMES = AGENT_HANDLER_NAMES;
+exports.askUserQuestionTool = askUserQuestionTool;
+exports.buildFileTreePrompt = buildFileTreePrompt;
+exports.createAgentStateManager = createAgentStateManager;
+exports.createPromptManager = createPromptManager;
+exports.createSession = createSession;
+exports.createTaskHandler = createTaskHandler;
+exports.createTaskTool = createTaskTool;
+exports.createToolRegistry = createToolRegistry;
+exports.createToolRouter = createToolRouter;
+exports.editTool = editTool;
+exports.fileContentToMessageContent = fileContentToMessageContent;
+exports.findNodeByPath = findNodeByPath;
+exports.flattenFileTree = flattenFileTree;
+exports.globTool = globTool;
+exports.grepTool = grepTool;
+exports.hasNoOtherToolCalls = hasNoOtherToolCalls;
+exports.hasTaskTool = hasTaskTool;
+exports.isPathInScope = isPathInScope;
+exports.isTerminalStatus = isTerminalStatus;
+exports.readTool = readTool;
+exports.withSubagentSupport = withSubagentSupport;
+exports.writeTool = writeTool;
+//# sourceMappingURL=workflow.js.map
+//# sourceMappingURL=workflow.js.map