Update proxy_server.py

proxy_server.py

@@ -270,29 +270,17 @@ async def stream_openai_response_to_claude_events(openai_response: httpx.Respons
                 'stop_reason': claude_stop_reason,
                 'stop_sequence': None
             }
-            # Removed usage from here
         }
         yield f"event: message_delta\ndata: {json.dumps(final_delta)}\n\n"
 
         # 7. Send message_stop (including final usage)
-        # --- FIX: Add usage info to the message_stop event ---
+        # --- FIX: Use simpler 'usage' structure in message_stop ---
         final_stop_event_data = {
             'type': 'message_stop',
-            # Claude API v1 examples sometimes show a nested 'message' object here,
-            # but let's try putting usage directly under the event first for simplicity,
-            # similar to how message_start includes a 'message' object.
-            # If this doesn't work, we might need the nested 'message' structure.
-             'amazon-bedrock-invocationMetrics': { # Mimic Bedrock's potential structure for usage
-                 'inputTokenCount': input_tokens,
-                 'outputTokenCount': output_tokens if output_tokens > 0 else (accumulated_content_len // 4), # Use estimate if needed
-                 'invocationLatency': 0, # Placeholder
-                 'firstByteLatency': 0 # Placeholder
-             }
-            # Alternative simpler structure (if the above fails):
-            # 'usage': {
-            #      'input_tokens': input_tokens,
-            #      'output_tokens': output_tokens if output_tokens > 0 else (accumulated_content_len // 4)
-            # }
+            'usage': {
+                 'input_tokens': input_tokens,
+                 'output_tokens': output_tokens if output_tokens > 0 else (accumulated_content_len // 4) # Use estimate if needed
+            }
         }
         yield f"event: message_stop\ndata: {json.dumps(final_stop_event_data)}\n\n"
         # --- End Fix ---
@@ -436,4 +424,3 @@ if __name__ == "__main__":
     log_config_level = log_level.lower() if log_level in ["CRITICAL", "ERROR", "WARNING", "INFO", "DEBUG", "TRACE"] else "info"
     logger.info(f"Starting Uvicorn server on {host}:{port}")
     uvicorn.run("proxy_server:app", host=host, port=port, reload=True, log_level=log_config_level)
-