agentstudio 0.1.0

package/src/types/index.ts

@@ -0,0 +1,248 @@
+// Agent configuration types
+export interface AgentTool {
+  name: string;
+  enabled: boolean;
+  permissions?: {
+    requireConfirmation?: boolean;
+    allowedPaths?: string[];
+    blockedPaths?: string[];
+  };
+}
+
+export interface AgentConfig {
+  id: string;
+  name: string;
+  description: string;
+  version: string;
+
+  // AI configuration
+  systemPrompt: string;
+  maxTurns: number;
+  permissionMode: 'default' | 'acceptEdits' | 'bypassPermissions' | 'plan';
+  model: string;
+
+  // Available tools
+  allowedTools: AgentTool[];
+
+  // UI configuration
+  ui: {
+    icon: string;
+    primaryColor: string;
+    headerTitle: string;
+    headerDescription: string;
+    welcomeMessage?: string; // Custom welcome message instead of title + description
+    componentType: 'slides' | 'chat' | 'documents' | 'code' | 'custom';
+    customComponent?: string; // Path to custom component
+  };
+
+  // File system integration
+  workingDirectory?: string;
+  dataDirectory?: string;
+  fileTypes?: string[]; // Supported file extensions
+
+  // Metadata
+  author: string;
+  homepage?: string;
+  tags: string[];
+  createdAt: string;
+  updatedAt: string;
+
+  // Enable/disable state
+  enabled: boolean;
+
+  // Project associations
+  projects?: string[]; // Array of project paths associated with this agent
+}
+
+export interface AgentSession {
+  id: string;
+  agentId: string;
+  title: string;
+  createdAt: number;
+  lastUpdated: number;
+  messages: AgentMessage[];
+  claudeVersionId?: string; // Claude version ID used for this session
+  customData?: Record<string, unknown>;
+}
+
+export interface AgentMessage {
+  id: string;
+  role: 'user' | 'assistant';
+  content: string;
+  timestamp: number;
+  messageParts?: MessagePart[];
+  agentId: string;
+}
+
+export interface MessagePart {
+  id: string;
+  type: 'text' | 'tool' | 'command' | 'compactSummary' | 'image';
+  content?: string;
+  toolData?: {
+    id: string;
+    toolName: string;
+    toolInput: Record<string, unknown>;
+    toolResult?: string;
+    isExecuting: boolean;
+    isError?: boolean;
+    claudeId?: string; // Claude's tool use ID for matching with results
+  };
+  imageData?: {
+    id: string;
+    data: string;
+    mediaType: string;
+    filename?: string;
+  };
+  order: number;
+  originalContent?: string; // For commands that need to preserve original content
+}
+
+// Built-in agent templates
+export const BUILTIN_AGENTS: Partial<AgentConfig>[] = [
+  {
+    id: 'ppt-editor',
+    name: 'PPT编辑助手',
+    description: '专门用于创建和编辑HTML演示文稿的AI助手',
+    systemPrompt: `You are an AI assistant specialized in helping users create and edit HTML presentations.
+You can help with:
+- Content creation and editing
+- Design suggestions
+- Structure improvements
+- HTML/CSS modifications
+- Presentation flow optimization
+- File operations for slide management
+
+The presentation uses HTML slides with embedded CSS styling. Each slide should be self-contained with a 1280x720 viewport.
+Slides are stored in the ../slides/ directory relative to the backend.
+
+Always provide helpful, specific suggestions and when possible, include code examples.
+Please respond in Chinese.`,
+    allowedTools: [
+      { name: 'Write', enabled: true },
+      { name: 'Read', enabled: true },
+      { name: 'Edit', enabled: true },
+      { name: 'Glob', enabled: true },
+      { name: 'MultiEdit', enabled: true },
+      { name: 'Bash', enabled: true }
+    ],
+    ui: {
+      icon: '🎯',
+      primaryColor: '#3B82F6',
+      headerTitle: 'AI PPT助手',
+      headerDescription: '与AI聊天来编辑你的演示文稿',
+      welcomeMessage: '你好！我是你的AI PPT助手，可以帮你创建、编辑和优化HTML演示文稿。有什么需要帮助的吗？',
+      componentType: 'slides'
+    },
+    workingDirectory: '../slides',
+    dataDirectory: '.ai-sessions',
+    fileTypes: ['.html', '.css', '.js'],
+    tags: ['presentation', 'html', 'css', 'slides'],
+    enabled: true
+  },
+  {
+    id: 'code-assistant',
+    name: '代码助手',
+    description: '通用代码开发和审查助手',
+    systemPrompt: `You are a professional software development assistant. You can help with:
+- Code review and optimization
+- Bug fixing and debugging
+- Architecture design
+- Best practices implementation
+- Documentation writing
+- Testing strategies
+
+You have access to file system operations and can directly modify code files.
+Always follow coding best practices and maintain clean, readable code.
+Please respond in Chinese.`,
+    allowedTools: [
+      { name: 'Write', enabled: true },
+      { name: 'Read', enabled: true },
+      { name: 'Edit', enabled: true },
+      { name: 'Glob', enabled: true },
+      { name: 'MultiEdit', enabled: true },
+      { name: 'Bash', enabled: true },
+      { name: 'Task', enabled: true }
+    ],
+    ui: {
+      icon: '💻',
+      primaryColor: '#10B981',
+      headerTitle: '代码助手',
+      headerDescription: '专业的软件开发和代码审查助手',
+      welcomeMessage: '你好！我是专业的代码助手，可以帮你进行代码开发、审查、调试和优化。请告诉我你想要解决的编程问题！',
+      componentType: 'code'
+    },
+    tags: ['coding', 'development', 'review', 'debugging'],
+    enabled: false
+  },
+  {
+    id: 'document-writer',
+    name: '文档助手',
+    description: '专注于文档创建和编辑的助手',
+    systemPrompt: `You are a professional document writing assistant. You can help with:
+- Creating and editing documentation
+- Technical writing
+- Content structuring
+- Markdown formatting
+- Research and information gathering
+- Proofreading and editing
+
+You work primarily with text files and markdown documents.
+Focus on clarity, accuracy, and professional presentation.
+Please respond in Chinese.`,
+    allowedTools: [
+      { name: 'Write', enabled: true },
+      { name: 'Read', enabled: true },
+      { name: 'Edit', enabled: true },
+      { name: 'Glob', enabled: true },
+      { name: 'WebFetch', enabled: true },
+      { name: 'WebSearch', enabled: true }
+    ],
+    ui: {
+      icon: '📝',
+      primaryColor: '#8B5CF6',
+      headerTitle: '文档助手',
+      headerDescription: '专业的文档创建和编辑助手',
+      welcomeMessage: '你好！我是文档助手，专门帮助你创建、编辑和优化各种文档。无论是技术文档还是普通文档，我都能为你提供专业建议！',
+      componentType: 'documents'
+    },
+    fileTypes: ['.md', '.txt', '.rst', '.adoc'],
+    tags: ['documentation', 'writing', 'markdown', 'content'],
+    enabled: true
+  },
+  {
+    id: 'general-chat',
+    name: '通用聊天助手',
+    description: '通用的AI聊天助手，适用于各种对话和咨询',
+    systemPrompt: `You are a general-purpose AI assistant. You can help with:
+- General questions and conversations
+- Problem-solving and brainstorming
+- Information and explanations
+- Creative tasks and writing
+- Analysis and research
+- File operations when needed
+
+You are helpful, harmless, and honest. Always strive to provide accurate and useful information.
+Please respond in Chinese unless the user specifically requests another language.`,
+    allowedTools: [
+      { name: 'Write', enabled: true },
+      { name: 'Read', enabled: true },
+      { name: 'Edit', enabled: true },
+      { name: 'Glob', enabled: true },
+      { name: 'MultiEdit', enabled: true },
+      { name: 'Bash', enabled: true },
+      { name: 'Task', enabled: true },
+      { name: 'WebFetch', enabled: true },
+      { name: 'WebSearch', enabled: true }
+    ],
+    ui: {
+      icon: '💬',
+      primaryColor: '#6366F1',
+      headerTitle: '通用聊天',
+      headerDescription: '与AI进行自由对话和咨询',
+      welcomeMessage: '你好！我是通用AI助手，可以帮你解答问题、进行对话、处理各种任务。有什么我可以帮助你的吗？',
+      componentType: 'chat'
+    },
+    tags: ['general', 'chat', 'conversation', 'assistant'],
+    enabled: true
+  }
+];

