@theia/ai-chat 1.56.0 → 1.57.0-next.112

package/src/common/chat-tool-request-service.ts

@@ -0,0 +1,59 @@
+// *****************************************************************************
+// Copyright (C) 2025 EclipseSource GmbH.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+
+import { ToolRequest } from '@theia/ai-core';
+import { injectable } from '@theia/core/shared/inversify';
+import { ChatRequestModelImpl } from './chat-model';
+
+export interface ChatToolRequest extends ToolRequest {
+    handler: (
+        arg_string: string,
+        context: ChatRequestModelImpl,
+    ) => Promise<unknown>;
+}
+
+/**
+ * Wraps tool requests in a chat context.
+ *
+ * This service extracts tool requests from a given chat request model and wraps their
+ * handler functions to provide additional context, such as the chat request model.
+ */
+@injectable()
+export class ChatToolRequestService {
+
+    getChatToolRequests(request: ChatRequestModelImpl): ChatToolRequest[] {
+        const toolRequests = request.message.toolRequests.size > 0 ? [...request.message.toolRequests.values()] : undefined;
+        if (!toolRequests) {
+            return [];
+        }
+        return this.toChatToolRequests(toolRequests, request);
+    }
+
+    toChatToolRequests(toolRequests: ToolRequest[] | undefined, request: ChatRequestModelImpl): ChatToolRequest[] {
+        if (!toolRequests) {
+            return [];
+        }
+        return toolRequests.map(toolRequest => this.toChatToolRequest(toolRequest, request));
+    }
+
+    protected toChatToolRequest(toolRequest: ToolRequest, request: ChatRequestModelImpl): ChatToolRequest {
+        return {
+            ...toolRequest,
+            handler: async (arg_string: string) => toolRequest.handler(arg_string, request)
+        };
+    }
+
+}

package/src/common/command-chat-agents.ts

@@ -315,7 +315,7 @@ export class CommandChatAgent extends AbstractTextToModelParsingChatAgent<Parsed
             const theiaCommand = this.commandRegistry.getCommand(parsedCommand.commandId);
             if (theiaCommand === undefined) {
                 console.error(`No Theia Command with id ${parsedCommand.commandId}`);
-                request.response.cancel();
+                request.cancel();
             }
             const args = parsedCommand.arguments !== undefined &&
                 parsedCommand.arguments.length > 0

package/lib/common/o1-chat-agent.d.ts

@@ -1,13 +0,0 @@
-import { ChatAgent, AbstractStreamParsingChatAgent, SystemMessageDescription } from './chat-agents';
-import { AgentSpecificVariables, PromptTemplate } from '@theia/ai-core';
-export declare class O1ChatAgent extends AbstractStreamParsingChatAgent implements ChatAgent {
-    name: string;
-    description: string;
-    promptTemplates: PromptTemplate[];
-    readonly agentSpecificVariables: AgentSpecificVariables[];
-    readonly variables: string[];
-    readonly functions: string[];
-    constructor();
-    protected getSystemMessageDescription(): Promise<SystemMessageDescription | undefined>;
-}
-//# sourceMappingURL=o1-chat-agent.d.ts.map

package/lib/common/o1-chat-agent.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"o1-chat-agent.d.ts","sourceRoot":"","sources":["../../src/common/o1-chat-agent.ts"],"names":[],"mappings":"AAgBA,OAAO,EACH,SAAS,EACT,8BAA8B,EAC9B,wBAAwB,EAC3B,MAAM,eAAe,CAAC;AAGvB,OAAO,EAAE,sBAAsB,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAExE,qBACa,WAAY,SAAQ,8BAA+B,YAAW,SAAS;IAEzE,IAAI,SAAgB;IACpB,WAAW,SAAsD;IACjE,eAAe,EAAE,cAAc,EAAE,CAAM;IAC9C,QAAQ,CAAC,sBAAsB,EAAE,sBAAsB,EAAE,CAAM;IAC/D,QAAQ,CAAC,SAAS,EAAE,MAAM,EAAE,CAAM;IAClC,QAAQ,CAAC,SAAS,EAAE,MAAM,EAAE,CAAM;;cAalB,2BAA2B,IAAI,OAAO,CAAC,wBAAwB,GAAG,SAAS,CAAC;CAI/F"}

package/lib/common/o1-chat-agent.js

