diff --git a/claude_mirror/server.py b/claude_mirror/server.py
index 16b5bc47..88d6b853 100644
--- a/claude_mirror/server.py
+++ b/claude_mirror/server.py
@@ -147,7 +147,9 @@ class Tool(BaseModel):
     input_schema: Dict[str, Any]
 
 class ThinkingConfig(BaseModel):
-    enabled: bool
+    enabled: Optional[bool] = None
+    budget_tokens: Optional[int] = None
+    type: Optional[str] = None
 
 class MessagesRequest(BaseModel):
     model: str
@@ -1110,4 +1112,4 @@ async def root():
     sys.stdout.flush()
     
     # Configure uvicorn to run with minimal logs
-    uvicorn.run(app, host="0.0.0.0", port=8082, log_level="error")
\ No newline at end of file
+    uvicorn.run(app, host="0.0.0.0", port=8082, log_level="error")