stk-codegen 1.0.7 → 1.0.9

package/dist/services/websocketClient.d.ts

@@ -1,17 +1,24 @@
 import { BackendMessage, ClientMessage } from '../types/index.js';
 type MessageHandler = (data: BackendMessage) => void;
-declare class WebSocketClient {
+type StatusChangeHandler = (status: 'connecting' | 'connected' | 'disconnected') => void;
+export declare class WebSocketClient {
     private ws;
     private messageHandler;
+    private onStatusChange;
     private sessionId;
     private handshakeSent;
     private isConnecting;
     private connectionPromise;
+    private messageQueue;
+    private isReadyForUserInput;
+    private status;
+    private setStatus;
     private createSession;
     private setupWebSocketConnection;
+    private flushMessageQueue;
     ensureConnected(): Promise<void>;
-    connect(messageHandler: MessageHandler): Promise<void>;
-    send(data: ClientMessage): void;
+    connect(messageHandler: MessageHandler, onStatusChange: StatusChangeHandler): Promise<void>;
+    send(data: ClientMessage): Promise<void>;
 }
 export declare const websocketClient: WebSocketClient;
 export {};

package/dist/tools/SearchText.d.ts

@@ -1,18 +1,8 @@
-export declare const SearchText: (parameters: {
-    pattern: string;
-    dir_path?: string;
-    include?: string;
-    case_sensitive?: boolean;
-    fixed_strings?: boolean;
-    context?: number;
-}) => Promise<{
+import { type SearchTextParams } from './searchTextFallback.js';
+export declare const SearchText: (parameters: SearchTextParams) => Promise<{
     matches: ({
         file: string;
         line: any;
         content: any;
     } | null)[];
-    error?: undefined;
-} | {
-    error: any;
-    matches?: undefined;
 }>;

package/dist/tools/searchTextFallback.d.ts

@@ -0,0 +1,20 @@
+export type SearchTextParams = {
+    pattern: string;
+    dir_path?: string;
+    include?: string;
+    case_sensitive?: boolean;
+    fixed_strings?: boolean;
+    context?: number;
+};
+export type SearchTextMatch = {
+    file: string;
+    line: number;
+    content: string;
+};
+/**
+ * Node-based SearchText fallback that avoids external binaries.
+ * Behavior is best-effort compatible with rg: returns line matches and line text.
+ */
+export declare const searchTextFallback: (params: SearchTextParams) => Promise<{
+    matches: SearchTextMatch[];
+}>;

package/dist/types/index.d.ts

@@ -9,6 +9,10 @@ export interface ConversationMessage {
         status: 'streaming' | 'finished';
         content?: string;
     };
+    highlightedBlock?: {
+        language?: string;
+        content: string;
+    };
 }
 export interface PendingShellCommand {
     tool_name: string;
@@ -33,21 +37,24 @@ export interface ToolResult {
     payload: {
         tool_name: string;
         tool_id: string;
-        generation_id: string;
+        generation_id: string | null;
         status: 'success' | 'error';
-        result: any;
+        result: any | null;
+        error?: string | null;
     };
 }
 export interface ClientCancelGeneration {
     type: 'client_cancel_generation';
     payload: {
         generation_id: string;
+        reason?: string;
     };
 }
 export interface AgentGenerationCancelled {
     type: 'agent_generation_cancelled';
     payload: {
         generation_id: string;
+        reason?: string | null;
     };
 }
 export interface UserMessage {
@@ -85,6 +92,7 @@ export interface HandshakeMessage {
             content: string;
             filePaths: string[];
         } | null;
+        initial_prompt?: string;
     };
 }
 export interface ClientCancelLast {
@@ -95,20 +103,38 @@ export type ClientMessage = UserMessage | MagicCommand | ClientCancelGeneration
 export type BackendMessage = RunTool | AgentGenerationCancelled | {
     type: 'agent_error';
     payload: {
+        generation_id?: string;
         message: string;
         details?: string;
     };
 } | {
     type: 'agent_message_delta';
     payload: {
+        generation_id?: string;
+        content: string;
+    };
+} | {
+    type: 'agent_message';
+    payload: {
+        generation_id?: string;
         content: string;
     };
 } | {
     type: 'agent_thought';
     payload: {
+        generation_id?: string;
         content: string;
     };
 } | {
     type: 'agent_finished';
+    payload: {
+        generation_id?: string;
+        content?: string;
+    };
+} | {
+    type: 'handshake_ack';
+    payload: Record<string, never>;
+} | {
+    type: 'bootstrap_finished';
     payload: Record<string, never>;
 };

package/dist/ui/agent/buildInkViewModel.d.ts

@@ -7,6 +7,10 @@ export interface StaticMessageView {
     toolParams?: Record<string, any>;
     status?: 'running' | 'finished' | 'error';
     isBootstrap?: boolean;
+    highlightedBlock?: {
+        language?: string;
+        content: string;
+    };
 }
 export interface ShellRunView {
     id: string;

package/dist/ui/agent/textFormatting.d.ts

@@ -0,0 +1,8 @@
+export type InlineSegment = {
+    type: 'text';
+    content: string;
+} | {
+    type: 'code';
+    content: string;
+};
+export declare function parseInlineCodeSegments(text: string): InlineSegment[];

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "stk-codegen",
-  "version": "1.0.7",
+  "version": "1.0.9",
   "description": "A CLI executor for the autonomous agent.",
   "main": "index.js",
   "bin": {