Upload main.py

main.py

@@ -19,6 +19,7 @@ create_table()
 class Request(BaseModel):
     enonce : str
     code : str
+    res_test : str
 
 app = FastAPI()
 
@@ -45,14 +46,15 @@ Tu dois t'adresser directement à l'élève.
 L'élève ne peut pas te poser des questions, il peut juste te proposer son code.
 Tu ne dois pas proposer à l'élève de te poser des questions
 Il est inutile de proposer à l'élève de tester son code avec les exemples proposés.
-Tu ne dois pas proposer aux élèves des modifications du programme qui sorte du cadre de l'exercice. Par exemple, pour l'exercice qui demande d'écrire une fonction moyenne, si dans l'énoncé il est précisé que l'on a un tableau non vide d'entier en paramètre, il est inutile de dire à l'élève que son programme doit gérer les tableaux vides. 
+Tu ne dois pas proposer aux élèves des modifications du programme qui sorte du cadre de l'exercice. Par exemple, pour l'exercice qui demande d'écrire une fonction moyenne, si dans l'énoncé il est précisé que l'on a un tableau non vide d'entier en paramètre, il est inutile de dire à l'élève que son programme doit gérer les tableaux vides.
+Des tests unitaires ont été prévus afin d'évaluer le code proposé par l'élève. Si les tests sont tous positifs, cela signifie que le code proposé par l'élève est correct (tu dois alors lui dire que son code est correct et éventuellement lui donner des conseils pour encore améliorer la qualité de son code). Si les tests unitaires sont négatifs, tu dois aider l'élève à trouver ces erreurs.  
 Tu dois t'exprimer en français
 """
 
 prompt = ChatPromptTemplate.from_messages(
     [
         ("system", system),
-        ("human", "Voici l'exercice proposé à l'élève: \n {enonce} \n\n Voici le programme proposé par l'élève : {code}"),
+        ("human", "Voici l'exercice proposé à l'élève: \n {enonce} \n Voici le programme proposé par l'élève : {code} \n Voici le résultat des tests unitaires : {test_unit}"),
     ]
 )
 
@@ -61,10 +63,18 @@ chain = prompt | llm | StrOutputParser()
 class GraphState(TypedDict):
     enonce : str
     messages: Annotated[list[AnyMessage], add_messages]
+    res_test : str
     
 
 def chatbot(state : GraphState):
-    response = chain.invoke({'enonce': state['enonce'].content, 'code' : state['messages'][-1].content})
+    msg_test = ""
+    if state['res_test'] == '0' :
+        msg_test = "Le code de l'élève n'a pas encore été testé avec les tests unitaires"
+    elif state['res_test'] == '1':
+        msg_test = "Le code de l'élève a bien été testé avec les tests unitaires, il a échoué à au moins un test unitaire"
+    elif state['res_test'] == '2':
+        msg_test = "Le code de l'élève a bien été testé avec les tests unitaires, il a réussi tous les tests unitaires"
+    response = chain.invoke({'enonce': state['enonce'].content, 'code' : state['messages'][-1].content, 'test_unit' : msg_test} )
     return {"messages": [AIMessage(content=response)]}
 
 workflow = StateGraph(GraphState)
@@ -77,7 +87,7 @@ app_chatbot = workflow.compile()
 
 @app.post('/request')
 def request(req: Request):
-    rep = app_chatbot.invoke({"enonce" : HumanMessage(content=req.enonce),"messages": [HumanMessage(content=req.code)]}, stream_mode="values")
+    rep = app_chatbot.invoke({"enonce" : HumanMessage(content=req.enonce),"messages": [HumanMessage(content=req.code)], "res_test" : [HumanMessage(content=req.res_test)]}, stream_mode="values")
     return {"response":rep['messages'][-1].content}