From 3de23e7f187de8e2ff4b12749371b9167b39ce4b Mon Sep 17 00:00:00 2001
From: Yujong Lee <yujong@berri.ai>
Date: Wed, 16 Sep 2026 23:10:10 +0000
Subject: [PATCH] test(rust_bridge): drop generated OCR route assertions from
 bridge_route tests

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
---
 .../python-bridge/src/routes/definition.rs    | 51 ++++++++-----------
 1 file changed, 21 insertions(+), 30 deletions(-)

diff --git a/litellm-rust/crates/python-bridge/src/routes/definition.rs b/litellm-rust/crates/python-bridge/src/routes/definition.rs
index 4c8d98ebe62..f846c7ea1f9 100644
--- a/litellm-rust/crates/python-bridge/src/routes/definition.rs
+++ b/litellm-rust/crates/python-bridge/src/routes/definition.rs
@@ -157,11 +157,6 @@ mod tests {
             let module = PyModule::new(py, "routes").expect("module should be created");
             crate::routes::register(&module).expect("routes should register");
             let routes = [
-                (
-                    "ocr",
-                    "aocr",
-                    "(model, document, api_key=None, api_base=None, custom_llm_provider=None, extra_headers=None, optional_params=None, input_sources=None, timeout_seconds=None)",
-                ),
                 (
                     "transcription",
                     "atranscription",
@@ -244,24 +239,22 @@ mod tests {
             kwargs
                 .set_item("extra_headers", &invalid_headers)
                 .expect("kwargs should accept extra_headers");
-            let document = PyDict::new(py);
+            let audio = PyDict::new(py);
 
-            for (sync_name, async_name) in [("ocr", "aocr"), ("transcription", "atranscription")] {
-                let sync_error = module
-                    .getattr(sync_name)
-                    .and_then(|function| function.call(("model", &document), Some(&kwargs)))
-                    .expect_err("sync route should reject non-dict extra_headers");
-                let async_error = module
-                    .getattr(async_name)
-                    .and_then(|function| function.call(("model", &document), Some(&kwargs)))
-                    .expect_err("async route should reject non-dict extra_headers");
+            let sync_error = module
+                .getattr("transcription")
+                .and_then(|function| function.call(("model", &audio), Some(&kwargs)))
+                .expect_err("sync route should reject non-dict extra_headers");
+            let async_error = module
+                .getattr("atranscription")
+                .and_then(|function| function.call(("model", &audio), Some(&kwargs)))
+                .expect_err("async route should reject non-dict extra_headers");
 
-                assert_eq!(
-                    sync_error.to_string(),
-                    "ValueError: extra_headers must be a dict"
-                );
-                assert_eq!(async_error.to_string(), sync_error.to_string());
-            }
+            assert_eq!(
+                sync_error.to_string(),
+                "ValueError: extra_headers must be a dict"
+            );
+            assert_eq!(async_error.to_string(), sync_error.to_string());
         });
     }
 
@@ -312,15 +305,13 @@ mod tests {
 
             let invalid_payload =
                 PyModule::new(py, "invalid_payload").expect("invalid payload should be created");
-            for name in ["ocr", "transcription"] {
-                let error = module
-                    .getattr(name)
-                    .and_then(|function| {
-                        function.call(("model", &invalid_payload), Some(&headers_kwargs))
-                    })
-                    .expect_err("payload should be validated before headers");
-                assert!(!error.to_string().contains("extra_headers"));
-            }
+            let error = module
+                .getattr("transcription")
+                .and_then(|function| {
+                    function.call(("model", &invalid_payload), Some(&headers_kwargs))
+                })
+                .expect_err("payload should be validated before headers");
+            assert!(!error.to_string().contains("extra_headers"));
         });
     }