Upload main.py

main.py

@@ -13,9 +13,13 @@ from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 from database import create_table, return_title, return_exercise
+import logging
 
 create_table()
 
+logger = logging.getLogger('uvicorn.error')
+logger.setLevel(logging.DEBUG)
+
 class Request(BaseModel):
     enonce : str
     code : str
@@ -35,7 +39,7 @@ load_dotenv()
 os.environ["GROQ_API_KEY"] = os.getenv('GROQ_API_KEY')
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
-llm = ChatGroq(model="llama-3.1-8b-instant", temperature=0.5)
+llm = ChatGroq(model="llama-3.1-70b-versatile", temperature=0.5)
 #llm = ChatGroq(model="mixtral-8x7b-32768", temperature=0.5)
 exercise = ""
 system = """
@@ -68,11 +72,12 @@ class GraphState(TypedDict):
 
 def chatbot(state : GraphState):
     msg_test = ""
-    if state['res_test'] == '0' :
+    res_test =state['res_test'].content
+    if 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':
+    elif 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':
+    elif 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)]}
@@ -87,7 +92,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)], "res_test" : [HumanMessage(content=req.res_test)]}, 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}