myavana-bot-test-core 2.0.7 → 2.0.9

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "myavana-bot-test-core",
-    "version": "2.0.7",
+    "version": "2.0.9",
     "description": "Shared bot functionality with enhanced features",
     "main": "index.js",
     "scripts": {

package/src/ai.js

@@ -1,13 +1,14 @@
 // ai.js
 const { genkit } = require('genkit');
-const { googleAI, gemini15Flash, gemini20FlashExp, gemini } = require('@genkit-ai/googleai');
+const { googleAI } = require('@genkit-ai/googleai');
+
 const config = require('./config');
 
 const ai = genkit({
     plugins: [googleAI({
         apiKey: config.genkitApiKey,
     })],
-    model: gemini('gemini-3-flash-preview'),
+    model: googleAI.model('gemini-3-flash-preview'),
 });
 
 module.exports = ai;

package/src/conversation.js

@@ -1,7 +1,7 @@
 // conversation.js
 const { pgClient, redisClient } = require('./database');
 const ai = require('./ai');
-const {gemini15Flash, gemini} = require("@genkit-ai/googleai");
+import { googleAI } from '@genkit-ai/google-genai';
 
 const saveConversationSummary = async (conversationId, userId, summary) => {
     const query = `
@@ -28,7 +28,7 @@ const generateConversationSummary = async (chatHistory) => {
     }
     const summaryPrompt = `Summarize the following conversation, getting all important information, especially the user's hair details, name, and location: ${JSON.stringify(chatHistory)}`;
     const { text } = await ai.chat({
-        model: gemini('gemini-2.5-flash'),
+        model: googleAI.model('gemini-2.5-flash'),
         system: summaryPrompt,
     });
     return text;

package/src/postResponseProcessing.js

@@ -1,6 +1,6 @@
 // postResponseProcessing.js
 const ai = require("./ai");
-const {gemini15Flash, gemini} = require("@genkit-ai/googleai");
+import { googleAI } from '@genkit-ai/google-genai';
 const DatabaseSessionStore = require("./session");
 const { pgClient, redisClient } = require('./database');
 const {generateAndSaveSummary, saveConversationSummary, saveConversation} = require("./conversation");
@@ -226,7 +226,7 @@ const postProcessConversation = async (chatHistory, message) => {
     //console.log('Extraction prompt: ', extractionPrompt);
     try {
         const chat = ai.chat({
-            model: gemini('gemini-2.5-flash'),
+            model: googleAI.model('gemini-2.5-flash'),
             config: {
                 temperature: 1.1,
             },
@@ -385,7 +385,7 @@ const postResponseProductCheck = async (chatHistory, message) => {
     // Try Gemini first
     try {
         const chat = ai.chat({
-            model: gemini('gemini-2.5-flash'), // or gemini20FlashExp
+            model: googleAI.model('gemini-2.5-flash'),
             config: {
                 temperature: 0.3, // Reduced temperature for more consistent JSON formatting
             },

package/src/unifiedChatHandler.js

@@ -1,6 +1,6 @@
 // unifiedChatHandler.js - Fixed version with proper model prioritization
 const { genkit, z } = require('genkit');
-const { gemini15Flash, gemini15Pro, gemini20FlashExp, gemini } = require('@genkit-ai/googleai');
+import { googleAI } from '@genkit-ai/google-genai';
 
 class UnifiedChatHandler {
     constructor(config) {
@@ -17,9 +17,9 @@ class UnifiedChatHandler {
         
         // Initialize model priority list - GEMINI FIRST!
         this.modelPriority = [
-            { name: 'gemini20FlashExp', model: gemini('gemini-3-flash-preview'), type: 'primary' },
-            { name: 'gemini15Flash', model: gemini('gemini-2.5-flash'), type: 'primary' },
-            { name: 'gemini15Pro', model: gemini('gemini-3-pro-preview'), type: 'backup' }
+            { name: 'gemini20FlashExp', model: googleAI.model('gemini-3-flash-preview'), type: 'primary' },
+            { name: 'gemini15Flash', model: googleAI.model('gemini-2.5-flash'), type: 'primary' },
+            { name: 'gemini15Pro', model: googleAI.model('gemini-3-pro-preview'), type: 'backup' }
         ];
 
         // Only initialize xAI as last resort if API key exists