test fix

Author: ishaan-jaff

tests/test_litellm/test_utils.py

@@ -546,6 +546,8 @@ def test_aaamodel_prices_and_context_window_json_is_valid():
                 },
                 "input_cost_per_video_per_second_above_128k_tokens": {"type": "number"},
                 "input_dbu_cost_per_token": {"type": "number"},
+                "annotation_cost_per_page": {"type": "number"},
+                "ocr_cost_per_page": {"type": "number"},
                 "litellm_provider": {"type": "string"},
                 "max_audio_length_hours": {"type": "number"},
                 "max_audio_per_prompt": {"type": "number"},
@@ -573,6 +575,7 @@ def test_aaamodel_prices_and_context_window_json_is_valid():
                         "moderation",
                         "rerank",
                         "responses",
+                        "ocr",
                     ],
                 },
                 "output_cost_per_audio_token": {"type": "number"},
@@ -629,6 +632,7 @@ def test_aaamodel_prices_and_context_window_json_is_valid():
                             "/v1/batch",
                             "/v1/audio/transcriptions",
                             "/v1/audio/speech",
+                            "/v1/ocr",
                         ],
                     },
                 },