@superblocksteam/vite-plugin-file-sync 2.0.72-next.8 → 2.0.73-next.0

package/dist/ai-service/context-upload.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context-upload.js","sourceRoot":"","sources":["../../src/ai-service/context-upload.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,EAAE,iCAAiC,EAAE,MAAM,uBAAuB,CAAC;AAC1E,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAG9D;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAAC,EAC7C,aAAa,EACb,QAAQ,EACR,UAAU,EACV,cAAc,EACd,MAAM,EACN,kBAAkB,EAClB,MAAM,EACN,GAAG,GAUJ;IACC,iDAAiD;IACjD,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAC1B,cAAc,EACd,cAAc,EACd,SAAS,EACT,MAAM,CACP,CAAC;IAEF,0CAA0C;IAC1C,IAAI,YAAsB,CAAC;IAC3B,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,UAAU,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;QACnE,iEAAiE;QACjE,YAAY,GAAG,OAAO;aACnB,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;aACjC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC;IAAC,OAAO,GAAQ,EAAE,CAAC;QAClB,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC1B,8DAA8D;YAC9D,MAAM,CAAC,IAAI,CACT,oEAAoE,MAAM,mBAAmB,CAC9F,CAAC;YACF,OAAO;QACT,CAAC;QACD,MAAM,GAAG,CAAC;IACZ,CAAC;IAED,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC9B,MAAM,CAAC,IAAI,CACT,gEAAgE,MAAM,mBAAmB,CAC1F,CAAC;QACF,OAAO;IACT,CAAC;IAED,wCAAwC;IACxC,IAAI,WAAmB,CAAC;IACxB,IAAI,CAAC;QACH,WAAW,GAAG,MAAM,cAAc,CAAC;YACjC,aAAa;YACb,kBAAkB;YAClB,GAAG;SACJ,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,KAAK,CACV,uDAAuD,EACvD,YAAY,CAAC,KAAK,CAAC,CACpB,CAAC;QACF,MAAM,KAAK,CAAC;IACd,CAAC;IAED,+CAA+C;IAC/C,MAAM,QAAQ,GAAG,IAAI,QAAQ,EAAE,CAAC;IAChC,MAAM,aAAa,GAAa,EAAE,CAAC;IAEnC,KAAK,MAAM,QAAQ,IAAI,YAAY,EAAE,CAAC;QACpC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QACjD,IAAI,CAAC;YACH,iDAAiD;YACjD,MAAM,UAAU,GAAG,MAAM,QAAQ,CAAC,QAAQ,CAAC,CAAC;YAE5C,wCAAwC;YACxC,4DAA4D;YAC5D,MAAM,UAAU,GAAG,IAAI,UAAU,CAAC,UAAU,CAAC,CAAC;YAC9C,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;YAEpC,sDAAsD;YACtD,QAAQ,CAAC,MAAM,CAAC,SAAS,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;YAC3C,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,CAAC,IAAI,CACT,0DAA0D,QAAQ,YAAY,EAC9E,YAAY,CAAC,GAAG,CAAC,CAClB,CAAC;YACF,8CAA8C;QAChD,CAAC;IACH,CAAC;IAED,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/B,MAAM,CAAC,IAAI,CACT,uEAAuE,CACxE,CAAC;QACF,OAAO;IACT,CAAC;IAED,8CAA8C;IAC9C,MAAM,YAAY,GAAG,iCAAiC,CAAC,kBAAkB,CAAC,CAAC;IAE3E,8DAA8D;IAC9D,MAAM,GAAG,GAAG,GAAG,YAAY,iBAAiB,aAAa,aAAa,kBAAkB,CAAC,UAAU,CAAC,YAAY,kBAAkB,CAAC,QAAQ,CAAC,UAAU,CAAC;IAEvJ,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;YAChC,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE;gBACP,aAAa,EAAE,UAAU,WAAW,EAAE;gBACtC,gEAAgE;aACjE;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,SAAS,GAAG,MAAM,QAAQ;iBAC7B,IAAI,EAAE;iBACN,KAAK,CAAC,GAAG,EAAE,CAAC,+BAA+B,CAAC,CAAC;YAChD,MAAM,CAAC,KAAK,CACV,oDAAoD,QAAQ,CAAC,MAAM,gBAAgB,QAAQ,CAAC,UAAU,WAAW,SAAS,sBAAsB,aAAa,cAAc,QAAQ,EAAE,CACtL,CAAC;YACF,MAAM,IAAI,KAAK,CAAC,0BAA0B,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;QACnE,CAAC;QAED,MAAM,CAAC,IAAI,CACT,oDAAoD,aAAa,CAAC,MAAM,qBAAqB,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,uBAAuB,aAAa,cAAc,QAAQ,EAAE,CAClL,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,KAAK,CACV,qDAAqD,EACrD,YAAY,CAAC,KAAK,CAAC,CACpB,CAAC;QACF,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC"}

package/dist/ai-service/index.d.ts

