style and ton

main.py

@@ -146,7 +146,10 @@ def generate_answer(user_input: UserInput):
         if not context:
             context = "No context found"
 
-        answer = generate_response_via_langchain(prompt, 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)
+        if user_input.style_tonality is None:
+            answer = generate_response_via_langchain(prompt, model="gpt-4o",stream=user_input.stream,context = context , messages=user_input.messages)
+        else:
+            answer = generate_response_via_langchain(prompt, 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)
         
         if user_input.stream:
             return StreamingResponse(answer, media_type="application/json")