Update src/marketResearchAgent.js

src/marketResearchAgent.js

@@ -1,29 +1,37 @@
-import crypto from "crypto";
-import { callLLM } from "./llm/callLLM.js";
-
-/* =========================================================
-   PUBLIC ENTRY
-========================================================= */
-
-export async function marketResearchAgent({ input, provider, model }) {
-  if (!input?.keyword) {
-    throw new Error("keyword is required");
-  }
-
-  const marketTitle = normalizeKeyword(input.keyword);
-  const normalizedTitle = `Global ${marketTitle} Market and Forecast 2026–2033`;
-
+import crypto from "crypto";
+import { callLLM } from "./llm/callLLM.js";
+import { validateMarketData, ValidationError, SCHEMA_VERSION } from "./schema/marketData.schema.js";
+import { createLogger } from "./utils/errorLogger.js";
+
+// Task 8.2: Add error logging to OpenClaw Agent
+const logger = createLogger('OpenClaw Agent');
+
+/* =========================================================
+   PUBLIC ENTRY
+========================================================= */
+
+export async function marketResearchAgent({ input, provider, model }) {
+  if (!input?.keyword) {
+    logger.log('Missing keyword in input', { input });
+    throw new Error("keyword is required");
+  }
+
+  logger.info('Starting market research', { keyword: input.keyword, provider, model });
+
+  const marketTitle = normalizeKeyword(input.keyword);
+  const normalizedTitle = `Global ${marketTitle} Market and Forecast 2026–2033`;
+
   /* =======================================================
      1. MARKET SIZE WILL COME FROM LLM ANALYSIS
      ======================================================= */
 
   // Market size data will be provided by LLM analysis
   // No random generation - all data comes from AI analysis
-
-  /* =======================================================
-     2. SINGLE AI RESEARCH PROMPT (MULTI-AXIS)
-     ======================================================= */
-
+
+  /* =======================================================
+     2. SINGLE AI RESEARCH PROMPT (MULTI-AXIS)
+     ======================================================= */
+
   const researchPrompt = `
 You are a seasoned healthcare market research analyst. Analyze the global market for: "${marketTitle}"
 
@@ -90,23 +98,31 @@ RULES:
 5. Include at least 5 companies
 6. JSON only, no markdown
 `;
-
+
   let ai;
 
   try {
     const raw = await callLLM({ provider, model, prompt: researchPrompt });
     ai = extractJsonObject(raw);
     validateAI(ai);
+    logger.info('LLM analysis successful', { marketTitle });
     console.log('OpenClaw: LLM analysis successful');
   } catch (error) {
+    logger.warn('LLM analysis failed, using fallback data', { 
+      error: error.message,
+      marketTitle 
+    });
     console.log('OpenClaw: LLM failed:', error.message);
     ai = generateFallbackData(marketTitle);
   }
-
-  /* =======================================================
-     3. DASHBOARD VIEW (INFOGRAPHIC-READY)
-     ======================================================= */
-
+  
+  // Ensure all required data is present
+  ai = ensureCompleteData(ai, marketTitle);
+
+  /* =======================================================
+     3. DASHBOARD VIEW (INFOGRAPHIC-READY)
+     ======================================================= */
+
   const dashboard_view = {
     marketTitle: ai?.marketTitle || `Global ${marketTitle} Market`,
 
@@ -124,6 +140,9 @@ RULES:
     ],
 
     marketSegments: ai?.marketSegments || [],
+    
+    // Add regional data for dashboard
+    regional: generateRegionalData(ai),
 
     drivers: (ai?.marketDrivers || []).map(d => ({
       driver: d,
@@ -136,14 +155,23 @@ RULES:
       company: c.company,
       share: c.player_marketShare_2025
     })),
+    
+    // Add segments data for dashboard charts
+    segments: (ai?.marketSegments || []).map(seg => ({
+      segment: seg.segmentName,
+      marketSize: calculateSegmentSize(seg, ai?.currentYear_2025 || 0),
+      growthRate: seg.segmentName_cagr_Forecast || 0,
+      marketShare: seg.subSegments?.[0]?.segment_marketShare_2025 || 0,
+      subSegments: seg.subSegments || []
+    })),
 
     citation: "AI Market Analysis"
   };
-
-  /* =======================================================
-     4. REPORT VIEW (PDF-READY)
-     ======================================================= */
-
+
+  /* =======================================================
+     4. REPORT VIEW (PDF-READY)
+     ======================================================= */
+
   const report_view = {
     marketTitle: ai?.marketTitle || marketTitle,
 
@@ -168,43 +196,44 @@ RULES:
 
     competitiveLandscape: ai?.competitiveLandscape || []
   };
-
-  /* =======================================================
-     5. FINAL RESPONSE
-     ======================================================= */
-
-  return {
-    meta: {
-      job_id: input.job_id || `job_${crypto.randomUUID()}`,
-      keyword: marketTitle,
-      normalized_title: normalizedTitle,
-      status: "completed",
-      timestamp: new Date().toISOString(),
-      provider_used: provider || "auto",
-      model_used: model || "auto",
-      confidence: "high"
-    },
-    dashboard_view,
-    report_view
-  };
-}
-
-/* =========================================================
-   HELPERS & VALIDATION
-========================================================= */
-
-function extractJsonObject(text) {
-  if (!text) return null;
-  const start = text.indexOf("{");
-  const end = text.lastIndexOf("}");
-  if (start === -1 || end === -1 || end <= start) return null;
-  try {
-    return JSON.parse(text.slice(start, end + 1));
-  } catch {
-    return null;
-  }
-}
-
+
+  /* =======================================================
+     5. FINAL RESPONSE
+     ======================================================= */
+
+  return {
+    meta: {
+      job_id: input.job_id || `job_${crypto.randomUUID()}`,
+      keyword: marketTitle,
+      normalized_title: normalizedTitle,
+      status: "completed",
+      timestamp: new Date().toISOString(),
+      provider_used: provider || "auto",
+      model_used: model || "auto",
+      confidence: "high",
+      schemaVersion: SCHEMA_VERSION  // Task 9.2: Include schema version
+    },
+    dashboard_view,
+    report_view
+  };
+}
+
+/* =========================================================
+   HELPERS & VALIDATION
+========================================================= */
+
+function extractJsonObject(text) {
+  if (!text) return null;
+  const start = text.indexOf("{");
+  const end = text.lastIndexOf("}");
+  if (start === -1 || end === -1 || end <= start) return null;
+  try {
+    return JSON.parse(text.slice(start, end + 1));
+  } catch {
+    return null;
+  }
+}
+
 function validateAI(obj) {
   if (!obj || typeof obj !== "object") {
     throw new Error("Invalid AI output - not an object");
@@ -220,19 +249,19 @@ function validateAI(obj) {
     }
   }
 }
