GH-40 Initial Agent View and Communication History

Co-authored-by: Olaf Lessenich <[email protected]>

Author: ndoschek

examples/browser-only/package.json

@@ -18,6 +18,8 @@
     "@theia/ai-chat": "1.49.0",
     "@theia/ai-chat-ui": "1.49.0",
     "@theia/ai-core": "1.49.0",
+    "@theia/ai-history": "1.49.0",
+    "@theia/ai-history-ui": "1.49.0",
     "@theia/ai-openai": "1.49.0",
     "@theia/api-samples": "1.49.0",
     "@theia/bulk-edit": "1.49.0",

examples/browser-only/tsconfig.json

@@ -17,6 +17,12 @@
     {
       "path": "../../packages/ai-core"
     },
+    {
+      "path": "../../packages/ai-history"
+    },
+    {
+      "path": "../../packages/ai-history-ui"
+    },
     {
       "path": "../../packages/ai-openai"
     },

examples/browser/package.json

@@ -23,6 +23,8 @@
     "@theia/ai-chat": "1.49.0",
     "@theia/ai-chat-ui": "1.49.0",
     "@theia/ai-core": "1.49.0",
+    "@theia/ai-history": "1.49.0",
+    "@theia/ai-history-ui": "1.49.0",
     "@theia/ai-openai": "1.49.0",
     "@theia/api-provider-sample": "1.49.0",
     "@theia/api-samples": "1.49.0",

examples/browser/tsconfig.json

@@ -17,6 +17,12 @@
     {
       "path": "../../packages/ai-core"
     },
+    {
+      "path": "../../packages/ai-history"
+    },
+    {
+      "path": "../../packages/ai-history-ui"
+    },
     {
       "path": "../../packages/ai-openai"
     },

examples/electron/package.json

@@ -29,6 +29,8 @@
     "@theia/ai-chat": "1.49.0",
     "@theia/ai-chat-ui": "1.49.0",
     "@theia/ai-core": "1.49.0",
+    "@theia/ai-history": "1.49.0",
+    "@theia/ai-history-ui": "1.49.0",
     "@theia/ai-openai": "1.49.0",
     "@theia/api-provider-sample": "1.49.0",
     "@theia/api-samples": "1.49.0",

examples/electron/tsconfig.json

@@ -20,6 +20,12 @@
     {
       "path": "../../packages/ai-core"
     },
+    {
+      "path": "../../packages/ai-history"
+    },
+    {
+      "path": "../../packages/ai-history-ui"
+    },
     {
       "path": "../../packages/ai-openai"
     },

packages/ai-chat/package.json

