agency-lang 0.0.93 → 0.0.94

package/dist/lib/agents/agency-agent/subagents/code.js

@@ -0,0 +1,628 @@
+import { print, __printTool, __printToolParams, printJSON, __printJSONTool, __printJSONToolParams, input, __inputTool, __inputToolParams, sleep, __sleepTool, __sleepToolParams, round, __roundTool, __roundToolParams, fetch, __fetchTool, __fetchToolParams, fetchJSON, __fetchJSONTool, __fetchJSONToolParams, read, __readTool, __readToolParams, write, __writeTool, __writeToolParams, readImage, __readImageTool, __readImageToolParams, notify, __notifyTool, __notifyToolParams } from "/Users/adityabhargava/agency-lang/stdlib/index.js";
+import { fileURLToPath } from "url";
+import process from "process";
+import { z } from "zod";
+import { color, nanoid } from "agency-lang";
+import path from "path";
+import {
+  RuntimeContext,
+  ThreadStore,
+  setupFunction,
+  callHook,
+  interrupt,
+  isInterrupt,
+  isDebugger,
+  debugStep,
+  respondToInterrupt as _respondToInterrupt,
+  approveInterrupt as _approveInterrupt,
+  rejectInterrupt as _rejectInterrupt,
+  resolveInterrupt as _resolveInterrupt,
+  modifyInterrupt as _modifyInterrupt,
+  rewindFrom as _rewindFrom,
+  ToolCallError,
+  RestoreSignal,
+  readSkill as _readSkillRaw,
+  readSkillTool as __readSkillTool,
+  readSkillToolParams as __readSkillToolParams,
+  _builtinTool as __builtinTool
+} from "agency-lang/runtime";
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+const __cwd = process.cwd();
+const getDirname = () => __dirname;
+const __globalCtx = new RuntimeContext({
+  statelogConfig: {
+    host: "https://agency-lang.com",
+    apiKey: process.env["STATELOG_API_KEY"] || "",
+    projectId: "agency-lang",
+    debugMode: false
+  },
+  smoltalkDefaults: {
+    openAiApiKey: process.env["OPENAI_API_KEY"] || "",
+    googleApiKey: process.env["GEMINI_API_KEY"] || "",
+    model: "gpt-4o-mini",
+    logLevel: "warn",
+    statelog: {
+      host: "https://agency-lang.com",
+      projectId: "smoltalk",
+      apiKey: process.env["STATELOG_SMOLTALK_API_KEY"] || "",
+      traceId: nanoid()
+    }
+  },
+  dirname: __dirname
+});
+const graph = __globalCtx.graph;
+function readSkill({ filepath }) {
+  return _readSkillRaw({ filepath, dirname: __dirname });
+}
+function tool(__name) {
+  return __builtinTool(__name, __toolRegistry);
+}
+function approve(value) {
+  return { type: "approved", value };
+}
+function reject(value) {
+  return { type: "rejected", value };
+}
+const respondToInterrupt = (interrupt2, response, opts) => _respondToInterrupt({ ctx: __globalCtx, interrupt: interrupt2, interruptResponse: response, overrides: opts?.overrides, metadata: opts?.metadata });
+const approveInterrupt = (interrupt2, opts) => _approveInterrupt({ ctx: __globalCtx, interrupt: interrupt2, overrides: opts?.overrides, metadata: opts?.metadata });
+const rejectInterrupt = (interrupt2, opts) => _rejectInterrupt({ ctx: __globalCtx, interrupt: interrupt2, overrides: opts?.overrides, metadata: opts?.metadata });
+const modifyInterrupt = (interrupt2, newArguments, opts) => _modifyInterrupt({ ctx: __globalCtx, interrupt: interrupt2, newArguments, overrides: opts?.overrides, metadata: opts?.metadata });
+const resolveInterrupt = (interrupt2, value, opts) => _resolveInterrupt({ ctx: __globalCtx, interrupt: interrupt2, value, overrides: opts?.overrides, metadata: opts?.metadata });
+const rewindFrom = (checkpoint2, overrides, opts) => _rewindFrom({ ctx: __globalCtx, checkpoint: checkpoint2, overrides, metadata: opts?.metadata });
+const __setDebugger = (dbg) => {
+  __globalCtx.debugger = dbg;
+};
+const __getCheckpoints = () => __globalCtx.checkpoints;
+function __initializeGlobals(__ctx) {
+  __ctx.globals.set("lib/agents/agency-agent/subagents/code.agency", "code", {
+    "contents": ``
+  });
+  __ctx.globals.markInitialized("lib/agents/agency-agent/subagents/code.agency");
+}
+const __writeCodeTool = {
+  name: "writeCode",
+  description: `Use this tool to write code to a file. The tool takes in a string of code content and replaces the existing code with it.
+    The tool returns the current state of the code after your modification.`,
+  schema: z.object({ "content": z.string() })
+};
+const __writeCodeToolParams = ["content"];
+const __appendCodeTool = {
+  name: "appendCode",
+  description: `Use this tool to incrementally write code to a file. The tool takes in a string of code content and appends it to the existing code.
+    The tool returns the current state of the code after your modification.`,
+  schema: z.object({ "content": z.string() })
+};
+const __appendCodeToolParams = ["content"];
+const __readCodeTool = {
+  name: "readCode",
+  description: `Use this tool to read the current code. This is useful to see what code has been written so far as you build it out over multiple steps.`,
+  schema: z.object({})
+};
+const __readCodeToolParams = [];
+const __toolRegistry = {
+  writeCode: {
+    definition: __writeCodeTool,
+    handler: {
+      name: "writeCode",
+      params: __writeCodeToolParams,
+      execute: writeCode,
+      isBuiltin: false
+    }
+  },
+  appendCode: {
+    definition: __appendCodeTool,
+    handler: {
+      name: "appendCode",
+      params: __appendCodeToolParams,
+      execute: appendCode,
+      isBuiltin: false
+    }
+  },
+  readCode: {
+    definition: __readCodeTool,
+    handler: {
+      name: "readCode",
+      params: __readCodeToolParams,
+      execute: readCode,
+      isBuiltin: false
+    }
+  },
+  print: {
+    definition: __printTool,
+    handler: {
+      name: "print",
+      params: __printToolParams,
+      execute: print,
+      isBuiltin: false
+    }
+  },
+  printJSON: {
+    definition: __printJSONTool,
+    handler: {
+      name: "printJSON",
+      params: __printJSONToolParams,
+      execute: printJSON,
+      isBuiltin: false
+    }
+  },
+  input: {
+    definition: __inputTool,
+    handler: {
+      name: "input",
+      params: __inputToolParams,
+      execute: input,
+      isBuiltin: false
+    }
+  },
+  sleep: {
+    definition: __sleepTool,
+    handler: {
+      name: "sleep",
+      params: __sleepToolParams,
+      execute: sleep,
+      isBuiltin: false
+    }
+  },
+  round: {
+    definition: __roundTool,
+    handler: {
+      name: "round",
+      params: __roundToolParams,
+      execute: round,
+      isBuiltin: false
+    }
+  },
+  fetch: {
+    definition: __fetchTool,
+    handler: {
+      name: "fetch",
+      params: __fetchToolParams,
+      execute: fetch,
+      isBuiltin: false
+    }
+  },
+  fetchJSON: {
+    definition: __fetchJSONTool,
+    handler: {
+      name: "fetchJSON",
+      params: __fetchJSONToolParams,
+      execute: fetchJSON,
+      isBuiltin: false
+    }
+  },
+  read: {
+    definition: __readTool,
+    handler: {
+      name: "read",
+      params: __readToolParams,
+      execute: read,
+      isBuiltin: false
+    }
+  },
+  write: {
+    definition: __writeTool,
+    handler: {
+      name: "write",
+      params: __writeToolParams,
+      execute: write,
+      isBuiltin: false
+    }
+  },
+  readImage: {
+    definition: __readImageTool,
+    handler: {
+      name: "readImage",
+      params: __readImageToolParams,
+      execute: readImage,
+      isBuiltin: false
+    }
+  },
+  notify: {
+    definition: __notifyTool,
+    handler: {
+      name: "notify",
+      params: __notifyToolParams,
+      execute: notify,
+      isBuiltin: false
+    }
+  },
+  readSkill: {
+    definition: __readSkillTool,
+    handler: {
+      name: "readSkill",
+      params: __readSkillToolParams,
+      execute: readSkill,
+      isBuiltin: true
+    }
+  }
+};
+async function writeCode(content, __state = void 0) {
+  const __setupData = setupFunction({
+    state: __state
+  });
+  const __stack = __setupData.stack;
+  const __step = __setupData.step;
+  const __self = __setupData.self;
+  const __threads = __setupData.threads;
+  const __ctx = __state?.ctx || __globalCtx;
+  const statelogClient = __ctx.statelogClient;
+  const __graph = __ctx.graph;
+  let __forked;
+  if (!__ctx.globals.isInitialized("lib/agents/agency-agent/subagents/code.agency")) {
+    __initializeGlobals(__ctx);
+  }
+  let __funcStartTime = performance.now();
+  await callHook({
+    callbacks: __ctx.callbacks,
+    name: "onFunctionStart",
+    data: {
+      functionName: "writeCode",
+      args: {
+        content
+      },
+      isBuiltin: false
+    }
+  });
+  await __ctx.audit({
+    type: "functionCall",
+    functionName: "writeCode",
+    args: {
+      content
+    },
+    result: void 0
+  });
+  __stack.args["content"] = content;
+  __self.__retryable = __self.__retryable ?? true;
+  try {
+    if (__step <= 0) {
+      __stack.step++;
+    }
+    if (__step <= 1) {
+      const __dbg = await debugStep(__ctx, __state, {
+        moduleId: "lib/agents/agency-agent/subagents/code.agency",
+        scopeName: "writeCode",
+        stepPath: "1",
+        label: null,
+        nodeContext: false
+      });
+      if (__dbg) {
+        return __dbg;
+      }
+      __stack.step++;
+    }
+    if (__step <= 2) {
+      const __funcResult = await print(color.green(`Overwriting code with new content`), {
+        ctx: __ctx,
+        threads: new ThreadStore(),
+        interruptData: __state?.interruptData
+      });
+      if (isInterrupt(__funcResult)) {
+        await __ctx.pendingPromises.awaitAll();
+        return {
+          data: __funcResult
+        };
+      }
+      await __ctx.audit({
+        type: "assignment",
+        variable: "__funcResult",
+        value: __funcResult
+      });
+      __stack.step++;
+    }
+    if (__step <= 3) {
+      const __dbg = await debugStep(__ctx, __state, {
+        moduleId: "lib/agents/agency-agent/subagents/code.agency",
+        scopeName: "writeCode",
+        stepPath: "3",
+        label: null,
+        nodeContext: false
+      });
+      if (__dbg) {
+        return __dbg;
+      }
+      __stack.step++;
+    }
+    if (__step <= 4) {
+      const __funcResult = await print(__stack.args.content, {
+        ctx: __ctx,
+        threads: new ThreadStore(),
+        interruptData: __state?.interruptData
+      });
+      if (isInterrupt(__funcResult)) {
+        await __ctx.pendingPromises.awaitAll();
+        return {
+          data: __funcResult
+        };
+      }
+      await __ctx.audit({
+        type: "assignment",
+        variable: "__funcResult",
+        value: __funcResult
+      });
+      __stack.step++;
+    }
+    if (__step <= 5) {
+      const __dbg = await debugStep(__ctx, __state, {
+        moduleId: "lib/agents/agency-agent/subagents/code.agency",
+        scopeName: "writeCode",
+        stepPath: "5",
+        label: null,
+        nodeContext: false
+      });
+      if (__dbg) {
+        return __dbg;
+      }
+      __stack.step++;
+    }
+    if (__step <= 6) {
+      __ctx.globals.get("lib/agents/agency-agent/subagents/code.agency", "code").contents = __stack.args.content;
+      await __ctx.audit({
+        type: "assignment",
+        variable: '__ctx.globals.get("lib/agents/agency-agent/subagents/code.agency", "code").contents',
+        value: __ctx.globals.get("lib/agents/agency-agent/subagents/code.agency", "code").contents
+      });
+      __stack.step++;
+    }
+    if (__step <= 7) {
+      const __dbg = await debugStep(__ctx, __state, {
+        moduleId: "lib/agents/agency-agent/subagents/code.agency",
+        scopeName: "writeCode",
+        stepPath: "7",
+        label: null,
+        nodeContext: false
+      });
+      if (__dbg) {
+        return __dbg;
+      }
+      __stack.step++;
+    }
+    if (__step <= 8) {
+      const __auditReturnValue = __ctx.globals.get("lib/agents/agency-agent/subagents/code.agency", "code").contents;
+      await __ctx.audit({
+        type: "return",
+        value: __auditReturnValue
+      });
+      return __auditReturnValue;
+      __stack.step++;
+    }
+  } catch (__error) {
+    if (__error instanceof RestoreSignal) {
+      throw __error;
+    }
+    if (__error instanceof ToolCallError) {
+      __error.retryable = __error.retryable && __self.__retryable;
+      throw __error;
+    }
+    throw new ToolCallError(__error, { retryable: __self.__retryable });
+  } finally {
+    if (!__state?.isForked) {
+      __ctx.stateStack.pop();
+    }
+  }
+  await callHook({
+    callbacks: __ctx.callbacks,
+    name: "onFunctionEnd",
+    data: {
+      functionName: "writeCode",
+      timeTaken: performance.now() - __funcStartTime
+    }
+  });
+}
+async function appendCode(content, __state = void 0) {
+  const __setupData = setupFunction({
+    state: __state
+  });
+  const __stack = __setupData.stack;
+  const __step = __setupData.step;
+  const __self = __setupData.self;
+  const __threads = __setupData.threads;
+  const __ctx = __state?.ctx || __globalCtx;
+  const statelogClient = __ctx.statelogClient;
+  const __graph = __ctx.graph;
+  let __forked;
+  if (!__ctx.globals.isInitialized("lib/agents/agency-agent/subagents/code.agency")) {
+    __initializeGlobals(__ctx);
+  }
+  let __funcStartTime = performance.now();
+  await callHook({
+    callbacks: __ctx.callbacks,
+    name: "onFunctionStart",
+    data: {
+      functionName: "appendCode",
+      args: {
+        content
+      },
+      isBuiltin: false
+    }
+  });
+  await __ctx.audit({
+    type: "functionCall",
+    functionName: "appendCode",
+    args: {
+      content
+    },
+    result: void 0
+  });
+  __stack.args["content"] = content;
+  __self.__retryable = __self.__retryable ?? true;
+  try {
+    if (__step <= 0) {
+      __stack.step++;
+    }
+    if (__step <= 1) {
+      const __dbg = await debugStep(__ctx, __state, {
+        moduleId: "lib/agents/agency-agent/subagents/code.agency",
+        scopeName: "appendCode",
+        stepPath: "1",
+        label: null,
+        nodeContext: false
+      });
+      if (__dbg) {
+        return __dbg;
+      }
+      __stack.step++;
+    }
+    if (__step <= 2) {
+      __ctx.globals.get("lib/agents/agency-agent/subagents/code.agency", "code").contents = __ctx.globals.get("lib/agents/agency-agent/subagents/code.agency", "code").contents + `
+${__stack.args.content}`;
+      await __ctx.audit({
+        type: "assignment",
+        variable: '__ctx.globals.get("lib/agents/agency-agent/subagents/code.agency", "code").contents',
+        value: __ctx.globals.get("lib/agents/agency-agent/subagents/code.agency", "code").contents
+      });
+      __stack.step++;
+    }
+    if (__step <= 3) {
+      const __dbg = await debugStep(__ctx, __state, {
+        moduleId: "lib/agents/agency-agent/subagents/code.agency",
+        scopeName: "appendCode",
+        stepPath: "3",
+        label: null,
+        nodeContext: false
+      });
+      if (__dbg) {
+        return __dbg;
+      }
+      __stack.step++;
+    }
+    if (__step <= 4) {
+      const __auditReturnValue = __ctx.globals.get("lib/agents/agency-agent/subagents/code.agency", "code").contents;
+      await __ctx.audit({
+        type: "return",
+        value: __auditReturnValue
+      });
+      return __auditReturnValue;
+      __stack.step++;
+    }
+  } catch (__error) {
+    if (__error instanceof RestoreSignal) {
+      throw __error;
+    }
+    if (__error instanceof ToolCallError) {
+      __error.retryable = __error.retryable && __self.__retryable;
+      throw __error;
+    }
+    throw new ToolCallError(__error, { retryable: __self.__retryable });
+  } finally {
+    if (!__state?.isForked) {
+      __ctx.stateStack.pop();
+    }
+  }
+  await callHook({
+    callbacks: __ctx.callbacks,
+    name: "onFunctionEnd",
+    data: {
+      functionName: "appendCode",
+      timeTaken: performance.now() - __funcStartTime
+    }
+  });
+}
+async function readCode(__state = void 0) {
+  const __setupData = setupFunction({
+    state: __state
+  });
+  const __stack = __setupData.stack;
+  const __step = __setupData.step;
+  const __self = __setupData.self;
+  const __threads = __setupData.threads;
+  const __ctx = __state?.ctx || __globalCtx;
+  const statelogClient = __ctx.statelogClient;
+  const __graph = __ctx.graph;
+  let __forked;
+  if (!__ctx.globals.isInitialized("lib/agents/agency-agent/subagents/code.agency")) {
+    __initializeGlobals(__ctx);
+  }
+  let __funcStartTime = performance.now();
+  await callHook({
+    callbacks: __ctx.callbacks,
+    name: "onFunctionStart",
+    data: {
+      functionName: "readCode",
+      args: {},
+      isBuiltin: false
+    }
+  });
+  await __ctx.audit({
+    type: "functionCall",
+    functionName: "readCode",
+    args: {},
+    result: void 0
+  });
+  __self.__retryable = __self.__retryable ?? true;
+  try {
+    if (__step <= 0) {
+      __stack.step++;
+    }
+    if (__step <= 1) {
+      const __dbg = await debugStep(__ctx, __state, {
+        moduleId: "lib/agents/agency-agent/subagents/code.agency",
+        scopeName: "readCode",
+        stepPath: "1",
+        label: null,
+        nodeContext: false
+      });
+      if (__dbg) {
+        return __dbg;
+      }
+      __stack.step++;
+    }
+    if (__step <= 2) {
+      const __auditReturnValue = __ctx.globals.get("lib/agents/agency-agent/subagents/code.agency", "code").contents;
+      await __ctx.audit({
+        type: "return",
+        value: __auditReturnValue
+      });
+      return __auditReturnValue;
+      __stack.step++;
+    }
+  } catch (__error) {
+    if (__error instanceof RestoreSignal) {
+      throw __error;
+    }
+    if (__error instanceof ToolCallError) {
+      __error.retryable = __error.retryable && __self.__retryable;
+      throw __error;
+    }
+    throw new ToolCallError(__error, { retryable: __self.__retryable });
+  } finally {
+    if (!__state?.isForked) {
+      __ctx.stateStack.pop();
+    }
+  }
+  await callHook({
+    callbacks: __ctx.callbacks,
+    name: "onFunctionEnd",
+    data: {
+      functionName: "readCode",
+      timeTaken: performance.now() - __funcStartTime
+    }
+  });
+}
+var stdin_default = graph;
+const __sourceMap = { "lib/agents/agency-agent/subagents/code.agency:writeCode": { "1": { "line": 14, "col": 2 }, "2": { "line": 14, "col": 2 }, "3": { "line": 15, "col": 2 }, "4": { "line": 15, "col": 2 }, "5": { "line": 16, "col": 2 }, "6": { "line": 16, "col": 2 }, "7": { "line": 17, "col": 2 }, "8": { "line": 17, "col": 2 } }, "lib/agents/agency-agent/subagents/code.agency:appendCode": { "1": { "line": 25, "col": 2 }, "2": { "line": 25, "col": 2 }, "3": { "line": 26, "col": 2 }, "4": { "line": 26, "col": 2 } }, "lib/agents/agency-agent/subagents/code.agency:readCode": { "1": { "line": 33, "col": 2 }, "2": { "line": 33, "col": 2 } } };
+export {
+  __appendCodeTool,
+  __appendCodeToolParams,
+  __getCheckpoints,
+  __readCodeTool,
+  __readCodeToolParams,
+  __setDebugger,
+  __sourceMap,
+  __writeCodeTool,
+  __writeCodeToolParams,
+  appendCode,
+  approveInterrupt,
+  stdin_default as default,
+  interrupt,
+  isDebugger,
+  isInterrupt,
+  modifyInterrupt,
+  readCode,
+  readSkill,
+  rejectInterrupt,
+  resolveInterrupt,
+  respondToInterrupt,
+  rewindFrom,
+  writeCode
+};