update

ts/classes.aidoc.ts

@@ -8,7 +8,7 @@ export class AiDoc {
   public npmextraKV: plugins.npmextra.KeyValueStore;
   public qenvInstance: plugins.qenv.Qenv;
   public aidocInteract: plugins.smartinteract.SmartInteract;
-  public openaiInstance: plugins.smartai.OpenAiProvider;
+  public smartAiInstance: plugins.smartai.SmartAi;
 
   argvArg: any;
 
@@ -85,20 +85,28 @@ export class AiDoc {
     }
 
     // lets assume we have an OPENAI_Token now
-    this.openaiInstance = new plugins.smartai.OpenAiProvider({
+    this.smartAiInstance = new plugins.smartai.SmartAi({
       openaiToken: this.openaiToken,
     });
-    await this.openaiInstance.start();
+    await this.smartAiInstance.start();
   }
 
   public async stop() {
-    if (this.openaiInstance) {
-      await this.openaiInstance.stop();
+    if (this.smartAiInstance) {
+      await this.smartAiInstance.stop();
     }
     // No explicit cleanup needed for npmextraKV or aidocInteract
     // They don't keep event loop alive
   }
 
+  /**
+   * Get the OpenAI provider for direct chat calls
+   * This is a convenience getter to access the provider from SmartAi
+   */
+  public get openaiProvider(): plugins.smartai.OpenAiProvider {
+    return this.smartAiInstance.openaiProvider;
+  }
+
   public getOpenaiToken(): string {
     return this.openaiToken;
   }
@@ -146,13 +154,12 @@ export class AiDoc {
   }
   
   /**
-   * Count tokens in a text string using GPT tokenizer
-   * @param text The text to count tokens for
-   * @param model The model to use for tokenization (default: gpt-3.5-turbo)
-   * @returns The number of tokens in the text
+   * Estimate token count in a text string
+   * @param text The text to estimate tokens for
+   * @returns Estimated number of tokens
    */
-  public countTokens(text: string, model: string = 'gpt-3.5-turbo'): number {
+  public countTokens(text: string): number {
     const projectContextInstance = new aiDocsClasses.ProjectContext('');
-    return projectContextInstance.countTokens(text, model);
+    return projectContextInstance.countTokens(text);
   }
 }