agency-lang 0.0.7 → 0.0.8

package/dist/lib/backends/agencyGenerator.d.ts

@@ -1,4 +1,4 @@
-import { AgencyComment, AgencyProgram, Assignment, Literal, PromptLiteral, TypeAlias, TypeHint } from "../types.js";
+import { AgencyComment, AgencyProgram, Assignment, Literal, PromptLiteral, TypeAlias, TypeHint, VariableType } from "../types.js";
 import { AccessExpression, DotFunctionCall, DotProperty, IndexAccess } from "../types/access.js";
 import { AgencyArray, AgencyObject } from "../types/dataStructures.js";
 import { FunctionCall, FunctionDefinition } from "../types/function.js";
@@ -20,6 +20,7 @@ export declare class AgencyGenerator extends BaseGenerator {
     protected generateImports(): string;
     protected preprocess(): string;
     protected postprocess(): string;
+    protected aliasedTypeToString(aliasedType: VariableType): string;
     protected processTypeAlias(node: TypeAlias): string;
     protected processTypeHint(node: TypeHint): string;
     protected processAssignment(node: Assignment): string;

package/dist/lib/backends/agencyGenerator.js

@@ -6,8 +6,8 @@ export class AgencyGenerator extends BaseGenerator {
     constructor() {
         super();
     }
-    indent() {
-        return " ".repeat(this.indentLevel * this.indentSize);
+    indent(level = this.indentLevel) {
+        return " ".repeat(level * this.indentSize);
     }
     increaseIndent() {
         this.indentLevel++;
@@ -28,10 +28,26 @@ export class AgencyGenerator extends BaseGenerator {
     postprocess() {
         return "";
     }
+    aliasedTypeToString(aliasedType) {
+        if (aliasedType.type === "objectType") {
+            const props = aliasedType.properties
+                .map((prop) => {
+                let str = `${this.indent(this.indentLevel + 1)}`;
+                str += `${prop.key}: ${this.aliasedTypeToString(prop.value)}`;
+                if (prop.description) {
+                    str += ` # ${prop.description}`;
+                }
+                return str;
+            })
+                .join(";\n");
+            return `{\n${props}\n}`;
+        }
+        return variableTypeToString(aliasedType, this.typeAliases);
+    }
     // Type system methods
     processTypeAlias(node) {
         this.typeAliases[node.aliasName] = node.aliasedType;
-        const aliasedTypeStr = variableTypeToString(node.aliasedType, this.typeAliases);
+        const aliasedTypeStr = this.aliasedTypeToString(node.aliasedType);
         return this.indentStr(`type ${node.aliasName} = ${aliasedTypeStr}\n`);
     }
     processTypeHint(node) {

package/dist/lib/backends/typescriptGenerator.js

@@ -19,7 +19,9 @@ export class TypeScriptGenerator extends BaseGenerator {
     processTypeAlias(node) {
         this.typeAliases[node.aliasName] = node.aliasedType;
         const typeAliasStr = this.typeAliasToString(node);
-        this.generatedTypeAliases.push(typeAliasStr);
+        if (!this.generatedTypeAliases.includes(typeAliasStr)) {
+            this.generatedTypeAliases.push(typeAliasStr);
+        }
         return "";
     }
     typeAliasToString(node) {

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

@@ -1,4 +1,4 @@
-export declare const template = "import OpenAI from \"openai\";\nimport { zodResponseFormat } from \"openai/helpers/zod\";\nimport { z } from \"zod\";\nimport * as readline from \"readline\";\nimport fs from \"fs\";\nimport { Graph, goToNode } from \"simplemachine\";\nimport { StatelogClient } from \"statelog-client\";\nimport { nanoid } from \"nanoid\";\n\nconst statelogHost = \"http://localhost:1065\";\nconst traceId = nanoid();\nconst statelogClient = new StatelogClient({host: statelogHost, tid: traceId});\nconst model = \"gpt-4.1-nano-2025-04-14\";\n\nconst openai = new OpenAI({\n  apiKey: process.env.OPENAI_API_KEY,\n});\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  statelogHost,\n  traceId\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 Graph<State, Node>(nodes, graphConfig);";
+export declare const template = "import OpenAI from \"openai\";\nimport { zodResponseFormat } from \"openai/helpers/zod\";\nimport { z } from \"zod\";\nimport * as readline from \"readline\";\nimport fs from \"fs\";\nimport { Graph, goToNode } from \"simplemachine\";\nimport { StatelogClient } from \"statelog-client\";\nimport { nanoid } from \"nanoid\";\nimport { assistantMessage, getClient, Message, userMessage } from \"smoltalk\";\n\nconst statelogHost = \"http://localhost:1065\";\nconst traceId = nanoid();\nconst statelogClient = new StatelogClient({host: statelogHost, tid: traceId});\nconst model = \"gpt-4o-mini\";\n\nconst client = getClient({\n  apiKey: process.env.OPENAI_API_KEY || \"\",\n  model,\n});\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  statelogHost,\n  traceId\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 Graph<State, Node>(nodes, graphConfig);";
 export type TemplateType = {
     nodes: string;
 };

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

@@ -10,14 +10,16 @@ import fs from "fs";
 import { Graph, goToNode } from "simplemachine";
 import { StatelogClient } from "statelog-client";
 import { nanoid } from "nanoid";
+import { assistantMessage, getClient, Message, userMessage } from "smoltalk";
 
 const statelogHost = "http://localhost:1065";
 const traceId = nanoid();
 const statelogClient = new StatelogClient({host: statelogHost, tid: traceId});
-const model = "gpt-4.1-nano-2025-04-14";
+const model = "gpt-4o-mini";
 
-const openai = new OpenAI({
-  apiKey: process.env.OPENAI_API_KEY,
+const client = getClient({
+  apiKey: process.env.OPENAI_API_KEY || "",
+  model,
 });
 
 type State = {

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:any[] = [{ role: \"user\", content: prompt }];\n  const tools = {{{tools}}};\n\n  let completion = await openai.chat.completions.create({\n    model,\n    messages,\n    tools,\n    response_format: zodResponseFormat(z.object({\n      value: {{{zodSchema:string}}}\n    }), \"{{{variableName:string}}}_response\"),\n  });\n  const endTime = performance.now();\n  statelogClient.promptCompletion({\n    messages,\n    completion,\n    model,\n    timeTaken: endTime - startTime,\n  });\n\n  let responseMessage = completion.choices[0].message;\n  // Handle function calls\n  while (responseMessage.tool_calls && responseMessage.tool_calls.length > 0) {\n    // Add assistant's response with tool calls to message history\n    messages.push(responseMessage);\n    let toolCallStartTime, toolCallEndTime;\n\n    // Process each tool call\n    for (const toolCall of responseMessage.tool_calls) {\n      {{{functionCalls:string}}}\n    }\n\n    const nextStartTime = performance.now();\n    // Get the next response from the model\n    completion = await openai.chat.completions.create({\n      model,\n      messages: messages,\n      tools: tools,\n    });\n    const nextEndTime = performance.now();\n\n    statelogClient.promptCompletion({\n      messages,\n      completion,\n      model,\n      timeTaken: nextEndTime - nextStartTime,\n    });\n\n    responseMessage = completion.choices[0].message;\n  }\n\n  // Add final assistant response to history\n  messages.push(responseMessage);\n\n  try {\n  const result = JSON.parse(completion.choices[0].message.content || \"\");\n  return result.value;\n  } catch (e) {\n    return completion.choices[0].message.content;\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}\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  const responseFormat = {{{zodSchema:string}}};\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,\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,\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  try {\n  const result = JSON.parse(responseMessage.output || \"\");\n  return result.value;\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}\n";
 export type TemplateType = {
     variableName: string;
     argsStr: string;

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

@@ -6,17 +6,17 @@ export const template = `
 async function _{{{variableName:string}}}({{{argsStr:string}}}): Promise<{{{typeString:string}}}> {
   const prompt = {{{promptCode:string}}};
   const startTime = performance.now();
-  const messages:any[] = [{ role: "user", content: prompt }];
+  const messages: Message[] = [userMessage(prompt)];
   const tools = {{{tools}}};
 
-  let completion = await openai.chat.completions.create({
-    model,
+  const responseFormat = {{{zodSchema:string}}};
+
+  let completion = await client.text({
     messages,
     tools,
-    response_format: zodResponseFormat(z.object({
-      value: {{{zodSchema:string}}}
-    }), "{{{variableName:string}}}_response"),
+    responseFormat,
   });
+
   const endTime = performance.now();
   statelogClient.promptCompletion({
     messages,
@@ -25,25 +25,32 @@ async function _{{{variableName:string}}}({{{argsStr:string}}}): Promise<{{{type
     timeTaken: endTime - startTime,
   });
 
-  let responseMessage = completion.choices[0].message;
+  if (!completion.success) {
+    throw new Error(
+      \`Error getting response from $\{model\}: $\{completion.error\}\`
+    );
+  }
+
+  let responseMessage = completion.value;
+
   // Handle function calls
-  while (responseMessage.tool_calls && responseMessage.tool_calls.length > 0) {
+  while (responseMessage.toolCalls.length > 0) {
     // Add assistant's response with tool calls to message history
-    messages.push(responseMessage);
+    messages.push(assistantMessage(responseMessage.output));
     let toolCallStartTime, toolCallEndTime;
 
     // Process each tool call
-    for (const toolCall of responseMessage.tool_calls) {
+    for (const toolCall of responseMessage.toolCalls) {
       {{{functionCalls:string}}}
     }
 
     const nextStartTime = performance.now();
-    // Get the next response from the model
-    completion = await openai.chat.completions.create({
-      model,
-      messages: messages,
-      tools: tools,
+    let completion = await client.text({
+      messages,
+      tools,
+      responseFormat,
     });
+
     const nextEndTime = performance.now();
 
     statelogClient.promptCompletion({
@@ -53,17 +60,22 @@ async function _{{{variableName:string}}}({{{argsStr:string}}}): Promise<{{{type
       timeTaken: nextEndTime - nextStartTime,
     });
 
-    responseMessage = completion.choices[0].message;
+    if (!completion.success) {
+      throw new Error(
+        \`Error getting response from $\{model\}: $\{completion.error\}\`
+      );
+    }
+    responseMessage = completion.value;
   }
 
   // Add final assistant response to history
-  messages.push(responseMessage);
+  messages.push(assistantMessage(responseMessage.output));
 
   try {
-  const result = JSON.parse(completion.choices[0].message.content || "");
+  const result = JSON.parse(responseMessage.output || "");
   return result.value;
   } catch (e) {
-    return completion.choices[0].message.content;
+    return responseMessage.output;
     // console.error("Error parsing response for variable '{{{variableName:string}}}':", e);
     // console.error("Full completion response:", JSON.stringify(completion, null, 2));
     // throw e;

package/dist/scripts/agency.js

@@ -155,6 +155,7 @@ async function main() {
             }
             format(fmtContents, verbose);
             break;
+        case "ast":
         case "parse":
             let contents;
             if (filteredArgs.length < 2) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agency-lang",
-  "version": "0.0.7",
+  "version": "0.0.8",
   "description": "The Agency language",
   "main": "lib/index.js",
   "scripts": {
@@ -43,6 +43,7 @@
     "nanoid": "^5.1.6",
     "openai": "^6.15.0",
     "simplemachine": "github:egonSchiele/simplemachine",
+    "smoltalk": "^0.0.4",
     "statelog-client": "^0.0.28",
     "tarsec": "^0.1.1",
     "typestache": "^0.4.4",