agency-lang 0.0.11 → 0.0.13

package/dist/lib/templates/backends/graphGenerator/imports.d.ts

@@ -1,4 +1,4 @@
-export declare const template = "import { z } from \"zod\";\nimport * as readline from \"readline\";\nimport fs from \"fs\";\nimport { PieMachine, goToNode } from \"piemachine\";\nimport { StatelogClient } from \"statelog-client\";\nimport { nanoid } from \"nanoid\";\nimport { assistantMessage, getClient, userMessage } from \"smoltalk\";\n\nconst statelogHost = \"https://statelog.adit.io\";\nconst traceId = nanoid();\nconst statelogConfig = {\n    host: statelogHost,\n    traceId: traceId,\n    apiKey: process.env.STATELOG_API_KEY || \"\",\n    projectId: \"agency-lang\",\n    debugMode: false,\n  };\nconst statelogClient = new StatelogClient(statelogConfig);\nconst model = \"gpt-4o-mini\";\n\n\nconst getClientWithConfig = (config = {}) => {\n  const defaultConfig = {\n    openAiApiKey: process.env.OPENAI_API_KEY || \"\",\n    googleApiKey: process.env.GEMINI_API_KEY || \"\",\n    model,\n    logLevel: \"warn\",\n  };\n\n  return getClient({ ...defaultConfig, ...config });\n};\n\nlet client = getClientWithConfig();\n\ntype State = {\n  messages: string[];\n  data: any;\n}\n\n// enable debug logging\nconst graphConfig = {\n  debug: {\n    log: true,\n    logData: true,\n  },\n  statelog: statelogConfig,\n};\n\n// Define the names of the nodes in the graph\n// Useful for type safety\nconst nodes = {{{nodes:string}}} as const;\ntype Node = (typeof nodes)[number];\n\nconst graph = new PieMachine<State, Node>(nodes, graphConfig);";
+export declare const template = "// @ts-nocheck\nimport { z } from \"zod\";\nimport * as readline from \"readline\";\nimport fs from \"fs\";\nimport { PieMachine, goToNode } from \"piemachine\";\nimport { StatelogClient } from \"statelog-client\";\nimport { nanoid } from \"nanoid\";\nimport { assistantMessage, getClient, userMessage } from \"smoltalk\";\n\nconst statelogHost = \"https://statelog.adit.io\";\nconst traceId = nanoid();\nconst statelogConfig = {\n    host: statelogHost,\n    traceId: traceId,\n    apiKey: process.env.STATELOG_API_KEY || \"\",\n    projectId: \"agency-lang\",\n    debugMode: false,\n  };\nconst statelogClient = new StatelogClient(statelogConfig);\nconst __model: ModelName = \"gpt-4o-mini\";\n\n\nconst getClientWithConfig = (config = {}) => {\n  const defaultConfig = {\n    openAiApiKey: process.env.OPENAI_API_KEY || \"\",\n    googleApiKey: process.env.GEMINI_API_KEY || \"\",\n    model: __model,\n    logLevel: \"warn\",\n  };\n\n  return getClient({ ...defaultConfig, ...config });\n};\n\nlet __client = getClientWithConfig();\n\ntype State = {\n  messages: string[];\n  data: any;\n}\n\n// enable debug logging\nconst graphConfig = {\n  debug: {\n    log: true,\n    logData: true,\n  },\n  statelog: statelogConfig,\n};\n\n// Define the names of the nodes in the graph\n// Useful for type safety\nconst __nodes = {{{nodes:string}}} as const;\ntype Node = (typeof __nodes)[number];\n\nconst graph = new PieMachine<State, Node>(__nodes, graphConfig);";
 export type TemplateType = {
     nodes: string;
 };

package/dist/lib/templates/backends/graphGenerator/imports.js

@@ -2,7 +2,8 @@
 // Source: lib/templates/backends/graphGenerator/imports.mustache
 // Any manual changes will be lost.
 import { apply } from "typestache";
-export const template = `import { z } from "zod";
+export const template = `// @ts-nocheck
+import { z } from "zod";
 import * as readline from "readline";
 import fs from "fs";
 import { PieMachine, goToNode } from "piemachine";
