@output.ai/cli 0.7.3 → 0.7.4

package/dist/services/workflow_planner.js

@@ -8,7 +8,7 @@ export async function generatePlanName(description, date = new Date()) {
     const month = String(date.getMonth() + 1).padStart(2, '0');
     const day = String(date.getDate()).padStart(2, '0');
     const datePrefix = `${year}_${month}_${day}`;
-    const planNameSlug = await generateText({
+    const { text: planNameSlug } = await generateText({
         prompt: 'generate_plan_name@v1',
         variables: { description }
     });

package/dist/services/workflow_planner.spec.js

@@ -6,13 +6,19 @@ import fs from 'node:fs/promises';
 vi.mock('./coding_agents.js');
 vi.mock('@output.ai/llm');
 vi.mock('node:fs/promises');
+const mockGenerateTextResult = (text) => ({
+    text,
+    result: text,
+    usage: { inputTokens: 10, outputTokens: 5, totalTokens: 15 },
+    finishReason: 'stop'
+});
 describe('workflow-planner service', () => {
     beforeEach(() => {
         vi.clearAllMocks();
     });
     describe('generatePlanName', () => {
         it('should generate plan name with date prefix using LLM', async () => {
-            vi.mocked(generateText).mockResolvedValue('customer_order_processing');
+            vi.mocked(generateText).mockResolvedValue(mockGenerateTextResult('customer_order_processing'));
             const testDate = new Date(2025, 9, 6);
             const planName = await generatePlanName('A workflow that processes customer orders', testDate);
             expect(planName).toMatch(/^2025_10_06_/);
@@ -23,7 +29,7 @@ describe('workflow-planner service', () => {
             });
         });
         it('should clean and validate LLM response', async () => {
-            vi.mocked(generateText).mockResolvedValue('  User-Auth & Security!@#  ');
+            vi.mocked(generateText).mockResolvedValue(mockGenerateTextResult('  User-Auth & Security!@#  '));
             const testDate = new Date(2025, 9, 6);
             const planName = await generatePlanName('User authentication workflow', testDate);
             expect(planName).toBe('2025_10_06_user_auth_security');
@@ -34,14 +40,14 @@ describe('workflow-planner service', () => {
             await expect(generatePlanName('Test workflow')).rejects.toThrow('API rate limit exceeded');
         });
         it('should limit plan name length to 50 characters', async () => {
-            vi.mocked(generateText).mockResolvedValue('this_is_an_extremely_long_plan_name_that_exceeds_the_maximum_allowed_length_for_file_names');
+            vi.mocked(generateText).mockResolvedValue(mockGenerateTextResult('this_is_an_extremely_long_plan_name_that_exceeds_the_maximum_allowed_length_for_file_names'));
             const testDate = new Date(2025, 9, 6);
             const planName = await generatePlanName('Long workflow description', testDate);
             const namePart = planName.replace(/^2025_10_06_/, '');
             expect(namePart.length).toBeLessThanOrEqual(50);
         });
         it('should handle multiple underscores correctly', async () => {
-            vi.mocked(generateText).mockResolvedValue('user___auth___workflow');
+            vi.mocked(generateText).mockResolvedValue(mockGenerateTextResult('user___auth___workflow'));
             const testDate = new Date(2025, 9, 6);
             const planName = await generatePlanName('Test', testDate);
             expect(planName).toBe('2025_10_06_user_auth_workflow');

package/dist/templates/project/src/workflows/example_question/steps.ts.template

@@ -7,10 +7,10 @@ export const answerQuestion = step( {
   inputSchema: z.string(),
   outputSchema: z.string(),
   fn: async question => {
-    const response = await generateText( {
+    const { result } = await generateText( {
       prompt: 'answer_question@v1',
       variables: { question }
     } );
-    return response;
+    return result;
   }
 } );

package/dist/templates/workflow/steps.ts.template

@@ -10,11 +10,11 @@ export const exampleLLMStep = step( {
   } ),
   outputSchema: z.string(),
   fn: async ( { userInput } ) => {
-    const response = await generateText( {
+    const { result } = await generateText( {
       prompt: 'prompt@v1',
       variables: { userInput }
     } );
-    return response;
+    return result;
   }
 } );
 

package/dist/test_helpers/mocks.js

@@ -16,7 +16,12 @@ export const mockClaudeAgentSDK = {
  * Mock for @output.ai/llm
  */
 export const mockLLM = {
-    generateText: vi.fn().mockResolvedValue('workflow_plan_name')
+    generateText: vi.fn().mockResolvedValue({
+        text: 'workflow_plan_name',
+        sources: [],
+        usage: { inputTokens: 10, outputTokens: 5, totalTokens: 15 },
+        finishReason: 'stop'
+    })
 };
 /**
  * Mock for child_process spawn (for agents init)
@@ -61,7 +66,12 @@ export function resetAllMocks() {
     mockFS.writeFile.mockResolvedValue(undefined);
     mockFS.readFile.mockResolvedValue('template content');
     mockFS.stat.mockRejectedValue({ code: 'ENOENT' });
-    mockLLM.generateText.mockResolvedValue('workflow_plan_name');
+    mockLLM.generateText.mockResolvedValue({
+        text: 'workflow_plan_name',
+        sources: [],
+        usage: { inputTokens: 10, outputTokens: 5, totalTokens: 15 },
+        finishReason: 'stop'
+    });
     mockInquirer.input.mockResolvedValue('Mock workflow description');
     mockInquirer.confirm.mockResolvedValue(true);
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@output.ai/cli",
-  "version": "0.7.3",
+  "version": "0.7.4",
   "description": "CLI for Output.ai workflow generation",
   "type": "module",
   "main": "dist/index.js",