@@ -5,7 +5,8 @@ import type { FileSystemManager } from "../file-system-manager.js";
 import type { SourceTracker } from "../source-tracker.js";
 import type { JudgeEvaluationOptions, JudgeEvaluationResult } from "./judge/judge-service.js";
 import type { AiServiceConfig, AiServiceEvents, EditorClient, UserChange, ApiUpdate } from "./types.ts";
-import type { AiGenerateRequest, AiExplainCodeRequest, LLMProviderConfig, PromptContext, RuntimeErrorData, ConsoleLogEntry } from "@superblocksteam/library-shared/types";
+import type { ContextId } from "./llm/context/types.js";
+import type { AiGenerateRequest, AiExplainCodeRequest, AiLabelAttachmentRequest, AiLabelAttachmentResponse, LLMProviderConfig, PromptContext, RuntimeErrorData, ConsoleLogEntry } from "@superblocksteam/library-shared/types";
 import type { ViteDevServer } from "vite";
 export { AiServiceFeatureFlags } from "./features.js";
 export declare class AiService extends TracedEventEmitter<AiServiceEvents> {
@@ -18,7 +19,7 @@ export declare class AiService extends TracedEventEmitter<AiServiceEvents> {
     private _appContext;
     private llmProviderSettings;
     private _logger;
-    private clark;
+    private readonly clark;
     private clarkProfiler;
     readonly chatSessionStore: ChatSessionStore;
     private readonly contextManager;
@@ -31,6 +32,22 @@ export declare class AiService extends TracedEventEmitter<AiServiceEvents> {
     private readonly mcpServerManager;
     get appContext(): AppContextStore;
     get clarkState(): string;
+    get appRootDirPath(): string;
+    get organizationId(): string;
+    get superblocksBaseUrl(): string;
+    /**
+     * Gets the contextId for the current conversation context.
+     * Returns null if JWT is missing or has no userId.
+     */
+    getContextId(): ContextId | null;
+    /**
+     * Gets the JWT from Clark context
+     */
+    getClarkJwt(): string | undefined;
+    /**
+     * Updates the JWT in Clark context
+     */
+    updateClarkJwt(jwt: string): void;
     constructor(config: AiServiceConfig);
     /**
      * Initialize the AI service. Must be called after construction.
@@ -48,10 +65,16 @@ export declare class AiService extends TracedEventEmitter<AiServiceEvents> {
     private createClarkTagProviderFactory;
     handleAiGenerate(request: AiGenerateRequest, peer: EditorClient, peerId: string): Promise<void>;
     handleAiExplainCode(request: AiExplainCodeRequest, peer: EditorClient): Promise<void>;
+    handleAiLabelAttachment(request: AiLabelAttachmentRequest): Promise<AiLabelAttachmentResponse>;
+    /**
+     * Converts an attachment to the appropriate LLM content part.
+     */
+    private attachmentToContentPart;
     private submitAcceptanceEvaluation;
     handleUserAcceptedDraft(peer: EditorClient, silently?: boolean): Promise<void>;
     handleUserRejectedDraft(peer: EditorClient, silently?: boolean): Promise<void>;
     handleUserCanceled(): Promise<void>;
+    handleSkipTestCase(testRunId: string, testCaseId: string): Promise<void>;
     handleUserConnected(peer: EditorClient, peerId: string): void;
     handleUserDisconnected(peerId: string): void;
     private clearPeerConnection;
@@ -87,9 +110,10 @@ export declare class AiService extends TracedEventEmitter<AiServiceEvents> {
     }): Promise<JudgeEvaluationResult>;
     handleUserChange(change: UserChange): void;
     handleNewChat(): Promise<void>;
-    handleTruncateChatAfterTimestamp(sinceTimestamp: number): Promise<void>;
+    handleRestoreToCommit(sinceTimestamp: number, commitId: string, applicationId: string, branchName: string): Promise<void>;
     handleApiUpdate(update: ApiUpdate): Promise<void>;
     isBusy(): boolean;
+    isAwaitingUser(): boolean;
     /**
      * Check if Clark is currently generating code.
      * Used by lock-service to determine if lock should be held during generation.

package/dist/ai-service/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/ai-service/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAEL,kBAAkB,EACnB,MAAM,yBAAyB,CAAC;AAgBjC,OAAO,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC;AAwC3D,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AACnE,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,KAAK,EACV,sBAAsB,EACtB,qBAAqB,EACtB,MAAM,0BAA0B,CAAC;AAElC,OAAO,KAAK,EAEV,eAAe,EACf,eAAe,EACf,YAAY,EACZ,UAAU,EACV,SAAS,EACV,MAAM,YAAY,CAAC;AAYpB,OAAO,KAAK,EACV,iBAAiB,EACjB,oBAAoB,EACpB,iBAAiB,EACjB,aAAa,EACb,gBAAgB,EAChB,eAAe,EAGhB,MAAM,uCAAuC,CAAC;AAC/C,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,MAAM,CAAC;AAE1C,OAAO,EAAE,qBAAqB,EAAE,MAAM,eAAe,CAAC;AA2EtD,qBAAa,SAAU,SAAQ,kBAAkB,CAAC,eAAe,CAAC;IA8BpD,OAAO,CAAC,QAAQ,CAAC,MAAM;IA7BnC,OAAO,CAAC,iBAAiB,CAAoB;IAC7C,OAAO,CAAC,gBAAgB,CAAmB;IAC3C,OAAO,CAAC,QAAQ,CAAW;IAC3B,OAAO,CAAC,mBAAmB,CAAsB;IACjD,OAAO,CAAC,gBAAgB,CAAmB;IAC3C,OAAO,CAAC,WAAW,CAAkB;IACrC,OAAO,CAAC,mBAAmB,CAAsB;IACjD,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,KAAK,CAAQ;IACrB,OAAO,CAAC,aAAa,CAAgB;IACrC,SAAgB,gBAAgB,EAAE,gBAAgB,CAAC;IACnD,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAiB;IAChD,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAY;IACtC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAoC;IACjE,OAAO,CAAC,QAAQ,CAAC,qBAAqB,CACD;IACrC,OAAO,CAAC,6BAA6B,CAA2B;IAChE,OAAO,CAAC,eAAe,CAAC,CAAS;IACjC,OAAO,CAAC,UAAU,CAAC,CAAgB;IACnC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAA6B;IAE9D,IAAW,UAAU,IAAI,eAAe,CAEvC;IAED,IAAW,UAAU,IAAI,MAAM,CAE9B;gBAE4B,MAAM,EAAE,eAAe;IA0IpD;;;OAGG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAKjC;;;OAGG;IACG,sBAAsB,IAAI,OAAO,CAAC,IAAI,CAAC;IAK7C,OAAO,CAAC,SAAS;IAIjB,OAAO,CAAC,4BAA4B;IAwGpC,OAAO,CAAC,MAAM;IAQd,OAAO,CAAC,6BAA6B;IAWxB,gBAAgB,CAC3B,OAAO,EAAE,iBAAiB,EAC1B,IAAI,EAAE,YAAY,EAClB,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,IAAI,CAAC;IA0CH,mBAAmB,CAC9B,OAAO,EAAE,oBAAoB,EAC7B,IAAI,EAAE,YAAY,GACjB,OAAO,CAAC,IAAI,CAAC;IAwNhB,OAAO,CAAC,0BAA0B;IAqBrB,uBAAuB,CAClC,IAAI,EAAE,YAAY,EAClB,QAAQ,CAAC,EAAE,OAAO,GACjB,OAAO,CAAC,IAAI,CAAC;IAkBH,uBAAuB,CAClC,IAAI,EAAE,YAAY,EAClB,QAAQ,CAAC,EAAE,OAAO,GACjB,OAAO,CAAC,IAAI,CAAC;IAqBH,kBAAkB,IAAI,OAAO,CAAC,IAAI,CAAC;IAUzC,mBAAmB,CAAC,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI;IAuE7D,sBAAsB,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAiBnD,OAAO,CAAC,mBAAmB;IAQpB,sBAAsB,CAAC,KAAK,EAAE,KAAK,GAAG,IAAI;IAyB1C,kBAAkB,CAAC,SAAS,EAAE,gBAAgB;IA8C9C,uBAAuB,CAAC,EAAE,EAAE,MAAM;IAYlC,gBAAgB,CAAC,QAAQ,EAAE,eAAe,GAAG,IAAI;IA0CjD,2BAA2B,IAAI,IAAI;IAUnC,4BAA4B,IAAI,IAAI;IAY9B,0BAA0B,CACrC,OAAO,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,GAAG,EAAE,MAAM,CAAC;QAAC,aAAa,CAAC,EAAE,aAAa,CAAA;KAAE,EACvE,IAAI,EAAE,YAAY,GACjB,OAAO,CAAC,IAAI,CAAC;IAQhB;;;;;;;;OAQG;IACU,qBAAqB,CAAC,OAAO,EAAE;QAC1C,QAAQ,EAAE,MAAM,CAAC;QACjB,MAAM,EAAE,MAAM,CAAC;QACf,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,GAAG,EAAE,MAAM,CAAC;QACZ,SAAS,CAAC,EAAE,iBAAiB,CAAC;QAC9B,OAAO,CAAC,EAAE,OAAO,CAAC,sBAAsB,CAAC,CAAC;QAC1C,gBAAgB,CAAC,EAAE,GAAG,CAAC;KACxB,GAAG,OAAO,CAAC,qBAAqB,CAAC;IA6H3B,gBAAgB,CAAC,MAAM,EAAE,UAAU;IAQ7B,aAAa;IAYb,gCAAgC,CAAC,cAAc,EAAE,MAAM;IAuBvD,eAAe,CAAC,MAAM,EAAE,SAAS;IAkBvC,MAAM,IAAI,OAAO;IAQxB;;;;OAIG;IACI,YAAY,IAAI,OAAO;IAS9B;;;;;;OAMG;IACU,2BAA2B,IAAI,OAAO,CAAC,IAAI,CAAC;IAYzD;;;OAGG;IACH,OAAO,CAAC,yBAAyB;IAQpB,gCAAgC,IAAI,OAAO,CAAC,OAAO,CAAC;IAY1D,gBAAgB,CAAC,aAAa,EAAE,aAAa,GAAG,SAAS;IAQzD,oBAAoB,CAAC,iBAAiB,EAAE,iBAAiB;IAMzD,aAAa,CAAC,MAAM,EAAE,aAAa,GAAG,IAAI;IAI1C,aAAa,IAAI,aAAa,GAAG,SAAS;IAIjD,OAAO,CAAC,sBAAsB;CA2C/B"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/ai-service/index.ts"],"names":[],"mappings":"AAGA,OAAO,EAEL,kBAAkB,EACnB,MAAM,yBAAyB,CAAC;AAiBjC,OAAO,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC;AAyC3D,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AACnE,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,KAAK,EACV,sBAAsB,EACtB,qBAAqB,EACtB,MAAM,0BAA0B,CAAC;AAElC,OAAO,KAAK,EAEV,eAAe,EACf,eAAe,EACf,YAAY,EACZ,UAAU,EACV,SAAS,EACV,MAAM,YAAY,CAAC;AAEpB,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AAWxD,OAAO,KAAK,EACV,iBAAiB,EACjB,oBAAoB,EACpB,wBAAwB,EACxB,yBAAyB,EACzB,iBAAiB,EACjB,aAAa,EACb,gBAAgB,EAChB,eAAe,EAIhB,MAAM,uCAAuC,CAAC;AAC/C,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,MAAM,CAAC;AAE1C,OAAO,EAAE,qBAAqB,EAAE,MAAM,eAAe,CAAC;AA4EtD,qBAAa,SAAU,SAAQ,kBAAkB,CAAC,eAAe,CAAC;IAsEpD,OAAO,CAAC,QAAQ,CAAC,MAAM;IArEnC,OAAO,CAAC,iBAAiB,CAAoB;IAC7C,OAAO,CAAC,gBAAgB,CAAmB;IAC3C,OAAO,CAAC,QAAQ,CAAW;IAC3B,OAAO,CAAC,mBAAmB,CAAsB;IACjD,OAAO,CAAC,gBAAgB,CAAmB;IAC3C,OAAO,CAAC,WAAW,CAAkB;IACrC,OAAO,CAAC,mBAAmB,CAAsB;IACjD,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAQ;IAC9B,OAAO,CAAC,aAAa,CAAgB;IACrC,SAAgB,gBAAgB,EAAE,gBAAgB,CAAC;IACnD,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAiB;IAChD,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAY;IACtC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAoC;IACjE,OAAO,CAAC,QAAQ,CAAC,qBAAqB,CACD;IACrC,OAAO,CAAC,6BAA6B,CAA2B;IAChE,OAAO,CAAC,eAAe,CAAC,CAAS;IACjC,OAAO,CAAC,UAAU,CAAC,CAAgB;IACnC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAA6B;IAE9D,IAAW,UAAU,IAAI,eAAe,CAEvC;IAED,IAAW,UAAU,IAAI,MAAM,CAE9B;IAED,IAAW,cAAc,IAAI,MAAM,CAElC;IAED,IAAW,cAAc,IAAI,MAAM,CAElC;IAED,IAAW,kBAAkB,IAAI,MAAM,CAEtC;IAED;;;OAGG;IACI,YAAY,IAAI,SAAS,GAAG,IAAI;IAQvC;;OAEG;IACI,WAAW,IAAI,MAAM,GAAG,SAAS;IAIxC;;OAEG;IACI,cAAc,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI;gBAMX,MAAM,EAAE,eAAe;IA6IpD;;;OAGG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAKjC;;;OAGG;IACG,sBAAsB,IAAI,OAAO,CAAC,IAAI,CAAC;IAK7C,OAAO,CAAC,SAAS;IAMjB,OAAO,CAAC,4BAA4B;IAwGpC,OAAO,CAAC,MAAM;IAQd,OAAO,CAAC,6BAA6B;IAWxB,gBAAgB,CAC3B,OAAO,EAAE,iBAAiB,EAC1B,IAAI,EAAE,YAAY,EAClB,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,IAAI,CAAC;IAoDH,mBAAmB,CAC9B,OAAO,EAAE,oBAAoB,EAC7B,IAAI,EAAE,YAAY,GACjB,OAAO,CAAC,IAAI,CAAC;IAwNH,uBAAuB,CAClC,OAAO,EAAE,wBAAwB,GAChC,OAAO,CAAC,yBAAyB,CAAC;IAsFrC;;OAEG;IACH,OAAO,CAAC,uBAAuB;IAyC/B,OAAO,CAAC,0BAA0B;IAqBrB,uBAAuB,CAClC,IAAI,EAAE,YAAY,EAClB,QAAQ,CAAC,EAAE,OAAO,GACjB,OAAO,CAAC,IAAI,CAAC;IAqBH,uBAAuB,CAClC,IAAI,EAAE,YAAY,EAClB,QAAQ,CAAC,EAAE,OAAO,GACjB,OAAO,CAAC,IAAI,CAAC;IAwBH,kBAAkB,IAAI,OAAO,CAAC,IAAI,CAAC;IAUnC,kBAAkB,CAC7B,SAAS,EAAE,MAAM,EACjB,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,IAAI,CAAC;IAiDT,mBAAmB,CAAC,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI;IA2E7D,sBAAsB,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAsBnD,OAAO,CAAC,mBAAmB;IASpB,sBAAsB,CAAC,KAAK,EAAE,KAAK,GAAG,IAAI;IAyB1C,kBAAkB,CAAC,SAAS,EAAE,gBAAgB;IA8C9C,uBAAuB,CAAC,EAAE,EAAE,MAAM;IAYlC,gBAAgB,CAAC,QAAQ,EAAE,eAAe,GAAG,IAAI;IA0CjD,2BAA2B,IAAI,IAAI;IAUnC,4BAA4B,IAAI,IAAI;IAY9B,0BAA0B,CACrC,OAAO,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,GAAG,EAAE,MAAM,CAAC;QAAC,aAAa,CAAC,EAAE,aAAa,CAAA;KAAE,EACvE,IAAI,EAAE,YAAY,GACjB,OAAO,CAAC,IAAI,CAAC;IAQhB;;;;;;;;OAQG;IACU,qBAAqB,CAAC,OAAO,EAAE;QAC1C,QAAQ,EAAE,MAAM,CAAC;QACjB,MAAM,EAAE,MAAM,CAAC;QACf,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,GAAG,EAAE,MAAM,CAAC;QACZ,SAAS,CAAC,EAAE,iBAAiB,CAAC;QAC9B,OAAO,CAAC,EAAE,OAAO,CAAC,sBAAsB,CAAC,CAAC;QAC1C,gBAAgB,CAAC,EAAE,GAAG,CAAC;KACxB,GAAG,OAAO,CAAC,qBAAqB,CAAC;IA6H3B,gBAAgB,CAAC,MAAM,EAAE,UAAU;IAQ7B,aAAa;IAYb,qBAAqB,CAChC,cAAc,EAAE,MAAM,EACtB,QAAQ,EAAE,MAAM,EAChB,aAAa,EAAE,MAAM,EACrB,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,IAAI,CAAC;IA8EH,eAAe,CAAC,MAAM,EAAE,SAAS;IAkBvC,MAAM,IAAI,OAAO;IAQjB,cAAc,IAAI,OAAO;IAIhC;;;;OAIG;IACI,YAAY,IAAI,OAAO;IAS9B;;;;;;OAMG;IACU,2BAA2B,IAAI,OAAO,CAAC,IAAI,CAAC;IAYzD;;;OAGG;IACH,OAAO,CAAC,yBAAyB;IAQpB,gCAAgC,IAAI,OAAO,CAAC,OAAO,CAAC;IAY1D,gBAAgB,CAAC,aAAa,EAAE,aAAa,GAAG,SAAS;IAQzD,oBAAoB,CAAC,iBAAiB,EAAE,iBAAiB;IAMzD,aAAa,CAAC,MAAM,EAAE,aAAa,GAAG,IAAI;IAI1C,aAAa,IAAI,aAAa,GAAG,SAAS;IAIjD,OAAO,CAAC,sBAAsB;CA2C/B"}

package/dist/ai-service/index.js

@@ -1,7 +1,9 @@
 import os from "node:os";
 import path from "node:path";
+import { isTextAttachment } from "@superblocksteam/library-shared/types";
 import { addTracingToMethods, TracedEventEmitter, } from "@superblocksteam/shared";
-import { hasToolCall } from "ai";
+import { generateObject, hasToolCall } from "ai";
+import { z } from "zod";
 import { getErrorMeta, getLogger, setDefaultLogger } from "../util/logger.js";
 import { OperationPassthrough } from "../util/operation-queue.js";
 import { readFile } from "./agent/tools/build-read-file.js";
@@ -15,6 +17,7 @@ import { AppShell } from "./app-interface/shell.js";
 import { BoltArtifactProcessor } from "./artifacts/bolt.js";
 import { ChatSessionStore } from "./chat/chat-session-store.js";
 import { AppContextStore } from "./context/app-context.js";
+import { restoreContextFromCheckpoint } from "./context-download.js";
 import { IntegrationStore } from "./integrations/store.js";
 import { JudgeService } from "./judge/judge-service.js";
 import { CsvJudgeStorage } from "./judge/storage/csv-storage.js";
@@ -100,6 +103,7 @@ const mergePlanContexts = (existing, incoming) => {
         plan: incoming?.plan ?? existing?.plan,
         approved: incoming?.approved !== undefined ? incoming.approved : existing?.approved,
         integrations: incoming?.integrations ?? existing?.integrations,
+        entities: incoming?.entities ?? existing?.entities,
         title: incoming?.title ?? existing?.title,
         attachments: incoming?.attachments ?? existing?.attachments,
         enableTesting: incoming?.enableTesting ?? existing?.enableTesting,
@@ -132,11 +136,48 @@ export class AiService extends TracedEventEmitter {
     get clarkState() {
         return this.clark.state;
     }
+    get appRootDirPath() {
+        return this.config.appRootDirPath;
+    }
+    get organizationId() {
+        return this.config.organizationId;
+    }
+    get superblocksBaseUrl() {
+        return this.config.superblocksBaseUrl;
+    }
+    /**
+     * Gets the contextId for the current conversation context.
+     * Returns null if JWT is missing or has no userId.
+     */
+    getContextId() {
+        try {
+            return getContextId(this.clark, this.config);
+        }
+        catch {
+            return null;
+        }
+    }
+    /**
+     * Gets the JWT from Clark context
+     */
+    getClarkJwt() {
+        return this.clark?.context?.jwt;
+    }
+    /**
+     * Updates the JWT in Clark context
+     */
+    updateClarkJwt(jwt) {
+        if (jwt) {
+            this.clark.updateContext({ jwt });
+        }
+    }
     constructor(config) {
         super(config.tracer);
         this.config = config;
-        this._logger = config.logger;
         setDefaultLogger(config.logger);
+        // Store the wrapped logger so getLogger() can return it
+        // The wrapped logger forwards logs to the browser console when Clark logs are enabled
+        this._logger = getLogger();
         // Prevent traced functions from creating traces when called outside of a Clark context
         llmobs.setDefaultRootPolicy("drop-if-no-parent");
         // todo: move to state handlers
@@ -234,6 +275,7 @@ export class AiService extends TracedEventEmitter {
             "handleUserAcceptedDraft",
             "handleUserRejectedDraft",
             "handleUserCanceled",
+            "handleSkipTestCase",
             "handleBuildSystemError",
             "handleRuntimeError",
             "handleClearRuntimeError",
@@ -261,6 +303,8 @@ export class AiService extends TracedEventEmitter {
         this.getLogger().info("[ai-service] Integration cache removed");
     }
     getLogger() {
+        // Return the wrapped logger that forwards logs to the browser console
+        // when Clark logs are enabled (via ?clark_logs=true or SUPERBLOCKS_CLI_ENV=local)
         return this._logger;
     }
     createClarkTransitionHandler() {
@@ -355,6 +399,7 @@ export class AiService extends TracedEventEmitter {
         };
     }
     async handleAiGenerate(request, peer, peerId) {
+        this.getLogger().info(`[ai-service] handleAiGenerate: peerId=${peerId}`);
         if (this.clark.state === ClarkStateNames.Dead) {
             throw new Error("Service is unavailable");
         }
@@ -364,6 +409,13 @@ export class AiService extends TracedEventEmitter {
         }
         this.clarkProfiler.startSession();
         getLogger().info("AI Service received request with llmConfig:", request.llmConfig || {});
+        getLogger().info("[ai-service] handleAiGenerate received request", {
+            requestMode: request.mode,
+            currentContextMode: this.clark.context.currentMode,
+            requestPlanContextApproved: request.planContext?.approved,
+            existingPlanContextApproved: this.clark.context.planContext?.approved,
+            responseMetadataType: request.responseMetadata?.type,
+        });
         this.clark.updateContext((context) => {
             return {
                 ...context,
@@ -520,6 +572,110 @@ export class AiService extends TracedEventEmitter {
             });
         }
     }
+    async handleAiLabelAttachment(request) {
+        const logger = getLogger();
+        const startTime = Date.now();
+        const timeout = 30_000;
+        const attachment = request.attachment;
+        logger.debug("[ai-service] Labeling attachment", {
+            fileName: attachment.fileName ?? "unnamed",
+            type: attachment.type,
+        });
+        const llmConfig = request.llmConfig
+            ? {
+                ...request.llmConfig,
+                interleavedThinking: false,
+                thinkingEnabled: false,
+            }
+            : undefined;
+        const llmProviderSettings = this.getLLMProviderSettings(false, llmConfig);
+        const llmProvider = createLLMProvider(llmProviderSettings, () => request.jwt, llmConfig);
+        const model = llmProvider.modelForTask("fast");
+        const LabelAttachmentSchema = z.object({
+            label: z
+                .string()
+                .min(1)
+                .describe('A brief 2-3 word description (e.g., "sales dashboard", "user profile", "data table", "customer data")'),
+            insight: z
+                .string()
+                .min(1)
+                .describe('A key observation about the content (e.g., "chart displays monthly revenue", "CSV contains customer records", "JSON config for API settings")'),
+        });
+        const timeoutController = new AbortController();
+        const timeoutId = setTimeout(() => timeoutController.abort(), timeout);
+        try {
+            // Build the appropriate content part based on attachment type
+            const contentPart = this.attachmentToContentPart(attachment);
+            const { object: result } = await generateObject({
+                model,
+                schema: LabelAttachmentSchema,
+                messages: [
+                    {
+                        role: "user",
+                        content: [
+                            {
+                                type: "text",
+                                text: "Analyze this attachment and provide a brief label and key insight about its content.",
+                            },
+                            contentPart,
+                        ],
+                    },
+                ],
+                abortSignal: timeoutController.signal,
+            });
+            const durationMs = Date.now() - startTime;
+            logger.debug("[ai-service] Attachment labeling complete", {
+                fileName: attachment.fileName,
+                type: attachment.type,
+                label: result.label,
+                insight: result.insight,
+                durationMs,
+            });
+            return result;
+        }
+        finally {
+            clearTimeout(timeoutId);
+        }
+    }
+    /**
+     * Converts an attachment to the appropriate LLM content part.
+     */
+    attachmentToContentPart(attachment) {
+        if (attachment.type === "image") {
+            return {
+                type: "image",
+                image: attachment.image,
+            };
+        }
+        else if (attachment.type === "pdf") {
+            // Parse data URL to extract base64 data
+            const match = attachment.data.match(/^data:([^;]+);base64,(.+)$/);
+            const data = match?.[2] ?? attachment.data;
+            const mediaType = match?.[1] ?? "application/pdf";
+            return {
+                type: "file",
+                data,
+                mediaType,
+                filename: attachment.fileName,
+            };
+        }
+        else if (isTextAttachment(attachment)) {
+            // Wrap text content in XML tags for structured presentation
+            const sanitizedFileName = attachment.fileName
+                .replace(/&/g, "&")
+                .replace(/"/g, """)
+                .replace(/'/g, "'")
+                .replace(/</g, "&lt;")
+                .replace(/>/g, "&gt;");
+            return {
+                type: "text",
+                text: `<${attachment.type}_file name="${sanitizedFileName}">\n${attachment.content}\n</${attachment.type}_file>`,
+            };
+        }
+        else {
+            throw new Error(`Unsupported attachment type: ${attachment.type}`);
+        }
+    }
     submitAcceptanceEvaluation(wasAccepted) {
         if (!this.clark.rootSpan) {
             this.getLogger().warn("[ai-service] No root span available for submitting feedback");
@@ -537,12 +693,14 @@ export class AiService extends TracedEventEmitter {
         }
     }
     async handleUserAcceptedDraft(peer, silently) {
+        this.getLogger().info(`[ai-service] handleUserAcceptedDraft`);
         this.submitAcceptanceEvaluation(true);
         if (this.clark.state !== ClarkStateNames.AwaitingUser) {
             throw new Error("Service is not awaiting user input");
         }
         if (!this.clark.context.peer) {
-            // this can happen if the service was started with a draft
+            // peer can be null if service was started with a draft
+            this.getLogger().info(`[ai-service] Setting peer from RPC caller`);
             this.clark.updateContext({
                 peer,
             });
@@ -551,12 +709,14 @@ export class AiService extends TracedEventEmitter {
         void transitionTo({ type: USER_ACCEPTED_DRAFT, silently });
     }
     async handleUserRejectedDraft(peer, silently) {
+        this.getLogger().info(`[ai-service] handleUserRejectedDraft`);
         this.submitAcceptanceEvaluation(false);
         if (this.clark.state !== ClarkStateNames.AwaitingUser) {
             throw new Error("Service is not awaiting user input");
         }
         if (!this.clark.context.peer) {
-            // this can happen if the service was started with a draft
+            // peer can be null if service was started with a draft
+            this.getLogger().info(`[ai-service] Setting peer from RPC caller`);
             this.clark.updateContext({
                 peer,
             });
@@ -574,8 +734,42 @@ export class AiService extends TracedEventEmitter {
         this.clark.context.abortController?.abort();
         void this.clark.send({ type: USER_CANCELED });
     }
+    async handleSkipTestCase(testRunId, testCaseId) {
+        const logger = this.getLogger();
+        logger.info(`[ai-service] Skip test case requested: testRunId=${testRunId}, testCaseId=${testCaseId}`);
+        // Verify this is the active test run
+        if (this.clark.context.currentTestRunId !== testRunId) {
+            logger.warn(`[ai-service] Skip request for testRunId=${testRunId} but current is ${this.clark.context.currentTestRunId}`);
+            return;
+        }
+        // Update context with skip request
+        this.clark.updateContext({
+            skipRequestedForTestCase: testCaseId,
+        });
+        // Update the test case status to skipped
+        const testCaseStatuses = this.clark.context.testCaseStatuses ??
+            new Map();
+        testCaseStatuses.set(testCaseId, "skipped");
+        this.clark.updateContext({
+            testCaseStatuses,
+        });
+        // Send test_case_update message to UI
+        const peer = this.clark.context.peer;
+        if (peer) {
+            void peer.call.aiPushMessage({
+                type: "test_case_update",
+                testRunId,
+                testCaseId,
+                testCaseStatus: "skipped",
+                role: "assistant",
+                content: "",
+            });
+        }
+        logger.info(`[ai-service] Test case ${testCaseId} marked as skipped, flag set for subagent`);
+    }
     handleUserConnected(peer, peerId) {
         const existingPeerId = this.clark.context.peerId;
+        this.getLogger().info(`[ai-service] handleUserConnected: peerId=${peerId}`);
         if (this.clark.context.peer) {
             if (existingPeerId === peerId) {
                 this.getLogger().info(`[ai-service] Existing peer ${peerId} reconnected; refreshing reference`);
@@ -592,6 +786,7 @@ export class AiService extends TracedEventEmitter {
             peer,
             peerId,
         }));
+        this.getLogger().info(`[ai-service] Peer set: ${peerId}`);
         // 1. Sync generation state if generation is in progress
         if (this.clark.context.lastGenerationState) {
             this.getLogger().info(`[ai-service] Syncing generation state on reconnect: ${this.clark.context.lastGenerationState.state}`);
@@ -627,18 +822,19 @@ export class AiService extends TracedEventEmitter {
     }
     handleUserDisconnected(peerId) {
         const currentPeerId = this.clark.context.peerId;
+        this.getLogger().info(`[ai-service] handleUserDisconnected: peerId=${peerId}`);
         if (!currentPeerId) {
             return;
         }
         if (currentPeerId !== peerId) {
-            this.getLogger().warn(`[ai-service] Disconnect event for ${peerId} but current peer is ${currentPeerId}; clearing stale peer`);
-        }
-        else {
-            this.getLogger().info(`[ai-service] peer disconnecting: ${peerId}`);
+            this.getLogger().info(`[ai-service] Stale disconnect ignored: peer ${peerId} disconnecting but current peer is ${currentPeerId}`);
+            return; // Don't clear the current peer on stale disconnect
         }
+        this.getLogger().info(`[ai-service] peer disconnecting: ${peerId}`);
         this.clearPeerConnection();
     }
     clearPeerConnection() {
+        this.getLogger().info(`[ai-service] Clearing peer connection`);
         this.clark.updateContext((context) => ({
             ...context,
             peer: undefined,
@@ -877,18 +1073,59 @@ export class AiService extends TracedEventEmitter {
             this.getLogger().error("[ai-service] Failed to clear contexts on new chat", getErrorMeta(error));
         }
     }
-    async handleTruncateChatAfterTimestamp(sinceTimestamp) {
-        // Truncate chat messages on server
+    async handleRestoreToCommit(sinceTimestamp, commitId, applicationId, branchName) {
+        // TODO: change this to truncate after commit once commit info is supported here, so we can truncate messages accurately.
+        // 1. Truncate chat messages on server (always do this first)
         await this.chatSessionStore.truncateAfterTimestamp(sinceTimestamp);
-        // Truncate LLM context
+        // Check JWT once at the beginning - needed for both restoration and truncation
+        // Note: clark.context.jwt should be set by the socket handler (socket-manager.ts) before
+        // this method is called. The socket handler updates it from peerAuthorization if missing.
+        const jwt = this.clark?.context?.jwt;
+        if (!jwt) {
+            this.getLogger().warn(`[handleRestoreToCommit] No JWT available, cannot restore or truncate context`);
+            return;
+        }
+        // 2. Try to restore context from checkpoint
+        try {
+            const contextId = getContextId(this.clark, this.config);
+            const result = await restoreContextFromCheckpoint({
+                applicationId,
+                commitId,
+                branchName,
+                contextId,
+                appRootDirPath: this.config.appRootDirPath,
+                superblocksBaseUrl: this.config.superblocksBaseUrl,
+                logger: this.getLogger(),
+                jwt,
+                contextManager: this.contextManager,
+            });
+            if (result.success) {
+                this.getLogger().info(`[handleRestoreToCommit] Successfully restored main.json from checkpoint commitId=${commitId}, skipping truncation`);
+                return; // Success - no need to truncate
+            }
+            else {
+                // main.json not found or failed - fall through to truncation
+                const reason = result.savedFiles.length > 0
+                    ? "main.json not found in checkpoint"
+                    : "no file saved from checkpoint";
+                this.getLogger().info(`[handleRestoreToCommit] Context restore failed: ${reason} for commitId=${commitId}, using truncation fallback`);
+            }
+        }
+        catch (error) {
+            // Error downloading - log and fall through to truncation
+            this.getLogger().warn(`[handleRestoreToCommit] Error restoring context from checkpoint, using truncation fallback`, getErrorMeta(error));
+        }
+        // 3. Fallback: Truncate LLM context (restore failed)
         const contextId = getContextId(this.clark, this.config);
-        const ownerId = `truncation-handler-${Date.now()}-${Math.random().toString(36).slice(2, 9)}`;
+        const ownerId = `restore-handler-${Date.now()}-${Math.random().toString(36).slice(2, 9)}`;
         const contextHandle = await this.contextManager.acquireContext(contextId, ownerId);
         try {
             contextHandle.context.truncateAfterTimestamp(sinceTimestamp);
+            this.getLogger().info(`[handleRestoreToCommit] Context truncated after timestamp ${sinceTimestamp}`);
         }
         catch (error) {
             this.getLogger().error(`[ai-service] Failed to truncate context after timestamp ${sinceTimestamp}`, getErrorMeta(error));
+            throw error;
         }
         finally {
             contextHandle.release();
@@ -914,6 +1151,9 @@ export class AiService extends TracedEventEmitter {
             this.clark.state !== ClarkStateNames.AwaitingUser &&
             this.clark.state !== ClarkStateNames.Dead);
     }
+    isAwaitingUser() {
+        return this.clark.state === ClarkStateNames.AwaitingUser;
+    }
     /**
      * Check if Clark is currently generating code.
      * Used by lock-service to determine if lock should be held during generation.