Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
e8288731d8 | ||
|
|
e0f504c8ee | ||
|
|
487da2ed4d | ||
|
|
108ba8f167 | ||
|
|
5dea7d3353 | ||
|
|
134312ebff | ||
|
|
83e70eadf6 | ||
|
|
60c0699c57 | ||
|
|
44bf0b868e | ||
|
|
c779c51a03 | ||
|
|
eaa0ff0564 | ||
|
|
ecd9854527 | ||
|
|
f840561123 | ||
|
|
48f929a8f3 | ||
|
|
fd82d77709 | ||
|
|
189a2e4bf6 | ||
|
|
e322435bd8 | ||
|
|
efbecada77 | ||
|
|
4a9a0d8ec6 | ||
|
|
b554c983a6 | ||
|
|
007e723e4e | ||
|
|
95125df3b2 | ||
|
|
36f205a56e | ||
|
|
2eb3acceea | ||
|
|
4d5046e8d0 | ||
|
|
ba2f37cf59 | ||
|
|
4805228f28 | ||
|
|
6e4aee1984 | ||
|
|
b6765a9bc8 | ||
|
|
efdda0ed00 | ||
|
|
8a32c16ff4 | ||
|
|
4ce11d0bfe | ||
|
|
b3d92558fc | ||
|
|
ec37ac06fe | ||
|
|
6f7e3efd6e | ||
|
|
f794937fd8 | ||
|
|
4b80ff404f | ||
|
|
9288e849af |
No files matched your search
@@ -5,56 +5,78 @@ Ce projet est un laboratoire d'intelligence artificielle personnel. Il agit comm
|
|||||||
L'interface visuelle est inspirée de l'ergonomie de Gemini, avec le logo stylisé A38 (cerveau en réseau neuronal néon) comme charte graphique centrale.
|
L'interface visuelle est inspirée de l'ergonomie de Gemini, avec le logo stylisé A38 (cerveau en réseau neuronal néon) comme charte graphique centrale.
|
||||||
|
|
||||||
## 🏗️ Stack Technique & Architecture
|
## 🏗️ Stack Technique & Architecture
|
||||||
- **Backend :** FastAPI (Python) modulaire (une fonction = un module).
|
|
||||||
- **Frontend :** Vue.js + Tailwind CSS (Responsive / Mobile-first).
|
* **Backend :** FastAPI (Python) modulaire asynchrone (httpx, asyncio)[cite: 2].
|
||||||
- **Base de Données :** PostgreSQL (gestion des utilisateurs, historiques, suivi financier).
|
* **Frontend :** Vue.js + Tailwind CSS (Responsive / Mobile-first)[cite: 2].
|
||||||
- **Traitement Documentaire :** LibreOffice Headless (génération/conversion Excel, Word, PDF).
|
* **Base de Données :** PostgreSQL (gestion des utilisateurs, rôles hiérarchiques Super-Admin/Admin/User, historiques, suivi financier)[cite: 2].
|
||||||
- **Déploiement :** Docker, géré via Dockge et Gitea, sécurisé sur grappe RAID.
|
* **Traitement Documentaire :** LibreOffice Headless (génération/conversion Excel, Word, PDF)[cite: 2].
|
||||||
|
* **Déploiement :** Docker, géré via Dockge et Gitea, sécurisé sur grappe RAID[cite: 2].
|
||||||
|
|
||||||
## 📌 Règles de Versioning (SemVer)
|
## 📌 Règles de Versioning (SemVer)
|
||||||
Le développement suit un versioning strict (format `vX.Y.Z`) synchronisé avec les tags Gitea et GitHub, affiché dynamiquement sur l'interface :
|
|
||||||
- **Majeur (X) :** Changement profond d'architecture ou refonte globale.
|
Le développement suit un versioning strict (format `vX.Y.Z`) synchronisé avec les tags Gitea et GitHub, affiché dynamiquement sur l'interface[cite: 2]:
|
||||||
- **Mineur (Y) :** Validation et ajout d'une nouvelle fonctionnalité de la roadmap.
|
|
||||||
- **Patch (Z) :** Corrections de bugs, correctifs de sécurité ou ajustements visuels mineurs.
|
* **Majeur (X) :** Changement profond d'architecture ou refonte globale[cite: 2].
|
||||||
|
* **Mineur (Y) :** Validation et ajout d'une nouvelle fonctionnalité de la roadmap[cite: 2].
|
||||||
|
* **Patch (Z) :** Corrections de bugs, correctifs de sécurité ou ajustements visuels mineurs[cite: 2].
|
||||||
|
|
||||||
|
## 📦 Récupération des Paquets Docker (Tags)
|
||||||
|
|
||||||
|
Les images Docker sont construites automatiquement par les pipelines CI/CD lors de la création d'un tag[cite: 2].
|
||||||
|
|
||||||
|
* **Sur Gitea :** Depuis la page principale de votre dépôt, naviguez dans l'onglet **Packages** (ou Paquets). Vous y trouverez le registre de conteneurs avec toutes les versions taguées (`v0.16.1`, `latest`, etc.)[cite: 2].
|
||||||
|
* **Sur GitHub :** Depuis la page principale du dépôt, regardez dans la colonne de droite la section **Packages**. Cliquez dessus pour accéder au GitHub Container Registry (`ghcr.io/votre-utilisateur/aethas38-multi-ia`) et visualiser la liste des tags disponibles à tirer via Docker/Dockge[cite: 2].
|
||||||
|
|
||||||
## 🚀 ROADMAP & CAHIER DES CHARGES
|
## 🚀 ROADMAP & CAHIER DES CHARGES
|
||||||
|
|
||||||
### Sécurité & Infrastructure
|
### Sécurité & Infrastructure
|
||||||
- [x] Installation stricte via Dockge et Gitea (création automatique des dossiers inclus).
|
|
||||||
- [x] Sécurisation de la totalité du projet (fichiers, BDD PostgreSQL) sur le stockage RAID.
|
* [x] Installation stricte via Dockge et Gitea (création automatique des dossiers inclus)[cite: 2].
|
||||||
- [x] Mots de passe cryptés (bcrypt), protection anti-brute force.
|
* [x] Sécurisation de la totalité du projet (fichiers, BDD PostgreSQL) sur le stockage RAID[cite: 2].
|
||||||
- [x] Authentification 2FA obligatoire (Keepassium, Authenticator, etc.).
|
* [x] Mots de passe cryptés (bcrypt), protection anti-brute force[cite: 2].
|
||||||
- [x] Déconnexion automatique (timer invisible / session JWT) après 60 minutes d'inactivité.
|
* [x] Authentification 2FA obligatoire (Keepassium, Authenticator, etc.)[cite: 2].
|
||||||
- [x] Configuration de GitHub Actions / Gitea Actions pour la création automatique des paquets Docker (Tags SemVer + latest).
|
* [x] Déconnexion automatique (timer invisible / session JWT) après 60 minutes d'inactivité[cite: 2].
|
||||||
- [x] Horodatage UTC en base de données pour immunité au changement d'heure français (Prévention crash 2FA).
|
* [x] Configuration de GitHub Actions / Gitea Actions pour la création automatique des paquets Docker (Tags SemVer + latest)[cite: 2].
|
||||||
- [ ] Procédure de réinitialisation/modification du mot de passe (Admin et Utilisateurs).
|
* [x] Horodatage UTC en base de données pour immunité au changement d'heure français (Prévention crash 2FA)[cite: 2].
|
||||||
|
* [x] Rotation automatique des logs système : génération d'un fichier log par jour avec purge stricte des fichiers de plus de 7 jours[cite: 2].
|
||||||
|
* [x] Procédure de modification sécurisée du mot de passe (Utilisateurs et Admins)[cite: 2].
|
||||||
|
|
||||||
### Administration & Gestion des Utilisateurs
|
### Administration & Gestion des Utilisateurs
|
||||||
- [x] Assistant de première installation (création du compte admin complet).
|
|
||||||
- [ ] Refonte du setup initial (Super Admin) : Obligation de configurer le serveur mail (aide opérateurs FR/Gmail) et proposition de saisie des clés API.
|
* [x] Assistant de première installation : l'initiateur du projet se voit attribuer automatiquement le rôle exclusif de **Super-Admin** (un seul par projet)[cite: 2].
|
||||||
- [ ] Intégration de la roue crantée (Paramètres) en bas de la barre latérale.
|
* [x] Refonte du setup initial (Super Admin) : configuration du serveur mail (opérateurs FR/Gmail) et paramétrage complet des clés fournisseurs IA[cite: 2].
|
||||||
- [ ] Page de configuration Utilisateur : changement d'email (avec validation via lien envoyé) et de mot de passe.
|
* [x] Intégration de la roue crantée (Paramètres) en bas de la barre latérale[cite: 2].
|
||||||
- [ ] Page de configuration Admin : ajout/modification de fournisseurs, clés API, et serveur mail en permanence.
|
* [x] Page de configuration par niveau (Super-Admin, Admin, Utilisateur)[cite: 2].
|
||||||
- [ ] Hiérarchie des rôles : Un Admin peut proposer l'élévation d'un User en Admin (formulaire soumis à la validation exclusive du Super Admin).
|
* [x] Upload et personnalisation dynamique d'avatar utilisateur[cite: 2].
|
||||||
- [ ] Export de l'activité Admin : Fichier Excel avec KPI, graphiques "camembert", et suivi financier par utilisateur.
|
* [ ] Hiérarchie des rôles : Un Admin peut proposer l'élévation d'un User en Admin (formulaire soumis à la validation exclusive du Super Admin)[cite: 2].
|
||||||
- [ ] Suivi financier en temps réel et vérification du budget API.
|
* [ ] Export de l'activité Admin : Fichier Excel avec KPI, graphiques "camembert", et suivi financier par utilisateur[cite: 2].
|
||||||
- [ ] Envoi de rapport d'activité automatisé chaque jeudi à 04h00.
|
* [x] Suivi financier en temps réel et vérification du budget API (mis à jour après chaque requête et affiché de manière non invasive sous le nom)[cite: 2].
|
||||||
|
* [ ] Envoi de rapport d'activité automatisé chaque jeudi à 04h00[cite: 2].
|
||||||
|
|
||||||
### Expérience Utilisateur (UI/UX)
|
### Expérience Utilisateur (UI/UX)
|
||||||
- [x] Thème moderne (Gemini-like) et 100% compatible smartphones.
|
|
||||||
- [x] Refonte de la gestion des projets : séparation Épinglés (en haut) / Non Épinglés (Récents).
|
* [x] Thème moderne (Gemini-like) et 100% compatible smartphones[cite: 2].
|
||||||
- [x] Arrêt de la création automatique de nouveau projet à l'ouverture.
|
* [x] Refonte de la gestion des projets : séparation Épinglés (en haut) / Non Épinglés (Récents)[cite: 2].
|
||||||
- [x] Intégration permanente du logo AETHAS38 en local (Setup / Login / Dashboard).
|
* [x] Arrêt de la création automatique de nouveau projet à l'ouverture[cite: 2].
|
||||||
- [ ] Indicateur visuel (Loader/Animation) signalant que l'IA AETHAS38 "réfléchit" ou travaille.
|
* [x] Intégration permanente du logo AETHAS38 en local (Setup / Login / Dashboard)[cite: 2].
|
||||||
- [ ] Fonctionnalité de renommage manuel des discussions/projets.
|
* [x] Indicateur visuel (Loader/Animation) signalant que l'IA AETHAS38 "réfléchit" ou travaille[cite: 2].
|
||||||
|
* [x] Fonctionnalité de renommage manuel des discussions/projets[cite: 2].
|
||||||
|
* [ ] Affichage natif et fluide des images directement dans le flux de la discussion[cite: 2].
|
||||||
|
* [ ] Système de partage des réponses sur les réseaux et messageries externes (Facebook, WhatsApp, etc.)[cite: 2].
|
||||||
|
|
||||||
### Moteur IA, Modèles & Fonctionnalités
|
### Moteur IA, Modèles & Fonctionnalités
|
||||||
- [ ] Pipeline multi-agents : 1 Rédacteur de prompt, jusqu'à 5 Travailleurs, 1 Concaténeur final.
|
|
||||||
- [ ] Extraction automatisée des modèles disponibles chez les fournisseurs (sauvegarde txt horodatée ex: "07/10/2026 à 10h00").
|
* [x] Cœur asynchrone pour la parallélisation des appels API (module `orchestrator.py`)[cite: 2].
|
||||||
- [ ] Tâche planifiée (CRON) pour la mise à jour des modèles toutes les 12h (le timer se reset si mise à jour manuelle).
|
* [x] Pipeline multi-agents complet avec sélection dynamique par requête :
|
||||||
- [ ] Classement, filtrage (Gratuit/Payant) et catégorisation des modèles (Texte, Code, Image, Audio, Vidéo).
|
* Sélection de 1 à 5 travailleurs IA en parallèle[cite: 2].
|
||||||
- [ ] Upload manuel du fichier d'extraction des modèles avec intégration dans l'interface filtrable.
|
* Si un seul travailleur est sélectionné, bypass automatique du prompteur et du concaténeur[cite: 2].
|
||||||
- [ ] Traduction automatisée des descriptions des modèles via Ollama (local) ou Gemini.
|
* Si > 1 travailleur, activation de la chaîne complète (1 Prompteur ➔ N Travailleurs ➔ 1 Concaténeur)[cite: 2].
|
||||||
- [ ] Interface d'aide à la décision pour le choix des modèles selon la tâche souhaitée.
|
* [x] Extraction universelle multi-fournisseurs (OpenRouter, Groq, DeepSeek, Mistral, Gemini, Cloudflare) avec timeout porté à 90 secondes[cite: 2].
|
||||||
- [ ] Mécanisme de fallback / termes génériques pour les appels API (protection contre le dépassement des quotas Gemini).
|
* [x] Tâche planifiée (CRON) pour la mise à jour des modèles toutes les 12h (00h00 et 12h00)[cite: 2].
|
||||||
- [ ] Téléchargement des requêtes, réponses et fichiers générés (Excel, PDF, TXT, Images).
|
* [x] Page spécifique dédiée aux modèles permettant l'export, l'import manuel d'un fichier d'extraction JSON, ainsi que le filtrage par prix, domaine et mots-clés[cite: 2].
|
||||||
|
* [x] Extraction unitaire des retours IA et téléchargement global d'une conversation (formats TXT et JSON)[cite: 2].
|
||||||
|
* [x] Détection et téléchargement ciblé de pièces jointes ou blocs spécifiques (Code brut, scripts, fichiers de données) générés par l'IA en respectant le nommage exact demandé[cite: 2].
|
||||||
|
* [x] Classement, filtrage (Gratuit/Payant) et catégorisation des modèles (Texte, Code, Vision, Audio)[cite: 2].
|
||||||
|
* [ ] Traduction automatisée des descriptions des modèles via Ollama (local) ou Gemini[cite: 2].
|
||||||
|
* [ ] Interface d'aide à la décision pour le choix des modèles selon la tâche souhaitée[cite: 2].
|
||||||
|
* [ ] Mécanisme de fallback / termes génériques pour les appels API (protection contre le dépassement des quotas)[cite: 2].
|
||||||
@@ -0,0 +1,20 @@
|
|||||||
|
# 📚 WIKI AETHAS38 - Configuration des Fournisseurs IA
|
||||||
|
|
||||||
|
Ce document détaille la procédure pour obtenir les clés API avec les droits complets (Inférence + Accès financier) nécessaires au fonctionnement de l'Orchestrateur AETHAS38.
|
||||||
|
|
||||||
|
## 1. OpenRouter (Prioritaire)
|
||||||
|
OpenRouter agit comme un agrégateur. C'est le fournisseur recommandé car il permet d'accéder à presque tous les modèles (y compris Gemini, Claude, Llama) avec un seul portefeuille prépayé.
|
||||||
|
* **Obtenir la clé API :** Allez sur [OpenRouter Keys](https://openrouter.ai/keys). Créez une clé standard sans limite de crédit pour permettre à AETHAS38 de requêter les modèles.
|
||||||
|
* **Accès Financier (Management Key) :** Pour que AETHAS38 lise votre solde en temps réel, vous devez générer une clé spécifique d'administration (Management Key) sur la même page, ou accorder les droits `read_balance` à votre clé API principale selon les dernières mises à jour de leur interface.
|
||||||
|
|
||||||
|
## 2. Google Gemini Studio
|
||||||
|
* **Obtenir la clé API :** Rendez-vous sur [Google AI Studio](https://aistudio.google.com/app/apikey).
|
||||||
|
* **Levée des quotas :** Par défaut, la clé est gratuite mais soumise à des limites strictes (ex: 15 requêtes/minute) qui bloqueront le travail en parallèle d'AETHAS38. Pour un accès total, liez votre projet AI Studio à un compte de facturation Google Cloud Platform (GCP).
|
||||||
|
|
||||||
|
## 3. DeepSeek & Groq
|
||||||
|
* **DeepSeek :** [DeepSeek Platform](https://platform.deepseek.com/api_keys). Facturation à l'usage (Token). Il faut provisionner le compte par carte bancaire (Top-up).
|
||||||
|
* **Groq :** [Groq Console](https://console.groq.com/keys). Actuellement en phase gratuite avec quotas de requêtes très élevés, idéal pour les travailleurs de base de l'orchestrateur.
|
||||||
|
|
||||||
|
## 4. Cloudflare Workers AI & HuggingFace
|
||||||
|
* **Cloudflare :** [Cloudflare Dash](https://dash.cloudflare.com/profile/api-tokens). Créez un token personnalisé avec la permission `Workers AI : Read/Write`. L'ID de compte (Account ID) est visible dans la barre latérale droite de l'aperçu de la zone.
|
||||||
|
* **HuggingFace :** [HF Settings](https://huggingface.co/settings/tokens). Générez un token *Fine-Grained* et cochez les permissions liées à l'`Inference API` pour permettre l'exécution des modèles serverless.
|
||||||
@@ -0,0 +1,52 @@
|
|||||||
|
from openai import OpenAI
|
||||||
|
from fastapi import HTTPException
|
||||||
|
from .models import SystemSettings
|
||||||
|
|
||||||
|
def get_ai_response(messages: list, settings: SystemSettings) -> str:
|
||||||
|
"""Route la conversation vers le premier fournisseur IA disponible, priorité à Gemini Direct."""
|
||||||
|
if not settings:
|
||||||
|
raise HTTPException(status_code=500, detail="Configuration système introuvable.")
|
||||||
|
|
||||||
|
formatted_messages = [{"role": msg.role, "content": msg.content} for msg in messages]
|
||||||
|
|
||||||
|
try:
|
||||||
|
# 1. Priorité absolue : Test Gemini direct
|
||||||
|
if settings.gemini_api_key:
|
||||||
|
client = OpenAI(
|
||||||
|
base_url="https://generativelanguage.googleapis.com/v1beta/openai/",
|
||||||
|
api_key=settings.gemini_api_key
|
||||||
|
)
|
||||||
|
# Correction : Utilisation exacte du modèle 3.5 Flash-Lite
|
||||||
|
model = "gemini-3.5-flash-lite"
|
||||||
|
response = client.chat.completions.create(model=model, messages=formatted_messages)
|
||||||
|
|
||||||
|
# 2. Test OpenRouter (Fallback)
|
||||||
|
elif settings.openrouter_api_key:
|
||||||
|
client = OpenAI(base_url="https://openrouter.ai/api/v1", api_key=settings.openrouter_api_key)
|
||||||
|
model = "google/gemini-3.5-flash-lite"
|
||||||
|
response = client.chat.completions.create(
|
||||||
|
model=model,
|
||||||
|
messages=formatted_messages,
|
||||||
|
extra_headers={"HTTP-Referer": "https://aethas38.duckdns.org", "X-Title": "AETHAS38"}
|
||||||
|
)
|
||||||
|
|
||||||
|
# 3. Test DeepSeek
|
||||||
|
elif settings.deepseek_api_key:
|
||||||
|
client = OpenAI(base_url="https://api.deepseek.com/v1", api_key=settings.deepseek_api_key)
|
||||||
|
model = "deepseek-chat"
|
||||||
|
response = client.chat.completions.create(model=model, messages=formatted_messages)
|
||||||
|
|
||||||
|
# 4. Test Groq
|
||||||
|
elif settings.groq_api_key:
|
||||||
|
client = OpenAI(base_url="https://api.groq.com/openai/v1", api_key=settings.groq_api_key)
|
||||||
|
model = "llama3-8b-8192"
|
||||||
|
response = client.chat.completions.create(model=model, messages=formatted_messages)
|
||||||
|
|
||||||
|
else:
|
||||||
|
raise HTTPException(status_code=400, detail="Aucune clé API IA n'est configurée.")
|
||||||
|
|
||||||
|
return response.choices[0].message.content
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
print(f"Erreur API IA : {str(e)}")
|
||||||
|
raise HTTPException(status_code=502, detail=f"Détail du fournisseur : {str(e)}")
|
||||||
@@ -0,0 +1,24 @@
|
|||||||
|
import logging
|
||||||
|
from logging.handlers import TimedRotatingFileHandler
|
||||||
|
import os
|
||||||
|
|
||||||
|
# Création du dossier logs à la racine du projet
|
||||||
|
log_dir = os.path.join(os.getcwd(), "logs")
|
||||||
|
os.makedirs(log_dir, exist_ok=True)
|
||||||
|
|
||||||
|
# Configuration du logger
|
||||||
|
system_logger = logging.getLogger("AETHAS38_Logger")
|
||||||
|
system_logger.setLevel(logging.INFO)
|
||||||
|
|
||||||
|
# Rotation tous les jours à minuit, conservation sur 7 jours
|
||||||
|
handler = TimedRotatingFileHandler(
|
||||||
|
filename=os.path.join(log_dir, "aethas38_system.log"),
|
||||||
|
when="midnight",
|
||||||
|
interval=1,
|
||||||
|
backupCount=7,
|
||||||
|
encoding="utf-8"
|
||||||
|
)
|
||||||
|
|
||||||
|
formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s', datefmt='%d/%m/%Y %H:%M:%S')
|
||||||
|
handler.setFormatter(formatter)
|
||||||
|
system_logger.addHandler(handler)
|
||||||
+306
-187
@@ -1,248 +1,367 @@
|
|||||||
from fastapi import FastAPI, Depends, HTTPException, status, Request, Response
|
from fastapi import FastAPI, Depends, HTTPException, status, Request, Response, UploadFile, File
|
||||||
from fastapi.responses import RedirectResponse, FileResponse
|
from fastapi.responses import RedirectResponse, FileResponse
|
||||||
from fastapi.staticfiles import StaticFiles
|
from fastapi.staticfiles import StaticFiles
|
||||||
from sqlalchemy.orm import Session
|
from sqlalchemy.orm import Session
|
||||||
from typing import List
|
from typing import List
|
||||||
import os
|
import os, json, asyncio, shutil, base64, io, csv, zipfile
|
||||||
|
from datetime import datetime, timedelta, timezone
|
||||||
|
import pytz
|
||||||
|
|
||||||
# Importation de nos modules locaux
|
from .database import engine, Base, get_db, SessionLocal
|
||||||
from .database import engine, Base, get_db
|
|
||||||
from .models import User, Project, Message
|
|
||||||
from .auth import get_password_hash, generate_totp_secret, get_totp_uri, verify_password, verify_totp, create_access_token, verify_token
|
from .auth import get_password_hash, generate_totp_secret, get_totp_uri, verify_password, verify_totp, create_access_token, verify_token
|
||||||
from .schemas import AdminCreate, LoginRequest, ProjectCreate, ProjectResponse, ProjectRename, MessageCreate, MessageResponse
|
from .schemas import AdminCreate, LoginRequest, ProjectCreate, ProjectResponse, ProjectRename, MessageCreate, MessageResponse, PasswordChange, ModelReplacementRequest, LogRequest, SystemSettingsUpdate, SystemSettingsResponse
|
||||||
from .models import User, Project, Message, SystemSettings
|
from .models import User, Project, Message, SystemSettings, AIModel, FinancialLog
|
||||||
|
from .orchestrator import run_orchestrator, sync_providers_models, sync_finances, activity_logs, log_activity
|
||||||
|
|
||||||
# Création des tables dans la base de données
|
|
||||||
Base.metadata.create_all(bind=engine)
|
Base.metadata.create_all(bind=engine)
|
||||||
|
app = FastAPI(title="AETHAS38")
|
||||||
|
|
||||||
app = FastAPI(title="AETHAS38 - Orchestrateur Multi-IA")
|
|
||||||
|
|
||||||
# Configuration des fichiers statiques (images, css locaux, etc.)
|
|
||||||
assets_path = os.path.join(os.getcwd(), "frontend", "assets")
|
assets_path = os.path.join(os.getcwd(), "frontend", "assets")
|
||||||
os.makedirs(assets_path, exist_ok=True) # Crée le dossier s'il n'existe pas
|
avatars_path = os.path.join(assets_path, "avatars")
|
||||||
|
os.makedirs(assets_path, exist_ok=True)
|
||||||
|
os.makedirs(avatars_path, exist_ok=True)
|
||||||
app.mount("/assets", StaticFiles(directory=assets_path), name="assets")
|
app.mount("/assets", StaticFiles(directory=assets_path), name="assets")
|
||||||
# Définition des routes du logo dans login
|
|
||||||
|
|
||||||
|
async def scheduler_task():
|
||||||
|
tz = pytz.timezone('Europe/Paris')
|
||||||
|
while True:
|
||||||
|
now = datetime.now(tz)
|
||||||
|
if (now.hour == 0 or now.hour == 12) and now.minute == 0:
|
||||||
|
db = SessionLocal()
|
||||||
|
try:
|
||||||
|
settings = db.query(SystemSettings).first()
|
||||||
|
if settings:
|
||||||
|
try: await sync_providers_models(db, settings, "Automatique")
|
||||||
|
except: pass
|
||||||
|
|
||||||
def is_setup_required(db: Session) -> bool:
|
cutoff = datetime.now(timezone.utc) - timedelta(days=7)
|
||||||
"""Vérifie si la base de données contient au moins un administrateur."""
|
old_projects = db.query(Project).filter(Project.is_pinned == False, Project.created_at < cutoff).all()
|
||||||
admin = db.query(User).filter(User.is_admin == True).first()
|
if old_projects:
|
||||||
return admin is None
|
log_activity(f"[Nettoyage] Suppression de {len(old_projects)} discussion(s) de plus de 7 jours.")
|
||||||
|
for op in old_projects:
|
||||||
|
db.delete(op)
|
||||||
|
db.commit()
|
||||||
|
except Exception as e:
|
||||||
|
db.rollback()
|
||||||
|
log_activity(f"[Erreur Nettoyage] {str(e)}")
|
||||||
|
finally:
|
||||||
|
db.close()
|
||||||
|
|
||||||
|
await asyncio.sleep(60)
|
||||||
|
await asyncio.sleep(30)
|
||||||
|
|
||||||
|
@app.on_event("startup")
|
||||||
|
async def startup_event(): asyncio.create_task(scheduler_task())
|
||||||
|
|
||||||
|
def is_setup_required(db: Session) -> bool: return db.query(User).filter(User.is_superadmin == True).first() is None
|
||||||
|
|
||||||
|
def get_current_user(request: Request, db: Session = Depends(get_db)):
|
||||||
|
token = request.cookies.get("session_token")
|
||||||
|
if not token: raise HTTPException(status_code=401, detail="Non authentifié")
|
||||||
|
payload = verify_token(token)
|
||||||
|
if not payload: raise HTTPException(status_code=401, detail="Session expirée")
|
||||||
|
user = db.query(User).filter(User.username == payload.get("sub")).first()
|
||||||
|
if not user: raise HTTPException(status_code=401, detail="Utilisateur introuvable")
|
||||||
|
return user
|
||||||
|
|
||||||
@app.get("/")
|
@app.get("/")
|
||||||
def read_root(db: Session = Depends(get_db)):
|
def read_root(db: Session = Depends(get_db)): return RedirectResponse(url="/setup") if is_setup_required(db) else RedirectResponse(url="/login")
|
||||||
if is_setup_required(db):
|
|
||||||
return RedirectResponse(url="/setup")
|
|
||||||
return RedirectResponse(url="/login")
|
|
||||||
|
|
||||||
@app.get("/setup")
|
@app.get("/setup")
|
||||||
def setup_page(db: Session = Depends(get_db)):
|
def setup_page(db: Session = Depends(get_db)): return RedirectResponse(url="/login") if not is_setup_required(db) else FileResponse(os.path.join(os.getcwd(), "frontend", "index.html"))
|
||||||
"""Affiche l'interface d'installation Vue.js."""
|
|
||||||
if not is_setup_required(db):
|
|
||||||
return RedirectResponse(url="/login")
|
|
||||||
|
|
||||||
frontend_path = os.path.join(os.getcwd(), "frontend", "index.html")
|
|
||||||
if not os.path.exists(frontend_path):
|
|
||||||
raise HTTPException(status_code=404, detail="Interface introuvable.")
|
|
||||||
|
|
||||||
return FileResponse(frontend_path)
|
|
||||||
|
|
||||||
@app.get("/login")
|
@app.get("/login")
|
||||||
def login_page(db: Session = Depends(get_db)):
|
def login_page(db: Session = Depends(get_db)): return RedirectResponse(url="/setup") if is_setup_required(db) else FileResponse(os.path.join(os.getcwd(), "frontend", "login.html"))
|
||||||
"""Affiche la page de connexion sécurisée."""
|
|
||||||
if is_setup_required(db):
|
|
||||||
return RedirectResponse(url="/setup")
|
|
||||||
|
|
||||||
frontend_path = os.path.join(os.getcwd(), "frontend", "login.html")
|
|
||||||
if not os.path.exists(frontend_path):
|
|
||||||
raise HTTPException(status_code=404, detail="Interface de connexion introuvable.")
|
|
||||||
|
|
||||||
return FileResponse(frontend_path)
|
|
||||||
|
|
||||||
@app.post("/api/setup")
|
@app.post("/api/setup")
|
||||||
def create_admin(admin_data: AdminCreate, db: Session = Depends(get_db)):
|
def create_admin(admin_data: AdminCreate, db: Session = Depends(get_db)):
|
||||||
"""Reçoit les données du frontend, crée l'admin, sauvegarde la configuration et retourne le QR Code 2FA."""
|
if not is_setup_required(db): raise HTTPException(status_code=403, detail="Déjà installé.")
|
||||||
if not is_setup_required(db):
|
new_admin = User(email=admin_data.email, username=admin_data.username, hashed_password=get_password_hash(admin_data.password), totp_secret=generate_totp_secret(), is_admin=True, is_superadmin=True)
|
||||||
raise HTTPException(
|
|
||||||
status_code=status.HTTP_403_FORBIDDEN,
|
|
||||||
detail="L'installation a déjà été effectuée."
|
|
||||||
)
|
|
||||||
|
|
||||||
# Création du Super Admin
|
|
||||||
hashed_pw = get_password_hash(admin_data.password)
|
|
||||||
totp_secret = generate_totp_secret()
|
|
||||||
new_admin = User(
|
|
||||||
email=admin_data.email,
|
|
||||||
username=admin_data.username,
|
|
||||||
hashed_password=hashed_pw,
|
|
||||||
totp_secret=totp_secret,
|
|
||||||
is_admin=True
|
|
||||||
)
|
|
||||||
db.add(new_admin)
|
db.add(new_admin)
|
||||||
|
new_settings = SystemSettings(smtp_host=admin_data.smtp_host, smtp_port=admin_data.smtp_port, smtp_user=admin_data.smtp_user, smtp_password=admin_data.smtp_password, openrouter_api_key=admin_data.openrouter_api_key, openrouter_management_key=admin_data.openrouter_management_key, groq_api_key=admin_data.groq_api_key, gemini_api_key=admin_data.gemini_api_key, deepseek_api_key=admin_data.deepseek_api_key, mistral_api_key=admin_data.mistral_api_key, cloudflare_account_id=admin_data.cloudflare_account_id, cloudflare_api_token=admin_data.cloudflare_api_token, huggingface_api_key=admin_data.huggingface_api_key)
|
||||||
# Sauvegarde des paramètres système
|
|
||||||
new_settings = SystemSettings(
|
|
||||||
smtp_host=admin_data.smtp_host,
|
|
||||||
smtp_port=admin_data.smtp_port,
|
|
||||||
smtp_user=admin_data.smtp_user,
|
|
||||||
smtp_password=admin_data.smtp_password,
|
|
||||||
openrouter_api_key=admin_data.openrouter_api_key,
|
|
||||||
openrouter_management_key=admin_data.openrouter_management_key,
|
|
||||||
groq_api_key=admin_data.groq_api_key,
|
|
||||||
gemini_api_key=admin_data.gemini_api_key,
|
|
||||||
deepseek_api_key=admin_data.deepseek_api_key,
|
|
||||||
mistral_api_key=admin_data.mistral_api_key,
|
|
||||||
cloudflare_account_id=admin_data.cloudflare_account_id,
|
|
||||||
cloudflare_api_token=admin_data.cloudflare_api_token,
|
|
||||||
huggingface_api_key=admin_data.huggingface_api_key
|
|
||||||
)
|
|
||||||
db.add(new_settings)
|
db.add(new_settings)
|
||||||
|
|
||||||
db.commit()
|
db.commit()
|
||||||
db.refresh(new_admin)
|
db.refresh(new_admin)
|
||||||
|
return {"message": "Succès", "totp_secret": new_admin.totp_secret, "totp_uri": get_totp_uri(new_admin.totp_secret, new_admin.username)}
|
||||||
totp_uri = get_totp_uri(totp_secret, new_admin.username)
|
|
||||||
return {
|
|
||||||
"message": "Configuration terminée avec succès.",
|
|
||||||
"totp_secret": totp_secret,
|
|
||||||
"totp_uri": totp_uri
|
|
||||||
}
|
|
||||||
|
|
||||||
@app.post("/api/login")
|
@app.post("/api/login")
|
||||||
def login(login_data: LoginRequest, response: Response, db: Session = Depends(get_db)):
|
def login(login_data: LoginRequest, response: Response, db: Session = Depends(get_db)):
|
||||||
"""Vérifie les identifiants et crée une session de 60 minutes."""
|
|
||||||
user = db.query(User).filter(User.username == login_data.username).first()
|
user = db.query(User).filter(User.username == login_data.username).first()
|
||||||
|
if not user or not verify_password(login_data.password, user.hashed_password): raise HTTPException(status_code=401, detail="Identifiants incorrects.")
|
||||||
|
if not verify_totp(user.totp_secret, login_data.totp_code): raise HTTPException(status_code=401, detail="2FA invalide.")
|
||||||
|
|
||||||
# Vérification de l'utilisateur et du mot de passe
|
# Session valide 8 heures (28800 secondes)
|
||||||
if not user or not verify_password(login_data.password, user.hashed_password):
|
response.set_cookie(key="session_token", value=create_access_token(data={"sub": user.username}), httponly=True, max_age=28800, samesite="lax")
|
||||||
raise HTTPException(
|
|
||||||
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
||||||
detail="Nom d'utilisateur ou mot de passe incorrect."
|
|
||||||
)
|
|
||||||
|
|
||||||
# Vérification du code 2FA
|
|
||||||
if not verify_totp(user.totp_secret, login_data.totp_code):
|
|
||||||
raise HTTPException(
|
|
||||||
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
||||||
detail="Code 2FA invalide."
|
|
||||||
)
|
|
||||||
|
|
||||||
# Création du jeton de session
|
|
||||||
access_token = create_access_token(data={"sub": user.username})
|
|
||||||
|
|
||||||
# Injection du jeton dans un cookie sécurisé (invisible pour le JavaScript côté client)
|
|
||||||
response.set_cookie(
|
|
||||||
key="session_token",
|
|
||||||
value=access_token,
|
|
||||||
httponly=True,
|
|
||||||
max_age=3600, # 3600 secondes = 60 minutes
|
|
||||||
samesite="lax"
|
|
||||||
)
|
|
||||||
|
|
||||||
return {"message": "Connexion réussie"}
|
return {"message": "Connexion réussie"}
|
||||||
|
|
||||||
@app.get("/dashboard")
|
@app.get("/dashboard")
|
||||||
def dashboard(request: Request):
|
def dashboard(request: Request):
|
||||||
"""Route protégée : affiche l'interface principale si le cookie est valide."""
|
|
||||||
token = request.cookies.get("session_token")
|
token = request.cookies.get("session_token")
|
||||||
if not token:
|
if not token or not verify_token(token): return RedirectResponse(url="/login")
|
||||||
return RedirectResponse(url="/login")
|
return FileResponse(os.path.join(os.getcwd(), "frontend", "dashboard.html"))
|
||||||
|
|
||||||
payload = verify_token(token)
|
@app.get("/api/logs")
|
||||||
if not payload:
|
def get_logs():
|
||||||
return RedirectResponse(url="/login")
|
return {"logs": activity_logs}
|
||||||
|
|
||||||
frontend_path = os.path.join(os.getcwd(), "frontend", "dashboard.html")
|
@app.post("/api/logs")
|
||||||
if not os.path.exists(frontend_path):
|
def add_frontend_log(req: LogRequest):
|
||||||
raise HTTPException(status_code=404, detail="Interface du tableau de bord introuvable.")
|
log_activity(f"[Système UI] {req.message}")
|
||||||
|
return {"status": "ok"}
|
||||||
|
|
||||||
return FileResponse(frontend_path)
|
@app.post("/api/models/suggest_replacement")
|
||||||
|
async def suggest_replacement(req: ModelReplacementRequest, db: Session = Depends(get_db)):
|
||||||
|
log_activity(f"⚠️ Modèle indisponible: {req.missing_model}. Demande de suggestion à Gemini...")
|
||||||
|
settings = db.query(SystemSettings).first()
|
||||||
|
if not settings or not settings.gemini_api_key:
|
||||||
|
log_activity("Clé Gemini non trouvée. Fallback forcé sur gemini-3.5-flash-lite.")
|
||||||
|
return {"suggestion": "gemini-3.5-flash-lite", "reason": "Clé API Gemini non configurée dans le système."}
|
||||||
|
|
||||||
# --- GESTION DES PROJETS ---
|
models = db.query(AIModel).all()
|
||||||
|
available = [m.model_id for m in models]
|
||||||
|
|
||||||
def get_current_user(request: Request, db: Session = Depends(get_db)):
|
prompt = f"Le modèle IA '{req.missing_model}' n'est plus disponible. Voici les modèles disponibles : {', '.join(available)}. Trouve le modèle le plus proche techniquement. Réponds UNIQUEMENT avec ce format strict : ID_DU_MODELE | Brève explication en français de 10 mots max. Si aucun ne correspond, renvoie gemini-3.5-flash-lite | Par défaut."
|
||||||
"""Extrait l'utilisateur actuel à partir du cookie de session JWT."""
|
|
||||||
token = request.cookies.get("session_token")
|
|
||||||
if not token:
|
|
||||||
raise HTTPException(status_code=401, detail="Non authentifié")
|
|
||||||
|
|
||||||
payload = verify_token(token)
|
try:
|
||||||
if not payload:
|
from openai import AsyncOpenAI
|
||||||
raise HTTPException(status_code=401, detail="Session expirée")
|
client = AsyncOpenAI(base_url="https://generativelanguage.googleapis.com/v1beta/openai/", api_key=settings.gemini_api_key)
|
||||||
|
resp = await client.chat.completions.create(model="gemini-3.5-flash-lite", messages=[{"role": "user", "content": prompt}], max_tokens=50)
|
||||||
|
res = resp.choices[0].message.content.strip()
|
||||||
|
|
||||||
user = db.query(User).filter(User.username == payload.get("sub")).first()
|
if "|" in res:
|
||||||
if not user:
|
parts = res.split("|")
|
||||||
raise HTTPException(status_code=401, detail="Utilisateur introuvable")
|
sugg = parts[0].strip()
|
||||||
return user
|
reason = parts[1].strip()
|
||||||
|
else:
|
||||||
|
sugg = res.strip()
|
||||||
|
reason = "Sélectionné par Gemini."
|
||||||
|
|
||||||
|
if sugg not in available and sugg != "gemini-3.5-flash-lite":
|
||||||
|
sugg = "gemini-3.5-flash-lite"
|
||||||
|
reason = "Gemini a suggéré un modèle invalide. Fallback par défaut."
|
||||||
|
|
||||||
|
log_activity(f"✅ Remplacement trouvé : {req.missing_model} -> {sugg}")
|
||||||
|
return {"suggestion": sugg, "reason": reason}
|
||||||
|
except Exception as e:
|
||||||
|
log_activity(f"Erreur d'interrogation Gemini: {str(e)}. Fallback par défaut.")
|
||||||
|
return {"suggestion": "gemini-3.5-flash-lite", "reason": f"Erreur API."}
|
||||||
|
|
||||||
|
@app.get("/api/users/me")
|
||||||
|
def get_me(current_user: User = Depends(get_current_user)):
|
||||||
|
return {"username": current_user.username, "is_admin": current_user.is_admin, "is_superadmin": current_user.is_superadmin, "avatar_path": current_user.avatar_path}
|
||||||
|
|
||||||
|
@app.put("/api/users/me/password")
|
||||||
|
def change_password(passwords: PasswordChange, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
|
||||||
|
if not verify_password(passwords.old_password, current_user.hashed_password): raise HTTPException(status_code=400, detail="L'ancien mot de passe est incorrect.")
|
||||||
|
if len(passwords.new_password) < 8: raise HTTPException(status_code=400, detail="8 caractères minimum.")
|
||||||
|
current_user.hashed_password = get_password_hash(passwords.new_password)
|
||||||
|
db.commit()
|
||||||
|
return {"message": "Mot de passe mis à jour."}
|
||||||
|
|
||||||
|
@app.post("/api/users/me/avatar")
|
||||||
|
async def upload_avatar(file: UploadFile = File(...), db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
|
||||||
|
file_location = os.path.join(avatars_path, f"user_{current_user.id}.jpg")
|
||||||
|
with open(file_location, "wb") as buffer: shutil.copyfileobj(file.file, buffer)
|
||||||
|
current_user.avatar_path = f"/assets/avatars/user_{current_user.id}.jpg?v={int(datetime.now().timestamp())}"
|
||||||
|
db.commit()
|
||||||
|
return {"message": "Avatar mis à jour", "avatar_path": current_user.avatar_path}
|
||||||
|
|
||||||
|
# --- ROUTES SUPER-ADMIN ---
|
||||||
|
@app.get("/api/settings", response_model=SystemSettingsResponse)
|
||||||
|
def get_settings(db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
|
||||||
|
if not current_user.is_superadmin: raise HTTPException(status_code=403, detail="Super-Admin requis.")
|
||||||
|
return db.query(SystemSettings).first()
|
||||||
|
|
||||||
|
@app.put("/api/settings")
|
||||||
|
def update_settings(settings_data: SystemSettingsUpdate, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
|
||||||
|
if not current_user.is_superadmin: raise HTTPException(status_code=403, detail="Super-Admin requis.")
|
||||||
|
s = db.query(SystemSettings).first()
|
||||||
|
if not s: raise HTTPException(status_code=404)
|
||||||
|
for k, v in settings_data.dict(exclude_unset=True).items():
|
||||||
|
setattr(s, k, v)
|
||||||
|
db.commit()
|
||||||
|
log_activity("Configuration système mise à jour par le Super-Admin.")
|
||||||
|
return {"message": "Paramètres mis à jour avec succès."}
|
||||||
|
|
||||||
@app.get("/api/projects", response_model=List[ProjectResponse])
|
@app.get("/api/projects", response_model=List[ProjectResponse])
|
||||||
def get_projects(db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
|
def get_projects(db: Session = Depends(get_db), current_user: User = Depends(get_current_user)): return db.query(Project).filter(Project.user_id == current_user.id).order_by(Project.created_at.desc()).all()
|
||||||
"""Récupère tous les projets de l'utilisateur connecté."""
|
|
||||||
return db.query(Project).filter(Project.user_id == current_user.id).order_by(Project.created_at.desc()).all()
|
|
||||||
|
|
||||||
@app.post("/api/projects", response_model=ProjectResponse)
|
@app.post("/api/projects", response_model=ProjectResponse)
|
||||||
def create_project(project: ProjectCreate, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
|
def create_project(project: ProjectCreate, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
|
||||||
"""Crée un nouveau projet pour l'utilisateur connecté."""
|
p = Project(title=project.title, user_id=current_user.id); db.add(p); db.commit(); db.refresh(p); return p
|
||||||
new_project = Project(title=project.title, user_id=current_user.id)
|
|
||||||
db.add(new_project)
|
|
||||||
db.commit()
|
|
||||||
db.refresh(new_project)
|
|
||||||
return new_project
|
|
||||||
|
|
||||||
@app.put("/api/projects/{project_id}/rename", response_model=ProjectResponse)
|
@app.put("/api/projects/{project_id}/rename", response_model=ProjectResponse)
|
||||||
def rename_project(project_id: int, project_data: ProjectRename, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
|
def rename_project(project_id: int, project_data: ProjectRename, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
|
||||||
"""Renomme un projet existant."""
|
p = db.query(Project).filter(Project.id == project_id, Project.user_id == current_user.id).first()
|
||||||
project = db.query(Project).filter(Project.id == project_id, Project.user_id == current_user.id).first()
|
if not p: raise HTTPException(status_code=404)
|
||||||
if not project:
|
p.title = project_data.title; db.commit(); db.refresh(p); return p
|
||||||
raise HTTPException(status_code=404, detail="Projet introuvable")
|
|
||||||
|
|
||||||
project.title = project_data.title
|
|
||||||
db.commit()
|
|
||||||
db.refresh(project)
|
|
||||||
return project
|
|
||||||
|
|
||||||
@app.put("/api/projects/{project_id}/pin", response_model=ProjectResponse)
|
@app.put("/api/projects/{project_id}/pin", response_model=ProjectResponse)
|
||||||
def toggle_pin_project(project_id: int, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
|
def toggle_pin_project(project_id: int, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
|
||||||
"""Bascule le statut épinglé (is_pinned) d'un projet."""
|
p = db.query(Project).filter(Project.id == project_id, Project.user_id == current_user.id).first()
|
||||||
project = db.query(Project).filter(Project.id == project_id, Project.user_id == current_user.id).first()
|
if not p: raise HTTPException(status_code=404)
|
||||||
if not project:
|
p.is_pinned = not p.is_pinned; db.commit(); db.refresh(p); return p
|
||||||
raise HTTPException(status_code=404, detail="Projet introuvable")
|
|
||||||
|
|
||||||
project.is_pinned = not project.is_pinned
|
|
||||||
db.commit()
|
|
||||||
db.refresh(project)
|
|
||||||
return project
|
|
||||||
|
|
||||||
@app.delete("/api/projects/{project_id}")
|
@app.delete("/api/projects/{project_id}")
|
||||||
def delete_project(project_id: int, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
|
def delete_project(project_id: int, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
|
||||||
"""Supprime définitivement un projet et tous ses messages associés."""
|
p = db.query(Project).filter(Project.id == project_id, Project.user_id == current_user.id).first()
|
||||||
project = db.query(Project).filter(Project.id == project_id, Project.user_id == current_user.id).first()
|
if not p: raise HTTPException(status_code=404)
|
||||||
if not project:
|
db.delete(p); db.commit(); return {"message": "Supprimé"}
|
||||||
raise HTTPException(status_code=404, detail="Projet introuvable")
|
|
||||||
|
|
||||||
db.delete(project)
|
@app.get("/api/projects/{project_id}/export")
|
||||||
db.commit()
|
def export_project(project_id: int, format: str = "txt", db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
|
||||||
return {"message": "Projet supprimé"}
|
p = db.query(Project).filter(Project.id == project_id, Project.user_id == current_user.id).first()
|
||||||
|
messages = db.query(Message).filter(Message.project_id == project_id).order_by(Message.created_at.asc()).all()
|
||||||
# --- GESTION DES MESSAGES ---
|
if format == "json":
|
||||||
|
data = [{"role": m.role, "content": m.content, "date": m.created_at.isoformat()} for m in messages]
|
||||||
|
return Response(content=json.dumps(data, indent=2), media_type="application/json", headers={"Content-Disposition": f"attachment; filename=export_{project_id}.json"})
|
||||||
|
text = f"--- HISTORIQUE : {p.title} ---\n\n"
|
||||||
|
for m in messages: text += f"[{m.created_at.strftime('%Y-%m-%d %H:%M:%S')}] {'VOUS' if m.role == 'user' else 'AETHAS38'}:\n{m.content}\n\n{'-'*50}\n\n"
|
||||||
|
return Response(content=text, media_type="text/plain;charset=utf-8", headers={"Content-Disposition": f"attachment; filename=export_{project_id}.txt"})
|
||||||
|
|
||||||
@app.get("/api/projects/{project_id}/messages", response_model=List[MessageResponse])
|
@app.get("/api/projects/{project_id}/messages", response_model=List[MessageResponse])
|
||||||
def get_messages(project_id: int, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
|
def get_messages(project_id: int, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)): return db.query(Message).filter(Message.project_id == project_id).order_by(Message.created_at.asc()).all()
|
||||||
"""Récupère tous les messages d'un projet spécifique."""
|
|
||||||
project = db.query(Project).filter(Project.id == project_id, Project.user_id == current_user.id).first()
|
@app.post("/api/projects/{project_id}/messages", response_model=List[MessageResponse])
|
||||||
if not project:
|
async def create_message(project_id: int, message: MessageCreate, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
|
||||||
raise HTTPException(status_code=404, detail="Projet introuvable")
|
p = db.query(Project).filter(Project.id == project_id).first()
|
||||||
|
if p and p.title == "Nouvelle discussion":
|
||||||
|
user_msgs = db.query(Message).filter(Message.project_id == project_id, Message.role == "user").order_by(Message.created_at.asc()).all()
|
||||||
|
if len(user_msgs) == 1:
|
||||||
|
first_content = user_msgs[0].content.split("\n\n[Fichiers joints")[0].strip()
|
||||||
|
new_title = first_content.split('\n')[0][:35].strip()
|
||||||
|
if not new_title: new_title = "Discussion"
|
||||||
|
p.title = new_title + ("..." if len(first_content) > 35 else "")
|
||||||
|
db.commit()
|
||||||
|
|
||||||
|
extracted_files_data = []
|
||||||
|
files_names = []
|
||||||
|
|
||||||
|
if message.files:
|
||||||
|
for f in message.files:
|
||||||
|
files_names.append(f.name)
|
||||||
|
content = f.content
|
||||||
|
if content.startswith("data:"):
|
||||||
|
try:
|
||||||
|
header, b64data = content.split(",", 1)
|
||||||
|
file_bytes = base64.b64decode(b64data)
|
||||||
|
ext = f.name.split('.')[-1].lower()
|
||||||
|
extracted_text = ""
|
||||||
|
|
||||||
|
if ext == 'pdf':
|
||||||
|
try:
|
||||||
|
import PyPDF2
|
||||||
|
reader = PyPDF2.PdfReader(io.BytesIO(file_bytes))
|
||||||
|
extracted_text = "\n".join([page.extract_text() for page in reader.pages if page.extract_text()])
|
||||||
|
except ImportError:
|
||||||
|
extracted_text = "[Erreur: L'administrateur doit exécuter 'pip install PyPDF2' sur le serveur pour lire les PDF.]"
|
||||||
|
|
||||||
|
elif ext in ['xls', 'xlsx', 'xlsm', 'xlsb']:
|
||||||
|
try:
|
||||||
|
import openpyxl
|
||||||
|
wb = openpyxl.load_workbook(io.BytesIO(file_bytes), data_only=False)
|
||||||
|
for sheet_name in wb.sheetnames:
|
||||||
|
sheet = wb[sheet_name]
|
||||||
|
extracted_text += f"\n--- Feuille : {sheet_name} ---\n"
|
||||||
|
for row in sheet.iter_rows(values_only=True):
|
||||||
|
row_vals = [str(cell) if cell is not None else "" for cell in row]
|
||||||
|
if any(row_vals):
|
||||||
|
extracted_text += "\t".join(row_vals) + "\n"
|
||||||
|
|
||||||
|
if ext in ['xlsm', 'xlsb', 'xls']:
|
||||||
|
try:
|
||||||
|
from oletools.olevba import VBA_Parser
|
||||||
|
vbaparser = VBA_Parser("filename", data=file_bytes)
|
||||||
|
if vbaparser.detect_vba_macros():
|
||||||
|
extracted_text += "\n\n--- MACROS VBA DETECTEES ---\n"
|
||||||
|
for (filename, stream_path, vba_filename, vba_code) in vbaparser.extract_macros():
|
||||||
|
extracted_text += f"\n// Module: {vba_filename}\n{vba_code}\n"
|
||||||
|
except ImportError:
|
||||||
|
extracted_text += "\n[Extraction VBA impossible: L'administrateur doit exécuter 'pip install oletools' sur le serveur.]\n"
|
||||||
|
except Exception as e:
|
||||||
|
extracted_text += f"\n[Erreur de lecture VBA interne: {str(e)}]\n"
|
||||||
|
|
||||||
|
except ImportError:
|
||||||
|
extracted_text = "[Erreur: L'administrateur doit exécuter 'pip install openpyxl' sur le serveur pour lire Excel.]"
|
||||||
|
else:
|
||||||
|
extracted_text = file_bytes.decode('utf-8', errors='replace')
|
||||||
|
|
||||||
|
extracted_files_data.append({"name": f.name, "content": extracted_text})
|
||||||
|
except Exception as e:
|
||||||
|
extracted_files_data.append({"name": f.name, "content": f"[ERREUR DE DECODAGE: {str(e)}]"})
|
||||||
|
else:
|
||||||
|
extracted_files_data.append({"name": f.name, "content": content})
|
||||||
|
|
||||||
|
db_content = message.content
|
||||||
|
if files_names:
|
||||||
|
db_content += f"\n\n[Fichiers joints pour analyse : {', '.join(files_names)}]"
|
||||||
|
|
||||||
|
db.add(Message(role=message.role, content=db_content, project_id=project_id))
|
||||||
|
db.commit()
|
||||||
|
|
||||||
|
history = db.query(Message).filter(Message.project_id == project_id).order_by(Message.created_at.asc()).all()
|
||||||
|
settings = db.query(SystemSettings).first()
|
||||||
|
conf = message.config.dict() if message.config else {"workers": ["gemini-3.5-flash-lite"]}
|
||||||
|
|
||||||
|
ai_resp = await run_orchestrator(db, history, settings, conf, extracted_files_data)
|
||||||
|
|
||||||
|
db.add(Message(role="assistant", content=ai_resp, project_id=project_id))
|
||||||
|
db.commit()
|
||||||
return db.query(Message).filter(Message.project_id == project_id).order_by(Message.created_at.asc()).all()
|
return db.query(Message).filter(Message.project_id == project_id).order_by(Message.created_at.asc()).all()
|
||||||
|
|
||||||
@app.post("/api/projects/{project_id}/messages", response_model=MessageResponse)
|
# --- ROUTES MODÈLES & FINANCES ---
|
||||||
def create_message(project_id: int, message: MessageCreate, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
|
@app.get("/api/models/info")
|
||||||
"""Ajoute un message à un projet."""
|
def get_models_info(db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
|
||||||
project = db.query(Project).filter(Project.id == project_id, Project.user_id == current_user.id).first()
|
settings = db.query(SystemSettings).first()
|
||||||
if not project:
|
return {
|
||||||
raise HTTPException(status_code=404, detail="Projet introuvable")
|
"last_sync_date": settings.last_sync_date.isoformat() if settings and settings.last_sync_date else None,
|
||||||
|
"last_sync_type": settings.last_sync_type if settings else None,
|
||||||
|
"models": db.query(AIModel).order_by(AIModel.name.asc()).all(),
|
||||||
|
"finances": db.query(FinancialLog).all()
|
||||||
|
}
|
||||||
|
|
||||||
new_message = Message(role=message.role, content=message.content, project_id=project_id)
|
@app.get("/api/finances")
|
||||||
db.add(new_message)
|
def get_finances(db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
|
||||||
db.commit()
|
return db.query(FinancialLog).all()
|
||||||
db.refresh(new_message)
|
|
||||||
return new_message
|
@app.post("/api/models/sync")
|
||||||
|
async def trigger_model_sync(db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
|
||||||
|
if not current_user.is_admin: raise HTTPException(status_code=403, detail="Accès admin requis.")
|
||||||
|
settings = db.query(SystemSettings).first()
|
||||||
|
return await sync_providers_models(db, settings, "Manuelle")
|
||||||
|
|
||||||
|
@app.get("/api/models/export")
|
||||||
|
def export_models(db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
|
||||||
|
if not current_user.is_admin: raise HTTPException(status_code=403, detail="Accès admin requis.")
|
||||||
|
models = db.query(AIModel).order_by(AIModel.provider.asc(), AIModel.name.asc()).all()
|
||||||
|
|
||||||
|
csv_io = io.StringIO()
|
||||||
|
writer = csv.writer(csv_io, delimiter=',')
|
||||||
|
writer.writerow(["Provider", "Model ID", "Name", "Domain", "Is Free", "Context Length", "Pricing Prompt", "Pricing Completion", "Description"])
|
||||||
|
for m in models:
|
||||||
|
writer.writerow([m.provider, m.model_id, m.name, m.domain, m.is_free, m.context_length, m.pricing_prompt, m.pricing_completion, m.description_fr])
|
||||||
|
|
||||||
|
md_content = f"# Extraction des Modèles IA - AETHAS38\n\n**Date d'extraction :** {datetime.now().strftime('%d/%m/%Y à %H:%M:%S')}\n\n"
|
||||||
|
providers = sorted(list(set(m.provider for m in models)))
|
||||||
|
for prov in providers:
|
||||||
|
md_content += f"## Fournisseur : {prov.upper()}\n\n"
|
||||||
|
prov_models = [m for m in models if m.provider == prov]
|
||||||
|
for m in prov_models:
|
||||||
|
price_info = "**GRATUIT**" if m.is_free else f"In: ${m.pricing_prompt:.2f} / Out: ${m.pricing_completion:.2f}"
|
||||||
|
ctx_info = f"{int(m.context_length/1000)}k"
|
||||||
|
desc = m.description_fr.replace('\n', ' ') if m.description_fr else ""
|
||||||
|
md_content += f"- **{m.name or m.model_id}** (`{m.model_id}`)\n"
|
||||||
|
md_content += f" - *Domaine :* {m.domain}\n"
|
||||||
|
md_content += f" - *Prix (1M tokens) :* {price_info}\n"
|
||||||
|
md_content += f" - *Contexte :* {ctx_info}\n"
|
||||||
|
md_content += f" - *Description :* {desc}\n\n"
|
||||||
|
|
||||||
|
zip_io = io.BytesIO()
|
||||||
|
with zipfile.ZipFile(zip_io, mode='w', compression=zipfile.ZIP_DEFLATED) as zf:
|
||||||
|
zf.writestr("models_export.csv", csv_io.getvalue().encode('utf-8'))
|
||||||
|
zf.writestr(f"{datetime.now().strftime('%Y%m%d')}-extraction-modeles.md", md_content.encode('utf-8'))
|
||||||
|
|
||||||
|
zip_io.seek(0)
|
||||||
|
return Response(
|
||||||
|
content=zip_io.getvalue(),
|
||||||
|
media_type="application/zip",
|
||||||
|
headers={"Content-Disposition": f"attachment; filename=aethas38_models_{datetime.now().strftime('%Y%m%d')}.zip"}
|
||||||
|
)
|
||||||
+37
-25
@@ -1,58 +1,46 @@
|
|||||||
from sqlalchemy import Column, Integer, String, Boolean, DateTime, ForeignKey, Text
|
from sqlalchemy import Column, Integer, String, Boolean, ForeignKey, DateTime, Float
|
||||||
from sqlalchemy.orm import relationship
|
from sqlalchemy.orm import relationship
|
||||||
from datetime import datetime, timezone
|
from datetime import datetime, timezone
|
||||||
from .database import Base
|
from .database import Base
|
||||||
|
|
||||||
class User(Base):
|
class User(Base):
|
||||||
__tablename__ = "users"
|
__tablename__ = "users"
|
||||||
|
|
||||||
id = Column(Integer, primary_key=True, index=True)
|
id = Column(Integer, primary_key=True, index=True)
|
||||||
email = Column(String, unique=True, index=True, nullable=False)
|
email = Column(String, unique=True, index=True)
|
||||||
username = Column(String, unique=True, index=True, nullable=False)
|
username = Column(String, unique=True, index=True)
|
||||||
hashed_password = Column(String, nullable=False)
|
hashed_password = Column(String)
|
||||||
totp_secret = Column(String, nullable=False)
|
totp_secret = Column(String)
|
||||||
is_admin = Column(Boolean, default=False)
|
is_admin = Column(Boolean, default=False)
|
||||||
|
is_superadmin = Column(Boolean, default=False)
|
||||||
# Relation : Un utilisateur peut avoir plusieurs projets
|
avatar_path = Column(String, nullable=True)
|
||||||
projects = relationship("Project", back_populates="owner", cascade="all, delete-orphan")
|
projects = relationship("Project", back_populates="owner")
|
||||||
|
|
||||||
|
|
||||||
class Project(Base):
|
class Project(Base):
|
||||||
__tablename__ = "projects"
|
__tablename__ = "projects"
|
||||||
|
|
||||||
id = Column(Integer, primary_key=True, index=True)
|
id = Column(Integer, primary_key=True, index=True)
|
||||||
title = Column(String, nullable=False)
|
title = Column(String, index=True)
|
||||||
created_at = Column(DateTime, default=lambda: datetime.now(timezone.utc))
|
created_at = Column(DateTime, default=lambda: datetime.now(timezone.utc))
|
||||||
is_pinned = Column(Boolean, default=False)
|
is_pinned = Column(Boolean, default=False)
|
||||||
user_id = Column(Integer, ForeignKey("users.id"), nullable=False)
|
user_id = Column(Integer, ForeignKey("users.id"))
|
||||||
|
|
||||||
# Relations
|
|
||||||
owner = relationship("User", back_populates="projects")
|
owner = relationship("User", back_populates="projects")
|
||||||
messages = relationship("Message", back_populates="project", cascade="all, delete-orphan")
|
messages = relationship("Message", back_populates="project", cascade="all, delete-orphan")
|
||||||
|
|
||||||
|
|
||||||
class Message(Base):
|
class Message(Base):
|
||||||
__tablename__ = "messages"
|
__tablename__ = "messages"
|
||||||
|
|
||||||
id = Column(Integer, primary_key=True, index=True)
|
id = Column(Integer, primary_key=True, index=True)
|
||||||
role = Column(String, nullable=False) # Ex: 'user', 'assistant', 'worker_1', 'system'
|
role = Column(String)
|
||||||
content = Column(Text, nullable=False)
|
content = Column(String)
|
||||||
created_at = Column(DateTime, default=lambda: datetime.now(timezone.utc))
|
created_at = Column(DateTime, default=lambda: datetime.now(timezone.utc))
|
||||||
project_id = Column(Integer, ForeignKey("projects.id"), nullable=False)
|
project_id = Column(Integer, ForeignKey("projects.id"))
|
||||||
|
|
||||||
# Relation
|
|
||||||
project = relationship("Project", back_populates="messages")
|
project = relationship("Project", back_populates="messages")
|
||||||
|
|
||||||
class SystemSettings(Base):
|
class SystemSettings(Base):
|
||||||
__tablename__ = "system_settings"
|
__tablename__ = "system_settings"
|
||||||
|
|
||||||
id = Column(Integer, primary_key=True, index=True)
|
id = Column(Integer, primary_key=True, index=True)
|
||||||
# Serveur Mail (Obligatoire)
|
|
||||||
smtp_host = Column(String, nullable=False)
|
smtp_host = Column(String, nullable=False)
|
||||||
smtp_port = Column(Integer, nullable=False)
|
smtp_port = Column(Integer, nullable=False)
|
||||||
smtp_user = Column(String, nullable=False)
|
smtp_user = Column(String, nullable=False)
|
||||||
smtp_password = Column(String, nullable=False)
|
smtp_password = Column(String, nullable=False)
|
||||||
# Clés API (Optionnelles)
|
|
||||||
openrouter_api_key = Column(String, nullable=True)
|
openrouter_api_key = Column(String, nullable=True)
|
||||||
openrouter_management_key = Column(String, nullable=True)
|
openrouter_management_key = Column(String, nullable=True)
|
||||||
groq_api_key = Column(String, nullable=True)
|
groq_api_key = Column(String, nullable=True)
|
||||||
@@ -62,3 +50,27 @@ class SystemSettings(Base):
|
|||||||
cloudflare_account_id = Column(String, nullable=True)
|
cloudflare_account_id = Column(String, nullable=True)
|
||||||
cloudflare_api_token = Column(String, nullable=True)
|
cloudflare_api_token = Column(String, nullable=True)
|
||||||
huggingface_api_key = Column(String, nullable=True)
|
huggingface_api_key = Column(String, nullable=True)
|
||||||
|
last_sync_date = Column(DateTime, nullable=True)
|
||||||
|
last_sync_type = Column(String, nullable=True)
|
||||||
|
|
||||||
|
class AIModel(Base):
|
||||||
|
__tablename__ = "ai_models"
|
||||||
|
id = Column(Integer, primary_key=True, index=True)
|
||||||
|
provider = Column(String, index=True)
|
||||||
|
model_id = Column(String, unique=True, index=True)
|
||||||
|
name = Column(String)
|
||||||
|
description_fr = Column(String, nullable=True)
|
||||||
|
domain = Column(String, default="Texte")
|
||||||
|
is_free = Column(Boolean, default=False)
|
||||||
|
context_length = Column(Integer)
|
||||||
|
pricing_prompt = Column(Float)
|
||||||
|
pricing_completion = Column(Float)
|
||||||
|
last_updated = Column(DateTime, default=lambda: datetime.now(timezone.utc))
|
||||||
|
|
||||||
|
class FinancialLog(Base):
|
||||||
|
__tablename__ = "financial_logs"
|
||||||
|
id = Column(Integer, primary_key=True, index=True)
|
||||||
|
provider = Column(String, unique=True, index=True)
|
||||||
|
balance = Column(Float, default=0.0)
|
||||||
|
total_usage = Column(Float, default=0.0)
|
||||||
|
checked_at = Column(DateTime, default=lambda: datetime.now(timezone.utc))
|
||||||
@@ -0,0 +1,316 @@
|
|||||||
|
import asyncio
|
||||||
|
import httpx
|
||||||
|
import urllib.parse
|
||||||
|
from openai import AsyncOpenAI
|
||||||
|
from fastapi import HTTPException
|
||||||
|
from sqlalchemy.orm import Session
|
||||||
|
from sqlalchemy.exc import IntegrityError
|
||||||
|
from .models import SystemSettings, AIModel, FinancialLog
|
||||||
|
from datetime import datetime, timezone
|
||||||
|
|
||||||
|
# --- LOGGER GLOBAL POUR LE TERMINAL ---
|
||||||
|
activity_logs = []
|
||||||
|
|
||||||
|
def log_activity(msg: str):
|
||||||
|
ts = datetime.now(timezone.utc).strftime('%H:%M:%S')
|
||||||
|
activity_logs.append(f"[{ts}] {msg}")
|
||||||
|
if len(activity_logs) > 100:
|
||||||
|
activity_logs.pop(0)
|
||||||
|
# --------------------------------------
|
||||||
|
|
||||||
|
def determine_domain(model_id: str) -> str:
|
||||||
|
mid = model_id.lower()
|
||||||
|
if "vision" in mid or "vl" in mid or "omni" in mid: return "Vision & Texte"
|
||||||
|
if "coder" in mid or "code" in mid or "math" in mid: return "Code & Logique"
|
||||||
|
if "audio" in mid or "whisper" in mid: return "Audio"
|
||||||
|
return "Texte Polyvalent"
|
||||||
|
|
||||||
|
async def translate_en_to_fr(client: httpx.AsyncClient, text: str) -> str:
|
||||||
|
if not text: return "Aucune description fournie."
|
||||||
|
try:
|
||||||
|
short_text = text[:300].strip()
|
||||||
|
url = f"https://translate.googleapis.com/translate_a/single?client=gtx&sl=en&tl=fr&dt=t&q={urllib.parse.quote(short_text)}"
|
||||||
|
resp = await client.get(url, timeout=4.0)
|
||||||
|
if resp.status_code == 200:
|
||||||
|
translated = "".join([s[0] for s in resp.json()[0]])
|
||||||
|
return translated + ("..." if len(text) > 300 else "")
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
|
return text[:200] + "..."
|
||||||
|
|
||||||
|
async def sync_finances(db: Session, settings: SystemSettings):
|
||||||
|
async with httpx.AsyncClient(timeout=30.0) as client:
|
||||||
|
if settings.openrouter_management_key or settings.openrouter_api_key:
|
||||||
|
try:
|
||||||
|
key = settings.openrouter_management_key or settings.openrouter_api_key
|
||||||
|
cred_resp = await client.get("https://openrouter.ai/api/v1/credits", headers={"Authorization": f"Bearer {key}"})
|
||||||
|
balance = 0.0
|
||||||
|
usage = 0.0
|
||||||
|
if cred_resp.status_code == 200 and cred_resp.json().get("data"):
|
||||||
|
c_data = cred_resp.json().get("data", {})
|
||||||
|
balance = float(c_data.get("total_credits") or 0.0) - float(c_data.get("total_usage") or 0.0)
|
||||||
|
usage = float(c_data.get("total_usage") or 0.0)
|
||||||
|
else:
|
||||||
|
key_resp = await client.get("https://openrouter.ai/api/v1/auth/key", headers={"Authorization": f"Bearer {key}"})
|
||||||
|
if key_resp.status_code == 200:
|
||||||
|
data = key_resp.json().get("data", {})
|
||||||
|
limit = data.get("limit")
|
||||||
|
usage = float(data.get("usage") or 0.0)
|
||||||
|
balance = (float(limit) - usage) if limit is not None else -usage
|
||||||
|
update_finance_db(db, "OpenRouter", balance, usage)
|
||||||
|
except Exception as e: print(f"Erreur Finance OR: {e}")
|
||||||
|
|
||||||
|
if settings.groq_api_key: update_finance_db(db, "Groq", 999.0, 0.0)
|
||||||
|
if settings.deepseek_api_key:
|
||||||
|
try:
|
||||||
|
resp = await client.get("https://api.deepseek.com/user/balance", headers={"Authorization": f"Bearer {settings.deepseek_api_key}"})
|
||||||
|
if resp.status_code == 200:
|
||||||
|
infos = resp.json().get("balance_infos", [{}])[0]
|
||||||
|
update_finance_db(db, "DeepSeek", float(infos.get("total_balance", 0)), 0.0)
|
||||||
|
except Exception: pass
|
||||||
|
if settings.mistral_api_key: update_finance_db(db, "Mistral", 0.0, 0.0)
|
||||||
|
if settings.gemini_api_key: update_finance_db(db, "Gemini", 0.0, 0.0)
|
||||||
|
|
||||||
|
try: db.commit()
|
||||||
|
except: db.rollback()
|
||||||
|
|
||||||
|
def update_finance_db(db, provider, balance, usage):
|
||||||
|
try:
|
||||||
|
log = db.query(FinancialLog).filter(FinancialLog.provider == provider).first()
|
||||||
|
if log:
|
||||||
|
log.balance = balance; log.total_usage = usage; log.checked_at = datetime.now(timezone.utc)
|
||||||
|
else:
|
||||||
|
db.add(FinancialLog(provider=provider, balance=balance, total_usage=usage))
|
||||||
|
except Exception as e: print(f"Finance DB Error: {e}")
|
||||||
|
|
||||||
|
async def sync_providers_models(db: Session, settings: SystemSettings, sync_type: str = "Automatique"):
|
||||||
|
log_activity(f"Lancement de la synchronisation des modèles ({sync_type})...")
|
||||||
|
added = 0
|
||||||
|
models_to_process = {}
|
||||||
|
async with httpx.AsyncClient(timeout=90.0) as client:
|
||||||
|
if settings.openrouter_api_key:
|
||||||
|
try:
|
||||||
|
resp = await client.get("https://openrouter.ai/api/v1/models")
|
||||||
|
if resp.status_code == 200:
|
||||||
|
for item in resp.json().get("data", []):
|
||||||
|
try:
|
||||||
|
m_id = item.get("id")
|
||||||
|
if not m_id: continue
|
||||||
|
pricing = item.get("pricing") or {}
|
||||||
|
pp = float(pricing.get("prompt") or 0.0) * 1000000
|
||||||
|
pc = float(pricing.get("completion") or 0.0) * 1000000
|
||||||
|
is_free = (pp == 0.0 and pc == 0.0)
|
||||||
|
desc_en = item.get("description", "Generic AI Model.")
|
||||||
|
models_to_process[m_id] = {"provider": "openrouter", "name": item.get("name", "Inconnu"), "desc_en": desc_en, "domain": determine_domain(m_id), "is_free": is_free, "ctx": item.get("context_length", 0), "pp": pp, "pc": pc}
|
||||||
|
except: pass
|
||||||
|
except Exception as e: print(f"Erreur OR Models: {e}")
|
||||||
|
|
||||||
|
if settings.groq_api_key:
|
||||||
|
try:
|
||||||
|
resp = await client.get("https://api.groq.com/openai/v1/models", headers={"Authorization": f"Bearer {settings.groq_api_key}"})
|
||||||
|
if resp.status_code == 200:
|
||||||
|
for item in resp.json().get("data", []):
|
||||||
|
m_id = item["id"]
|
||||||
|
models_to_process[m_id] = {"provider": "groq", "name": m_id.capitalize(), "desc": "Modèle très rapide hébergé sur LPU Groq.", "domain": determine_domain(m_id), "is_free": True, "ctx": 8192, "pp": 0.0, "pc": 0.0}
|
||||||
|
except Exception: pass
|
||||||
|
|
||||||
|
if settings.deepseek_api_key:
|
||||||
|
try:
|
||||||
|
resp = await client.get("https://api.deepseek.com/models", headers={"Authorization": f"Bearer {settings.deepseek_api_key}"})
|
||||||
|
if resp.status_code == 200:
|
||||||
|
for item in resp.json().get("data", []):
|
||||||
|
m_id = item["id"]
|
||||||
|
models_to_process[m_id] = {"provider": "deepseek", "name": m_id.capitalize(), "desc": "Modèle officiel du fournisseur DeepSeek.", "domain": determine_domain(m_id), "is_free": False, "ctx": 64000, "pp": 0.14, "pc": 0.28}
|
||||||
|
except Exception: pass
|
||||||
|
|
||||||
|
if settings.mistral_api_key:
|
||||||
|
try:
|
||||||
|
resp = await client.get("https://api.mistral.ai/v1/models", headers={"Authorization": f"Bearer {settings.mistral_api_key}"})
|
||||||
|
if resp.status_code == 200:
|
||||||
|
for item in resp.json().get("data", []):
|
||||||
|
m_id = item["id"]
|
||||||
|
models_to_process[m_id] = {"provider": "mistral", "name": m_id.capitalize(), "desc": "Modèle officiel développé par Mistral AI.", "domain": determine_domain(m_id), "is_free": False, "ctx": 32000, "pp": 0.2, "pc": 0.6}
|
||||||
|
except Exception: pass
|
||||||
|
|
||||||
|
if settings.gemini_api_key:
|
||||||
|
try:
|
||||||
|
resp = await client.get(f"https://generativelanguage.googleapis.com/v1beta/models?key={settings.gemini_api_key}")
|
||||||
|
if resp.status_code == 200:
|
||||||
|
for item in resp.json().get("models", []):
|
||||||
|
m_id = item["name"].replace("models/", "")
|
||||||
|
models_to_process[m_id] = {"provider": "gemini", "name": item.get("displayName", m_id), "desc": "Modèle natif de l'écosystème Google Gemini.", "domain": determine_domain(m_id), "is_free": True, "ctx": item.get("inputTokenLimit", 32000), "pp": 0.0, "pc": 0.0}
|
||||||
|
except Exception: pass
|
||||||
|
|
||||||
|
if settings.cloudflare_account_id and settings.cloudflare_api_token:
|
||||||
|
try:
|
||||||
|
url = f"https://api.cloudflare.com/client/v4/accounts/{settings.cloudflare_account_id}/ai/models/search"
|
||||||
|
resp = await client.get(url, headers={"Authorization": f"Bearer {settings.cloudflare_api_token}"})
|
||||||
|
if resp.status_code == 200:
|
||||||
|
for item in resp.json().get("result", []):
|
||||||
|
m_id = item.get("name")
|
||||||
|
models_to_process[m_id] = {"provider": "cloudflare", "name": m_id.split("/")[-1], "desc": "Modèle Serverless Cloudflare Workers AI.", "domain": determine_domain(m_id), "is_free": True, "ctx": 4096, "pp": 0.0, "pc": 0.0}
|
||||||
|
except Exception: pass
|
||||||
|
|
||||||
|
sem = asyncio.Semaphore(15)
|
||||||
|
async def process_and_translate(m_id, data, client_session):
|
||||||
|
async with sem:
|
||||||
|
if "desc_en" in data:
|
||||||
|
data["desc"] = await translate_en_to_fr(client_session, data["desc_en"])
|
||||||
|
return m_id, data
|
||||||
|
|
||||||
|
async with httpx.AsyncClient(timeout=30.0) as client_trans:
|
||||||
|
tasks = [process_and_translate(m_id, data, client_trans) for m_id, data in models_to_process.items()]
|
||||||
|
translated_results = await asyncio.gather(*tasks)
|
||||||
|
|
||||||
|
for m_id, data in translated_results:
|
||||||
|
try:
|
||||||
|
existing = db.query(AIModel).filter(AIModel.model_id == m_id).first()
|
||||||
|
if existing:
|
||||||
|
existing.pricing_prompt = data["pp"]; existing.pricing_completion = data["pc"]; existing.is_free = data["is_free"]; existing.description_fr = data["desc"]; existing.last_updated = datetime.now(timezone.utc)
|
||||||
|
else:
|
||||||
|
db.add(AIModel(provider=data["provider"], model_id=m_id, name=data["name"], description_fr=data["desc"], domain=data["domain"], is_free=data["is_free"], context_length=data["ctx"], pricing_prompt=data["pp"], pricing_completion=data["pc"]))
|
||||||
|
added += 1
|
||||||
|
if added % 50 == 0: db.commit()
|
||||||
|
except IntegrityError: db.rollback()
|
||||||
|
except Exception: db.rollback()
|
||||||
|
|
||||||
|
try:
|
||||||
|
settings.last_sync_date = datetime.now(timezone.utc)
|
||||||
|
settings.last_sync_type = sync_type
|
||||||
|
db.commit()
|
||||||
|
except Exception as e:
|
||||||
|
db.rollback()
|
||||||
|
|
||||||
|
await sync_finances(db, settings)
|
||||||
|
log_activity(f"Synchronisation terminée : {added} modèles analysés.")
|
||||||
|
return {"status": "success", "models_processed": added}
|
||||||
|
|
||||||
|
def get_client_for_model(db: Session, model_id: str, settings: SystemSettings):
|
||||||
|
model_db = db.query(AIModel).filter(AIModel.model_id == model_id).first()
|
||||||
|
provider = model_db.provider if model_db else "openrouter"
|
||||||
|
|
||||||
|
if provider == "gemini" and settings.gemini_api_key:
|
||||||
|
return AsyncOpenAI(base_url="https://generativelanguage.googleapis.com/v1beta/openai/", api_key=settings.gemini_api_key), "gemini"
|
||||||
|
elif provider == "groq" and settings.groq_api_key:
|
||||||
|
return AsyncOpenAI(base_url="https://api.groq.com/openai/v1", api_key=settings.groq_api_key), "groq"
|
||||||
|
elif provider == "deepseek" and settings.deepseek_api_key:
|
||||||
|
return AsyncOpenAI(base_url="https://api.deepseek.com/v1", api_key=settings.deepseek_api_key), "deepseek"
|
||||||
|
elif provider == "mistral" and settings.mistral_api_key:
|
||||||
|
return AsyncOpenAI(base_url="https://api.mistral.ai/v1", api_key=settings.mistral_api_key), "mistral"
|
||||||
|
elif provider == "cloudflare" and settings.cloudflare_account_id and settings.cloudflare_api_token:
|
||||||
|
return AsyncOpenAI(base_url=f"https://api.cloudflare.com/client/v4/accounts/{settings.cloudflare_account_id}/ai/v1", api_key=settings.cloudflare_api_token), "cloudflare"
|
||||||
|
|
||||||
|
return AsyncOpenAI(base_url="https://openrouter.ai/api/v1", api_key=settings.openrouter_api_key), "openrouter"
|
||||||
|
|
||||||
|
async def ask_agent(client, model_id, messages, provider="openrouter"):
|
||||||
|
kwargs = {"model": model_id, "messages": messages}
|
||||||
|
if provider == "openrouter":
|
||||||
|
kwargs["extra_headers"] = {"HTTP-Referer": "https://aethas38.duckdns.org", "X-Title": "AETHAS38 Orchestrator"}
|
||||||
|
# Activation native du plugin web-search et de compression
|
||||||
|
kwargs["extra_body"] = {"plugins": [{"id": "context-compression"}, {"id": "web-search"}]}
|
||||||
|
|
||||||
|
resp = await client.chat.completions.create(**kwargs)
|
||||||
|
return resp.choices[0].message.content
|
||||||
|
|
||||||
|
async def run_orchestrator(db: Session, history: list, settings: SystemSettings, config: dict, extracted_files: list = None) -> str:
|
||||||
|
workers = config.get("workers", ["gemini-3.5-flash-lite"])
|
||||||
|
user_prompt = history[-1].content
|
||||||
|
original_user_text = user_prompt.split("\n\n[Fichiers joints")[0] if "[Fichiers joints" in user_prompt else user_prompt
|
||||||
|
formatted_history = [{"role": msg.role, "content": msg.content} for msg in history[:-1]]
|
||||||
|
|
||||||
|
final_response = ""
|
||||||
|
try:
|
||||||
|
p_mod = config.get("prompter")
|
||||||
|
if not p_mod: p_mod = "gemini-3.5-flash-lite"
|
||||||
|
|
||||||
|
log_activity(f"Démarrage de l'orchestration. Modèle Prompteur: {p_mod}")
|
||||||
|
p_client, p_prov = get_client_for_model(db, p_mod, settings)
|
||||||
|
|
||||||
|
files_context = ""
|
||||||
|
if extracted_files:
|
||||||
|
log_activity(f"Traitement Map-Reduce de {len(extracted_files)} fichier(s) joint(s).")
|
||||||
|
async def process_single_file(f):
|
||||||
|
file_sys = "You are an expert data analyst and senior developer. Extract the most important technical information from the file without losing critical code syntax."
|
||||||
|
content = f['content']
|
||||||
|
chunk_size = 150000
|
||||||
|
|
||||||
|
if len(content) > chunk_size:
|
||||||
|
chunks = [content[i:i+chunk_size] for i in range(0, len(content), chunk_size)]
|
||||||
|
chunk_analyses = []
|
||||||
|
log_activity(f"Fichier lourd ({f['name']}): Chunking en {len(chunks)} morceaux.")
|
||||||
|
for idx, chunk in enumerate(chunks):
|
||||||
|
file_prompt = f"Demande de l'utilisateur : '{original_user_text}'.\n\nPartie {idx+1}/{len(chunks)} du fichier '{f['name']}'. Analysez, extrayez et résumez le code, VBA, SQL ou les données pertinentes.\n\nContenu :\n```\n{chunk}\n```"
|
||||||
|
try:
|
||||||
|
analysis = await ask_agent(p_client, p_mod, [{"role": "system", "content": file_sys}, {"role": "user", "content": file_prompt}], p_prov)
|
||||||
|
chunk_analyses.append(analysis)
|
||||||
|
log_activity(f"Analyse chunk {idx+1}/{len(chunks)} pour {f['name']} réussie.")
|
||||||
|
except Exception as e:
|
||||||
|
chunk_analyses.append(f"[Erreur sur la partie {idx+1}: {str(e)}]")
|
||||||
|
log_activity(f"Erreur chunk {idx+1}/{len(chunks)} pour {f['name']}: {str(e)}")
|
||||||
|
|
||||||
|
await asyncio.sleep(1.5)
|
||||||
|
|
||||||
|
return f"\n\n--- Extraction du fichier {f['name']} (en {len(chunks)} parties) ---\n" + "\n".join(chunk_analyses)
|
||||||
|
else:
|
||||||
|
file_prompt = f"Demande de l'utilisateur : '{original_user_text}'.\n\nAnalysez le fichier ci-dessous. Extrayez, résumez et conservez méticuleusement tout le code, les macros VBA, les requêtes SQL, ou les données métier pertinentes pour répondre à la demande.\n\nFichier : {f['name']}\nContenu :\n```\n{content}\n```"
|
||||||
|
try:
|
||||||
|
analysis = await ask_agent(p_client, p_mod, [{"role": "system", "content": file_sys}, {"role": "user", "content": file_prompt}], p_prov)
|
||||||
|
log_activity(f"Analyse intégrale de {f['name']} réussie.")
|
||||||
|
return f"\n\n--- Extraction du fichier {f['name']} ---\n{analysis}"
|
||||||
|
except Exception as e:
|
||||||
|
log_activity(f"Erreur d'analyse sur {f['name']}: {str(e)}")
|
||||||
|
return f"\n\n--- Erreur sur {f['name']} ---\n{str(e)}"
|
||||||
|
|
||||||
|
file_analyses = []
|
||||||
|
for f in extracted_files:
|
||||||
|
analysis = await process_single_file(f)
|
||||||
|
file_analyses.append(analysis)
|
||||||
|
await asyncio.sleep(1.5)
|
||||||
|
|
||||||
|
files_context = "".join(file_analyses)
|
||||||
|
user_prompt = f"{original_user_text}\n\nVoici les données pré-traitées des fichiers joints :\n{files_context}"
|
||||||
|
|
||||||
|
log_activity(f"Optimisation/Traduction de la requête via Prompteur...")
|
||||||
|
prompt_system = "You are an expert prompt engineer. Translate and optimize the user request and any file context into clear, precise English tailored for AI execution. Keep all code blocks intact."
|
||||||
|
optimized = await ask_agent(p_client, p_mod, [{"role": "system", "content": prompt_system}, {"role": "user", "content": user_prompt}], p_prov)
|
||||||
|
|
||||||
|
log_activity(f"Lancement de {len(workers)} travailleur(s) en parallèle...")
|
||||||
|
if len(workers) == 1:
|
||||||
|
w_mod = workers[0]
|
||||||
|
client, provider = get_client_for_model(db, w_mod, settings)
|
||||||
|
worker_response = await ask_agent(client, w_mod, formatted_history + [{"role": "user", "content": optimized}], provider)
|
||||||
|
responses = [worker_response]
|
||||||
|
log_activity(f"Travailleur 1 ({w_mod}) a terminé.")
|
||||||
|
else:
|
||||||
|
w_tasks = []
|
||||||
|
for w in workers:
|
||||||
|
w_client, w_prov = get_client_for_model(db, w, settings)
|
||||||
|
w_tasks.append(ask_agent(w_client, w, formatted_history + [{"role": "user", "content": optimized}], w_prov))
|
||||||
|
responses = await asyncio.gather(*w_tasks, return_exceptions=True)
|
||||||
|
log_activity("Tous les travailleurs ont terminé leur analyse.")
|
||||||
|
|
||||||
|
c_mod = config.get("concatenator")
|
||||||
|
if not c_mod: c_mod = "gemini-3.5-flash-lite"
|
||||||
|
|
||||||
|
c_client, c_prov = get_client_for_model(db, c_mod, settings)
|
||||||
|
|
||||||
|
log_activity(f"Synthèse et traduction finale via Concaténeur ({c_mod})...")
|
||||||
|
concat_system = (
|
||||||
|
"You are a master lead developer and technical synthesizer. "
|
||||||
|
"Synthesize the provided expert responses into a single cohesive response. "
|
||||||
|
"Translate all explanatory text, descriptions, and user-facing prose into natural French. "
|
||||||
|
"CRITICAL: Do NOT translate code blocks, programming keywords, or source code contents. "
|
||||||
|
"You may translate code comments into French if appropriate, but leave code syntax strictly intact."
|
||||||
|
)
|
||||||
|
|
||||||
|
synth = f"User Request: {original_user_text}\n\n" + "\n".join([f"--- EXPERT {i+1} ---\n{str(r)}" for i, r in enumerate(responses)])
|
||||||
|
final_response = await ask_agent(c_client, c_mod, [{"role": "system", "content": concat_system}, {"role": "user", "content": synth}], c_prov)
|
||||||
|
|
||||||
|
log_activity("Orchestration terminée avec succès.")
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
final_response = f"L'IA a rencontré une erreur critique: {str(e)}"
|
||||||
|
log_activity(f"ERREUR CRITIQUE: {str(e)}")
|
||||||
|
|
||||||
|
await sync_finances(db, settings)
|
||||||
|
return final_response
|
||||||
+46
-12
@@ -1,17 +1,15 @@
|
|||||||
from pydantic import BaseModel, EmailStr
|
from pydantic import BaseModel, EmailStr
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
from typing import Optional
|
from typing import Optional, List
|
||||||
|
|
||||||
class AdminCreate(BaseModel):
|
class AdminCreate(BaseModel):
|
||||||
email: EmailStr
|
email: EmailStr
|
||||||
username: str
|
username: str
|
||||||
password: str
|
password: str
|
||||||
# Serveur Mail (Obligatoire)
|
|
||||||
smtp_host: str
|
smtp_host: str
|
||||||
smtp_port: int
|
smtp_port: int
|
||||||
smtp_user: str
|
smtp_user: str
|
||||||
smtp_password: str
|
smtp_password: str
|
||||||
# Clés API (Optionnelles)
|
|
||||||
openrouter_api_key: Optional[str] = None
|
openrouter_api_key: Optional[str] = None
|
||||||
openrouter_management_key: Optional[str] = None
|
openrouter_management_key: Optional[str] = None
|
||||||
groq_api_key: Optional[str] = None
|
groq_api_key: Optional[str] = None
|
||||||
@@ -22,38 +20,74 @@ class AdminCreate(BaseModel):
|
|||||||
cloudflare_api_token: Optional[str] = None
|
cloudflare_api_token: Optional[str] = None
|
||||||
huggingface_api_key: Optional[str] = None
|
huggingface_api_key: Optional[str] = None
|
||||||
|
|
||||||
|
class SystemSettingsBase(BaseModel):
|
||||||
|
smtp_host: Optional[str] = None
|
||||||
|
smtp_port: Optional[int] = None
|
||||||
|
smtp_user: Optional[str] = None
|
||||||
|
smtp_password: Optional[str] = None
|
||||||
|
openrouter_api_key: Optional[str] = None
|
||||||
|
openrouter_management_key: Optional[str] = None
|
||||||
|
groq_api_key: Optional[str] = None
|
||||||
|
gemini_api_key: Optional[str] = None
|
||||||
|
deepseek_api_key: Optional[str] = None
|
||||||
|
mistral_api_key: Optional[str] = None
|
||||||
|
cloudflare_account_id: Optional[str] = None
|
||||||
|
cloudflare_api_token: Optional[str] = None
|
||||||
|
huggingface_api_key: Optional[str] = None
|
||||||
|
|
||||||
|
class SystemSettingsUpdate(SystemSettingsBase): pass
|
||||||
|
|
||||||
|
class SystemSettingsResponse(SystemSettingsBase):
|
||||||
|
class Config: from_attributes = True
|
||||||
|
|
||||||
class LoginRequest(BaseModel):
|
class LoginRequest(BaseModel):
|
||||||
username: str
|
username: str
|
||||||
password: str
|
password: str
|
||||||
totp_code: str
|
totp_code: str
|
||||||
|
|
||||||
# --- NOUVEAU : Schémas pour les Projets ---
|
|
||||||
class ProjectBase(BaseModel):
|
class ProjectBase(BaseModel):
|
||||||
title: str
|
title: str
|
||||||
|
|
||||||
class ProjectCreate(ProjectBase):
|
class ProjectCreate(ProjectBase): pass
|
||||||
pass
|
|
||||||
|
|
||||||
class ProjectResponse(ProjectBase):
|
class ProjectResponse(ProjectBase):
|
||||||
id: int
|
id: int
|
||||||
created_at: datetime
|
created_at: datetime
|
||||||
is_pinned: bool
|
is_pinned: bool
|
||||||
|
class Config: from_attributes = True
|
||||||
|
|
||||||
class Config:
|
class ProjectRename(BaseModel):
|
||||||
from_attributes = True
|
title: str
|
||||||
|
|
||||||
|
class OrchestratorConfig(BaseModel):
|
||||||
|
workers: List[str]
|
||||||
|
prompter: Optional[str] = "gemini-3.5-flash-lite"
|
||||||
|
concatenator: Optional[str] = "gemini-3.5-flash-lite"
|
||||||
|
|
||||||
|
class AttachedFile(BaseModel):
|
||||||
|
name: str
|
||||||
|
content: str
|
||||||
|
|
||||||
# --- NOUVEAU : Schémas pour les Messages ---
|
|
||||||
class MessageBase(BaseModel):
|
class MessageBase(BaseModel):
|
||||||
role: str
|
role: str
|
||||||
content: str
|
content: str
|
||||||
|
|
||||||
class MessageCreate(MessageBase):
|
class MessageCreate(MessageBase):
|
||||||
pass
|
config: Optional[OrchestratorConfig] = None
|
||||||
|
files: Optional[List[AttachedFile]] = None
|
||||||
|
|
||||||
class MessageResponse(MessageBase):
|
class MessageResponse(MessageBase):
|
||||||
id: int
|
id: int
|
||||||
created_at: datetime
|
created_at: datetime
|
||||||
project_id: int
|
project_id: int
|
||||||
|
class Config: from_attributes = True
|
||||||
|
|
||||||
class ProjectRename(BaseModel):
|
class PasswordChange(BaseModel):
|
||||||
title: str
|
old_password: str
|
||||||
|
new_password: str
|
||||||
|
|
||||||
|
class ModelReplacementRequest(BaseModel):
|
||||||
|
missing_model: str
|
||||||
|
|
||||||
|
class LogRequest(BaseModel):
|
||||||
|
message: str
|
||||||
+977
-179
File diff suppressed because it is too large.
Load diff
@@ -11,3 +11,10 @@ python-multipart
|
|||||||
bcrypt
|
bcrypt
|
||||||
email-validator
|
email-validator
|
||||||
PyJWT
|
PyJWT
|
||||||
|
openai
|
||||||
|
httpx
|
||||||
|
python-multipart
|
||||||
|
pytz
|
||||||
|
PyPDF2
|
||||||
|
openpyxl
|
||||||
|
oletools
|
||||||
Reference in new issue
Block a user