-
-function normalizeKeyword(k) {
-  return String(k).replace(/market/gi, "").trim();
-}
-
-function round(n) {
-  return Math.round(n * 100) / 100;
-}
-
-function randomFloat(min, max) {
-  return round(min + Math.random() * (max - min));
-}
-
+
+function normalizeKeyword(k) {
+  return String(k).replace(/market/gi, "").trim();
+}
+
+function round(n) {
+  return Math.round(n * 100) / 100;
+}
+
+function randomFloat(min, max) {
+  return round(min + Math.random() * (max - min));
+}
+
 function randomInt(min, max) {
   return Math.floor(min + Math.random() * (max - min + 1));
 }
@@ -282,3 +311,64 @@ function generateFallbackData(marketTitle) {
     strategicRecommendations: ["Invest in R&D", "Expand to emerging markets", "Focus on digital transformation"]
   };
 }
+
+/**
+ * Ensure all required data fields are present
+ */
+function ensureCompleteData(data, marketTitle) {
+  // Ensure marketSegments have subSegments arrays
+  if (Array.isArray(data.marketSegments)) {
+    data.marketSegments = data.marketSegments.map(segment => ({
+      ...segment,
+      subSegments: Array.isArray(segment.subSegments) ? segment.subSegments : []
+    }));
+  }
+  
+  // Ensure minimum required fields
+  return {
+    ...data,
+    marketTitle: data.marketTitle || `Global ${marketTitle} Market`,
+    executiveOverview: data.executiveOverview || `Market analysis for ${marketTitle}.`,
+    marketSegments: data.marketSegments || [],
+    marketDrivers: data.marketDrivers || [],
+    competitiveLandscape: data.competitiveLandscape || []
+  };
+}
+
+/**
+ * Generate regional market data
+ */
+function generateRegionalData(aiData) {
+  const totalMarket = aiData?.currentYear_2025 || 0;
+  
+  // Default regional distribution
+  const regions = [
+    { region: 'North America', percentage: 35 },
+    { region: 'Europe', percentage: 28 },
+    { region: 'Asia Pacific', percentage: 25 },
+    { region: 'Latin America', percentage: 7 },
+    { region: 'Middle East & Africa', percentage: 5 }
+  ];
+  
+  return regions.map(r => ({
+    region: r.region,
+    share: r.percentage,
+    marketSize: round((totalMarket * r.percentage) / 100),
+    growthRate: aiData?.global_cagr_Forecast || 0
+  }));
+}
+
+/**
+ * Calculate segment market size
+ */
+function calculateSegmentSize(segment, totalMarket) {
+  if (!segment.subSegments || segment.subSegments.length === 0) {
+    return round(totalMarket * 0.2); // Default 20% if no data
+  }
+  
+  const avgShare = segment.subSegments.reduce((sum, sub) => {
+    return sum + (sub.segment_marketShare_2025 || 0);
+  }, 0) / segment.subSegments.length;
+  
+  return round((totalMarket * avgShare) / 100);
+}