feat: (BETA) Amélioration de l'assistant IA logisticien (Gemini) et support des documents
- **Amélioration de l'IA (Cloud Functions)** :
- Mise à jour du modèle vers `gemini-3.1-flash-lite` et augmentation de la limite des résultats de recherche à 50.
- Optimisation de la gestion des outils : augmentation du nombre d'appels simultanés (`MAX_TOOL_CALLS_PER_ITERATION`) à 40.
- Refonte du système de recherche d'équipements avec une stratégie en deux passes (recherche précise puis catégorielle avec normalisation agressive).
- Nouvelles consignes strictes pour la gestion des unités uniques (quantité de 1 par ID) et priorité aux flight cases (containers).
- Ajout d'une gestion de retry avec temporisation pour les erreurs de quota (429) et de surcharge (503).
- Support de l'analyse de documents joints (devis, listes) envoyés en `inlineData`.
- **Interface de l'Assistant (`AiEquipmentAssistantDialog`)** :
- Ajout de la possibilité de joindre des documents (PDF, images, texte) via `FilePicker` pour analyse par l'IA.
- Implémentation d'une vue de logs de debug détaillée pour suivre le raisonnement de l'IA et les appels d'outils.
- Amélioration visuelle de la discussion : bulles de message stylisées et structuration automatique des réponses (sections "Matériel ajouté" vs "Matériel non trouvé").
- Nouvelles options de confirmation : "Tout ajouter" ou "Ajouter sans alternatives".
- **Modèles et Services** :
- Mise à jour de `EventEquipment` pour inclure un champ `rationale` (justification du choix de l'équipement).
- Correction dans `EventAssignedEquipmentSection` pour ajouter automatiquement les équipements enfants lors de l'ajout d'un container proposé par l'IA.
- Ajout de la gestion des logs et des documents dans `AiEquipmentAssistantService`.
- **UI Divers** :
- Mise à jour de `EquipmentFormPage` pour clarifier le comportement de l'identifiant (auto-génération recommandée).
This commit is contained in:
ElPoyo
@@ -9,13 +9,13 @@ const { GoogleGenerativeAI } = require('@google/generative-ai');
|
||||
const admin = require('firebase-admin');
|
||||
const logger = require('firebase-functions/logger');
|
||||
|
||||
const GEMINI_MODEL = 'gemini-3.1-flash-lite-preview';
|
||||
const GEMINI_MODEL = 'gemini-3.1-flash-lite'; //Ne pas changer de versio, celle ci existe
|
||||
const GEMINI_API_KEY = 'AIzaSyB0hOvBjWeWjdrxVARzfErZ_uGuArlvmQc';
|
||||
const MAX_TOOL_ITERATIONS = 20;
|
||||
const PAST_EVENTS_LIMIT = 5;
|
||||
const SEARCH_RESULTS_LIMIT = 20;
|
||||
const SEARCH_RESULTS_LIMIT = 50; // Augmenté pour permettre de trouver plusieurs unités du même modèle
|
||||
const EVENT_SEARCH_SCAN_LIMIT = 100;
|
||||
const MAX_TOOL_CALLS_PER_ITERATION = 5;
|
||||
const MAX_TOOL_CALLS_PER_ITERATION = 40;
|
||||
const AVAILABILITY_EVENTS_SCAN_LIMIT = 500;
|
||||
const MAX_BATCH_AVAILABILITY_ITEMS = 50;
|
||||
|
||||
@@ -25,7 +25,6 @@ const getDb = () => admin.firestore();
|
||||
// ============================================================================
|
||||
// Déclarations des tools Gemini
|
||||
// ============================================================================
|
||||
|
||||
const AI_TOOLS = [
|
||||
{
|
||||
functionDeclarations: [
|
||||
@@ -167,33 +166,58 @@ const AI_TOOLS = [
|
||||
const SYSTEM_PROMPT = `Tu es un expert logisticien IA pour la gestion de materiel evenementiel.
|
||||
Tu dois proposer une liste de materiel et de flight cases adaptee a l evenement decrit.
|
||||
|
||||
Regles absolues :
|
||||
- Tu ne dois JAMAIS ecrire en base de donnees.
|
||||
- Utilise search_equipment pour trouver du materiel, check_availability_batch en priorite pour verifier la disponibilite, check_availability pour un cas isole, get_past_events pour t inspirer.
|
||||
- Si une recherche precise echoue, utilise list_equipment_by_category pour explorer les categories ou trouver corriger d'eventuelles fautes de frappe de l'utilisateur.
|
||||
- Si l utilisateur cite un evenement precis (nom/date), appelle d abord search_event_reference pour retrouver cet evenement et reutiliser son materiel ET ses flight cases.
|
||||
- La sous-categorie du materiel est tres importante. Prends-la en compte en priorite.
|
||||
- Sois libre d'ajouter du materiel pertinent par rapport aux options ou a la description de l'evenement si cela te semble justifie. Explique tes choix dans rationale.
|
||||
Regles sur les UNITES et QUANTITES (CRITIQUE) :
|
||||
- La plupart des equipements (Lumiere, Son, Video, Structure) sont des UNITES UNIQUES. Chaque ID (ex: "BARRE_LED_#1") represente un seul appareil physique.
|
||||
- Si l'utilisateur demande 6 barres LED, tu dois trouver et ajouter 6 IDs DIFFERENTS (ex: "BARRE_LED_#1", "BARRE_LED_#2", etc.) dans proposal.items, chacun avec "quantity": 1.
|
||||
- Ne jamais mettre "quantity": 6 pour un item dont totalQuantity est 1 ou null.
|
||||
- Les seules exceptions sont les categories "CONSUMABLE" (Consommable) et "CABLE" (Cable) qui peuvent avoir une quantite > 1 pour un seul ID.
|
||||
|
||||
Regles sur les flight cases (PRIORITAIRES) :
|
||||
- Apres avoir identifie les equipements necessaires, appelle TOUJOURS search_containers avec la liste de leurs IDs.
|
||||
- Si un flight case contient tous les equipements necessaires, propose le flight case uniquement. Ne mets PAS ces equipements dans items.
|
||||
- Si un flight case contient plus d equipements que necessaire (ex: 5 blinders dans le case mais on en veut 2), propose les equipements individuellement, pas le flight case.
|
||||
- N ajoute du materiel individuel dans items que pour les equipements sans flight case couvrant exactement le besoin.
|
||||
- Si un equipement est indisponible, cherche une alternative via search_equipment, verifie sa dispo, puis search_containers sur l alternative.
|
||||
- Verifie la disponibilite de CHAQUE equipement ET container propose.
|
||||
- Quand une proposition courante existe deja, renvoie toujours la LISTE FINALE COMPLETE.
|
||||
Regles sur les flight cases (PRIORITAIRE) :
|
||||
- Les flight cases (containers) sont le moyen privilege de transporter le materiel.
|
||||
- Strategie :
|
||||
1. Identifie d'abord tous les equipements individuels necessaires.
|
||||
2. Appelle search_containers avec la liste de TOUS ces IDs d'equipements.
|
||||
3. Si un ou plusieurs flight cases couvrent une partie ou la totalite du materiel :
|
||||
- Ajoute les flight cases dans "proposal.containers".
|
||||
- RETIRE les equipements contenus dans ces flight cases de ta liste "proposal.items".
|
||||
- Les equipements restants (non contenus dans des flight cases) doivent rester dans "proposal.items".
|
||||
- Un flight case est toujours preferable a des items individuels.
|
||||
|
||||
Regles de recherche (STRATEGIE EN DEUX PASSES) :
|
||||
1. PREMIER PASSE (RECHERCHE PRECISE) :
|
||||
- Cherche chaque item demande avec son nom complet (ex: "dB Technologies Ignenia IG5TR").
|
||||
- Si ca ne donne rien, REESSAYE immediatement avec des mots-clefs plus courts (ex: "Ignenia", "IG5TR", "VIO S118").
|
||||
- Sois perseverant : les noms en base sont souvent au format "MARQUE_MODELE_#1".
|
||||
|
||||
2. DEUXIEME PASSE (ANALYSE CONTEXTUELLE ET CATEGORIELLE) :
|
||||
- Si la recherche precise echoue, analyse le contexte de l'item dans le devis ou la demande.
|
||||
- Identifie la categorie probable (ex: "dB Technologies" -> "SONO" ou "SOUND", "Robert Juliat" -> "LUMIERE" ou "LIGHTING").
|
||||
- Utilise list_equipment_by_category pour recuperer tout le materiel de cette categorie.
|
||||
- Parcoure la liste recuperee pour trouver un item qui "match" logiquement (ex: l'utilisateur demande "Opera 15" mais l'identifiant est "DBTE_OPERA_15_#1").
|
||||
- Si tu trouves une correspondance logique dans la liste de la categorie, utilise l'ID exact trouve en base.
|
||||
|
||||
Regles de transparence (OBLIGATOIRE) :
|
||||
- Ton assistantMessage doit TOUJOURS contenir deux sections claires et distinctes si applicable :
|
||||
1. "Matériel ajouté :" (Utilise exactement ce titre) suivi de la liste des équipements et flight cases que tu as réussi à trouver et qui sont disponibles. Explique brièvement pourquoi.
|
||||
2. "Matériel non trouvé ou indisponible :" (Utilise exactement ce titre) suivi de la liste précise des éléments de la demande utilisateur que tu n'as pas pu intégrer. Pour chaque élément, précise s'il est "Absent de la base de données" ou "Indisponible (rupture de stock)".
|
||||
- N'hésite pas à être pédagogue dans tes explications, mais garde ces titres de section exacts pour permettre un affichage structuré dans l'application.
|
||||
|
||||
Regles absolues :
|
||||
- Tu ne dois JAMAIS ecrire en base de donnees.
|
||||
- Si l utilisateur cite un evenement precis (nom/date), appelle d abord search_event_reference.
|
||||
- Rend toujours la LISTE FINALE COMPLETE y compris le materiel deja assigne s'il est toujours pertinent.
|
||||
- Verifie la disponibilite de CHAQUE equipement ET container propose via check_availability_batch.
|
||||
- Si un materiel est en rupture de stock, cherche une alternative (autre ID du meme modele ou autre modele similaire). Si aucune alternative n'est trouvee, deplace l'item dans la section "indisponible".
|
||||
|
||||
Reponse finale : du JSON valide strict, sans markdown, avec ce format exact :
|
||||
{"assistantMessage":"...","proposal":{"summary":"...","containers":[{"containerId":"ID-EXACT","rationale":"..."}],"items":[{"equipmentId":"ID-EXACT","quantity":1,"rationale":"..."}]}}
|
||||
- containers : flight cases proposes (les equipements couverts par un container ne doivent pas etre dans items)
|
||||
- items : equipements individuels non couverts par un container
|
||||
- containers : flight cases proposes.
|
||||
- items : equipements individuels NON contenus dans les flight cases proposes ci-dessus.
|
||||
- Si impossible, renvoie proposal a null.`;
|
||||
|
||||
// ============================================================================
|
||||
// Implémentation des tools côté serveur (Firestore Admin SDK)
|
||||
// ============================================================================
|
||||
|
||||
/**
|
||||
* Recherche des équipements dans Firestore par mot-clé.
|
||||
*/
|
||||
@@ -202,26 +226,37 @@ async function toolSearchEquipment(query) {
|
||||
return { query, count: 0, results: [] };
|
||||
}
|
||||
|
||||
const normalizedQuery = query.trim().toLowerCase();
|
||||
// Normalisation agressive pour la recherche (sans accents, sans ponctuation)
|
||||
const normalize = (str) =>
|
||||
(str || '')
|
||||
.toLowerCase()
|
||||
.normalize('NFD')
|
||||
.replace(/[\u0300-\u036f]/g, '')
|
||||
.replace(/[^a-z0-9]/g, '');
|
||||
|
||||
const snapshot = await getDb().collection('equipments').limit(200).get();
|
||||
const normalizedQuery = normalize(query);
|
||||
|
||||
const snapshot = await getDb().collection('equipments').limit(500).get();
|
||||
const results = [];
|
||||
|
||||
snapshot.docs.forEach((doc) => {
|
||||
const data = doc.data();
|
||||
const searchableText = [
|
||||
const searchableFields = [
|
||||
data.name,
|
||||
doc.id,
|
||||
data.model,
|
||||
data.brand,
|
||||
data.category,
|
||||
data.subCategory,
|
||||
]
|
||||
.filter(Boolean)
|
||||
.join(' ')
|
||||
.toLowerCase();
|
||||
];
|
||||
|
||||
if (searchableText.includes(normalizedQuery)) {
|
||||
const isMatch = searchableFields.some(field => {
|
||||
if (!field) return false;
|
||||
const normalizedField = normalize(String(field));
|
||||
return normalizedField.includes(normalizedQuery) || normalizedQuery.includes(normalizedField);
|
||||
});
|
||||
|
||||
if (isMatch) {
|
||||
results.push({
|
||||
id: doc.id,
|
||||
name: data.name || doc.id,
|
||||
@@ -837,28 +872,35 @@ async function toolSearchEventReference(query, dateHint) {
|
||||
async function toolListEquipmentByCategory(category, subCategory) {
|
||||
let queryDb = getDb().collection('equipments');
|
||||
|
||||
if (category) {
|
||||
queryDb = queryDb.where('category', '==', category);
|
||||
}
|
||||
// Si on a des critères précis, on filtre. Sinon on scanne pour matcher flou.
|
||||
const snapshot = await queryDb.limit(1000).get();
|
||||
|
||||
if (subCategory) {
|
||||
queryDb = queryDb.where('subCategory', '==', subCategory);
|
||||
}
|
||||
const normalize = (str) =>
|
||||
(str || '').toLowerCase().normalize('NFD').replace(/[\u0300-\u036f]/g, '').trim();
|
||||
|
||||
const snapshot = await queryDb.limit(50).get();
|
||||
const normCat = category ? normalize(category) : null;
|
||||
const normSub = subCategory ? normalize(subCategory) : null;
|
||||
|
||||
const results = snapshot.docs.map((doc) => {
|
||||
const data = doc.data();
|
||||
return {
|
||||
id: doc.id,
|
||||
name: data.name || doc.id,
|
||||
category: data.category || '',
|
||||
subCategory: data.subCategory || '',
|
||||
brand: data.brand || null,
|
||||
model: data.model || null,
|
||||
status: data.status || '',
|
||||
};
|
||||
});
|
||||
const results = snapshot.docs
|
||||
.map((doc) => {
|
||||
const data = doc.data();
|
||||
return {
|
||||
id: doc.id,
|
||||
name: data.name || doc.id,
|
||||
category: data.category || '',
|
||||
subCategory: data.subCategory || '',
|
||||
brand: data.brand || null,
|
||||
model: data.model || null,
|
||||
status: data.status || '',
|
||||
};
|
||||
})
|
||||
.filter(item => {
|
||||
let match = true;
|
||||
if (normCat) match = match && normalize(item.category).includes(normCat);
|
||||
if (normSub) match = match && normalize(item.subCategory).includes(normSub);
|
||||
return match;
|
||||
})
|
||||
.slice(0, 50);
|
||||
|
||||
return {
|
||||
category: category || 'all',
|
||||
@@ -922,7 +964,6 @@ async function executeTool(toolCall, excludeEventId, sharedContext) {
|
||||
// ============================================================================
|
||||
// Gestionnaire principal
|
||||
// ============================================================================
|
||||
|
||||
/**
|
||||
* Construit le prompt utilisateur avec le contexte de l'événement.
|
||||
*/
|
||||
@@ -965,6 +1006,7 @@ function buildUserPrompt({
|
||||
'Demande :',
|
||||
finalMessage,
|
||||
'',
|
||||
'Si un fichier (document/devis) a été joint à ma demande (sous forme inlineData by Gemini), tu dois absolument l analyser, en deduire le materiel demande, et l ajouter a ta proposition.',
|
||||
'Si la demande cite un evenement precis (nom/date), commence par search_event_reference avant de proposer du materiel.',
|
||||
'Si une proposition courante existe, traite-la comme base de travail et renvoie toujours la liste finale complete apres modification.',
|
||||
'Rappel : vérifier la disponibilité avant de recommander. Privilégier check_availability_batch pour contrôler plusieurs équipements en un appel.',
|
||||
@@ -972,19 +1014,56 @@ function buildUserPrompt({
|
||||
].join('\n');
|
||||
}
|
||||
|
||||
/**
|
||||
* Construit un tableau de parts pour Gemini permettant d'inclure des documents.
|
||||
*/
|
||||
function buildMessageParts(params, logDebug) {
|
||||
const textPrompt = buildUserPrompt(params);
|
||||
const parts = [{ text: textPrompt }];
|
||||
|
||||
if (params.document && params.document.data && params.document.mimeType) {
|
||||
if (logDebug) logDebug(`[AI] Attaching document ${params.document.mimeType}`);
|
||||
else logger.info(`[AI] Attaching document ${params.document.mimeType}`);
|
||||
parts.push({
|
||||
inlineData: {
|
||||
data: params.document.data,
|
||||
mimeType: params.document.mimeType,
|
||||
}
|
||||
});
|
||||
const fileNameInfo = params.document.fileName ? ` (nom du fichier: ${params.document.fileName})` : ``;
|
||||
parts.push({
|
||||
text: `J'ai joint un document de type ${params.document.mimeType}${fileNameInfo}. Merci d'extraire le materiel et de l'ajouter a la proposition.`,
|
||||
});
|
||||
}
|
||||
|
||||
return parts;
|
||||
}
|
||||
|
||||
/**
|
||||
* Extrait un texte exploitable depuis la réponse Gemini, même si response.text() est vide.
|
||||
*/
|
||||
function extractResponseText(modelResponse) {
|
||||
if (!modelResponse) return '';
|
||||
function extractResponseText(modelResponse, logDebug, logError) {
|
||||
const info = logDebug || logger.info.bind(logger);
|
||||
const warn = logError || logger.warn.bind(logger);
|
||||
|
||||
if (!modelResponse) {
|
||||
warn('[AI] extractResponseText: modelResponse est vide ou undefined');
|
||||
return '';
|
||||
}
|
||||
|
||||
info('[AI] extractResponseText: analyse de la réponse brut', {
|
||||
hasTextFunc: typeof modelResponse.text === 'function',
|
||||
candidatesCount: modelResponse.candidates?.length || 0
|
||||
});
|
||||
|
||||
try {
|
||||
const directText = modelResponse.text?.();
|
||||
if (directText && directText.trim().length > 0) {
|
||||
info('[AI] extractResponseText: Texte extrait via modelResponse.text()');
|
||||
return directText.trim();
|
||||
}
|
||||
} catch (_) {
|
||||
// Fallback via candidates/parts ci-dessous.
|
||||
} catch (err) {
|
||||
warn('[AI] extractResponseText: erreur lors de l appel a modelResponse.text()', err?.message);
|
||||
}
|
||||
|
||||
const candidates = Array.isArray(modelResponse.candidates) ? modelResponse.candidates : [];
|
||||
@@ -1001,17 +1080,24 @@ function extractResponseText(modelResponse) {
|
||||
}
|
||||
}
|
||||
|
||||
return textParts.join('\n').trim();
|
||||
const fallbackText = textParts.join('\n').trim();
|
||||
info(`[AI] extractResponseText: Texte extrait en fallback (longueur: ${fallbackText.length})`);
|
||||
return fallbackText;
|
||||
}
|
||||
|
||||
/**
|
||||
* Extrait et parse le JSON de la réponse IA.
|
||||
*/
|
||||
function parseAiResponse(rawText) {
|
||||
function parseAiResponse(rawText, logDebug, logError) {
|
||||
const info = logDebug || logger.info.bind(logger);
|
||||
const err = logError || logger.error.bind(logger);
|
||||
|
||||
if (!rawText || rawText.trim().length === 0) {
|
||||
err('[AI] parseAiResponse: rawText recu est vide !');
|
||||
throw new Error('Réponse IA vide.');
|
||||
}
|
||||
|
||||
info('[AI] parseAiResponse tentatives de parsing sur le texte (apercu):', rawText.substring(0, 100) + '...');
|
||||
// Tentative directe
|
||||
try {
|
||||
const parsed = JSON.parse(rawText.trim());
|
||||
@@ -1045,10 +1131,55 @@ function parseAiResponse(rawText) {
|
||||
throw new Error('JSON IA invalide ou introuvable dans la réponse.');
|
||||
}
|
||||
|
||||
/**
|
||||
* Envoie un message à Gemini avec une temporisation en cas de Rate Limit (429).
|
||||
*/
|
||||
async function sendMessageWithRetry(chat, messagePayload, logDebug, logError) {
|
||||
const maxRetries = 5;
|
||||
|
||||
for (let attempt = 1; attempt <= maxRetries; attempt++) {
|
||||
try {
|
||||
return await chat.sendMessage(messagePayload);
|
||||
} catch (error) {
|
||||
const status = error?.status || error?.response?.status;
|
||||
const errorMessage = error.message || '';
|
||||
const isRateLimit = status === 429 || errorMessage.includes('429') || errorMessage.includes('quota');
|
||||
const isOverloaded = status === 503 || errorMessage.includes('503') || errorMessage.includes('overloaded') || errorMessage.includes('high demand');
|
||||
|
||||
if ((isRateLimit || isOverloaded) && attempt < maxRetries) {
|
||||
let delayMs = isOverloaded ? 5000 : 30000; // Plus court pour 503, long pour 429
|
||||
|
||||
const retryMatch = errorMessage.match(/retry in\s+([\d.]+)\s*s/i);
|
||||
if (retryMatch && !isNaN(parseFloat(retryMatch[1]))) {
|
||||
delayMs = Math.ceil(parseFloat(retryMatch[1]) * 1000) + 2000;
|
||||
}
|
||||
|
||||
const reason = isOverloaded ? 'Modèle surchargé (503)' : 'Quota atteint (429)';
|
||||
logDebug(`[AI] ${reason}. Temporisation de ${delayMs / 1000}s avant la tentative ${attempt + 1}/${maxRetries}...`);
|
||||
await new Promise((resolve) => setTimeout(resolve, delayMs));
|
||||
} else {
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Handler principal de la Cloud Function aiEquipmentProposal.
|
||||
*/
|
||||
async function handleAiEquipmentProposal(req, res) {
|
||||
const debugLogs = [];
|
||||
const logDebug = (msg, data) => {
|
||||
const dataStr = data ? ` | ${JSON.stringify(data)}` : '';
|
||||
debugLogs.push(`[INFO] ${msg}${dataStr}`);
|
||||
logger.info(msg, data);
|
||||
};
|
||||
const logError = (msg, data) => {
|
||||
const dataStr = data ? ` | ${JSON.stringify(data)}` : '';
|
||||
debugLogs.push(`[ERROR] ${msg}${dataStr}`);
|
||||
logger.error(msg, data);
|
||||
};
|
||||
|
||||
const {
|
||||
eventName,
|
||||
eventTypeId,
|
||||
@@ -1062,6 +1193,7 @@ async function handleAiEquipmentProposal(req, res) {
|
||||
currentEquipment = [],
|
||||
workingProposal = [],
|
||||
excludeEventId,
|
||||
document,
|
||||
} = req.body.data || {};
|
||||
|
||||
if (!startDate || !endDate) {
|
||||
@@ -1075,9 +1207,11 @@ async function handleAiEquipmentProposal(req, res) {
|
||||
systemInstruction: SYSTEM_PROMPT,
|
||||
tools: AI_TOOLS,
|
||||
toolConfig: { functionCallingConfig: { mode: 'AUTO' } },
|
||||
generationConfig: { temperature: 0.2, responseMimeType: 'application/json' },
|
||||
generationConfig: { temperature: 0.2 },
|
||||
});
|
||||
|
||||
logDebug('[AI] Reconstruction de l historique de conversation. Elements :', history?.length || 0);
|
||||
|
||||
// Reconstruire l'historique de conversation
|
||||
const chatHistory = (history || [])
|
||||
.filter((turn) => turn.text && turn.text.trim().length > 0)
|
||||
@@ -1086,13 +1220,15 @@ async function handleAiEquipmentProposal(req, res) {
|
||||
parts: [{ text: turn.text.trim() }],
|
||||
}));
|
||||
|
||||
logDebug('[AI] Historique formaté :', chatHistory);
|
||||
|
||||
const chat = model.startChat({ history: chatHistory });
|
||||
const toolResultCache = new Map();
|
||||
const sharedToolContext = {
|
||||
availabilityCandidatesByWindow: new Map(),
|
||||
};
|
||||
|
||||
const userPrompt = buildUserPrompt({
|
||||
const messageParts = buildMessageParts({
|
||||
userMessage,
|
||||
eventName,
|
||||
eventTypeId,
|
||||
@@ -1103,25 +1239,38 @@ async function handleAiEquipmentProposal(req, res) {
|
||||
eventOptions,
|
||||
currentEquipment,
|
||||
workingProposal,
|
||||
document,
|
||||
}, logDebug);
|
||||
|
||||
// Sécuriser les logs de base64
|
||||
const safeMessagePartsForLogs = JSON.parse(JSON.stringify(messageParts));
|
||||
safeMessagePartsForLogs.forEach(part => {
|
||||
if (part.inlineData && part.inlineData.data) {
|
||||
part.inlineData.data = `<BASE64 TRONQUE : ${part.inlineData.data.length} chars>`;
|
||||
}
|
||||
});
|
||||
|
||||
logger.info('[AI] Starting conversation', { eventTypeId, startDate, endDate });
|
||||
logDebug('[AI] Starting conversation. Parts envoys:', safeMessagePartsForLogs);
|
||||
|
||||
let response;
|
||||
|
||||
try {
|
||||
response = await chat.sendMessage(userPrompt);
|
||||
response = await sendMessageWithRetry(chat, messageParts, logDebug, logError);
|
||||
logDebug('[AI] Premiere reponse recue du sendMessage.');
|
||||
|
||||
// Boucle de function calling avec cache local.
|
||||
for (let iteration = 0; iteration < MAX_TOOL_ITERATIONS; iteration++) {
|
||||
const functionCalls = response.response.functionCalls();
|
||||
|
||||
if (!functionCalls || functionCalls.length === 0) {
|
||||
logDebug(`[AI] (Iteration ${iteration + 1}) Fin des tool calls, l IA a retourné une réponse (ou un appel vide).`);
|
||||
break;
|
||||
}
|
||||
|
||||
const limitedCalls = functionCalls.slice(0, MAX_TOOL_CALLS_PER_ITERATION);
|
||||
const availabilityCalls = limitedCalls.filter(
|
||||
logDebug(`[AI] (Iteration ${iteration + 1}) Tool calls demands par l IA:`, functionCalls);
|
||||
|
||||
const callsToProcess = functionCalls.slice(0, MAX_TOOL_CALLS_PER_ITERATION);
|
||||
const availabilityCalls = callsToProcess.filter(
|
||||
(call) => call.name === 'check_availability'
|
||||
&& call.args?.equipmentId
|
||||
&& call.args?.startDate
|
||||
@@ -1164,7 +1313,7 @@ async function handleAiEquipmentProposal(req, res) {
|
||||
batchResult.results.map((item) => [String(item.equipmentId), item]),
|
||||
);
|
||||
batchWindow = { startDate: firstStartDate, endDate: firstEndDate };
|
||||
logger.info('[AI] Consolidated check_availability calls into one batch call', {
|
||||
logDebug('[AI] Consolidated check_availability calls into one batch call', {
|
||||
iteration: iteration + 1,
|
||||
itemCount: equipmentIds.length,
|
||||
});
|
||||
@@ -1172,10 +1321,19 @@ async function handleAiEquipmentProposal(req, res) {
|
||||
}
|
||||
}
|
||||
|
||||
logger.info(`[AI] Tool calls (iteration ${iteration + 1}):`, limitedCalls.map((c) => c.name));
|
||||
logDebug(`[AI] Tool calls executes. Preparation du renvoi des resultats...`);
|
||||
|
||||
const toolResults = await Promise.all(
|
||||
limitedCalls.map(async (toolCall) => {
|
||||
functionCalls.map(async (toolCall, ind) => {
|
||||
if (ind >= MAX_TOOL_CALLS_PER_ITERATION) {
|
||||
return {
|
||||
functionResponse: {
|
||||
name: toolCall.name,
|
||||
response: { error: 'Limite de tool calls simultanes atteinte. (Ignore)' },
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
const cacheKey = buildToolCacheKey(toolCall.name, toolCall.args);
|
||||
if (toolResultCache.has(cacheKey)) {
|
||||
return {
|
||||
@@ -1205,7 +1363,7 @@ async function handleAiEquipmentProposal(req, res) {
|
||||
toolResult = await executeTool(toolCall, excludeEventId, sharedToolContext);
|
||||
}
|
||||
} catch (toolError) {
|
||||
logger.warn('[AI] Tool call failed, returning degraded tool response', {
|
||||
logError('[AI] Tool call failed, returning degraded tool response', {
|
||||
tool: toolCall.name,
|
||||
message: toolError?.message || 'unknown',
|
||||
});
|
||||
@@ -1214,6 +1372,8 @@ async function handleAiEquipmentProposal(req, res) {
|
||||
|
||||
toolResultCache.set(cacheKey, toolResult);
|
||||
|
||||
logDebug(`[AI] Resultat du tool ${toolCall.name} (apercu):`, JSON.stringify(toolResult).substring(0, 150) + '...');
|
||||
|
||||
return {
|
||||
functionResponse: {
|
||||
name: toolCall.name,
|
||||
@@ -1223,24 +1383,30 @@ async function handleAiEquipmentProposal(req, res) {
|
||||
}),
|
||||
);
|
||||
|
||||
response = await chat.sendMessage(toolResults);
|
||||
logDebug(`[AI] Envoi de ${toolResults.length} resultats au modèle pour la suite de la conversation.`);
|
||||
response = await sendMessageWithRetry(chat, toolResults, logDebug, logError);
|
||||
logDebug(`[AI] Reponse recue du modele suite aux resultats des tools.`);
|
||||
}
|
||||
} catch (error) {
|
||||
logger.error('[AI] Conversation timeout/error', {
|
||||
logError('[AI] Conversation timeout/error, DETAIL COMPLET:', {
|
||||
message: error?.message || 'unknown',
|
||||
status: error?.status,
|
||||
details: error?.details,
|
||||
stack: error?.stack
|
||||
});
|
||||
res.status(200).json({
|
||||
assistantMessage: 'Je suis désolé, je n\'ai pas réussi à traiter votre demande en raison d\'un temps trop long ou d\'une erreur technique. Veuillez réessayer avec des requêtes plus spécifiques.',
|
||||
proposal: null,
|
||||
debugLogs,
|
||||
});
|
||||
return;
|
||||
}
|
||||
|
||||
const rawText = extractResponseText(response.response);
|
||||
logger.info('[AI] Raw response received, parsing...', {
|
||||
logDebug('[AI] Structure finale du response.response (apercu):', JSON.stringify(response.response).substring(0, 300) + '...');
|
||||
const rawText = extractResponseText(response.response, logDebug, logError);
|
||||
logDebug('[AI] Raw response extracted:', {
|
||||
hasText: rawText.length > 0,
|
||||
candidateCount: response.response?.candidates?.length || 0,
|
||||
rawTextExcerpt: rawText.substring(0, 200) + '...'
|
||||
});
|
||||
|
||||
// Fallback non bloquant: éviter un 500 quand Gemini ne renvoie pas de texte exploitable.
|
||||
@@ -1248,18 +1414,20 @@ async function handleAiEquipmentProposal(req, res) {
|
||||
res.status(200).json({
|
||||
assistantMessage: 'Je n\'ai pas pu générer une réponse exploitable pour le moment. Réessaie avec une consigne plus précise.',
|
||||
proposal: null,
|
||||
debugLogs,
|
||||
});
|
||||
return;
|
||||
}
|
||||
|
||||
let payload;
|
||||
try {
|
||||
payload = parseAiResponse(rawText);
|
||||
payload = parseAiResponse(rawText, logDebug, logError);
|
||||
} catch (error) {
|
||||
logger.error('[AI] JSON parsing failed, returning degraded response', error);
|
||||
logError('[AI] JSON parsing failed, returning degraded response', error?.message);
|
||||
res.status(200).json({
|
||||
assistantMessage: rawText,
|
||||
proposal: null,
|
||||
debugLogs,
|
||||
});
|
||||
return;
|
||||
}
|
||||
@@ -1296,8 +1464,7 @@ async function handleAiEquipmentProposal(req, res) {
|
||||
}
|
||||
}
|
||||
|
||||
res.status(200).json({ assistantMessage, proposal });
|
||||
res.status(200).json({ assistantMessage, proposal, debugLogs });
|
||||
}
|
||||
|
||||
module.exports = { handleAiEquipmentProposal };
|
||||
|
||||
|
||||
Reference in New Issue
Block a user