mirror of
https://github.com/Ladebeze66/ragflow_preprocess.git
synced 2026-02-04 06:00:27 +01:00
54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
Configuration des agents dans l'application
|
|
"""
|
|
|
|
# Configuration des agents activés/désactivés
|
|
ACTIVE_AGENTS = {
|
|
"ocr": True, # Agent de reconnaissance optique de caractères
|
|
"vision": True, # Agent d'analyse d'images
|
|
"translation": True, # Agent de traduction
|
|
"summary": False, # Agent de résumé (désactivé)
|
|
"rewriter": True # Agent de reformulation
|
|
}
|
|
|
|
# Configuration des modèles par défaut pour chaque agent
|
|
DEFAULT_MODELS = {
|
|
"ocr": None, # OCR utilise Tesseract, pas de modèle LLM
|
|
"vision": "llava:34b-v1.6-fp16", # Modèle par défaut pour l'analyse d'images
|
|
"translation": "mistral:latest", # Modèle par défaut pour la traduction
|
|
"summary": "mistral:latest", # Modèle par défaut pour le résumé
|
|
"rewriter": "mistral:latest" # Modèle par défaut pour la reformulation
|
|
}
|
|
|
|
# Configuration de l'endpoint Ollama par défaut
|
|
DEFAULT_ENDPOINT = "http://217.182.105.173:11434"
|
|
|
|
# Configuration de journalisation détaillée
|
|
VERBOSE_LOGGING = True # Enregistre tous les détails des entrées/sorties
|
|
|
|
def is_agent_enabled(agent_name: str) -> bool:
|
|
"""
|
|
Vérifie si un agent est activé dans la configuration
|
|
|
|
Args:
|
|
agent_name (str): Nom de l'agent à vérifier
|
|
|
|
Returns:
|
|
bool: True si l'agent est activé, False sinon
|
|
"""
|
|
return ACTIVE_AGENTS.get(agent_name, False)
|
|
|
|
def get_default_model(agent_name: str) -> str:
|
|
"""
|
|
Renvoie le modèle par défaut pour un agent donné
|
|
|
|
Args:
|
|
agent_name (str): Nom de l'agent
|
|
|
|
Returns:
|
|
str: Nom du modèle par défaut
|
|
"""
|
|
return DEFAULT_MODELS.get(agent_name, "mistral") |