Update aiEngine.js

aiEngine.js

@@ -2,20 +2,31 @@ import OpenAI from "openai";
 import { BedrockRuntimeClient, ConverseCommand } from "@aws-sdk/client-bedrock-runtime";
 import { NodeHttpHandler } from "@smithy/node-http-handler";
 import fs from 'fs';
+import path from 'path';
 
-// Load prompts
-const prompts = JSON.parse(fs.readFileSync('./prompts.json', 'utf8'));
+// Load prompts safely
+let prompts = {};
+try {
+    const promptsPath = path.resolve('./prompts.json');
+    if (fs.existsSync(promptsPath)) {
+        prompts = JSON.parse(fs.readFileSync(promptsPath, 'utf8'));
+    }
+} catch (e) {
+    console.error("Error loading prompts:", e);
+}
 
 // --- CLIENT INITIALIZATION ---
+// 1. AWS Bedrock (Claude)
 const bedrockClient = new BedrockRuntimeClient({
     region: process.env.AWS_REGION || "us-east-1",
     requestHandler: new NodeHttpHandler({ http2Handler: undefined }),
     credentials: {
-        accessKey_id: process.env.AWS_ACCESS_KEY_ID,
+        accessKeyId: process.env.AWS_ACCESS_KEY_ID,
         secretAccessKey: process.env.AWS_SECRET_ACCESS_KEY,
     }
 });
 
