@jupyterlite/ai 0.14.0 → 0.16.0

package/src/chat-model-handler.ts

@@ -1,16 +1,17 @@
 import { ActiveCellManager } from '@jupyter/chat';
-import { TranslationBundle } from '@jupyterlab/translation';
-import { AgentManagerFactory } from './agent';
+import { IDocumentManager } from '@jupyterlab/docmanager';
+import { IRenderMimeRegistry } from '@jupyterlab/rendermime';
+import { Contents } from '@jupyterlab/services';
+
 import { AIChatModel } from './chat-model';
-import { AISettingsModel } from './models/settings-model';
-import {
+import type {
+  IAgentManagerFactory,
+  IAISettingsModel,
   IChatModelHandler,
+  ICreateChatOptions,
   IProviderRegistry,
-  ITokenUsage,
   IToolRegistry
 } from './tokens';
-import { IDocumentManager } from '@jupyterlab/docmanager';
-import { IRenderMimeRegistry } from '@jupyterlab/rendermime';
 
 /**
  * The chat model handler.
@@ -24,14 +25,12 @@ export class ChatModelHandler implements IChatModelHandler {
     this._providerRegistry = options.providerRegistry;
     this._rmRegistry = options.rmRegistry;
     this._activeCellManager = options.activeCellManager;
-    this._trans = options.trans;
+    this._contentsManager = options.contentsManager;
   }
 
-  createModel(
-    name: string,
-    activeProvider: string,
-    tokenUsage?: ITokenUsage
-  ): AIChatModel {
+  createModel(options: ICreateChatOptions): AIChatModel {
+    const { name, activeProvider, tokenUsage, messages, autosave } = options;
+
     // Create Agent Manager first so it can be shared
     const agentManager = this._agentManagerFactory.createAgent({
       settingsModel: this._settingsModel,
@@ -49,8 +48,13 @@ export class ChatModelHandler implements IChatModelHandler {
       agentManager,
       activeCellManager: this._activeCellManager,
       documentManager: this._docManager,
-      trans: this._trans
+      contentsManager: this._contentsManager
+    });
+
+    messages?.forEach(message => {
+      model.messageAdded({ ...message.content });
     });
+    model.autosave = autosave ?? false;
 
     model.name = name;
 
@@ -68,13 +72,13 @@ export class ChatModelHandler implements IChatModelHandler {
   }
 
   private _docManager: IDocumentManager;
-  private _agentManagerFactory: AgentManagerFactory;
-  private _settingsModel: AISettingsModel;
+  private _agentManagerFactory: IAgentManagerFactory;
+  private _settingsModel: IAISettingsModel;
   private _toolRegistry?: IToolRegistry;
   private _providerRegistry?: IProviderRegistry;
   private _rmRegistry: IRenderMimeRegistry;
   private _activeCellManager?: ActiveCellManager;
-  private _trans: TranslationBundle;
+  private _contentsManager?: Contents.IManager;
 }
 
 export namespace ChatModelHandler {
@@ -86,11 +90,11 @@ export namespace ChatModelHandler {
     /**
      * The agent manager factory.
      */
-    agentManagerFactory: AgentManagerFactory;
+    agentManagerFactory: IAgentManagerFactory;
     /**
      * AI settings model for configuration
      */
-    settingsModel: AISettingsModel;
+    settingsModel: IAISettingsModel;
     /**
      * Optional tool registry for managing available tools
      */
@@ -108,8 +112,8 @@ export namespace ChatModelHandler {
      */
     activeCellManager?: ActiveCellManager | undefined;
     /**
-     * The application language translation bundle.
+     * The contents manager.
      */
-    trans: TranslationBundle;
+    contentsManager?: Contents.IManager;
   }
 }