updated

app.py

@@ -107,7 +107,9 @@ async def generate_story(story_request: StoryRequest):
     # Extract the raw string from the `ChatPromptValue`
     # formatted_text = formatted_prompt.to_string()
 
-    response = llm(formatted_prompt)
+    chain = prompt | llm
+    response = chain.invoke(formatted_prompt)
+    # response = llm(formatted_prompt)
     
     # final_prompt = prompt_template.format(text=story)