File size: 2,875 Bytes
5db4cdd d7a0bf7 516d7a6 1083cf8 69b529d d7a0bf7 516d7a6 5db4cdd 53bdc35 5db4cdd 53bdc35 5db4cdd 5471363 516d7a6 91e3360 516d7a6 d7a0bf7 5db4cdd 516d7a6 d7a0bf7 5db4cdd d7a0bf7 516d7a6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
import logging
from langchain_core.tools import tool
from src.services.analysis_service import AnalysisService
import json
import os
from datetime import datetime
from pydantic.v1 import BaseModel, Field
from typing import List, Dict, Any
from src.models import load_all_models
from pymongo import MongoClient
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
class InterviewAnalysisArgs(BaseModel):
"""Arguments for the trigger_interview_analysis tool."""
user_id: str = Field(..., description="The unique identifier for the user, provided in the system prompt.")
job_offer_id: str = Field(..., description="The unique identifier for the job offer, provided in the system prompt.")
job_description: str = Field(..., description="The full JSON string of the job offer description.")
conversation_history: List[Dict[str, Any]] = Field(..., description="The complete conversation history between the user and the agent.")
@tool("trigger_interview_analysis", args_schema=InterviewAnalysisArgs)
def trigger_interview_analysis(user_id: str, job_offer_id: str, job_description: str, conversation_history: List[Dict[str, Any]]):
"""
Call this tool to end the interview and launch the final analysis.
You MUST provide all arguments: user_id, job_offer_id, job_description, and the complete conversation_history.
"""
try:
logger.info(f"Outil 'trigger_interview_analysis' appelé pour user_id: {user_id} et job_offer_id: {job_offer_id}")
if '@' in user_id or ' ' in job_offer_id:
logger.error(f"Appel de l'outil avec des données invalides. User ID: {user_id}, Job Offer ID: {job_offer_id}")
return "Erreur: Appel de l'outil avec des paramètres invalides. L'analyse n'a pas pu être lancée."
mongo_client = MongoClient(os.getenv("MONGO_URI"))
db = mongo_client[os.getenv("MONGO_DB_NAME")]
collection = db[os.getenv("MONGO_FEEDBACK")]
models = load_all_models()
analysis_service = AnalysisService(models=models)
feedback_data = analysis_service.run_analysis(
conversation_history=conversation_history,
job_description=job_description
)
mongo_document = {
"user_id": user_id,
"job_offer_id": job_offer_id,
"feedback_data": feedback_data,
"updated_at": datetime.utcnow()
}
result = collection.insert_one(mongo_document)
logger.info(f"Analyse pour l'utilisateur {user_id} terminée et sauvegardée dans MongoDB avec l'ID: {result.inserted_id}")
return "L'analyse a été déclenchée et terminée avec succès."
except Exception as e:
logger.error(f"Erreur dans l'outil d'analyse : {e}", exc_info=True)
return "Une erreur est survenue lors du lancement de l'analyse." |