@polka-codes/core 0.9.28 → 0.9.29

package/dist/_tsup-dts-rollup.d.ts

@@ -1985,6 +1985,7 @@ declare type WorkflowContext = {
     parameters: Record<string, any>;
     verbose?: number;
     agentCallback?: TaskEventCallback;
+    logger?: Console;
 };
 export { WorkflowContext }
 export { WorkflowContext as WorkflowContext_alias_1 }

package/dist/index.js

@@ -3760,8 +3760,9 @@ var makeAgentStepSpecHandler = (getModelFn) => ({
     return {
       ...step2,
       async run(input, context, resumedState) {
+        const logger = context.logger ?? console;
         if (context.verbose && context.verbose >= 1) {
-          console.log(`[agent-step] Running agent step '${step2.id}' with input:`, input);
+          logger.log(`[agent-step] Running agent step '${step2.id}' with input:`, input);
         }
         try {
           const model = await getModelFn(step2, context);
@@ -3783,7 +3784,7 @@ var makeAgentStepSpecHandler = (getModelFn) => ({
                 throw new Error(`Unknown agent: ${agentName}`);
               }
               if (context.verbose && context.verbose >= 1) {
-                console.log(`[agent-step] Using agent: ${agentName}`);
+                logger.log(`[agent-step] Using agent: ${agentName}`);
               }
               const agentOptions = {
                 ai: model,
@@ -3803,7 +3804,7 @@ var makeAgentStepSpecHandler = (getModelFn) => ({
                 throw new Error("No system prompt specified for the agent step.");
               }
               if (context.verbose && context.verbose >= 1) {
-                console.log(`[agent-step] Using generic WorkflowAgent`);
+                logger.log(`[agent-step] Using generic WorkflowAgent`);
               }
               const systemPrompt = resolveTemplatedString(step2.systemPrompt, input);
               return new WorkflowAgent("agent", model, {
@@ -3841,11 +3842,11 @@ var makeAgentStepSpecHandler = (getModelFn) => ({
             }
           }
           if (context.verbose && context.verbose >= 1) {
-            console.log(`[agent-step] Starting agent with content:`, JSON.stringify(combinedContentParts, null, 2));
+            logger.log(`[agent-step] Starting agent with content:`, JSON.stringify(combinedContentParts, null, 2));
           }
           const exitReason = await agent.start(combinedContentParts);
           if (context.verbose && context.verbose >= 1) {
-            console.log(`[agent-step] Agent exited with reason:`, exitReason);
+            logger.log(`[agent-step] Agent exited with reason:`, exitReason);
           }
           const handleExitReason = (reason) => {
             switch (reason.type) {
@@ -4012,15 +4013,16 @@ var command = (id, ...command2) => ({
 
 // src/workflow/runStep.ts
 var runStep = async (step2, input, context, resumedState, allOutputs) => {
+  const logger = context.logger ?? console;
   if (context.verbose && context.verbose >= 1) {
-    console.log(`[workflow] running step: ${step2.id}`);
-    console.log(`[workflow] input: ${JSON.stringify(input, null, 2)}`);
+    logger.log(`[workflow] running step: ${step2.id}`);
+    logger.log(`[workflow] input: ${JSON.stringify(input, null, 2)}`);
   }
   try {
     const validatedInput = step2.inputSchema?.parse(input) ?? input;
     const result = await step2.run({ ...validatedInput, $: allOutputs }, context, resumedState);
     if (context.verbose && context.verbose >= 1) {
-      console.log(`[workflow] step result: ${step2.id}`, JSON.stringify(result, null, 2));
+      logger.log(`[workflow] step result: ${step2.id}`, JSON.stringify(result, null, 2));
     }
     if (result.type === "success") {
       const validatedOutput = step2.outputSchema?.parse(result.output) ?? result.output;
@@ -4032,7 +4034,7 @@ var runStep = async (step2, input, context, resumedState, allOutputs) => {
     return result;
   } catch (error) {
     if (context.verbose && context.verbose >= 1) {
-      console.error(`[workflow] step error: ${step2.id}`, error);
+      logger.error(`[workflow] step error: ${step2.id}`, error);
     }
     return { type: "error", error };
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polka-codes/core",
-  "version": "0.9.28",
+  "version": "0.9.29",
   "license": "AGPL-3.0",
   "author": "github@polka.codes",
   "type": "module",