genaicode 0.11.1 → 0.12.1

package/dist/main/knowledge/knowledge-store.js

@@ -0,0 +1,128 @@
+import { readCache, writeCache } from '../../files/cache-file.js';
+import { randomUUID } from 'crypto';
+import { ModelType, } from '../../ai-service/common-types.js';
+const KNOWLEDGE_CACHE_KEY = 'container-task-knowledge-v1';
+// Function definition for providing context to the LLM.
+export const knowledgeContextDef = {
+    name: 'provideKnowledgeContext',
+    description: 'Provides the user query and knowledge base corpus for ranking.',
+    parameters: {
+        type: 'object',
+        properties: {},
+        required: [],
+    },
+};
+// Function definition for the LLM to call to return ranked results.
+const respondKnowledgeQueryDef = {
+    name: 'respondKnowledgeQuery',
+    description: 'Returns a result of the knowledge query.',
+    parameters: {
+        type: 'object',
+        properties: {
+            synthesizedResponse: {
+                type: 'string',
+                description: 'The synthesized response generated by the LLM based on the knowledge query.',
+            },
+            sourceEntries: {
+                type: 'array',
+                description: 'An array of knowledge base entries that are very relevant to the query.',
+                maxItems: 3,
+                items: {
+                    type: 'object',
+                    properties: {
+                        id: {
+                            type: 'string',
+                            description: 'The unique identifier of the knowledge base entry.',
+                        },
+                    },
+                    required: ['id'],
+                },
+            },
+        },
+        required: ['synthesizedResponse', 'sourceEntries'],
+    },
+};
+export async function getKnowledgeBase() {
+    const entries = await readCache(KNOWLEDGE_CACHE_KEY, []);
+    return entries;
+}
+async function writeKnowledgeBase(entries) {
+    await writeCache(KNOWLEDGE_CACHE_KEY, entries);
+}
+export async function appendKnowledge(entryData) {
+    const knowledgeBase = await getKnowledgeBase();
+    const newEntry = {
+        id: randomUUID(),
+        createdAt: new Date().toISOString(),
+        ...entryData,
+    };
+    knowledgeBase.push(newEntry);
+    await writeKnowledgeBase(knowledgeBase);
+    return newEntry;
+}
+export async function queryKnowledge(args, generateContentFn) {
+    const { query } = args;
+    const knowledgeBase = await getKnowledgeBase();
+    if (knowledgeBase.length === 0) {
+        return {};
+    }
+    const corpus = knowledgeBase.map((entry) => ({
+        id: entry.id,
+        prompt: entry.prompt,
+        answer: entry.answer,
+    }));
+    const systemPrompt = `
+    You are an AI assistant that ranks knowledge base entries based on relevance to a user query.
+    You will be given the context (query and corpus) as a function response to 'respondKnowledgeQuery'.
+    Your task is to analyze this context and call the 'respondKnowledgeQuery' function with the results.
+  `;
+    const prompt = [
+        {
+            type: 'systemPrompt',
+            systemPrompt,
+        },
+        {
+            type: 'user',
+            text: 'Hello!',
+        },
+        {
+            type: 'assistant',
+            text: 'Please provide me with context: corpus, query',
+            functionCalls: [
+                {
+                    name: 'provideKnowledgeContext',
+                },
+            ],
+        },
+        {
+            type: 'user',
+            text: `Query: ${query}`,
+            functionResponses: [
+                {
+                    name: 'provideKnowledgeContext',
+                    content: JSON.stringify({ query, corpus }),
+                },
+            ],
+        },
+    ];
+    const response = await generateContentFn(prompt, {
+        modelType: ModelType.LITE,
+        temperature: 0.1,
+        functionDefs: [knowledgeContextDef, respondKnowledgeQueryDef],
+        requiredFunctionName: 'respondKnowledgeQuery',
+        expectedResponseType: {
+            text: false,
+            functionCall: true,
+        },
+    }, { disableCache: true });
+    const functionCall = response.find((part) => part.type === 'functionCall')?.functionCall;
+    const sourceEntries = [];
+    for (const id of functionCall?.args?.sourceEntries?.map((e) => e.id) || []) {
+        const entry = knowledgeBase.find((e) => e.id === id);
+        if (entry) {
+            sourceEntries.push(entry);
+        }
+    }
+    return { synthesizedResponse: functionCall?.args?.synthesizedResponse, sourceEntries };
+}
+//# sourceMappingURL=knowledge-store.js.map

