Update app.py

app.py

@@ -307,7 +307,7 @@ def rewrite(state: AgentState):
             timeout=30
         )
         response.raise_for_status()
-        return {"messages": [AIMessage(content=response.json()['choices'][0]['message']['content'])}
+        return {"messages": [AIMessage(content=response.json()['choices'][0]['message']['content'])]}
     except Exception as e:
         return {"messages": [AIMessage(content=f"Rewrite Error: {str(e)}")]}
 
@@ -482,4 +482,4 @@ def main():
         """)
 
 if __name__ == "__main__":
-    main()
+    main()