Skip to content
Extraits de code Groupes Projets
Valider 05603caa rédigé par DavePk04's avatar DavePk04
Parcourir les fichiers

Black: fix linting

parent 976f2096
Aucune branche associée trouvée
Aucune étiquette associée trouvée
Aucune requête de fusion associée trouvée
......@@ -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:
......
0% Chargement en cours ou .
You are about to add 0 people to the discussion. Proceed with caution.
Terminez d'abord l'édition de ce message.
Veuillez vous inscrire ou vous pour commenter