@superblocksteam/vite-plugin-file-sync 2.0.72-next.12 → 2.0.72-next.15

package/dist/ai-service/features.js.map

@@ -1 +1 @@
-{"version":3,"file":"features.js","sourceRoot":"","sources":["../../src/ai-service/features.ts"],"names":[],"mappings":"AAAA,OAAO,CAAC,MAAM,KAAK,CAAC;AAsBpB;;;;;;;;;;;;GAYG;AACH,MAAM,gBAAgB,GAAG;IACvB,2BAA2B,EAAE;QAC3B,GAAG,EAAE,2BAA2B;QAChC,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;KACnB;IACD,gCAAgC,EAAE;QAChC,GAAG,EAAE,0BAA0B;QAC/B,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;KACnB;IACD,oBAAoB,EAAE;QACpB,GAAG,EAAE,yBAAyB;QAC9B,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;KAC9B;IACD,kBAAkB,EAAE;QAClB,GAAG,EAAE,0BAA0B;QAC/B,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;KAC9B;IACD,uBAAuB,EAAE;QACvB,GAAG,EAAE,+BAA+B;QACpC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;KAC7B;IACD,sBAAsB,EAAE;QACtB,GAAG,EAAE,iCAAiC;QACtC,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE;KACpB;CAC6C,CAAC;AAUjD,MAAM,CAAC,MAAM,qBAAqB,GAAG;IACnC;;OAEG;IACH,MAAM,CAAC,QAAiC,EAAE;QACxC,OAAO,IAAI,KAAK,CAAC,EAA2B,EAAE;YAC5C,GAAG,CAAC,OAAO,EAAE,IAAqB;gBAChC,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,CAAC,CAAC,IAAI,IAAI,gBAAgB,CAAC,EAAE,CAAC;oBAC5D,OAAO,SAAS,CAAC;gBACnB,CAAC;gBAED,MAAM,GAAG,GAAG,gBAAgB,CAAC,IAAqC,CAAC,CAAC;gBACpE,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBAE3B,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;oBACtB,OAAO,SAAS,CAAC;gBACnB,CAAC;gBAED,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;gBACzC,OAAO,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;YAClD,CAAC;SACF,CAAC,CAAC;IACL,CAAC;CACF,CAAC"}
+{"version":3,"file":"features.js","sourceRoot":"","sources":["../../src/ai-service/features.ts"],"names":[],"mappings":"AAAA,OAAO,CAAC,MAAM,KAAK,CAAC;AAsBpB;;;;;;;;;;;;GAYG;AACH,MAAM,gBAAgB,GAAG;IACvB,2BAA2B,EAAE;QAC3B,GAAG,EAAE,2BAA2B;QAChC,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;KACnB;IACD,gCAAgC,EAAE;QAChC,GAAG,EAAE,0BAA0B;QAC/B,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;KACnB;IACD,oBAAoB,EAAE;QACpB,GAAG,EAAE,yBAAyB;QAC9B,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;KAC9B;IACD,kBAAkB,EAAE;QAClB,GAAG,EAAE,0BAA0B;QAC/B,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;KAC9B;IACD,uBAAuB,EAAE;QACvB,GAAG,EAAE,+BAA+B;QACpC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;KAC7B;IACD,sBAAsB,EAAE;QACtB,GAAG,EAAE,iCAAiC;QACtC,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE;KACpB;IACD,oBAAoB,EAAE;QACpB,GAAG,EAAE,2BAA2B;QAChC,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;KAC9B;CAC6C,CAAC;AAUjD,MAAM,CAAC,MAAM,qBAAqB,GAAG;IACnC;;OAEG;IACH,MAAM,CAAC,QAAiC,EAAE;QACxC,OAAO,IAAI,KAAK,CAAC,EAA2B,EAAE;YAC5C,GAAG,CAAC,OAAO,EAAE,IAAqB;gBAChC,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,CAAC,CAAC,IAAI,IAAI,gBAAgB,CAAC,EAAE,CAAC;oBAC5D,OAAO,SAAS,CAAC;gBACnB,CAAC;gBAED,MAAM,GAAG,GAAG,gBAAgB,CAAC,IAAqC,CAAC,CAAC;gBACpE,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBAE3B,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;oBACtB,OAAO,SAAS,CAAC;gBACnB,CAAC;gBAED,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;gBACzC,OAAO,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;YAClD,CAAC;SACF,CAAC,CAAC;IACL,CAAC;CACF,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;AA0EtD,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;IAuIpD;;;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;IAsGpC,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;IAwCH,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;IAwE7D,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;IA4CH,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,12 +1,14 @@
 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-files.js";
+import { readFile } from "./agent/tools/build-read-file.js";
 import { debugCache } from "./agent/tools/debug-cache.js";
-import { buildReadFilesToolFactory } from "./agent/tools/index.js";
+import { buildReadFileToolFactory } from "./agent/tools/index.js";
 import { SessionEntityPermissionStore, } from "./agent/tools2/entity-permissions.js";
 import { ToolRegistry } from "./agent/tools2/registry.js";
 import { explainCodeFinalizeToolFactory } from "./agent/tools2/tools/explain-code-finalize.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,8 +103,10 @@ 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,
     };
 };
 export class AiService extends TracedEventEmitter {
@@ -131,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
@@ -198,9 +240,10 @@ export class AiService extends TracedEventEmitter {
             appShell: this.appShell,
             retryOptions: config.retryOptions,
             unifiedProviderEnabled: config.features.unifiedProviderEnabled ?? false,
+            organizationId: config.organizationId,
+            applicationId: config.applicationId,
+            getJwt: () => this.getJwt(),
         });