@@ -4,6 +4,7 @@
   "description": "Theia - AI Chat Extension",
   "dependencies": {
     "@theia/ai-core": "1.49.0",
+    "@theia/ai-history": "1.49.0",
     "@theia/core": "1.49.0",
     "@theia/filesystem": "1.49.0",
     "@theia/workspace": "1.49.0",

packages/ai-chat/src/browser/agent-frontend-module.ts

@@ -15,7 +15,7 @@
 // *****************************************************************************
 
 import { Agent } from '@theia/ai-core/lib/common';
-import { bindContributionProvider } from '@theia/core';
+import { bindContributionProvider, CommandContribution } from '@theia/core';
 import { ContainerModule } from '@theia/core/shared/inversify';
 import {
     ChatAgent,
@@ -27,8 +27,10 @@ import {
     ChatServiceImpl,
     ChatVariablesService,
     DefaultChatAgent,
+    DummyChatAgent,
     DummyChatVariablesService
 } from '../common';
+import { DummyCommandContribution } from './dummy-command-contribution';
 
 export default new ContainerModule(bind => {
     bindContributionProvider(bind, Agent);
@@ -49,4 +51,10 @@ export default new ContainerModule(bind => {
     bind(DefaultChatAgent).toSelf().inSingletonScope();
     bind(Agent).toService(DefaultChatAgent);
     bind(ChatAgent).toService(DefaultChatAgent);
+
+    bind(DummyChatAgent).toSelf().inSingletonScope();
+    bind(Agent).toService(DummyChatAgent);
+    bind(ChatAgent).toService(DummyChatAgent);
+
+    bind(CommandContribution).to(DummyCommandContribution);
 });

packages/ai-chat/src/browser/dummy-command-contribution.ts

@@ -0,0 +1,37 @@
+// *****************************************************************************
+// 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 { CommandContribution, CommandRegistry } from '@theia/core';
+import { inject, injectable } from '@theia/core/shared/inversify';
+import { ChatAgentService, DummyChatAgent } from '../common';
+
+export const TRIGGER_DUMMY_AGENT = ({
+    id: 'trigger.dummy.agent',
+    label: 'Trigger Dummy Agent',
+});
+
+@injectable()
+export class DummyCommandContribution implements CommandContribution {
+    @inject(ChatAgentService) agentService: ChatAgentService;
+
+    registerCommands(registry: CommandRegistry): void {
+        registry.registerCommand(TRIGGER_DUMMY_AGENT, {
+            execute: () => {
+                (this.agentService.getAgentsByName('Dummy Chat Agent')[0] as DummyChatAgent).invoke();
+            }
+        });
+    }
+}

packages/ai-chat/src/common/chat-agents.ts

@@ -19,6 +19,7 @@
  *--------------------------------------------------------------------------------------------*/
 // Partially copied from https://github.com/microsoft/vscode/blob/a2cab7255c0df424027be05d58e1b7b941f4ea60/src/vs/workbench/contrib/chat/common/chatAgents.ts
 
+import { CommunicationRecordingService } from '@theia/ai-core';
 import {
     Agent,
     isLanguageModelStreamResponse,
@@ -28,7 +29,7 @@ import {
     LanguageModelStreamResponsePart,
     PromptTemplate
 } from '@theia/ai-core/lib/common';
-import { ILogger, isArray } from '@theia/core';
+import { generateUuid, ILogger, isArray } from '@theia/core';
 import { inject, injectable } from '@theia/core/shared/inversify';
 import { ChatRequestModelImpl, ChatResponseContent, MarkdownChatResponseContentImpl } from './chat-model';
 import { getMessages } from './chat-util';
@@ -67,6 +68,9 @@ export class DefaultChatAgent implements ChatAgent {
     @inject(ILogger)
     protected logger: ILogger;
 
+    @inject(CommunicationRecordingService)
+    protected recordingService: CommunicationRecordingService;
+
     id: string = 'DefaultChatAgent';
     name: string = 'Default Chat Agent';
     description: string = 'The default chat agent provided by Theia.';
@@ -95,6 +99,13 @@ export class DefaultChatAgent implements ChatAgent {
     locations: ChatAgentLocation[] = [];
 
     async invoke(request: ChatRequestModelImpl): Promise<void> {
+        this.recordingService.recordRequest({
+            agentId: this.id,
+            sessionId: request.session.id,
+            timestamp: Date.now(),
+            requestId: request.id,
+            request: request.request.text
+        });
         const selector = this.languageModelRequirements.find(req => req.purpose === 'chat')!;
         const languageModels = await this.languageModelRegistry.selectLanguageModels({ agent: this.id, ...selector });
         if (languageModels.length === 0) {
@@ -106,6 +117,13 @@ export class DefaultChatAgent implements ChatAgent {
                 new MarkdownChatResponseContentImpl(languageModelResponse.text)
             );
             request.response.complete();
+            this.recordingService.recordResponse({
+                agentId: this.id,
+                sessionId: request.session.id,
+                timestamp: Date.now(),
+                requestId: request.response.requestId,
+                response: request.response.response.asString()
+            });
             return;
         }
         if (isLanguageModelStreamResponse(languageModelResponse)) {
@@ -118,6 +136,13 @@ export class DefaultChatAgent implements ChatAgent {
                 }
             }
             request.response.complete();
+            this.recordingService.recordResponse({
+                agentId: this.id,
+                sessionId: request.session.id,
+                timestamp: Date.now(),
+                requestId: request.response.requestId,
+                response: request.response.response.asString()
+            });
             return;
         }
         this.logger.error(
@@ -129,6 +154,13 @@ export class DefaultChatAgent implements ChatAgent {
             )
         );
         request.response.complete();
+        this.recordingService.recordResponse({
+            agentId: this.id,
+            sessionId: request.session.id,
+            timestamp: Date.now(),
+            requestId: request.response.requestId,
+            response: request.response.response.asString()
+        });
     }
 
     private parse(token: LanguageModelStreamResponsePart, previousContent: ChatResponseContent[]): ChatResponseContent | ChatResponseContent[] {
@@ -141,3 +173,37 @@ export class DefaultChatAgent implements ChatAgent {
         return new MarkdownChatResponseContentImpl(token.content ?? '');
     }
 }
+
+@injectable()
+export class DummyChatAgent implements ChatAgent {
+
+    @inject(CommunicationRecordingService)
+    protected recordingService: CommunicationRecordingService;
+
+    id: string = 'DummyChatAgent';
+    name: string = 'Dummy Chat Agent';
+    description: string = 'The dummy chat agent provided by ES.';
+    variables: string[] = [];
+    promptTemplates: PromptTemplate[] = [];
+    languageModelRequirements: Omit<LanguageModelSelector, 'agentId'>[] = [];
+    locations: ChatAgentLocation[] = [];
+
+    async invoke(request?: ChatRequestModelImpl): Promise<void> {
+        const requestUuid = generateUuid();
+        const sessionId = 'dummy-session';
+        this.recordingService.recordRequest({
+            agentId: this.id,
+            sessionId: sessionId,
+            timestamp: Date.now(),
+            requestId: requestUuid,
+            request: 'Dummy request'
+        });
+        this.recordingService.recordResponse({
+            agentId: this.id,
+            sessionId: sessionId,
+            timestamp: Date.now(),
+            requestId: requestUuid,
+            response: 'Dummy response'
+        });
+    }
+}