@@ -20,21 +21,21 @@ const statelogConfig = {
     debugMode: false,
   };
 const statelogClient = new StatelogClient(statelogConfig);
-const model = "gpt-4o-mini";
+const __model: ModelName = "gpt-4o-mini";
 
 
 const getClientWithConfig = (config = {}) => {
   const defaultConfig = {
     openAiApiKey: process.env.OPENAI_API_KEY || "",
     googleApiKey: process.env.GEMINI_API_KEY || "",
-    model,
+    model: __model,
     logLevel: "warn",
   };
 
   return getClient({ ...defaultConfig, ...config });
 };
 
-let client = getClientWithConfig();
+let __client = getClientWithConfig();
 
 type State = {
   messages: string[];
@@ -52,10 +53,10 @@ const graphConfig = {
 
 // Define the names of the nodes in the graph
 // Useful for type safety
-const nodes = {{{nodes:string}}} as const;
-type Node = (typeof nodes)[number];
+const __nodes = {{{nodes:string}}} as const;
+type Node = (typeof __nodes)[number];
 
-const graph = new PieMachine<State, Node>(nodes, graphConfig);`;
+const graph = new PieMachine<State, Node>(__nodes, graphConfig);`;
 const render = (args) => {
     return apply(template, args);
 };

package/dist/lib/templates/backends/graphGenerator/runNodeFunction.d.ts

@@ -1,4 +1,4 @@
-export declare const template = "export async function {{{nodeName:string}}}(data) {\n  const result = await graph.run(\"{{{nodeName:string}}}\", { messages: [], data });\n  return result.data;\n}\n";
+export declare const template = "export async function {{{nodeName:string}}}(data:any): Promise<any> {\n  const result = await graph.run(\"{{{nodeName:string}}}\", { messages: [], data });\n  return result.data;\n}\n";
 export type TemplateType = {
     nodeName: string;
 };

package/dist/lib/templates/backends/graphGenerator/runNodeFunction.js

@@ -2,7 +2,7 @@
 // Source: lib/templates/backends/graphGenerator/runNodeFunction.mustache
 // Any manual changes will be lost.
 import { apply } from "typestache";
-export const template = `export async function {{{nodeName:string}}}(data) {
+export const template = `export async function {{{nodeName:string}}}(data:any): Promise<any> {
   const result = await graph.run("{{{nodeName:string}}}", { messages: [], data });
   return result.data;
 }

package/dist/lib/templates/backends/graphGenerator/specialVar.d.ts

@@ -1,4 +1,4 @@
-export declare const template = "client = getClientWithConfig({ {{{name}}}: {{{value}}} });";
+export declare const template = "__client = getClientWithConfig({ {{{name}}}: {{{value}}} });";
 export type TemplateType = {
     name: string | boolean | number;
     value: string | boolean | number;

package/dist/lib/templates/backends/graphGenerator/specialVar.js

@@ -2,7 +2,7 @@
 // Source: lib/templates/backends/graphGenerator/specialVar.mustache
 // Any manual changes will be lost.
 import { apply } from "typestache";
-export const template = `client = getClientWithConfig({ {{{name}}}: {{{value}}} });`;
+export const template = `__client = getClientWithConfig({ {{{name}}}: {{{value}}} });`;
 const render = (args) => {
     return apply(template, args);
 };

package/dist/lib/templates/backends/typescriptGenerator/promptFunction.d.ts

@@ -1,4 +1,4 @@
-export declare const template = "\nasync function _{{{variableName:string}}}({{{argsStr:string}}}): Promise<{{{typeString:string}}}> {\n  const prompt = {{{promptCode:string}}};\n  const startTime = performance.now();\n  const messages: Message[] = [userMessage(prompt)];\n  const tools = {{{tools}}};\n\n  {{#hasResponseFormat}}\n  // Need to make sure this is always an object\n  const responseFormat = z.object({\n     response: {{{zodSchema:string}}}\n  });\n  {{/hasResponseFormat}}\n  {{^hasResponseFormat}}\n  const responseFormat = undefined;\n  {{/hasResponseFormat}}\n\n  let completion = await client.text({\n    messages,\n    tools,\n    responseFormat,\n  });\n\n  const endTime = performance.now();\n  statelogClient.promptCompletion({\n    messages,\n    completion,\n    model: client.getModel(),\n    timeTaken: endTime - startTime,\n  });\n\n  if (!completion.success) {\n    throw new Error(\n      `Error getting response from ${model}: ${completion.error}`\n    );\n  }\n\n  let responseMessage = completion.value;\n\n  // Handle function calls\n  while (responseMessage.toolCalls.length > 0) {\n    // Add assistant's response with tool calls to message history\n    messages.push(assistantMessage(responseMessage.output));\n    let toolCallStartTime, toolCallEndTime;\n\n    // Process each tool call\n    for (const toolCall of responseMessage.toolCalls) {\n      {{{functionCalls:string}}}\n    }\n\n    const nextStartTime = performance.now();\n    let completion = await client.text({\n      messages,\n      tools,\n      responseFormat,\n    });\n\n    const nextEndTime = performance.now();\n\n    statelogClient.promptCompletion({\n      messages,\n      completion,\n      model: client.getModel(),\n      timeTaken: nextEndTime - nextStartTime,\n    });\n\n    if (!completion.success) {\n      throw new Error(\n        `Error getting response from ${model}: ${completion.error}`\n      );\n    }\n    responseMessage = completion.value;\n  }\n\n  // Add final assistant response to history\n  messages.push(assistantMessage(responseMessage.output));\n\n  {{#hasResponseFormat}}\n  try {\n  const result = JSON.parse(responseMessage.output || \"\");\n  return result.response;\n  } catch (e) {\n    return responseMessage.output;\n    // console.error(\"Error parsing response for variable '{{{variableName:string}}}':\", e);\n    // console.error(\"Full completion response:\", JSON.stringify(completion, null, 2));\n    // throw e;\n  }\n  {{/hasResponseFormat}}\n\n  {{^hasResponseFormat}}\n  return responseMessage.output;\n  {{/hasResponseFormat}}\n}\n";
+export declare const template = "\nasync function _{{{variableName:string}}}({{{argsStr:string}}}): Promise<{{{typeString:string}}}> {\n  const __prompt = {{{promptCode:string}}};\n  const startTime = performance.now();\n  const __messages: Message[] = [userMessage(__prompt)];\n  const __tools = {{{tools}}};\n\n  {{#hasResponseFormat}}\n  // Need to make sure this is always an object\n  const __responseFormat = z.object({\n     response: {{{zodSchema:string}}}\n  });\n  {{/hasResponseFormat}}\n  {{^hasResponseFormat}}\n  const __responseFormat = undefined;\n  {{/hasResponseFormat}}\n\n  let __completion = await __client.text({\n    messages: __messages,\n    tools: __tools,\n    responseFormat: __responseFormat,\n  });\n\n  const endTime = performance.now();\n  statelogClient.promptCompletion({\n    messages: __messages,\n    completion: __completion,\n    model: __client.getModel(),\n    timeTaken: endTime - startTime,\n  });\n\n  if (!__completion.success) {\n    throw new Error(\n      `Error getting response from ${__model}: ${__completion.error}`\n    );\n  }\n\n  let responseMessage = __completion.value;\n\n  // Handle function calls\n  while (responseMessage.toolCalls.length > 0) {\n    // Add assistant's response with tool calls to message history\n    __messages.push(assistantMessage(responseMessage.output));\n    let toolCallStartTime, toolCallEndTime;\n\n    // Process each tool call\n    for (const toolCall of responseMessage.toolCalls) {\n      {{{functionCalls:string}}}\n    }\n  \n    const nextStartTime = performance.now();\n    let __completion = await __client.text({\n      messages: __messages,\n      tools: __tools,\n      responseFormat: __responseFormat,\n    });\n\n    const nextEndTime = performance.now();\n\n    statelogClient.promptCompletion({\n      messages: __messages,\n      completion: __completion,\n      model: __client.getModel(),\n      timeTaken: nextEndTime - nextStartTime,\n    });\n\n    if (!__completion.success) {\n      throw new Error(\n        `Error getting response from ${__model}: ${__completion.error}`\n      );\n    }\n    responseMessage = __completion.value;\n  }\n\n  // Add final assistant response to history\n  __messages.push(assistantMessage(responseMessage.output));\n  {{#hasResponseFormat}}\n  try {\n  const result = JSON.parse(responseMessage.output || \"\");\n  return result.response;\n  } catch (e) {\n    return responseMessage.output;\n    // console.error(\"Error parsing response for variable '{{{variableName:string}}}':\", e);\n    // console.error(\"Full completion response:\", JSON.stringify(__completion, null, 2));\n    // throw e;\n  }\n  {{/hasResponseFormat}}\n\n  {{^hasResponseFormat}}\n  return responseMessage.output;\n  {{/hasResponseFormat}}\n}\n";
 export type TemplateType = {
     variableName: string;
     argsStr: string;

package/dist/lib/templates/backends/typescriptGenerator/promptFunction.js

@@ -4,81 +4,80 @@
 import { apply } from "typestache";
 export const template = `
 async function _{{{variableName:string}}}({{{argsStr:string}}}): Promise<{{{typeString:string}}}> {
-  const prompt = {{{promptCode:string}}};
+  const __prompt = {{{promptCode:string}}};
   const startTime = performance.now();
-  const messages: Message[] = [userMessage(prompt)];
-  const tools = {{{tools}}};
+  const __messages: Message[] = [userMessage(__prompt)];
+  const __tools = {{{tools}}};
 
   {{#hasResponseFormat}}
   // Need to make sure this is always an object
-  const responseFormat = z.object({
+  const __responseFormat = z.object({
      response: {{{zodSchema:string}}}
   });
   {{/hasResponseFormat}}
   {{^hasResponseFormat}}
-  const responseFormat = undefined;
+  const __responseFormat = undefined;
   {{/hasResponseFormat}}
 
-  let completion = await client.text({
-    messages,
-    tools,
-    responseFormat,
+  let __completion = await __client.text({
+    messages: __messages,
+    tools: __tools,
+    responseFormat: __responseFormat,
   });
 
   const endTime = performance.now();
   statelogClient.promptCompletion({
-    messages,
-    completion,
-    model: client.getModel(),
+    messages: __messages,
+    completion: __completion,
+    model: __client.getModel(),
     timeTaken: endTime - startTime,
   });
 
-  if (!completion.success) {
+  if (!__completion.success) {
     throw new Error(
-      \`Error getting response from $\{model\}: $\{completion.error\}\`
+      \`Error getting response from $\{__model\}: $\{__completion.error\}\`
     );
   }
 
-  let responseMessage = completion.value;
+  let responseMessage = __completion.value;
 
   // Handle function calls
   while (responseMessage.toolCalls.length > 0) {
     // Add assistant's response with tool calls to message history
-    messages.push(assistantMessage(responseMessage.output));
+    __messages.push(assistantMessage(responseMessage.output));
     let toolCallStartTime, toolCallEndTime;
 
     // Process each tool call
     for (const toolCall of responseMessage.toolCalls) {
       {{{functionCalls:string}}}
     }
-
+  
     const nextStartTime = performance.now();
-    let completion = await client.text({
-      messages,
-      tools,
-      responseFormat,
+    let __completion = await __client.text({
+      messages: __messages,
+      tools: __tools,
+      responseFormat: __responseFormat,
     });
 
     const nextEndTime = performance.now();
 
     statelogClient.promptCompletion({
-      messages,
-      completion,
-      model: client.getModel(),
+      messages: __messages,
+      completion: __completion,
+      model: __client.getModel(),
       timeTaken: nextEndTime - nextStartTime,
     });
 
-    if (!completion.success) {
+    if (!__completion.success) {
       throw new Error(
-        \`Error getting response from $\{model\}: $\{completion.error\}\`
+        \`Error getting response from $\{__model\}: $\{__completion.error\}\`
       );
     }
-    responseMessage = completion.value;
+    responseMessage = __completion.value;
   }
 
   // Add final assistant response to history
-  messages.push(assistantMessage(responseMessage.output));
-
+  __messages.push(assistantMessage(responseMessage.output));
   {{#hasResponseFormat}}
   try {
   const result = JSON.parse(responseMessage.output || "");
@@ -86,7 +85,7 @@ async function _{{{variableName:string}}}({{{argsStr:string}}}): Promise<{{{type
   } catch (e) {
     return responseMessage.output;
     // console.error("Error parsing response for variable '{{{variableName:string}}}':", e);
-    // console.error("Full completion response:", JSON.stringify(completion, null, 2));
+    // console.error("Full completion response:", JSON.stringify(__completion, null, 2));
     // throw e;
   }
   {{/hasResponseFormat}}

package/dist/lib/templates/backends/typescriptGenerator/toolCall.d.ts

@@ -1,4 +1,4 @@
-export declare const template = "if (toolCall.type === \"function\" &&\n  toolCall.function.name === \"{{{name:string}}}\"\n) {\n  const args = JSON.parse(toolCall.function.arguments);\n\n  toolCallStartTime = performance.now();\n  const result = await {{{name}}}(args);\n  toolCallEndTime = performance.now();\n\n  console.log(\"Tool '{{{name:string}}}' called with arguments:\", args);\n  console.log(\"Tool '{{{name:string}}}' returned result:\", result);\n\nstatelogClient.toolCall({\n    toolName: \"{{{name:string}}}\",\n    args,\n    output: result,\n    model,\n    timeTaken: toolCallEndTime - toolCallStartTime,\n  });\n\n  // Add function result to messages\n  messages.push({\n    role: \"tool\",\n    tool_call_id: toolCall.id,\n    content: JSON.stringify(result),\n  });\n}";
+export declare const template = "if (toolCall.type === \"function\" &&\n  toolCall.function.name === \"{{{name:string}}}\"\n) {\n  const args = JSON.parse(toolCall.function.arguments);\n\n  toolCallStartTime = performance.now();\n  const result = await {{{name}}}(args);\n  toolCallEndTime = performance.now();\n\n  console.log(\"Tool '{{{name:string}}}' called with arguments:\", args);\n  console.log(\"Tool '{{{name:string}}}' returned result:\", result);\n\nstatelogClient.toolCall({\n    toolName: \"{{{name:string}}}\",\n    args,\n    output: result,\n    model: __client.getModel(),\n    timeTaken: toolCallEndTime - toolCallStartTime,\n  });\n\n  // Add function result to messages\n  __messages.push({\n    role: \"tool\",\n    tool_call_id: toolCall.id,\n    content: JSON.stringify(result),\n  });\n}";
 export type TemplateType = {
     name: string;
 };

package/dist/lib/templates/backends/typescriptGenerator/toolCall.js

@@ -18,12 +18,12 @@ statelogClient.toolCall({
     toolName: "{{{name:string}}}",
     args,
     output: result,
-    model,
+    model: __client.getModel(),
     timeTaken: toolCallEndTime - toolCallStartTime,
   });
 
   // Add function result to messages
-  messages.push({
+  __messages.push({
     role: "tool",
     tool_call_id: toolCall.id,
     content: JSON.stringify(result),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agency-lang",
-  "version": "0.0.11",
+  "version": "0.0.13",
   "description": "The Agency language",
   "main": "lib/index.js",
   "scripts": {
@@ -41,18 +41,18 @@
   "homepage": "https://github.com/egonSchiele/agency-lang#readme",
   "dependencies": {
     "egonlog": "^0.0.2",
-    "nanoid": "^5.1.6",
-    "piemachine": "^0.0.2",
-    "smoltalk": "^0.0.11",
-    "statelog-client": "^0.0.31",
+    "tarsec": "^0.1.1",
+    "typestache": "^0.4.4",
     "zod": "^4.3.5"
   },
   "devDependencies": {
     "@types/node": "^25.0.3",
-    "tarsec": "^0.1.1",
+    "nanoid": "^5.1.6",
+    "piemachine": "^0.0.2",
+    "smoltalk": "^0.0.11",
+    "statelog-client": "^0.0.31",
     "tsc-alias": "^1.8.16",
     "typescript": "^5.9.3",
-    "typestache": "^0.4.4",
     "vitest": "^4.0.16"
   }
 }