-        // Initialize judge service
-        this.mcpServerManager = new PlaywrightMcpServerManager(config.logger);
         this.clark = new Clark({
             initialContext: {
                 abortController: new AbortController(),
@@ -209,6 +252,8 @@ export class AiService extends TracedEventEmitter {
             createTagProvider: this.createClarkTagProviderFactory(),
             sessionId,
         });
+        // Initialize judge service
+        this.mcpServerManager = new PlaywrightMcpServerManager(this.clark.logger);
         this.config.draftInterface
             .hasLocalDraftChanges()
             .then((hasDraft) => {
@@ -230,6 +275,7 @@ export class AiService extends TracedEventEmitter {
             "handleUserAcceptedDraft",
             "handleUserRejectedDraft",
             "handleUserCanceled",
+            "handleSkipTestCase",
             "handleBuildSystemError",
             "handleRuntimeError",
             "handleClearRuntimeError",
@@ -257,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() {
@@ -290,6 +338,7 @@ export class AiService extends TracedEventEmitter {
                     llmClient: this.llmClient,
                     toolRegistry: this.toolRegistry,
                     entityPermissionStore: this.entityPermissionStore,
+                    mcpServerManager: this.mcpServerManager,
                     viteServer: this.viteServer,
                 };
                 const clarkStateHandlers = Object.fromEntries(Object.entries(clarkStateHandlerFactories).map(([state, handlerFactory]) => [state, handlerFactory(clark, params)]));
@@ -300,6 +349,7 @@ export class AiService extends TracedEventEmitter {
                 }
                 const handler = clarkStateHandlers[to];
                 if (!handler) {
+                    this.getLogger().error(`[ai-service] No handler for state ${to}`);
                     throw new Error(`No handler for state ${to}`);
                 }
                 this.getLogger().info(`[ai-service] Transitioned from ${from} to ${to} with event ${event.type}`);
@@ -349,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");
         }
@@ -363,6 +414,7 @@ export class AiService extends TracedEventEmitter {
                 ...context,
                 useMockGeneration: request.mockGeneration ?? false,
                 planContext: mergePlanContexts(context.planContext, request.planContext),
+                browserContext: request.browserContext,
             };
         });
         const transitionTo = transitionFrom(this.clark);
@@ -404,7 +456,7 @@ export class AiService extends TracedEventEmitter {
              * Tools expect passing a Clark instance to handle shared state (e.g: search for `requiredFileReads` and its usage)
              * We don't need it here since we only use a single read tool.
              */
-            const build_readFiles = buildReadFilesToolFactory.create({
+            const build_readFile = buildReadFileToolFactory.create({
                 clark: this.clark,
                 services: {
                     appShell: this.appShell,
@@ -425,7 +477,7 @@ export class AiService extends TracedEventEmitter {
                     code,
                     pageContent,
                     finalizeToolName: explainCodeFinalize.name,
-                    readFilesToolName: build_readFiles.name,
+                    readFileToolName: build_readFile.name,
                 });
             }
             else {
@@ -458,7 +510,7 @@ export class AiService extends TracedEventEmitter {
                         model,
                         stopWhen: hasToolCall(explainCodeFinalize.name),
                         tools: {
-                            build_readFiles,
+                            build_readFile,
                             explainCodeFinalize,
                         },
                         user: { role: "user", content: prompt },
@@ -513,6 +565,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");
@@ -530,12 +686,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,
             });
@@ -544,12 +702,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,
             });
@@ -567,8 +727,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`);
@@ -585,6 +779,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}`);
@@ -612,7 +807,6 @@ export class AiService extends TracedEventEmitter {
                 formattedInput: request.formattedInput ?? undefined,
                 rawInput: request.rawInput,
                 permissions: request.permissions ?? [],
-                toolCallId: request.toolCallId,
             })
                 .catch((error) => {
                 this.getLogger().error(`[ai-service] Failed to resend tool permission request: ${error}`);
@@ -621,18 +815,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,
@@ -871,18 +1066,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();
@@ -908,6 +1144,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.