@@ -1,45 +0,0 @@
-"use strict";
-// *****************************************************************************
-// Copyright (C) 2024 EclipseSource GmbH.
-//
-// This program and the accompanying materials are made available under the
-// terms of the Eclipse Public License v. 2.0 which is available at
-// http://www.eclipse.org/legal/epl-2.0.
-//
-// This Source Code may also be made available under the following Secondary
-// Licenses when the conditions for such availability set forth in the Eclipse
-// Public License v. 2.0 are satisfied: GNU General Public License, version 2
-// with the GNU Classpath Exception which is available at
-// https://www.gnu.org/software/classpath/license.html.
-//
-// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
-// *****************************************************************************
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.O1ChatAgent = void 0;
-const tslib_1 = require("tslib");
-const chat_agents_1 = require("./chat-agents");
-const inversify_1 = require("@theia/core/shared/inversify");
-let O1ChatAgent = class O1ChatAgent extends chat_agents_1.AbstractStreamParsingChatAgent {
-    constructor() {
-        super('o1-preview', [{
-                purpose: 'chat',
-                identifier: 'openai/o1-preview',
-            }], 'chat');
-        this.name = 'O1-Preview';
-        this.description = 'An agent for interacting with ChatGPT o1-preview';
-        this.promptTemplates = [];
-        this.agentSpecificVariables = [];
-        this.variables = [];
-        this.functions = [];
-    }
-    async getSystemMessageDescription() {
-        // O1 currently does not support system prompts
-        return undefined;
-    }
-};
-exports.O1ChatAgent = O1ChatAgent;
-exports.O1ChatAgent = O1ChatAgent = tslib_1.__decorate([
-    (0, inversify_1.injectable)(),
-    tslib_1.__metadata("design:paramtypes", [])
-], O1ChatAgent);
-//# sourceMappingURL=o1-chat-agent.js.map

package/lib/common/o1-chat-agent.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"o1-chat-agent.js","sourceRoot":"","sources":["../../src/common/o1-chat-agent.ts"],"names":[],"mappings":";AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;;;;AAEhF,+CAIuB;AAEvB,4DAA0D;AAInD,IAAM,WAAW,GAAjB,MAAM,WAAY,SAAQ,4CAA8B;IAS3D;QACI,KAAK,CACD,YAAY,EACZ,CAAC;gBACG,OAAO,EAAE,MAAM;gBACf,UAAU,EAAE,mBAAmB;aAClC,CAAC,EACF,MAAM,CACT,CAAC;QAfC,SAAI,GAAG,YAAY,CAAC;QACpB,gBAAW,GAAG,kDAAkD,CAAC;QACjE,oBAAe,GAAqB,EAAE,CAAC;QACrC,2BAAsB,GAA6B,EAAE,CAAC;QACtD,cAAS,GAAa,EAAE,CAAC;QACzB,cAAS,GAAa,EAAE,CAAC;IAWlC,CAAC;IAES,KAAK,CAAC,2BAA2B;QACvC,+CAA+C;QAC/C,OAAO,SAAS,CAAC;IACrB,CAAC;CACJ,CAAA;AAxBY,kCAAW;sBAAX,WAAW;IADvB,IAAA,sBAAU,GAAE;;GACA,WAAW,CAwBvB"}

package/src/common/o1-chat-agent.ts

@@ -1,51 +0,0 @@
-// *****************************************************************************
-// Copyright (C) 2024 EclipseSource GmbH.
-//
-// This program and the accompanying materials are made available under the
-// terms of the Eclipse Public License v. 2.0 which is available at
-// http://www.eclipse.org/legal/epl-2.0.
-//
-// This Source Code may also be made available under the following Secondary
-// Licenses when the conditions for such availability set forth in the Eclipse
-// Public License v. 2.0 are satisfied: GNU General Public License, version 2
-// with the GNU Classpath Exception which is available at
-// https://www.gnu.org/software/classpath/license.html.
-//
-// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
-// *****************************************************************************
-
-import {
-    ChatAgent,
-    AbstractStreamParsingChatAgent,
-    SystemMessageDescription
-} from './chat-agents';
-
-import { injectable } from '@theia/core/shared/inversify';
-import { AgentSpecificVariables, PromptTemplate } from '@theia/ai-core';
-
-@injectable()
-export class O1ChatAgent extends AbstractStreamParsingChatAgent implements ChatAgent {
-
-    public name = 'O1-Preview';
-    public description = 'An agent for interacting with ChatGPT o1-preview';
-    public promptTemplates: PromptTemplate[] = [];
-    readonly agentSpecificVariables: AgentSpecificVariables[] = [];
-    readonly variables: string[] = [];
-    readonly functions: string[] = [];
-
-    constructor() {
-        super(
-            'o1-preview',
-            [{
-                purpose: 'chat',
-                identifier: 'openai/o1-preview',
-            }],
-            'chat'
-        );
-    }
-
-    protected async getSystemMessageDescription(): Promise<SystemMessageDescription | undefined> {
-        // O1 currently does not support system prompts
-        return undefined;
-    }
-}