refactor: extract model config info logic to service layer and improve type exports

Move getModelConfigsInfo implementation from SimpleCodeGeneratorAgent to ModelConfigService for better separation of concerns. Export ModelConfigsInfoMessage, AgentDisplayConfig, and ModelConfigsInfo types from websocketTypes for frontend consumption. Remove duplicate AgentDisplayConfig interface from model-config-tabs.tsx. Add blueprint_chunk WebSocket message type. Remove 'advanced' workflow tab and update agent

Author: AshishKumar4

src/api-types.ts

@@ -115,7 +115,10 @@ export type {
 export type { 
   WebSocketMessage, 
   WebSocketMessageData,
-  CodeFixEdits 
+  CodeFixEdits,
+  ModelConfigsInfoMessage,
+  AgentDisplayConfig,
+  ModelConfigsInfo
 } from 'worker/api/websocketTypes';
 
 // Database/Schema Types commonly used in frontend

src/components/config-modal.tsx

@@ -29,9 +29,9 @@ import type {
   UserModelConfigWithMetadata, 
   ModelConfigUpdate, 
   AIModels,
-  ByokProvidersData
+  ByokProvidersData,
+  AgentDisplayConfig
 } from '@/api-types';
-import type { AgentDisplayConfig } from './model-config-tabs';
 
 interface ConfigModalProps {
   isOpen: boolean;

src/components/model-config-tabs.tsx

@@ -2,7 +2,6 @@ import { useState, useCallback } from 'react';
 import { 
   Settings, 
   Rocket, 
-  Wrench, 
   Code, 
   Bug,
   Brain,
@@ -17,7 +16,7 @@ import { Badge } from '@/components/ui/badge';
 import { toast } from 'sonner';
 import { ConfigCard } from './config-card';
 import { ConfigModal } from './config-modal';
-import type { ModelConfig, UserModelConfigWithMetadata, ModelConfigUpdate } from '@/api-types';
+import type { ModelConfig, UserModelConfigWithMetadata, ModelConfigUpdate, AgentDisplayConfig } from '@/api-types';
 
 // Define workflow-based tab structure with dynamic agent categorization
 export const WORKFLOW_TABS = {
@@ -49,13 +48,6 @@ export const WORKFLOW_TABS = {
     description: 'Code fixing and review',
     patterns: ['fixer', 'fix', 'review', 'debug']
   },
-  advanced: {
-    id: 'advanced',
-    label: 'Advanced',
-    icon: Wrench,
-    description: 'Specialized operations',
-    patterns: ['screenshot', 'analysis', 'image', 'vision']
-  }
 } as const;
 
 // Helper function to categorize agents dynamically with specific mappings
@@ -64,11 +56,11 @@ const categorizeAgent = (agentKey: string): string => {
   const specificMappings: Record<string, string> = {
     // Quick Start - Most commonly used
     'templateSelection': 'quickstart',
-    'blueprint': 'quickstart', 
     'conversationalResponse': 'quickstart',
 
     // Planning - Project planning and setup
     'phaseGeneration': 'planning',
+    'blueprint': 'quickstart', 
     'projectSetup': 'planning',
 
     // Coding - Development and implementation 
@@ -77,12 +69,7 @@ const categorizeAgent = (agentKey: string): string => {
     'fileRegeneration': 'coding',           // Fix: was going to planning due to "generation"
 
     // Debugging - Code fixing and review
-    'realtimeCodeFixer': 'debugging',
-    'fastCodeFixer': 'debugging',
-    'codeReview': 'debugging',
-    
-    // Advanced - Specialized operations
-    'screenshotAnalysis': 'advanced'
+    'deepDebugger': 'debugging',
   };
 
   // Check specific mappings first
@@ -117,17 +104,6 @@ const categorizeAgent = (agentKey: string): string => {
   return 'advanced';
 };
 
-// Frontend-specific agent display interface
-export interface AgentDisplayConfig {
-  key: string;
-  name: string;
-  description: string;
-  constraint?: {
-    enabled: boolean;
-    allowedModels: string[];
-  };
-}
-
 interface ModelConfigTabsProps {
   agentConfigs: AgentDisplayConfig[];
   modelConfigs: Record<string, UserModelConfigWithMetadata>;

worker/agents/core/simpleGeneratorAgent.ts

@@ -30,8 +30,7 @@ import { ScreenshotAnalysisOperation } from '../operations/ScreenshotAnalysis';
 // Database schema imports removed - using zero-storage OAuth flow
 import { BaseSandboxService } from '../../services/sandbox/BaseSandboxService';
 import { WebSocketMessageData, WebSocketMessageType } from '../../api/websocketTypes';
-import { InferenceContext, AgentActionKey, ModelConfig } from '../inferutils/config.types';
-import { AGENT_CONFIG, AGENT_CONSTRAINTS } from '../inferutils/config';
+import { InferenceContext, ModelConfig } from '../inferutils/config.types';
 import { ModelConfigService } from '../../database/services/ModelConfigService';
 import { fixProjectIssues } from '../../services/code-fixer';
 import { GitVersionControl } from '../git';
@@ -1344,73 +1343,9 @@ export class SimpleCodeGeneratorAgent extends Agent<Env, CodeGenState> {
         };
     }
 
-    /**
-     * Get current model configurations (defaults + user overrides)
-     * Used by WebSocket to provide configuration info to frontend
-     */
-    async getModelConfigsInfo() {
-        const userId = this.state.inferenceContext.userId;
-        if (!userId) {
-            throw new Error('No user session available for model configurations');
-        }
-
-        try {
-            const modelConfigService = new ModelConfigService(this.env);
-            
-            // Get all user configs
-            const userConfigsRecord = await modelConfigService.getUserModelConfigs(userId);
-            
-            // Transform to match frontend interface with constraint info
-            const agents = Object.entries(AGENT_CONFIG).map(([key, config]) => {
-                const constraint = AGENT_CONSTRAINTS.get(key as AgentActionKey);
-                return {
-                    key,
-                    name: config.name,
-                    description: config.description,
-                    constraint: constraint ? {
-                        enabled: constraint.enabled,
-                        allowedModels: Array.from(constraint.allowedModels)
-                    } : undefined
-                };
-            });
-
-            const userConfigs: Record<string, any> = {};
-            const defaultConfigs: Record<string, any> = {};
-
-            for (const [actionKey, mergedConfig] of Object.entries(userConfigsRecord)) {
-                if (mergedConfig.isUserOverride) {
-                    userConfigs[actionKey] = {
-                        name: mergedConfig.name,
-                        max_tokens: mergedConfig.max_tokens,
-                        temperature: mergedConfig.temperature,
-                        reasoning_effort: mergedConfig.reasoning_effort,
-                        fallbackModel: mergedConfig.fallbackModel,
-                        isUserOverride: true
-                    };
-                }
-                
-                // Always include default config
-                const defaultConfig = AGENT_CONFIG[actionKey as AgentActionKey];
-                if (defaultConfig) {
-                    defaultConfigs[actionKey] = {
-                        name: defaultConfig.name,
-                        max_tokens: defaultConfig.max_tokens,
-                        temperature: defaultConfig.temperature,
-                        reasoning_effort: defaultConfig.reasoning_effort,
-                        fallbackModel: defaultConfig.fallbackModel
-                    };
-                }
-            }
-
-            return {
-                agents,
-                userConfigs,
-                defaultConfigs
-            };
-        } catch (error) {
-            this.logger().error('Error fetching model configs info:', error);
-            throw error;
-        }
+    getModelConfigsInfo() {
+        const modelService = new ModelConfigService(this.env);
+        return modelService.getModelConfigsInfo(this.state.inferenceContext.userId);
     }
 
     getTotalFiles(): number {

worker/api/websocketTypes.ts

@@ -352,6 +352,11 @@ type BlueprintUpdatedMessage = {
 	updatedKeys: string[];
 };
 
+type BlueprintChunkMessage = {
+	type: 'blueprint_chunk';
+	chunk: string;
+};
+
 type DeterministicCodeFixStartedMessage = {
 	type: 'deterministic_code_fix_started';
 	message: string;
@@ -365,14 +370,18 @@ type DeterministicCodeFixCompletedMessage = {
     issues: CodeIssue[];
 };
 
-type ModelConfigsInfoMessage = {
+export type ModelConfigsInfoMessage = {
 	type: 'model_configs_info';
 	message: string;
 	configs: {
 		agents: Array<{
 			key: string;
 			name: string;
 			description: string;
+            constraint?: {
+                enabled: boolean;
+                allowedModels: string[];
+            };
 		}>;
 		userConfigs: Record<string, {
 			name?: string;
@@ -392,6 +401,9 @@ type ModelConfigsInfoMessage = {
 	};
 };
 
+export type AgentDisplayConfig = ModelConfigsInfoMessage['configs']['agents'][number];
+export type ModelConfigsInfo = ModelConfigsInfoMessage['configs'];
+
 type TerminalCommandMessage = {
 	type: 'terminal_command';
 	command: string;
@@ -462,6 +474,7 @@ export type WebSocketMessage =
 	| ConversationClearedMessage
     | ProjectNameUpdatedMessage
     | BlueprintUpdatedMessage
+    | BlueprintChunkMessage
     | DeterministicCodeFixStartedMessage
     | DeterministicCodeFixCompletedMessage
 	| ModelConfigsInfoMessage
@@ -478,4 +491,4 @@ type WebSocketMessagePayload<T extends WebSocketMessageType> = Extract<WebSocket
 
 // A utility type to get only the data part of the payload, excluding the 'type' property.
 // This is what your 'data' parameter will be.
-export type WebSocketMessageData<T extends WebSocketMessageType> = Omit<WebSocketMessagePayload<T>, 'type'>;
+export type WebSocketMessageData<T extends WebSocketMessageType> = Omit<WebSocketMessagePayload<T>, 'type'>;