@theia/ai-core 1.55.0-next.37 → 1.55.0

package/src/common/agent-service.ts

@@ -13,10 +13,11 @@
 //
 // SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
 // *****************************************************************************
-import { inject, injectable, named, optional, postConstruct } from '@theia/core/shared/inversify';
-import { ContributionProvider } from '@theia/core';
+import { inject, injectable, optional, postConstruct } from '@theia/core/shared/inversify';
+import { Emitter, Event } from '@theia/core';
 import { Agent } from './agent';
 import { AISettingsService } from './settings-service';
+import { PromptService } from './prompt-service';
 
 export const AgentService = Symbol('AgentService');
 
@@ -48,21 +49,42 @@ export interface AgentService {
      * @return true if the agent is enabled, false otherwise.
      */
     isEnabled(agentId: string): boolean;
+
+    /**
+     * Allows to register an agent programmatically.
+     * @param agent the agent to register
+     */
+    registerAgent(agent: Agent): void;
+
+    /**
+     * Allows to unregister an agent programmatically.
+     * @param agentId the agent id to unregister
+     */
+    unregisterAgent(agentId: string): void;
+
+    /**
+     * Emitted when the list of agents changes.
+     * This can be used to update the UI when agents are added or removed.
+     */
+    onDidChangeAgents: Event<void>;
 }
 
 @injectable()
 export class AgentServiceImpl implements AgentService {
 
-    @inject(ContributionProvider) @named(Agent)
-    protected readonly agentsProvider: ContributionProvider<Agent>;
-
     @inject(AISettingsService) @optional()
     protected readonly aiSettingsService: AISettingsService | undefined;
 
+    @inject(PromptService)
+    protected readonly promptService: PromptService;
+
     protected disabledAgents = new Set<string>();
 
     protected _agents: Agent[] = [];
 
+    private readonly onDidChangeAgentsEmitter = new Emitter<void>();
+    readonly onDidChangeAgents = this.onDidChangeAgentsEmitter.event;
+
     @postConstruct()
     protected init(): void {
         this.aiSettingsService?.getSettings().then(settings => {
@@ -74,22 +96,29 @@ export class AgentServiceImpl implements AgentService {
         });
     }
 
-    private get agents(): Agent[] {
-        // We can't collect the contributions at @postConstruct because this will lead to a circular dependency
-        // with agents reusing the chat agent service (e.g. orchestrator) which in turn injects the agent service
-        return [...this.agentsProvider.getContributions(), ...this._agents];
-    }
-
     registerAgent(agent: Agent): void {
         this._agents.push(agent);
+        agent.promptTemplates.forEach(
+            template => this.promptService.storePrompt(template.id, template.template)
+        );
+        this.onDidChangeAgentsEmitter.fire();
+    }
+
+    unregisterAgent(agentId: string): void {
+        const agent = this._agents.find(a => a.id === agentId);
+        this._agents = this._agents.filter(a => a.id !== agentId);
+        this.onDidChangeAgentsEmitter.fire();
+        agent?.promptTemplates.forEach(
+            template => this.promptService.removePrompt(template.id)
+        );
     }
 
     getAgents(): Agent[] {
-        return this.agents.filter(agent => this.isEnabled(agent.id));
+        return this._agents.filter(agent => this.isEnabled(agent.id));
     }
 
     getAllAgents(): Agent[] {
-        return this.agents;
+        return this._agents;
     }
 
     enableAgent(agentId: string): void {

package/src/common/communication-recording-service.ts

@@ -42,6 +42,8 @@ export interface CommunicationRecordingService {
 
     getHistory(agentId: string): CommunicationHistory;
 
+    getSessionHistory(sessionId: string): CommunicationHistory;
+
     clearHistory(): void;
     readonly onStructuralChange: Event<void>;
 }

package/src/common/prompt-service.ts

@@ -58,17 +58,46 @@ export interface PromptService {
      */
     getPrompt(id: string, args?: { [key: string]: unknown }): Promise<ResolvedPromptTemplate | undefined>;
     /**
-     * Manually add a prompt to the list of prompts.
+     * Adds a prompt to the list of prompts.
      * @param id the id of the prompt
      * @param prompt the prompt template to store
      */
     storePrompt(id: string, prompt: string): void;
+    /**
+     * Removes a prompt from the list of prompts.
+     * @param id the id of the prompt
+     */
+    removePrompt(id: string): void;
     /**
      * Return all known prompts as a {@link PromptMap map}.
      */
     getAllPrompts(): PromptMap;
 }
 
+export interface CustomAgentDescription {
+    id: string;
+    name: string;
+    description: string;
+    prompt: string;
+    defaultLLM: string;
+}
+export namespace CustomAgentDescription {
+    export function is(entry: unknown): entry is CustomAgentDescription {
+        // eslint-disable-next-line no-null/no-null
+        return typeof entry === 'object' && entry !== null
+            && 'id' in entry && typeof entry.id === 'string'
+            && 'name' in entry && typeof entry.name === 'string'
+            && 'description' in entry && typeof entry.description === 'string'
+            && 'prompt' in entry
+            && typeof entry.prompt === 'string'
+            && 'defaultLLM' in entry
+            && typeof entry.defaultLLM === 'string';
+    }
+    export function equals(a: CustomAgentDescription, b: CustomAgentDescription): boolean {
+        return a.id === b.id && a.name === b.name && a.description === b.description && a.prompt === b.prompt && a.defaultLLM === b.defaultLLM;
+    }
+}
+
 export const PromptCustomizationService = Symbol('PromptCustomizationService');
 export interface PromptCustomizationService {
     /**
@@ -89,9 +118,9 @@ export interface PromptCustomizationService {
      * on the implementation. Implementation may for example decide to
      * open an editor, or request more information from the user, ...
      * @param id the template id.
-     * @param content optional content to customize the template.
+     * @param content optional default content to initialize the template
      */
-    editTemplate(id: string, content?: string): void;
+    editTemplate(id: string, defaultContent?: string): void;
 
     /**
      * Reset the template to its default value.
@@ -109,6 +138,22 @@ export interface PromptCustomizationService {
      * Event which is fired when the prompt template is changed.
      */
     readonly onDidChangePrompt: Event<string>;
+
+    /**
+     * Return all custom agents.
+     * @returns all custom agents
+     */
+    getCustomAgents(): Promise<CustomAgentDescription[]>;
+
+    /**
+     * Event which is fired when custom agents are modified.
+     */
+    readonly onDidChangeCustomAgents: Event<void>;
+
+    /**
+     * Open the custom agent yaml file.
+     */
+    openCustomAgentYaml(): void;
 }
 
 @injectable()
@@ -210,4 +255,7 @@ export class PromptServiceImpl implements PromptService {
     storePrompt(id: string, prompt: string): void {
         this._prompts[id] = { id, template: prompt };
     }
+    removePrompt(id: string): void {
+        delete this._prompts[id];
+    }
 }