@locusai/sdk 0.4.4 → 0.4.5

package/src/ai/claude-runner.ts

@@ -1,5 +1,5 @@
 import { spawn } from "node:child_process";
-import { DEFAULT_MODEL } from "../core/config";
+import { DEFAULT_MODEL } from "../core/config.js";
 
 export class ClaudeRunner {
   constructor(
@@ -7,7 +7,32 @@ export class ClaudeRunner {
     private model: string = DEFAULT_MODEL
   ) {}
 
-  run(prompt: string, _isPlanning = false): Promise<string> {
+  async run(prompt: string, _isPlanning = false): Promise<string> {
+    const maxRetries = 3;
+    let lastError: Error | null = null;
+
+    for (let attempt = 1; attempt <= maxRetries; attempt++) {
+      try {
+        return await this.executeRun(prompt);
+      } catch (error) {
+        const err = error as Error;
+        lastError = err;
+        const isLastAttempt = attempt === maxRetries;
+
+        if (!isLastAttempt) {
+          const delay = Math.pow(2, attempt) * 1000; // 2s, 4s, 8s
+          console.warn(
+            `Claude CLI attempt ${attempt} failed: ${err.message}. Retrying in ${delay}ms...`
+          );
+          await new Promise((resolve) => setTimeout(resolve, delay));
+        }
+      }
+    }
+
+    throw lastError || new Error("Claude CLI failed after multiple attempts");
+  }
+
+  private executeRun(prompt: string): Promise<string> {
     return new Promise((resolve, reject) => {
       const args = [
         "--dangerously-skip-permissions",
@@ -28,11 +53,12 @@ export class ClaudeRunner {
 
       claude.stdout.on("data", (data) => {
         output += data.toString();
-        process.stdout.write(data.toString());
+        // Only write to stdout if we're not retrying or if logic dictates
+        // process.stdout.write(data.toString());
       });
       claude.stderr.on("data", (data) => {
         errorOutput += data.toString();
-        process.stderr.write(data.toString());
+        // process.stderr.write(data.toString());
       });
 
       claude.on("error", (err) =>

package/src/ai/index.ts

@@ -1,2 +1,2 @@
-export { AnthropicClient } from "./anthropic-client";
-export { ClaudeRunner } from "./claude-runner";
+export { AnthropicClient } from "./anthropic-client.js";
+export { ClaudeRunner } from "./claude-runner.js";

package/src/core/index.ts

@@ -1,3 +1,3 @@
-export { DEFAULT_MODEL, getLocusPath, LOCUS_CONFIG } from "./config";
-export { type CodebaseIndex, CodebaseIndexer } from "./indexer";
-export { PromptBuilder } from "./prompt-builder";
+export { DEFAULT_MODEL, getLocusPath, LOCUS_CONFIG } from "./config.js";
+export { type CodebaseIndex, CodebaseIndexer } from "./indexer.js";
+export { PromptBuilder } from "./prompt-builder.js";

package/src/core/prompt-builder.ts

@@ -1,7 +1,6 @@
 import { existsSync, readFileSync } from "node:fs";
-import { Task } from "@locusai/shared";
-import { getLocusPath } from "./config";
-import { CodebaseIndex } from "./indexer";
+import { AssigneeRole, Task } from "@locusai/shared";
+import { getLocusPath } from "./config.js";
 
 export class PromptBuilder {
   constructor(private projectPath: string) {}
@@ -9,8 +8,9 @@ export class PromptBuilder {
   async build(task: Task): Promise<string> {
     let prompt = `# Task: ${task.title}\n\n`;
 
-    if (task.assigneeRole) {
-      prompt += `## Role\nYou are acting as a ${task.assigneeRole} engineer.\n\n`;
+    const roleText = this.roleToText(task.assigneeRole);
+    if (roleText) {
+      prompt += `## Role\nYou are acting as a ${roleText}.\n\n`;
     }
 
     prompt += `## Description\n${task.description || "No description provided."}\n\n`;
@@ -29,13 +29,7 @@ export class PromptBuilder {
     // 2. Add Codebase Index context
     const indexPath = getLocusPath(this.projectPath, "indexFile");
     if (existsSync(indexPath)) {
-      try {
-        const indexContent = readFileSync(indexPath, "utf-8");
-        const index = JSON.parse(indexContent) as CodebaseIndex;
-        prompt += this.formatIndex(index, task);
-      } catch (err) {
-        console.warn(`Warning: Could not read codebase index: ${err}`);
-      }
+      prompt += `## Codebase Overview\nThere is an index file in the .locus/codebase-index.json and if you need you can check it.\n\n`;
     }
 
     // 3. Add Documents
@@ -57,9 +51,10 @@ export class PromptBuilder {
 
     // 5. Add Comments & Feedback
     if (task.comments && task.comments.length > 0) {
+      const comments = task.comments.slice(0, 5);
       prompt += `## Task History & Feedback\n`;
       prompt += `Review the following comments for context or rejection feedback:\n\n`;
-      for (const comment of task.comments) {
+      for (const comment of comments) {
         const date = new Date(comment.createdAt).toLocaleString();
         prompt += `### ${comment.author} (${date})\n${comment.text}\n\n`;
       }
@@ -73,28 +68,24 @@ export class PromptBuilder {
     return prompt;
   }
 
-  private formatIndex(index: CodebaseIndex, task: Task): string {
-    let section = `## Codebase Overview\nThis codebase has been indexed to help you navigate.\n\n`;
-
-    // Structural directories
-    const structuralDirs = Object.entries(index.responsibilities || {})
-      .filter(([path]) => !path.includes(".") || path.split("/").length <= 2)
-      .slice(0, 15);
-
-    if (structuralDirs.length > 0) {
-      section += `### Project Structure\n${structuralDirs.map(([p, d]) => `- \`${p}\`: ${d}`).join("\n")}\n\n`;
+  roleToText(role: Task["assigneeRole"]): string | null {
+    if (!role) {
+      return null;
     }
 
-    // Relevant symbols
-    const keywords = `${task.title} ${task.description}`.toLowerCase();
-    const symbols = Object.entries(index.symbols || {})
-      .filter(([symbol]) => keywords.includes(symbol.toLowerCase()))
-      .slice(0, 10);
-
-    if (symbols.length > 0) {
-      section += `### Potentially Relevant Symbols\n${symbols.map(([s, f]) => `- \`${s}\` is defined in: ${Array.isArray(f) ? f.join(", ") : f}`).join("\n")}\n\n`;
+    switch (role) {
+      case AssigneeRole.BACKEND:
+        return "Backend Engineer";
+      case AssigneeRole.FRONTEND:
+        return "Frontend Engineer";
+      case AssigneeRole.PM:
+        return "Product Manager";
+      case AssigneeRole.QA:
+        return "QA Engineer";
+      case AssigneeRole.DESIGN:
+        return "Product Designer";
+      default:
+        return "engineer";
     }
-
-    return section;
   }
 }

package/src/events.ts

@@ -1,6 +1,6 @@
 import { EventEmitter } from "events";
 
-import { RetryOptions } from "./utils/retry";
+import { RetryOptions } from "./utils/retry.js";
 
 export enum LocusEvent {
   TOKEN_EXPIRED = "TOKEN_EXPIRED",

package/src/index-node.ts

@@ -8,16 +8,16 @@
  */
 
 // Node.js-only: Agent system
-export * from "./agent";
+export * from "./agent/index.js";
 // Node.js-only: AI clients
-export * from "./ai";
+export * from "./ai/index.js";
 // Node.js-only: Core utilities (uses fs)
-export * from "./core";
+export * from "./core/index.js";
 // Re-export everything from main index (browser-safe)
-export * from "./index";
+export * from "./index.js";
 
 // Node.js-only: Orchestrator
-export { AgentOrchestrator, type OrchestratorConfig } from "./orchestrator";
+export { AgentOrchestrator, type OrchestratorConfig } from "./orchestrator.js";
 
 // Utilities
-export { c } from "./utils/colors";
+export { c } from "./utils/colors.js";

package/src/index.ts

@@ -1,25 +1,25 @@
 import axios, { AxiosInstance, InternalAxiosRequestConfig } from "axios";
-import { LocusConfig, LocusEmitter, LocusEvent } from "./events";
-import { AuthModule } from "./modules/auth";
-import { CiModule } from "./modules/ci";
-import { DocsModule } from "./modules/docs";
-import { InvitationsModule } from "./modules/invitations";
-import { OrganizationsModule } from "./modules/organizations";
-import { SprintsModule } from "./modules/sprints";
-import { TasksModule } from "./modules/tasks";
-import { WorkspacesModule } from "./modules/workspaces";
-import { RetryOptions } from "./utils/retry";
+import { LocusConfig, LocusEmitter, LocusEvent } from "./events.js";
+import { AuthModule } from "./modules/auth.js";
+import { CiModule } from "./modules/ci.js";
+import { DocsModule } from "./modules/docs.js";
+import { InvitationsModule } from "./modules/invitations.js";
+import { OrganizationsModule } from "./modules/organizations.js";
+import { SprintsModule } from "./modules/sprints.js";
+import { TasksModule } from "./modules/tasks.js";
+import { WorkspacesModule } from "./modules/workspaces.js";
+import { RetryOptions } from "./utils/retry.js";
 
 // Browser-safe exports only
-export * from "./events";
-export * from "./modules/auth";
-export * from "./modules/ci";
-export * from "./modules/docs";
-export * from "./modules/invitations";
-export * from "./modules/organizations";
-export * from "./modules/sprints";
-export * from "./modules/tasks";
-export * from "./modules/workspaces";
+export * from "./events.js";
+export * from "./modules/auth.js";
+export * from "./modules/ci.js";
+export * from "./modules/docs.js";
+export * from "./modules/invitations.js";
+export * from "./modules/organizations.js";
+export * from "./modules/sprints.js";
+export * from "./modules/tasks.js";
+export * from "./modules/workspaces.js";
 
 export class LocusClient {
   private readonly api: AxiosInstance;

package/src/modules/auth.ts

@@ -4,7 +4,7 @@ import {
   User,
   VerifyOtp,
 } from "@locusai/shared";
-import { BaseModule } from "./base";
+import { BaseModule } from "./base.js";
 
 export class AuthModule extends BaseModule {
   async getMe(): Promise<User> {

package/src/modules/base.ts

@@ -1,5 +1,5 @@
 import { AxiosInstance } from "axios";
-import { LocusEmitter } from "../events";
+import { LocusEmitter } from "../events.js";
 
 export abstract class BaseModule {
   constructor(

package/src/modules/ci.ts

@@ -1,5 +1,5 @@
 import { ReportCiResult } from "@locusai/shared";
-import { BaseModule } from "./base";
+import { BaseModule } from "./base.js";
 
 export class CiModule extends BaseModule {
   async report(body: ReportCiResult): Promise<{ success: boolean }> {

package/src/modules/docs.ts

@@ -10,7 +10,7 @@ import {
   UpdateDoc,
   UpdateDocGroup,
 } from "@locusai/shared";
-import { BaseModule } from "./base";
+import { BaseModule } from "./base.js";
 
 export class DocsModule extends BaseModule {
   async create(workspaceId: string, body: CreateDoc): Promise<Doc> {

package/src/modules/invitations.ts

@@ -6,7 +6,7 @@ import {
   InvitationResponse,
   InvitationsResponse,
 } from "@locusai/shared";
-import { BaseModule } from "./base";
+import { BaseModule } from "./base.js";
 
 export class InvitationsModule extends BaseModule {
   async create(orgId: string, body: CreateInvitation): Promise<Invitation> {

package/src/modules/organizations.ts

@@ -7,7 +7,7 @@ import {
   OrganizationResponse,
   OrganizationsResponse,
 } from "@locusai/shared";
-import { BaseModule } from "./base";
+import { BaseModule } from "./base.js";
 
 export interface ApiKey {
   id: string;

package/src/modules/sprints.ts

@@ -5,7 +5,7 @@ import {
   SprintsResponse,
   UpdateSprint,
 } from "@locusai/shared";
-import { BaseModule } from "./base";
+import { BaseModule } from "./base.js";
 
 export class SprintsModule extends BaseModule {
   async list(workspaceId: string): Promise<Sprint[]> {

package/src/modules/tasks.ts

@@ -9,7 +9,7 @@ import {
   TasksResponse,
   UpdateTask,
 } from "@locusai/shared";
-import { BaseModule } from "./base";
+import { BaseModule } from "./base.js";
 
 export interface TaskListOptions {
   sprintId?: string;

package/src/modules/workspaces.ts

@@ -10,7 +10,7 @@ import {
   WorkspaceStats,
   WorkspacesResponse,
 } from "@locusai/shared";
-import { BaseModule } from "./base";
+import { BaseModule } from "./base.js";
 
 export class WorkspacesModule extends BaseModule {
   async listAll(): Promise<Workspace[]> {

package/src/orchestrator.ts

@@ -3,8 +3,8 @@ import { existsSync } from "node:fs";
 import { dirname, join } from "node:path";
 import { Task, TaskPriority, TaskStatus } from "@locusai/shared";
 import { EventEmitter } from "events";
-import { LocusClient } from "./index";
-import { c } from "./utils/colors";
+import { LocusClient } from "./index.js";
+import { c } from "./utils/colors.js";
 
 export interface AgentConfig {
   id: string;