style and ton optionnal

main.py

@@ -69,8 +69,8 @@ app = FastAPI(dependencies=[Depends(api_key_auth)])
 #     return response
 
 class StyleWriter(BaseModel):
-    style: str
-    tonality: str
+    style: Optional[str] = "neutral"
+    tonality: Optional[str] = "formal"
 
 models = ["gpt-4o","gpt-4o-mini","mistral-large-latest"]
 
@@ -212,8 +212,20 @@ def generate_answer(user_input: UserInput):
             prompt_formated = prompt_reformatting(template_prompt,context,prompt,enterprise_name=getattr(user_input,"marque",""))
             answer = generate_response_via_langchain(prompt, model=getattr(user_input,"model","gpt-4o"),stream=user_input.stream,context = context , messages=user_input.messages,template=template_prompt,enterprise_name=getattr(user_input,"marque",""))
         else:
-            prompt_formated = prompt_reformatting(template_prompt,context,prompt,style=user_input.style_tonality.style,tonality=user_input.style_tonality.tonality,enterprise_name=getattr(user_input,"marque",""))
-            answer = generate_response_via_langchain(prompt,model=getattr(user_input,"model","gpt-4o"),stream=user_input.stream,context = context , messages=user_input.messages,style=user_input.style_tonality.style,tonality=user_input.style_tonality.tonality,template=template_prompt,enterprise_name=getattr(user_input,"marque",""))
+            prompt_formated = prompt_reformatting(template_prompt,
+                                                  context,
+                                                  prompt,
+                                                  style=getattr(user_input.style_tonality,"style","neutral"),
+                                                  tonality=getattr(user_input.style_tonality,"tonality","formal"),
+                                                  enterprise_name=getattr(user_input,"marque",""))
+            
+            answer = generate_response_via_langchain(prompt,model=getattr(user_input,"model","gpt-4o"),
+                                                     stream=user_input.stream,context = context , 
+                                                     messages=user_input.messages,
+                                                     style=getattr(user_input.style_tonality,"style","neutral"),
+                                                     tonality=getattr(user_input.style_tonality,"tonality","formal"),
+                                                     template=template_prompt,
+                                                     enterprise_name=getattr(user_input,"marque",""))
         
         if user_input.stream:
             return StreamingResponse(stream_generator(answer,prompt_formated), media_type="application/json")