package/dist/main/knowledge/knowledge-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"knowledge-store.js","sourceRoot":"","sources":["../../../src/main/knowledge/knowledge-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAClE,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAEpC,OAAO,EAEL,SAAS,GAIV,MAAM,kCAAkC,CAAC;AAE1C,MAAM,mBAAmB,GAAG,6BAA6B,CAAC;AAE1D,wDAAwD;AACxD,MAAM,CAAC,MAAM,mBAAmB,GAAgB;IAC9C,IAAI,EAAE,yBAAyB;IAC/B,WAAW,EAAE,gEAAgE;IAC7E,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE,EAAE;QACd,QAAQ,EAAE,EAAE;KACb;CACF,CAAC;AAEF,oEAAoE;AACpE,MAAM,wBAAwB,GAAgB;IAC5C,IAAI,EAAE,uBAAuB;IAC7B,WAAW,EAAE,0CAA0C;IACvD,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,mBAAmB,EAAE;gBACnB,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,6EAA6E;aAC3F;YACD,aAAa,EAAE;gBACb,IAAI,EAAE,OAAO;gBACb,WAAW,EAAE,yEAAyE;gBACtF,QAAQ,EAAE,CAAC;gBACX,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,EAAE,EAAE;4BACF,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,oDAAoD;yBAClE;qBACF;oBACD,QAAQ,EAAE,CAAC,IAAI,CAAC;iBACjB;aACF;SACF;QACD,QAAQ,EAAE,CAAC,qBAAqB,EAAE,eAAe,CAAC;KACnD;CACF,CAAC;AAEF,MAAM,CAAC,KAAK,UAAU,gBAAgB;IACpC,MAAM,OAAO,GAAG,MAAM,SAAS,CAAY,mBAAmB,EAAE,EAAE,CAAC,CAAC;IACpE,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,KAAK,UAAU,kBAAkB,CAAC,OAAkB;IAClD,MAAM,UAAU,CAAC,mBAAmB,EAAE,OAAO,CAAC,CAAC;AACjD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,SAA4B;IAChE,MAAM,aAAa,GAAG,MAAM,gBAAgB,EAAE,CAAC;IAC/C,MAAM,QAAQ,GAAY;QACxB,EAAE,EAAE,UAAU,EAAE;QAChB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACnC,GAAG,SAAS;KACb,CAAC;IACF,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC7B,MAAM,kBAAkB,CAAC,aAAa,CAAC,CAAC;IACxC,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,IAAwB,EACxB,iBAA0C;IAE1C,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC;IACvB,MAAM,aAAa,GAAG,MAAM,gBAAgB,EAAE,CAAC;IAE/C,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,MAAM,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC3C,EAAE,EAAE,KAAK,CAAC,EAAE;QACZ,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,MAAM,EAAE,KAAK,CAAC,MAAM;KACrB,CAAC,CAAC,CAAC;IAEJ,MAAM,YAAY,GAAG;;;;GAIpB,CAAC;IAEF,MAAM,MAAM,GAAiB;QAC3B;YACE,IAAI,EAAE,cAAc;YACpB,YAAY;SACb;QACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,QAAQ;SACf;QACD;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,+CAA+C;YACrD,aAAa,EAAE;gBACb;oBACE,IAAI,EAAE,yBAAyB;iBAChC;aACF;SACF;QACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,UAAU,KAAK,EAAE;YACvB,iBAAiB,EAAE;gBACjB;oBACE,IAAI,EAAE,yBAAyB;oBAC/B,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;iBAC3C;aACF;SACF;KACF,CAAC;IAEF,MAAM,QAAQ,GAAG,MAAM,iBAAiB,CACtC,MAAM,EACN;QACE,SAAS,EAAE,SAAS,CAAC,IAAI;QACzB,WAAW,EAAE,GAAG;QAChB,YAAY,EAAE,CAAC,mBAAmB,EAAE,wBAAwB,CAAC;QAC7D,oBAAoB,EAAE,uBAAuB;QAC7C,oBAAoB,EAAE;YACpB,IAAI,EAAE,KAAK;YACX,YAAY,EAAE,IAAI;SACnB;KACF,EACD,EAAE,YAAY,EAAE,IAAI,EAAE,CACvB,CAAC;IAEF,MAAM,YAAY,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAC,EAAE,YAE/D,CAAC;IAEd,MAAM,aAAa,GAAc,EAAE,CAAC;IACpC,KAAK,MAAM,EAAE,IAAI,YAAY,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC;QAC3E,MAAM,KAAK,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;QACrD,IAAI,KAAK,EAAE,CAAC;YACV,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5B,CAAC;IACH,CAAC;IAED,OAAO,EAAE,mBAAmB,EAAE,YAAY,EAAE,IAAI,EAAE,mBAAmB,EAAE,aAAa,EAAE,CAAC;AACzF,CAAC"}

