whatif

main.py

@@ -137,8 +137,8 @@ async def generate(user_input: UserInput):
 @app.post("/whatif")
 async def generate_whatif(whatif_input: WhatifInput):
     try:
-        print(whatif_input)
-        return generate_whatif_stream(whatif_input["question"],whatif_input["response"])
+        print(f"whatif_input: {whatif_input}")
+        return generate_whatif_stream(whatif_input["question"],whatif_input["answer"])
     except Exception as e:
         return {"message": str(e)}
     

rag.py

@@ -174,7 +174,7 @@ def generate_stream(query:str,messages = [], model = "gpt-4o-mini", max_tokens =
         print(e)
         return False
     
-def generate_whatif_stream(question:str,response:str, stream:bool = True) -> str:
+def generate_whatif_stream(question:str,response:str, stream:bool = False) -> str:
     try:
         prompt = PromptTemplate.from_template(template_whatif)
         llm_chain = prompt | llm | StrOutputParser()