package/src/types/subagents.ts

@@ -0,0 +1,106 @@
+// Subagent configuration types
+export interface SubagentTool {
+  name: string;
+  enabled: boolean;
+  permissions?: {
+    requireConfirmation?: boolean;
+    allowedPaths?: string[];
+    blockedPaths?: string[];
+  };
+}
+
+export interface Subagent {
+  id: string;
+  name: string;
+  description: string;
+  version: string;
+
+  // AI configuration
+  systemPrompt: string;
+  maxTurns: number;
+  permissionMode: 'default' | 'acceptEdits' | 'bypassPermissions' | 'plan';
+  model: string;
+
+  // Available tools
+  allowedTools: SubagentTool[];
+
+  // UI configuration
+  ui: {
+    icon: string;
+    primaryColor: string;
+    headerTitle: string;
+    headerDescription: string;
+    welcomeMessage?: string;
+    componentType: 'slides' | 'chat' | 'documents' | 'code' | 'custom';
+    customComponent?: string;
+  };
+
+  // File system integration
+  workingDirectory?: string;
+  dataDirectory?: string;
+  fileTypes?: string[];
+
+  // Metadata
+  author: string;
+  homepage?: string;
+  tags: string[];
+  createdAt: string;
+  updatedAt: string;
+
+  // Enable/disable state
+  enabled: boolean;
+
+  // Project associations
+  projects?: string[];
+}
+
+export interface SubagentCreate {
+  name: string;
+  description: string;
+  systemPrompt: string;
+  model: string;
+  allowedTools: SubagentTool[];
+  ui: {
+    icon: string;
+    primaryColor: string;
+    headerTitle: string;
+    headerDescription: string;
+    welcomeMessage?: string;
+    componentType: 'slides' | 'chat' | 'documents' | 'code' | 'custom';
+    customComponent?: string;
+  };
+  workingDirectory?: string;
+  dataDirectory?: string;
+  fileTypes?: string[];
+  author: string;
+  tags: string[];
+}
+
+export interface SubagentUpdate {
+  name?: string;
+  description?: string;
+  systemPrompt?: string;
+  model?: string;
+  allowedTools?: SubagentTool[];
+  ui?: {
+    icon?: string;
+    primaryColor?: string;
+    headerTitle?: string;
+    headerDescription?: string;
+    welcomeMessage?: string;
+    componentType?: 'slides' | 'chat' | 'documents' | 'code' | 'custom';
+    customComponent?: string;
+  };
+  workingDirectory?: string;
+  dataDirectory?: string;
+  fileTypes?: string[];
+  author?: string;
+  tags?: string[];
+  enabled?: boolean;
+}
+
+export interface SubagentFilter {
+  search?: string;
+  tags?: string[];
+  enabled?: boolean;
+}