+// 2. Azure OpenAI (GPT-5 Mini)
 const azureOpenAI = new OpenAI({
     apiKey: process.env.AZURE_OPENAI_API_KEY,
     baseURL: `${process.env.AZURE_OPENAI_ENDPOINT}/openai/deployments/${process.env.AZURE_DEPLOYMENT_NAME}`,
@@ -23,82 +34,177 @@ const azureOpenAI = new OpenAI({
     defaultHeaders: { "api-key": process.env.AZURE_OPENAI_API_KEY }
 });
 
-// Helper to convert Google history format to Standard Chat format
-function transformHistory(history) {
+// --- HELPER: History Transformer ---
+// Converts your App's { role: 'user', parts: [{text: ''}] } format 
+// to formats acceptable by Azure/Bedrock
+function toAzureHistory(history) {
     return history.map(h => ({
-        role: h.role === 'model' ? 'assistant' : h.role,
+        role: h.role === 'model' ? 'assistant' : 'user',
         content: h.parts[0].text
     }));
 }
 
+function toBedrockHistory(history) {
+    return history.map(h => ({
+        role: h.role === 'model' ? 'assistant' : 'user',
+        content: [{ text: h.parts[0].text }]
+    }));
+}
+
 export const AIEngine = {
-    // PM -> Claude Sonnet 4.6 (via Bedrock)
+    /**
+     * 1. PROJECT MANAGER (Reasoning & Delegation)
+     * Uses AWS Bedrock -> Claude 3.5 Sonnet (v2/4.6)
+     */
     callPM: async (history, input) => {
-        const chatHistory = transformHistory(history);
+        // Prepare History
+        const chatHistory = toBedrockHistory(history);
+        const sysPrompt = prompts.pm_system_prompt || "You are a Senior Project Manager.";
+
         const command = new ConverseCommand({
             modelId: "arn:aws:bedrock:us-east-1:106774395747:inference-profile/global.anthropic.claude-sonnet-4-6",
-            system: [{ text: prompts.pm_system_prompt }],
+            system: [{ text: sysPrompt }],
             messages: [...chatHistory, { role: "user", content: [{ text: input }] }],
-            inferenceConfig: { maxTokens: 50000, temperature: 0.7 },
+            inferenceConfig: { maxTokens: 4000, temperature: 1 },
             additionalModelRequestFields: { 
                 thinking: { type: "adaptive" },
                 output_config: { effort: "high" }
             }
         });
 
-        const response = await bedrockClient.send(command);
-        const text = response.output.message.content.find(b => b.text)?.text;
-        
-        return { 
-            text, 
-            usage: { totalTokenCount: response.usage?.totalTokens || 0 } 
-        };
+        try {
+            const response = await bedrockClient.send(command);
+            const text = response.output.message.content.find(b => b.text)?.text;
+            
+            // Extract usage
+            const inputTokens = response.usage?.inputTokens || 0;
+            const outputTokens = response.usage?.outputTokens || 0;
+
+            console.log(`[AI-PM] Claude Usage: ${inputTokens + outputTokens}`);
+
+            return { 
+                text, 
+                usage: { totalTokenCount: inputTokens + outputTokens } 
+            };
+        } catch (error) {
+            console.error("PM AI Error (Bedrock):", error);
+            throw error;
+        }
     },
 
-    // Worker -> GPT-5 Mini (via Azure)
+    /**
+     * 2. WORKER (Coding & Execution)
+     * Uses Azure OpenAI -> GPT-5 Mini
+     */
     callWorker: async (history, input, images = []) => {
-        const chatHistory = transformHistory(history);
+        const chatHistory = toAzureHistory(history);
+        const sysPrompt = prompts.worker_system_prompt || "You are a specialized Worker.";
         
-        // Note: GPT-5 Mini handles images via the content array if needed
-        const response = await azureOpenAI.chat.completions.create({
-            model: process.env.AZURE_DEPLOYMENT_NAME,
-            messages: [
-                { role: "system", content: prompts.worker_system_prompt },
-                ...chatHistory,
-                { role: "user", content: input }
-            ],
-            reasoning_effort: "high"
-        });
+        // Azure GPT-5 Mini Setup
+        // Note: For now, we append images to the user content if needed, 
+        // but simplified here for text-parity with your script.
+        
+        try {
+            const response = await azureOpenAI.chat.completions.create({
+                model: process.env.AZURE_DEPLOYMENT_NAME, // "gpt-5-mini"
+                messages: [
+                    { role: "system", content: sysPrompt },
+                    ...chatHistory,
+                    { role: "user", content: input }
+                ],
+                reasoning_effort: "high"
+            });
+
+            const text = response.choices[0].message.content;
+            const usage = response.usage?.total_tokens || 0;
+
+            console.log(`[AI-WORKER] GPT-5 Usage: ${usage}`);
 
-        return { 
-            text: response.choices[0].message.content, 
-            usage: { totalTokenCount: response.usage?.total_tokens || 0 } 
-        };
+            return { 
+                text, 
+                usage: { totalTokenCount: usage } 
+            };
+        } catch (error) {
+            console.error("Worker AI Error (Azure):", error);
+            throw error;
+        }
     },
 
-    // Standard tasks use the faster Worker model
+    /**
+     * 3. ONBOARDING ANALYST (Question Generation)
+     * Returns STRICT JSON
+     */
     generateEntryQuestions: async (description) => {
-        const response = await azureOpenAI.chat.completions.create({
-            model: process.env.AZURE_DEPLOYMENT_NAME,
-            messages: [
-                { role: "system", content: prompts.analyst_system_prompt },
-                { role: "user", content: `[MODE 1: QUESTIONS]\nIdea: "${description}"` }
-            ],
-            response_format: { type: "json_object" }
-        });
-        const parsed = JSON.parse(response.choices[0].message.content);
-        return { ...parsed, usage: { totalTokenCount: response.usage.total_tokens } };
+        const sysPrompt = prompts.analyst_system_prompt || "Return JSON only.";
+        const input = `[MODE 1: QUESTIONS]\nAnalyze this game idea: "${description}". Check for TOS violations or nonsense. If good, ask 3 questions. Output ONLY raw JSON.`;
+
+        try {
+            const response = await azureOpenAI.chat.completions.create({
+                model: process.env.AZURE_DEPLOYMENT_NAME,
+                messages: [
+                    { role: "system", content: sysPrompt },
+                    { role: "user", content: input }
+                ],
+                response_format: { type: "json_object" }
+            });
+
+            const parsed = JSON.parse(response.choices[0].message.content);
+            return { 
+                ...parsed, 
+                usage: { totalTokenCount: response.usage.total_tokens } 
+            };
+        } catch (e) {
+            console.error("Analyst Error:", e);
+            // Fallback structure
+            return { 
+                status: "ACCEPTED", 
+                questions: [{ id: "fallback", label: "Elaborate?", type: "textarea" }], 
+                usage: { totalTokenCount: 0 } 
+            };
+        }
     },
 
+    /**
+     * 4. PROJECT GRADER (Feasibility Check)
+     * Returns STRICT JSON
+     */
     gradeProject: async (description, answers) => {
-        const response = await azureOpenAI.chat.completions.create({
-            model: process.env.AZURE_DEPLOYMENT_NAME,
-            messages: [
-                { role: "system", content: prompts.analyst_system_prompt },
-                { role: "user", content: `[MODE 2: GRADING]\nIdea: ${description}\nAnswers: ${JSON.stringify(answers)}` }
-            ],
-            response_format: { type: "json_object" }
-        });
-        return { ...JSON.parse(response.choices[0].message.content), usage: { totalTokenCount: response.usage.total_tokens } };
+        const sysPrompt = prompts.analyst_system_prompt || "Return JSON only.";
+        const input = `[MODE 2: GRADING]\nIdea: "${description}"\nUser Answers: ${JSON.stringify(answers)}\n\nAssess feasibility. Output JSON with title and rating.`;
+
+        try {
+            const response = await azureOpenAI.chat.completions.create({
+                model: process.env.AZURE_DEPLOYMENT_NAME,
+                messages: [
+                    { role: "system", content: sysPrompt },
+                    { role: "user", content: input }
+                ],
+                response_format: { type: "json_object" }
+            });
+
+            const parsed = JSON.parse(response.choices[0].message.content);
+            return { 
+                ...parsed, 
+                usage: { totalTokenCount: response.usage.total_tokens } 
+            };
+        } catch (e) {
+            console.error("Grading Error:", e);
+            return { 
+                feasibility: 50, 
+                rating: 'C', 
+                title: "Untitled", 
+                usage: { totalTokenCount: 0 } 
+            };
+        }
+    },
+
+    /**
+     * 5. IMAGE GENERATOR 
+     * Kept as placeholder since you said "not being used" for now,
+     * but prevents crashes if App.js calls it.
+     */
+    generateImage: async (prompt) => {
+        console.log("Image generation currently skipped in Azure/Bedrock migration.");
+        return null;
     }
 };