Skip to content
GitLab
Explorer
Connexion
S'inscrire
Navigation principale
Rechercher ou aller à…
Projet
LanguageLab
Gestion
Activité
Membres
Labels
Programmation
Tickets
Tableaux des tickets
Jalons
Wiki
Code
Requêtes de fusion
Dépôt
Branches
Validations
Étiquettes
Graphe du dépôt
Comparer les révisions
Extraits de code
Déploiement
Releases
Registre de paquets
Registre de conteneur
Registre de modèles
Opération
Modules Terraform
Surveillance
Incidents
Analyse
Données d'analyse des chaînes de valeur
Analyse des contributeurs
Données d'analyse du dépôt
Expériences du modèle
Aide
Aide
Support
Documentation de GitLab
Comparer les forfaits GitLab
Forum de la communauté
Contribuer à GitLab
Donner votre avis
Conditions générales et politique de confidentialité
Raccourcis clavier
?
Extraits de code
Groupes
Projets
Afficher davantage de fils d'Ariane
Serge Bibauw
LanguageLab
Validations
05603caa
Valider
05603caa
rédigé
Il y a 2 mois
par
DavePk04
Parcourir les fichiers
Options
Téléchargements
Correctifs
Plain Diff
Black: fix linting
parent
976f2096
Aucune branche associée trouvée
Branches contenant la validation
Aucune étiquette associée trouvée
Aucune requête de fusion associée trouvée
Modifications
1
Masquer les modifications d'espaces
En ligne
Côte à côte
Affichage de
1 fichier modifié
backend/app/routes/chat.py
+5
-16
5 ajouts, 16 suppressions
backend/app/routes/chat.py
avec
5 ajouts
et
16 suppressions
backend/app/routes/chat.py
+
5
−
16
Voir le fichier @
05603caa
...
...
@@ -4,49 +4,38 @@ import requests
import
os
from
dotenv
import
load_dotenv
# Load API key from .env
load_dotenv
()
API_KEY
=
os
.
getenv
(
"
OPENROUTER_API_KEY
"
)
# Mistral API details
MODEL_NAME
=
"
mistralai/mistral-small-24b-instruct-2501:free
"
API_URL
=
"
https://openrouter.ai/api/v1/chat/completions
"
# Initialize FastAPI router
chatRouter
=
APIRouter
(
prefix
=
"
/chat
"
,
tags
=
[
"
chat
"
])
# Message schema
class
ChatMessage
(
BaseModel
):
session_id
:
str
role
:
str
content
:
str
# Temporary chat history storage
chat_sessions
=
{}
@chatRouter.post
(
"
/
"
)
async
def
chat_with_ai
(
message
:
ChatMessage
):
session_id
=
message
.
session_id
user_message
=
{
"
role
"
:
message
.
role
,
"
content
"
:
message
.
content
}
# Store chat history per session
if
session_id
not
in
chat_sessions
:
chat_sessions
[
session_id
]
=
[]
chat_sessions
[
session_id
].
append
(
user_message
)
# Prepare request
headers
=
{
"
Authorization
"
:
f
"
Bearer
{
API_KEY
}
"
,
"
Content-Type
"
:
"
application/json
"
}
headers
=
{
"
Authorization
"
:
f
"
Bearer
{
API_KEY
}
"
,
"
Content-Type
"
:
"
application/json
"
}
data
=
{
"
model
"
:
MODEL_NAME
,
"
messages
"
:
chat_sessions
[
session_id
]
}
data
=
{
"
model
"
:
MODEL_NAME
,
"
messages
"
:
chat_sessions
[
session_id
]}
# Call Mistral AI
response
=
requests
.
post
(
API_URL
,
headers
=
headers
,
json
=
data
)
if
response
.
status_code
==
200
:
...
...
Ce diff est replié.
Cliquez pour l'agrandir.
Aperçu
0%
Chargement en cours
Veuillez réessayer
ou
joindre un nouveau fichier
.
Annuler
You are about to add
0
people
to the discussion. Proceed with caution.
Terminez d'abord l'édition de ce message.
Enregistrer le commentaire
Annuler
Veuillez vous
inscrire
ou vous
se connecter
pour commenter