package/src/utils/agentStorage.ts

@@ -0,0 +1,126 @@
+import fs from 'fs-extra';
+import path from 'path';
+import { fileURLToPath } from 'url';
+import { AgentConfig } from '../types/index.js';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+
+export class AgentStorage {
+  private dataDir: string;
+
+  constructor(dataDir?: string) {
+    this.dataDir = dataDir || path.join(__dirname, '../../data/agents');
+  }
+
+  private getAgentPath(agentId: string): string {
+    return path.join(this.dataDir, `${agentId}.json`);
+  }
+
+  async saveAgent(agent: AgentConfig): Promise<void> {
+    await fs.ensureDir(this.dataDir);
+    const agentPath = this.getAgentPath(agent.id);
+    await fs.writeJson(agentPath, agent, { spaces: 2 });
+  }
+
+  async loadAgent(agentId: string): Promise<AgentConfig | null> {
+    const agentPath = this.getAgentPath(agentId);
+    try {
+      if (await fs.pathExists(agentPath)) {
+        return await fs.readJson(agentPath);
+      }
+    } catch (error) {
+      console.error(`Error loading agent ${agentId}:`, error);
+    }
+    return null;
+  }
+
+  async deleteAgent(agentId: string): Promise<void> {
+    const agentPath = this.getAgentPath(agentId);
+    try {
+      if (await fs.pathExists(agentPath)) {
+        await fs.remove(agentPath);
+      }
+    } catch (error) {
+      console.error(`Error deleting agent ${agentId}:`, error);
+      throw error;
+    }
+  }
+
+  async listAgents(): Promise<AgentConfig[]> {
+    await fs.ensureDir(this.dataDir);
+    const files = await fs.readdir(this.dataDir);
+    const agents: AgentConfig[] = [];
+
+    for (const file of files) {
+      if (file.endsWith('.json')) {
+        const agentId = path.basename(file, '.json');
+        try {
+          const agent = await this.loadAgent(agentId);
+          if (agent) {
+            agents.push(agent);
+          }
+        } catch (error) {
+          console.error(`Error loading agent from file ${file}:`, error);
+        }
+      }
+    }
+
+    return agents;
+  }
+
+  async agentExists(agentId: string): Promise<boolean> {
+    const agentPath = this.getAgentPath(agentId);
+    return await fs.pathExists(agentPath);
+  }
+
+  // Alias for listAgents to match route usage
+  getAllAgents(): Promise<AgentConfig[]> {
+    return this.listAgents();
+  }
+
+  // Alias for loadAgent to match route usage
+  getAgent(agentId: string): Promise<AgentConfig | null> {
+    return this.loadAgent(agentId);
+  }
+
+  // Create agent method
+  async createAgent(agentData: Omit<AgentConfig, 'createdAt' | 'updatedAt'>): Promise<AgentConfig> {
+    const now = new Date().toISOString();
+    const agent: AgentConfig = {
+      ...agentData,
+      createdAt: now,
+      updatedAt: now
+    };
+
+    await this.saveAgent(agent);
+    return agent;
+  }
+
+  // Session-related methods (stubs for now)
+  getAgentSessions(agentId: string, search?: string): Promise<any[]> {
+    // This would need session storage implementation
+    return Promise.resolve([]);
+  }
+
+  getSession(agentId: string, sessionId: string): Promise<any | null> {
+    // This would need session storage implementation
+    return Promise.resolve(null);
+  }
+
+  createSession(agentId: string, title: string): Promise<any> {
+    // This would need session storage implementation
+    const session = {
+      id: `session-${Date.now()}`,
+      title,
+      createdAt: new Date().toISOString(),
+      updatedAt: new Date().toISOString()
+    };
+    return Promise.resolve(session);
+  }
+
+  deleteSession(agentId: string, sessionId: string): Promise<boolean> {
+    // This would need session storage implementation
+    return Promise.resolve(true);
+  }
+}

