mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-22 00:31:44 +00:00
904c679595
litellm-types mirrors litellm/types (chat, Anthropic Messages and Responses websocket data) and litellm-core-utils mirrors litellm/litellm_core_utils (provider resolution, prompt factory, core helpers, call arguments). Route request types move up to their core route module, the transform contracts move into base_llm, and the three duplicated provider error enums become one Error in base_llm/chat/transformation.rs, mirroring BaseLLMException. The empty-text placeholder goes back to the value Python's factory.py uses; the provider extraction had changed it to a single space. Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
51 lines
1.3 KiB
Rust
51 lines
1.3 KiB
Rust
use std::{
|
|
fs,
|
|
path::{Path, PathBuf},
|
|
};
|
|
|
|
const DISALLOWED_OUTSIDE_INTEROP: &[&str] = &[
|
|
"py.import(\"json\")",
|
|
"pythonize::",
|
|
"serde_json::to_string",
|
|
"serde_json::from_str",
|
|
];
|
|
|
|
fn source_root() -> PathBuf {
|
|
Path::new(env!("CARGO_MANIFEST_DIR")).join("src")
|
|
}
|
|
|
|
fn rust_sources(directory: &Path) -> Vec<PathBuf> {
|
|
fs::read_dir(directory)
|
|
.expect("bridge source directory should be readable")
|
|
.map(|entry| {
|
|
entry
|
|
.expect("bridge source entry should be readable")
|
|
.path()
|
|
})
|
|
.flat_map(|path| {
|
|
if path.is_dir() {
|
|
rust_sources(&path)
|
|
} else if path.extension().is_some_and(|extension| extension == "rs") {
|
|
vec![path]
|
|
} else {
|
|
Vec::new()
|
|
}
|
|
})
|
|
.collect()
|
|
}
|
|
|
|
#[test]
|
|
fn serialization_uses_the_interop_boundary() {
|
|
let root = source_root();
|
|
|
|
for path in rust_sources(&root) {
|
|
let source = fs::read_to_string(&path).expect("bridge source should be readable");
|
|
for disallowed in DISALLOWED_OUTSIDE_INTEROP {
|
|
assert!(
|
|
!source.contains(disallowed),
|
|
"{} bypasses litellm-host-python with `{disallowed}`",
|
|
path.display()
|
|
);
|
|
}
|
|
}
|
|
}
|