tango-app-api-audio-analytics 1.0.15 → 1.0.16

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "tango-app-api-audio-analytics",
-    "version": "1.0.15",
+    "version": "1.0.16",
     "description": "audioAnalytics",
     "main": "index.js",
     "type": "module",
@@ -14,6 +14,7 @@
     "license": "ISC",
     "dependencies": {
         "@aws-sdk/client-secrets-manager": "^3.1003.0",
+        "@google/generative-ai": "^0.24.1",
         "aws-sdk": "^2.1693.0",
         "body-parser": "^2.2.2",
         "cors": "^2.8.6",

package/src/controllers/audioAnalytics.controller.js

@@ -1,5 +1,6 @@
 import { insertWithId, updateOpenSearchData, searchOpenSearchData, logger } from 'tango-app-api-middleware';
 import { randomUUID } from 'crypto';
+import { GoogleGenerativeAI } from '@google/generative-ai';
 
 const EXTERNAL_STREAM_API = 'http://172.236.179.51:8000/stream';
 
@@ -31,7 +32,7 @@ export async function createCohort( req, res ) {
       updatedAt: new Date().toISOString(),
     };
 
-    const result = await insertWithId( 'tango-audio-config', cohortId, cohortData );
+    const result = await insertWithId( 'tango-audio-config-test', cohortId, cohortData );
     logger.info( { result } );
 
     if ( result && result.body && result.body.result === 'created' ) {
@@ -67,7 +68,7 @@ export async function createBulkCohort( req, res ) {
         updatedAt: new Date().toISOString(),
       };
 
-      const result = await insertWithId( 'tango-audio-config', cohortId, cohortData );
+      const result = await insertWithId( 'tango-audio-config-test', cohortId, cohortData );
       logger.info( { result } );
 
       if ( result && result.body && result.body.result === 'created' ) {
@@ -118,7 +119,7 @@ export async function updateCohort( req, res ) {
       doc_as_upsert: true,
     };
 
-    const result = await updateOpenSearchData( 'tango-audio-config', cohortId, updatePayload );
+    const result = await updateOpenSearchData( 'tango-audio-config-test', cohortId, updatePayload );
     logger.info( { result } );
 
     if ( result && result.body && result.body.result === 'updated' ) {
@@ -144,7 +145,7 @@ export async function deleteCohort( req, res ) {
       },
     };
 
-    const result = await updateOpenSearchData( 'tango-audio-config', cohortId, updatePayload );
+    const result = await updateOpenSearchData( 'tango-audio-config-test', cohortId, updatePayload );
     logger.info( { result } );
 
     if ( result && result.body && result.body.result === 'updated' ) {
@@ -169,7 +170,7 @@ export async function getCohort( req, res ) {
       },
     };
 
-    const result = await searchOpenSearchData( 'tango-audio-config', query );
+    const result = await searchOpenSearchData( 'tango-audio-config-test', query );
     logger.info( { result } );
 
     if ( !result || result?.body?.hits?.hits?.length === 0 ) {
@@ -204,7 +205,7 @@ export async function listCohortsByClient( req, res ) {
       sort: [ { createdAt: { order: 'desc' } } ],
     };
 
-    const result = await searchOpenSearchData( 'tango-audio-config', query );
+    const result = await searchOpenSearchData( 'tango-audio-config-test', query );
     logger.info( { result } );
 
     const total = result?.body?.hits?.total?.value || 0;
@@ -393,6 +394,52 @@ export const callExternalStreamAPI = async ( req, res ) => {
   }
 };
 
+export const getGeminiResponse = async ( req, res ) => {
+  try {
+    const { context, topic, sessionData } = req.body;
+
+    // Build a prompt to get suggestions from Gemini
+    const suggestionPrompt = `Based on the following context, provide helpful suggestion prompts that the user could ask for autocomplete:
+Context: ${context || 'General assistance'}
+${topic ? `Topic: ${topic}` : ''}
+${sessionData ? `Session Information: ${JSON.stringify( sessionData )}` : ''}
+
+Generate 5-8 concise and relevant suggestion prompts. Format exactly as a JSON array of objects with this structure:
+[
+  {"id": 1, "text": "suggestion text here", "category": "category_name"},
+  {"id": 2, "text": "suggestion text here", "category": "category_name"}
+]
+Only return the JSON array, no other text.`;
+
+    const genAI = new GoogleGenerativeAI( process.env.GOOGLE_API_KEY );
+
+    const model = genAI.getGenerativeModel( {
+      model: 'gemini-2.5-pro',
+    } );
+
+    const result = await model.generateContent( suggestionPrompt );
+    const responseText = result.response.text().trim();
+    logger.info( { responseText } );
+    // Parse the JSON array response
+    let suggestionsArray = [];
+    suggestionsArray = JSON.parse( responseText );
+    logger.info( { suggestionsArray } );
+
+    if ( !Array.isArray( suggestionsArray ) ) {
+      suggestionsArray = [];
+    }
+
+
+    logger.info( { message: 'Gemini autocomplete suggestions generated', context, suggestionCount: suggestionsArray.length } );
+
+    return res.sendSuccess( { suggestions: suggestionsArray } );
+  } catch ( error ) {
+    logger.error( `Error getting Gemini suggestions: ${error.message}` );
+    const err = error.message || 'Internal server error while getting Gemini suggestions';
+    return res.sendError( err, 500 );
+  }
+};
+
 // ======================= CHAT STREAM API =======================
 
 const CHAT_STREAM_API = 'http://13.232.38.210:8000/api/chat/stream';

package/src/dtos/audioAnalytics.dtos.js

@@ -716,3 +716,30 @@ const getChatSchema = joi.object( {
 export const getChatValid = {
   body: getChatSchema,
 };
+
+// ======================= GEMINI SUGGESTION PROMPTS API SCHEMA =======================
+
+/**
+ * Gemini Suggestion Prompts Request Schema
+ * For generating autocomplete suggestions based on context using Gemini AI
+ *
+ * Response format:
+ * {
+ *   "status": "success",
+ *   "data": {
+ *     "suggestions": [
+ *       { "id": 1, "text": "suggestion text", "category": "category_name" },
+ *       { "id": 2, "text": "suggestion text", "category": "category_name" }
+ *     ]
+ *   }
+ * }
+ */
+const getGeminiResponseSchema = joi.object( {
+  context: joi.string().optional().description( 'Context for generating suggestions' ),
+  topic: joi.string().optional().description( 'Topic for suggestion generation' ),
+  sessionData: joi.object().optional().unknown( true ).description( 'Session data for context' ),
+} ).strict();
+
+export const getGeminiResponseValid = {
+  body: getGeminiResponseSchema,
+};

package/src/routes/audioAnalytics.routes.js

@@ -2,8 +2,8 @@
 import express from 'express';
 import { validate } from 'tango-app-api-middleware';
 import { createCohortValid, createBulkCohortValid, updateCohortValid, getCohortValid, listCohortsByClientValid, deleteCohortValid } from '../dtos/audioAnalytics.dtos.js';
-import { cohortAnalysisCardValid, conversationsListValid, conversationDetailsValid, chatStreamValid, chatHistoryListValid, getChatValid } from '../dtos/audioAnalytics.dtos.js';
-import { createCohort, createBulkCohort, updateCohort, deleteCohort, getCohort, listCohortsByClient, chatStream, chatHistoryList, getChat } from '../controllers/audioAnalytics.controller.js';
+import { cohortAnalysisCardValid, conversationsListValid, conversationDetailsValid, chatStreamValid, chatHistoryListValid, getChatValid, getGeminiResponseValid } from '../dtos/audioAnalytics.dtos.js';
+import { createCohort, createBulkCohort, updateCohort, deleteCohort, getCohort, listCohortsByClient, chatStream, chatHistoryList, getChat, getGeminiResponse } from '../controllers/audioAnalytics.controller.js';
 import { getCohortAnalysisCard } from '../controllers/cohortAnalytics.controller.js';
 import { getConversationsList, getConversationDetails } from '../controllers/conversationAnalytics.controller.js';
 
@@ -35,5 +35,8 @@ audioAnalyticsrouter.post( '/conversations/:conversationId', validate( conversat
 // Chat Stream API Route
 audioAnalyticsrouter.post( '/chat/stream', validate( chatStreamValid ), chatStream );
 
+// Gemini Suggestion Prompts API Route
+audioAnalyticsrouter.post( '/gemini/suggestions', validate( getGeminiResponseValid ), getGeminiResponse );
+
 
 export default audioAnalyticsrouter;