package/src/utils/claudeVersionStorage.ts

@@ -0,0 +1,199 @@
+import fs from 'fs-extra';
+import path from 'path';
+import { fileURLToPath } from 'url';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+
+export interface ClaudeVersion {
+  id: string;
+  name: string;
+  model: string;
+  description: string;
+  isDefault: boolean;
+  executablePath: string;
+  environmentVariables: Record<string, string>;
+  alias?: string;
+  createdAt: string;
+  updatedAt: string;
+}
+
+export class ClaudeVersionStorage {
+  private dataDir: string;
+
+  constructor(dataDir?: string) {
+    this.dataDir = dataDir || path.join(__dirname, '../../data/claude-versions');
+  }
+
+  private getVersionsPath(): string {
+    return path.join(this.dataDir, 'versions.json');
+  }
+
+  async saveVersions(versions: ClaudeVersion[]): Promise<void> {
+    await fs.ensureDir(this.dataDir);
+    const versionsPath = this.getVersionsPath();
+    await fs.writeJson(versionsPath, versions, { spaces: 2 });
+  }
+
+  async loadVersions(): Promise<ClaudeVersion[]> {
+    const versionsPath = this.getVersionsPath();
+    try {
+      if (await fs.pathExists(versionsPath)) {
+        return await fs.readJson(versionsPath);
+      }
+    } catch (error) {
+      console.error('Error loading Claude versions:', error);
+    }
+
+    // Return default versions if file doesn't exist
+    const now = new Date().toISOString();
+    return [
+      {
+        id: 'claude-3-5-sonnet-20241022',
+        name: 'Claude 3.5 Sonnet',
+        model: 'claude-3-5-sonnet-20241022',
+        description: 'Latest Claude 3.5 Sonnet model',
+        isDefault: true,
+        executablePath: '',
+        environmentVariables: {},
+        alias: 'claude-3-5-sonnet',
+        createdAt: now,
+        updatedAt: now
+      },
+      {
+        id: 'claude-3-opus-20240229',
+        name: 'Claude 3 Opus',
+        model: 'claude-3-opus-20240229',
+        description: 'Claude 3 Opus model',
+        isDefault: false,
+        executablePath: '',
+        environmentVariables: {},
+        alias: 'claude-3-opus',
+        createdAt: now,
+        updatedAt: now
+      },
+      {
+        id: 'claude-3-sonnet-20240229',
+        name: 'Claude 3 Sonnet',
+        model: 'claude-3-sonnet-20240229',
+        description: 'Claude 3 Sonnet model',
+        isDefault: false,
+        executablePath: '',
+        environmentVariables: {},
+        alias: 'claude-3-sonnet',
+        createdAt: now,
+        updatedAt: now
+      }
+    ];
+  }
+
+  async getDefaultVersionId(): Promise<string> {
+    const versions = await this.loadVersions();
+    const defaultVersion = versions.find(v => v.isDefault);
+    return defaultVersion?.id || versions[0]?.id || 'claude-3-5-sonnet-20241022';
+  }
+
+  async setDefaultVersion(versionId: string): Promise<void> {
+    const versions = await this.loadVersions();
+    const updatedVersions = versions.map(v => ({
+      ...v,
+      isDefault: v.id === versionId
+    }));
+    await this.saveVersions(updatedVersions);
+  }
+
+  async createVersion(version: Omit<ClaudeVersion, 'id' | 'createdAt' | 'updatedAt'>): Promise<ClaudeVersion> {
+    const versions = await this.loadVersions();
+    const now = new Date().toISOString();
+    const newVersion: ClaudeVersion = {
+      ...version,
+      id: `custom-${Date.now()}`,
+      createdAt: now,
+      updatedAt: now
+    };
+    versions.push(newVersion);
+    await this.saveVersions(versions);
+    return newVersion;
+  }
+
+  async updateVersion(versionId: string, updates: Partial<ClaudeVersion>): Promise<ClaudeVersion | null> {
+    const versions = await this.loadVersions();
+    const versionIndex = versions.findIndex(v => v.id === versionId);
+    if (versionIndex === -1) return null;
+
+    versions[versionIndex] = {
+      ...versions[versionIndex],
+      ...updates,
+      updatedAt: new Date().toISOString()
+    };
+    await this.saveVersions(versions);
+    return versions[versionIndex];
+  }
+
+  async deleteVersion(versionId: string): Promise<boolean> {
+    const versions = await this.loadVersions();
+    const filteredVersions = versions.filter(v => v.id !== versionId);
+    if (filteredVersions.length === versions.length) return false;
+
+    await this.saveVersions(filteredVersions);
+    return true;
+  }
+
+  async initializeSystemVersion(): Promise<void> {
+    const versions = await this.loadVersions();
+    // Ensure at least one version exists
+    if (versions.length === 0) {
+      const now = new Date().toISOString();
+      await this.saveVersions([
+        {
+          id: 'claude-3-5-sonnet-20241022',
+          name: 'Claude 3.5 Sonnet',
+          model: 'claude-3-5-sonnet-20241022',
+          description: 'Latest Claude 3.5 Sonnet model',
+          isDefault: true,
+          executablePath: '',
+          environmentVariables: {},
+          alias: 'claude-3-5-sonnet',
+          createdAt: now,
+          updatedAt: now
+        }
+      ]);
+    }
+  }
+}
+
+// Export singleton functions for backward compatibility
+export const getAllVersions = async (): Promise<ClaudeVersion[]> => {
+  const storage = new ClaudeVersionStorage();
+  return storage.loadVersions();
+};
+
+export const getDefaultVersionId = async (): Promise<string> => {
+  const storage = new ClaudeVersionStorage();
+  return storage.getDefaultVersionId();
+};
+
+export const setDefaultVersion = async (versionId: string): Promise<void> => {
+  const storage = new ClaudeVersionStorage();
+  return storage.setDefaultVersion(versionId);
+};
+
+export const createVersion = async (version: Omit<ClaudeVersion, 'id' | 'createdAt' | 'updatedAt'>): Promise<ClaudeVersion> => {
+  const storage = new ClaudeVersionStorage();
+  return storage.createVersion(version);
+};
+
+export const updateVersion = async (versionId: string, updates: Partial<ClaudeVersion>): Promise<ClaudeVersion | null> => {
+  const storage = new ClaudeVersionStorage();
+  return storage.updateVersion(versionId, updates);
+};
+
+export const deleteVersion = async (versionId: string): Promise<boolean> => {
+  const storage = new ClaudeVersionStorage();
+  return storage.deleteVersion(versionId);
+};
+
+export const initializeSystemVersion = async (): Promise<void> => {
+  const storage = new ClaudeVersionStorage();
+  return storage.initializeSystemVersion();
+};