package/dist/main/knowledge/types.d.ts

@@ -0,0 +1,20 @@
+export interface KBEntry {
+    id: string;
+    createdAt: string;
+    prompt: string;
+    answer: string;
+    metadata?: Record<string, unknown>;
+}
+export type GainKnowledgeArgs = {
+    prompt: string;
+    answer: string;
+    metadata?: Record<string, unknown>;
+};
+export type QueryKnowledgeArgs = {
+    query: string;
+    explanation: string;
+};
+export interface QueryResult {
+    synthesizedResponse?: string;
+    sourceEntries?: KBEntry[];
+}

package/dist/main/knowledge/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/main/knowledge/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/main/knowledge/types.ts"],"names":[],"mappings":""}

package/dist/main/ui/backend/endpoints/interrupt-execution-endpoint.js

@@ -1,9 +1,23 @@
 import { registerEndpoint } from '../api-handlers.js';
+import { requestInterrupt } from '../../../../prompt/steps/step-ask-question/handlers/container-task/commands/interrupt-controller.js';
 registerEndpoint((router, service) => {
-    router.post('/interrupt-execution', async (_, res) => {
+    router.post('/interrupt-execution', async (req, res) => {
         try {
-            await service.interruptExecution();
-            res.json({ message: 'Execution interrupted successfully' });
+            const { scope } = req.body;
+            if (scope === 'container-command') {
+                const result = requestInterrupt();
+                if (result.accepted) {
+                    res.json({ message: 'Interrupt for container command requested successfully' });
+                }
+                else {
+                    res.status(409).json({ error: 'Interrupt already in progress or no command running' });
+                }
+            }
+            else {
+                // Fallback to interrupting the whole iteration
+                await service.interruptExecution();
+                res.json({ message: 'Execution interrupted successfully' });
+            }
         }
         catch (error) {
             console.error('Error interrupting execution:', error);

package/dist/main/ui/backend/endpoints/interrupt-execution-endpoint.js.map

@@ -1 +1 @@
-{"version":3,"file":"interrupt-execution-endpoint.js","sourceRoot":"","sources":["../../../../../src/main/ui/backend/endpoints/interrupt-execution-endpoint.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAEtD,gBAAgB,CAAC,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE;IACnC,MAAM,CAAC,IAAI,CAAC,sBAAsB,EAAE,KAAK,EAAE,CAAC,EAAE,GAAG,EAAE,EAAE;QACnD,IAAI,CAAC;YACH,MAAM,OAAO,CAAC,kBAAkB,EAAE,CAAC;YACnC,GAAG,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,oCAAoC,EAAE,CAAC,CAAC;QAC9D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;YACtD,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,gDAAgD,EAAE,CAAC,CAAC;QACpF,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
+{"version":3,"file":"interrupt-execution-endpoint.js","sourceRoot":"","sources":["../../../../../src/main/ui/backend/endpoints/interrupt-execution-endpoint.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAE,MAAM,qGAAqG,CAAC;AAEvI,gBAAgB,CAAC,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE;IACnC,MAAM,CAAC,IAAI,CAAC,sBAAsB,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE;QACrD,IAAI,CAAC;YACH,MAAM,EAAE,KAAK,EAAE,GAAG,GAAG,CAAC,IAAI,CAAC;YAE3B,IAAI,KAAK,KAAK,mBAAmB,EAAE,CAAC;gBAClC,MAAM,MAAM,GAAG,gBAAgB,EAAE,CAAC;gBAClC,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;oBACpB,GAAG,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,wDAAwD,EAAE,CAAC,CAAC;gBAClF,CAAC;qBAAM,CAAC;oBACN,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,qDAAqD,EAAE,CAAC,CAAC;gBACzF,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,+CAA+C;gBAC/C,MAAM,OAAO,CAAC,kBAAkB,EAAE,CAAC;gBACnC,GAAG,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,oCAAoC,EAAE,CAAC,CAAC;YAC9D,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;YACtD,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,gDAAgD,EAAE,CAAC,CAAC;QACpF,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/main/ui/common/api-types.d.ts

@@ -10,10 +10,11 @@ export interface CodegenResult {
     message?: string;
 }
 export type ConfirmationProps = {
-    includeAnswer: boolean;
-    confirmLabel: string;
-    declineLabel: string;
-    defaultValue: boolean;
+    includeAnswer?: boolean;
+    confirmLabel?: string;
+    declineLabel?: string;
+    defaultValue?: boolean;
+    secret?: boolean;
 } | undefined;
 export interface Question {
     id: string;
@@ -73,6 +74,68 @@ export interface ServiceConfigUpdate<T extends AiServiceType = AiServiceType> {
     serviceType: T;
     config: ServiceConfig<T>;
 }
+export declare enum ChatMessageType {
+    USER = "user",
+    ASSISTANT = "assistant",
+    SYSTEM = "system"
+}
+export declare enum ChatMessageFlags {
+    CONVERSATION_SUMMARY = "conversation-summary",
+    MESSAGE_EDITABLE = "editable"
+}
+export interface ChatMessageImage {
+    base64url: string;
+    mediaType: 'image/jpeg' | 'image/png' | 'image/gif' | 'image/webp';
+    originalName?: string;
+}
+export interface ChatMessage {
+    id: string;
+    iterationId: string;
+    type: ChatMessageType;
+    flags?: ChatMessageFlags[];
+    content: string;
+    timestamp: string;
+    data?: Record<string, unknown>;
+    images?: ChatMessageImage[];
+}
+export type LogLevel = 'info' | 'warn' | 'error' | 'success' | 'debug';
+export interface TerminalEvent {
+    id: string;
+    iterationId: string;
+    level: LogLevel;
+    source: 'docker' | 'container-task' | 'command' | 'copy' | 'system';
+    text: string;
+    timestamp: string;
+    data?: Record<string, unknown>;
+}
+/**
+ * Represents a single content item from the backend, which can be
+ * either a chat message or a terminal event. This is the shape of items
+ * returned by the /api/content endpoint.
+ */
+export type ApiContentProps = {
+    message: ChatMessage;
+    terminalEvent?: never;
+    data?: unknown;
+} | {
+    message?: never;
+    terminalEvent: TerminalEvent;
+    data?: unknown;
+};
+export type ExecutionPlanStepState = 'pending' | 'in-progress' | 'completed' | 'failed' | 'skipped';
+export interface ExecutionPlanStep {
+    id: string;
+    description: string;
+    state?: ExecutionPlanStepState;
+    statusUpdate?: string;
+    dependencies?: string[];
+}
+export type ExecutionPlan = ExecutionPlanStep[];
+export type ExecutionPlanUpdate = {
+    id: string;
+    state: ExecutionPlanStepState;
+    statusUpdate?: string;
+};
 /**
  * Represents a node in the conversation graph.
  */

package/dist/main/ui/common/api-types.js

@@ -1,2 +1,13 @@
-export {};
+// --- Content Types ---
+export var ChatMessageType;
+(function (ChatMessageType) {
+    ChatMessageType["USER"] = "user";
+    ChatMessageType["ASSISTANT"] = "assistant";
+    ChatMessageType["SYSTEM"] = "system";
+})(ChatMessageType || (ChatMessageType = {}));
+export var ChatMessageFlags;
+(function (ChatMessageFlags) {
+    ChatMessageFlags["CONVERSATION_SUMMARY"] = "conversation-summary";
+    ChatMessageFlags["MESSAGE_EDITABLE"] = "editable";
+})(ChatMessageFlags || (ChatMessageFlags = {}));
 //# sourceMappingURL=api-types.js.map

package/dist/main/ui/common/api-types.js.map

@@ -1 +1 @@
-{"version":3,"file":"api-types.js","sourceRoot":"","sources":["../../../../src/main/ui/common/api-types.ts"],"names":[],"mappings":""}
+{"version":3,"file":"api-types.js","sourceRoot":"","sources":["../../../../src/main/ui/common/api-types.ts"],"names":[],"mappings":"AA0FA,wBAAwB;AAExB,MAAM,CAAN,IAAY,eAIX;AAJD,WAAY,eAAe;IACzB,gCAAa,CAAA;IACb,0CAAuB,CAAA;IACvB,oCAAiB,CAAA;AACnB,CAAC,EAJW,eAAe,KAAf,eAAe,QAI1B;AAED,MAAM,CAAN,IAAY,gBAGX;AAHD,WAAY,gBAAgB;IAC1B,iEAA6C,CAAA;IAC7C,iDAA6B,CAAA;AAC/B,CAAC,EAHW,gBAAgB,KAAhB,gBAAgB,QAG3B"}