ajout d'un prompt de base

app.py

@@ -16,12 +16,20 @@ class PromptRequest(BaseModel):
     prompt: str
     model: str = "gemini/gemini-1.5-pro"  # Valeur par défaut
 
+# Prompt de base
+base_prompt = """
+Tu es un développeur logiciel expérimenté. Tu dois toujours répondre en français d'une manière réfléchie et concise. 
+"""
+
 @app.post("/")
 async def predict(request: PromptRequest):
     try:
+        # Ajout du prompt de base à la requête de l'utilisateur
+        full_prompt = base_prompt + request.prompt 
+
         response = litellm.completion(
             model=request.model, 
-            messages=[{"role": "user", "content": request.prompt}]
+            messages=[{"role": "user", "content": full_prompt}]
         )
         return {"response": response.choices[0].message.content, "model": request.model}
     except Exception as e: