Update main.py

main.py

@@ -77,7 +77,11 @@ class ZaiClient:
 
     # ── chat CRUD ───────────────────────────────────────────────────
 
-    async def create_chat(self, user_message: str, model: str = DEFAULT_MODEL) -> dict:
+    async def create_chat(
+        self,
+        user_message: str,
+        model: str = DEFAULT_MODEL,
+    ) -> dict:
         """POST /api/v1/chats/new — creates a new chat session."""
         msg_id = str(uuid.uuid4())
         ts = int(time.time())
@@ -353,6 +357,7 @@ async def main() -> None:
         # 3. Create chat
         user_message = "Hello"
         print(f"\n[3] Creating chat with first message: {user_message!r}")
+        messages = [{"role": "user", "content": user_message}]
         chat = await client.create_chat(user_message)
         chat_id = chat["id"]
         print(f"    chat_id : {chat_id}")