feat: Refactor AI chat processing models and types for improved structure and clarity

Author: Marvin Zhang

packages/ai/src/__tests__/models.test.ts

@@ -3,7 +3,7 @@
  */
 
 import { describe, it, expect } from 'vitest';
-import { ChatMessageData, ChatSessionData, WorkspaceDataContainer } from '../models/index.js';
+import { ChatMessageData, ChatSessionData, WorkspaceDataContainer } from '../types/index.js';
 
 describe('MessageData', () => {
   it('should create a message with required fields', () => {
@@ -78,11 +78,11 @@ describe('ChatSessionData', () => {
     const session = new ChatSessionData({
       agent: 'GitHub Copilot',
       messages,
-      session_id: 'session-1',
+      id: 'session-1',
     });
 
     expect(session.messages).toHaveLength(2);
-    expect(session.session_id).toBe('session-1');
+    expect(session.id).toBe('session-1');
   });
 });
 
@@ -99,8 +99,8 @@ describe('WorkspaceDataContainer', () => {
 
   it('should handle chat sessions correctly', () => {
     const sessions = [
-      new ChatSessionData({ agent: 'GitHub Copilot', session_id: 'session-1' }),
-      new ChatSessionData({ agent: 'GitHub Copilot', session_id: 'session-2' }),
+      new ChatSessionData({ agent: 'GitHub Copilot', id: 'session-1' }),
+      new ChatSessionData({ agent: 'GitHub Copilot', id: 'session-2' }),
     ];
 
     const workspace = new WorkspaceDataContainer({

packages/ai/src/index.ts

@@ -4,8 +4,8 @@
  * Main entry point for the TypeScript implementation
  */
 
-// Export all models
-export * from './models/index.js';
+// Export all types
+export * from './types/index.js';
 
 // Export all parsers
 export * from './parsers/index.js';

packages/ai/src/models/application.ts

@@ -3,11 +3,11 @@
  *
  * Provides a common interface for parsing chat history from various AI coding assistants
  * like GitHub Copilot, Cursor, Claude Code, etc.
- * 
+ *
  * Supports three-level hierarchy: Application -> Workspace -> Session
  */
 
-import type { ChatSession, ApplicationInfo, WorkspaceInfo } from '../models/index.js';
+import type { ChatSession, Application, Workspace, ParserType } from '../types/index.js';
 import { Logger, ConsoleLogger } from './utils.js';
 
 /**
@@ -20,17 +20,23 @@ export abstract class BaseParser {
     this.logger = logger || new ConsoleLogger();
   }
 
+  /**
+   * Get the parser type (e.g. 'github-copilot', 'cursor', etc.)
+   * Used for metadata and identification
+   */
+  abstract getParserType(): ParserType;
+
   /**
    * Get all available applications (VS Code installations) for this AI assistant
    * Returns lightweight application info without workspaces or sessions
    */
-  abstract getApplications(): Promise<ApplicationInfo[]>;
+  abstract getApplications(): Promise<Application[]>;
 
   /**
    * Get all workspaces from a specific application
    * Returns lightweight workspace info without sessions
    */
-  abstract getWorkspaces(applicationId: string): Promise<WorkspaceInfo[]>;
+  abstract getWorkspaces(applicationId: string): Promise<Workspace[]>;
 
   /**
    * Get all chat sessions from a specific workspace within an application
@@ -40,5 +46,9 @@ export abstract class BaseParser {
   /**
    * Parse a single chat session by session ID
    */
-  protected abstract parseChatSession(applicationId: string, workspaceId: string, sessionId: string): Promise<ChatSession | null>;
+  protected abstract parseChatSession(
+    applicationId: string,
+    workspaceId: string,
+    sessionId: string,
+  ): Promise<ChatSession | null>;
 }