remove StudentPersonality enum

src/app/api/agent/chat/route.ts

@@ -1,7 +1,7 @@
 import { NextRequest, NextResponse } from 'next/server';
 import { createOpenAI } from '@ai-sdk/openai';
 import { AIAgent } from '@/lib/agent';
-import { AgentType, StudentPersonality, AgentConfig } from '@/lib/types/agent';
+import { AgentType, AgentConfig } from '@/lib/types/agent';
 
 const customOpenAI = createOpenAI({
   apiKey: process.env.CZ_OPENAI_API_KEY || '',
@@ -11,7 +11,7 @@ const customOpenAI = createOpenAI({
 const defaultConfig: AgentConfig = {
   id: 'legacy-agent',
   type: AgentType.STUDENT,
-  personality: StudentPersonality.ADHD_COMBINED,
+  personality: 'adhd_combined',
   name: 'Default Student',
   description: 'A helpful AI student for legacy API compatibility',
   systemPrompt: 'You are a helpful AI assistant that can help with educational tasks. You have access to tools for calculations and note-taking.',

src/app/api/agent/history/[id]/route.ts

@@ -1,7 +1,7 @@
 import { NextRequest, NextResponse } from 'next/server';
 import { createOpenAI } from '@ai-sdk/openai';
 import { AIAgent } from '@/lib/agent';
-import { AgentType, StudentPersonality, AgentConfig } from '@/lib/types/agent';
+import { AgentType, AgentConfig } from '@/lib/types/agent';
 
 const customOpenAI = createOpenAI({
   apiKey: process.env.CZ_OPENAI_API_KEY || '',
@@ -10,7 +10,7 @@ const customOpenAI = createOpenAI({
 const defaultConfig: AgentConfig = {
   id: 'legacy-agent',
   type: AgentType.STUDENT,
-  personality: StudentPersonality.ADHD_COMBINED,
+  personality: 'adhd_combined',
   name: 'Default Student',
   description: 'A helpful AI student for legacy API compatibility',
   systemPrompt: 'You are a helpful AI assistant that can help with educational tasks.',

src/app/api/agents/personalities/route.ts

@@ -0,0 +1,16 @@
+import { NextResponse } from 'next/server';
+import { studentPrompts } from '@/lib/prompts/student-prompts';
+
+export async function GET() {
+  const personalities = Object.entries(studentPrompts).map(([key, config]) => ({
+    id: key,
+    name: config.name,
+    description: config.description,
+    tools: config.tools
+  }));
+
+  return NextResponse.json({
+    personalities,
+    total: personalities.length
+  });
+}

src/app/api/agents/route.ts

@@ -1,10 +1,14 @@
 import { NextRequest, NextResponse } from 'next/server';
 import { AgentRepository } from '@/lib/repositories/agent-repository';
 import { requireBasicAuth, createUnauthorizedResponse } from '@/lib/middleware/basic-auth';
+import { getAvailablePersonalities } from '@/lib/prompts/student-prompts';
 import { z } from 'zod';
 
 const createAgentSchema = z.object({
-  personality: z.enum(['adhd_inattentive', 'adhd_hyperactive', 'adhd_combined']),
+  personality: z.string().refine(
+    (p) => getAvailablePersonalities().includes(p),
+    { message: 'Invalid personality type' }
+  ),
   name: z.string().optional(),
   customPrompt: z.string().optional(),
 });

src/app/dashboard/page.tsx

@@ -124,10 +124,15 @@ export default function Dashboard() {
 
   const createAllStudents = async () => {
     try {
-      // Create all three students sequentially
-      await createAgent('adhd_inattentive');  // 小明
-      await createAgent('adhd_hyperactive');  // 小華
-      await createAgent('adhd_combined');     // 小美
+      // Fetch available personalities from API
+      const response = await fetch(`${API_URL}/api/agents/personalities`);
+      const data = await response.json();
+
+      // Create each personality sequentially to prevent race conditions
+      for (const personality of data.personalities) {
+        await createAgent(personality.id);
+      }
+
       // Fetch agents once after all are created
       await fetchAgents();
     } catch (error) {

src/lib/agent-configs.ts

@@ -1,10 +1,9 @@
 import {
   AgentConfig,
   AgentType,
-  StudentPersonality,
   CreateAgentRequest
 } from './types/agent';
-import { getStudentConfig, getStudentPrompt } from './prompts/student-prompts';
+import { getStudentConfig, getStudentPrompt, studentPrompts, isValidPersonality } from './prompts/student-prompts';
 import { allTools } from './tools';
 
 export class AgentConfigFactory {
@@ -20,7 +19,12 @@ export class AgentConfigFactory {
     let config: AgentConfig;
 
     if (request.type === AgentType.STUDENT) {
-      const studentPersonality = request.personality as StudentPersonality;
+      const studentPersonality = request.personality;
+
+      if (!isValidPersonality(studentPersonality)) {
+        throw new Error(`Invalid student personality: ${studentPersonality}`);
+      }
+
       const template = getStudentConfig(studentPersonality);
 
       config = {
@@ -113,7 +117,7 @@ If you don't need a tool, just respond normally with helpful information.`;
       if (updatedConfig.metadata?.templateUsed) {
         if (config.type === AgentType.STUDENT) {
           updatedConfig.systemPrompt = getStudentPrompt(
-            config.personality as StudentPersonality,
+            config.personality,
             updates.tools
           );
         }
@@ -143,14 +147,11 @@ If you don't need a tool, just respond normally with helpful information.`;
     return [
       {
         type: AgentType.STUDENT,
-        personalities: Object.values(StudentPersonality).map(personality => {
-          const config = getStudentConfig(personality);
-          return {
-            personality,
-            name: config.name,
-            description: config.description
-          };
-        })
+        personalities: Object.entries(studentPrompts).map(([key, config]) => ({
+          personality: key,
+          name: config.name,
+          description: config.description
+        }))
       },
       {
         type: AgentType.ASSISTANT,
@@ -184,8 +185,8 @@ If you don't need a tool, just respond normally with helpful information.`;
       errors.push('Invalid agent type');
     }
 
-    if (config.type === AgentType.STUDENT && !Object.values(StudentPersonality).includes(config.personality as StudentPersonality)) {
-      errors.push('Invalid student personality');
+    if (config.type === AgentType.STUDENT && !isValidPersonality(config.personality)) {
+      errors.push(`Invalid student personality: ${config.personality}`);
     }
 
 

src/lib/prompts/student-prompts.ts

@@ -1,16 +1,14 @@
-import { StudentPersonality } from '../types/agent';
-
 export interface StudentPromptTemplate {
-  personality: StudentPersonality;
+  personality: string;
   name: string;
   description: string;
   systemPrompt: string;
   tools: string[];
 }
 
-export const studentPrompts: Record<StudentPersonality, StudentPromptTemplate> = {
-  [StudentPersonality.ADHD_INATTENTIVE]: {
-    personality: StudentPersonality.ADHD_INATTENTIVE,
+export const studentPrompts: Record<string, StudentPromptTemplate> = {
+  'adhd_inattentive': {
+    personality: 'adhd_inattentive',
     name: '小明 - 注意力不集中型學生',
     description: '患有ADHD注意力不足型的學生，在專注力、組織能力和遵循指示方面有困難',
     systemPrompt: `IMPORTANT: You MUST respond in Traditional Chinese (繁體中文) using Mandarin. All your responses should be in Chinese.
@@ -55,8 +53,8 @@ IMPORTANT: Keep your responses SHORT and CONCISE. Aim for 1-3 sentences maximum
     tools: ['calculate', 'save_note']
   },
 
-  [StudentPersonality.ADHD_HYPERACTIVE]: {
-    personality: StudentPersonality.ADHD_HYPERACTIVE,
+  'adhd_hyperactive': {
+    personality: 'adhd_hyperactive',
     name: '小華 - 過動型學生',
     description: '患有ADHD過動型的學生，精力充沛、衝動且難以自我控制',
     systemPrompt: `IMPORTANT: You MUST respond in Traditional Chinese (繁體中文) using Mandarin. All your responses should be in Chinese.
@@ -100,8 +98,8 @@ IMPORTANT: Keep your responses SHORT and CONCISE. Aim for 1-3 sentences maximum
     tools: ['calculate', 'save_note']
   },
 
-  [StudentPersonality.ADHD_COMBINED]: {
-    personality: StudentPersonality.ADHD_COMBINED,
+  'adhd_combined': {
+    personality: 'adhd_combined',
     name: '小美 - 混合型學生',
     description: '患有ADHD混合型的學生，同時有注意力不足和過動問題，並在社交線索和情緒調節方面有困難',
     systemPrompt: `IMPORTANT: You MUST respond in Traditional Chinese (繁體中文) using Mandarin. All your responses should be in Chinese.
@@ -146,13 +144,28 @@ IMPORTANT: Keep your responses SHORT and CONCISE. Aim for 1-3 sentences maximum
   }
 };
 
-export function getStudentPrompt(personality: StudentPersonality, availableTools: string[]): string {
+export function getAvailablePersonalities(): string[] {
+  return Object.keys(studentPrompts);
+}
+
+export function isValidPersonality(personality: string): boolean {
+  return personality in studentPrompts;
+}
+
+export function getStudentPrompt(personality: string, availableTools: string[]): string {
   const template = studentPrompts[personality];
+  if (!template) {
+    throw new Error(`Unknown personality: ${personality}`);
+  }
   const toolDescriptions = availableTools.map(tool => `- ${tool}: Available for use`).join('\n');
 
   return template.systemPrompt.replace('{{TOOLS}}', toolDescriptions);
 }
 
-export function getStudentConfig(personality: StudentPersonality) {
-  return studentPrompts[personality];
+export function getStudentConfig(personality: string) {
+  const config = studentPrompts[personality];
+  if (!config) {
+    throw new Error(`Unknown personality: ${personality}`);
+  }
+  return config;
 }

src/lib/repositories/agent-repository.ts

@@ -1,7 +1,6 @@
 import { FileStore } from '../persistence/file-store';
 import { StudentAgent } from '../types/models';
 import { getStudentPrompt, getStudentConfig } from '../prompts/student-prompts';
-import { StudentPersonality } from '../types/agent';
 
 export class AgentRepository {
   private store: FileStore<StudentAgent>;
@@ -12,11 +11,11 @@ export class AgentRepository {
 
   async createAgent(
     userId: string,
-    personality: 'adhd_inattentive' | 'adhd_hyperactive' | 'adhd_combined',
+    personality: string,
     name?: string,
     customPrompt?: string
   ): Promise<StudentAgent> {
-    const config = getStudentConfig(personality as StudentPersonality);
+    const config = getStudentConfig(personality);
     const tools = config.tools || [];
 
     const agentData = {
@@ -24,7 +23,7 @@ export class AgentRepository {
       personality,
       name: name || config.name,
       description: config.description,
-      systemPrompt: customPrompt || getStudentPrompt(personality as StudentPersonality, tools),
+      systemPrompt: customPrompt || getStudentPrompt(personality, tools),
       isActive: true,
       createdAt: new Date().toISOString(),
       updatedAt: new Date().toISOString(),

src/lib/types/agent.ts

@@ -15,17 +15,10 @@ export enum AgentType {
   ASSISTANT = 'assistant' // Legacy support
 }
 
-export enum StudentPersonality {
-  ADHD_INATTENTIVE = 'adhd_inattentive',
-  ADHD_HYPERACTIVE = 'adhd_hyperactive',
-  ADHD_COMBINED = 'adhd_combined'
-}
-
-
 export interface AgentConfig {
   id: string;
   type: AgentType;
-  personality: StudentPersonality;
+  personality: string;
   name: string;
   description: string;
   systemPrompt: string;
@@ -45,7 +38,7 @@ export interface AgentInstance {
 
 export interface CreateAgentRequest {
   type: AgentType;
-  personality: StudentPersonality;
+  personality: string;
   name?: string;
   customPrompt?: string;
   tools?: string[];

src/lib/types/models.ts

@@ -9,7 +9,7 @@ export interface User {
 export interface StudentAgent {
   id: string;
   userId: string;
-  personality: 'adhd_inattentive' | 'adhd_hyperactive' | 'adhd_combined';
+  personality: string;
   name: string;
   description: string;
   systemPrompt: string;