Update app.py

app.py

@@ -25,7 +25,12 @@ def generate_response(prompt):
     response = client.chat.completions.create(
         model="o1-preview",
         messages=st.session_state.messages + [{"role": "user", "content": prompt}],
-        max_completion_tokens=1000,
+        total_tokens: 1000,
+        prompt_tokens: 400,
+        completion_tokens: 600,
+        completion_tokens_details: {
+            reasoning_tokens: 500
+        }
     )
     return response.choices[0].message.content