apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-07-05 15:52:12 +02:00
Code Issues Projects Releases Packages Wiki Activity

refactored changes to support enum dispatch

Browse source
This commit is contained in:
Salman Paracha 2025-08-19 14:53:30 -07:00
parent 7253a0f203
commit 327b29ec6f
10 changed files with 526 additions and 478 deletions
Download patch file Download diff file Expand all files Collapse all files

15
crates/hermesllm/src/lib.rs
View file

@ -6,13 +6,10 @@ pub mod apis;
pub mod clients;
// Re-export important types and traits
pub use providers::{
ProviderId, ConversionMode,
ProviderRequest, ProviderResponse, ProviderStreamResponse, ProviderStreamResponseIter,
TokenUsage,
try_request_from_bytes, try_response_from_bytes, try_streaming_from_bytes,
has_compatible_api, supported_apis
};
pub use providers::request::{ProviderRequestType, ProviderRequest, ProviderRequestError};
pub use providers::response::{ProviderResponseType, ProviderResponse, ProviderStreamResponse, ProviderResponseError, TokenUsage, try_streaming_from_bytes};
pub use providers::id::ProviderId;
pub use providers::adapters::{has_compatible_api, supported_apis};
#[cfg(test)]
mod tests {
@ -58,7 +55,7 @@ mod tests {
]
}"#;
let result = try_request_from_bytes(json_request.as_bytes(), &ProviderId::OpenAI);
let result: Result<ProviderRequestType, std::io::Error> = ProviderRequestType::try_from(json_request.as_bytes());
assert!(result.is_ok());
let request = result.unwrap();
@ -74,7 +71,7 @@ mod tests {
data: [DONE]
"#;
let result = try_streaming_from_bytes(sse_data.as_bytes(), &ProviderId::OpenAI, ConversionMode::Passthrough);
let result = try_streaming_from_bytes(sse_data.as_bytes(), &ProviderId::OpenAI);
assert!(result.is_ok());
let mut streaming_response = result.unwrap();