scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-30 01:52:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 70

test(rust-bridge): give the logger test machine its own route type

Browse source 
Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
This commit is contained in:
Yujong Lee 2026-09-24 12:37:34 -07:00
parent 2f421ac770
commit 4cfcbc704a
1 changed files with 11 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
litellm-rust/crates/python-bridge/src/logger/tests.rs
View file

@ -1,4 +1,4 @@
use std::{process::Command, task::Poll};
use std::{convert::Infallible, process::Command, task::Poll};
use litellm_host::{
host::HostResult,
@ -8,22 +8,24 @@ use litellm_host::{
use pyo3::{prelude::*, types::PyDict};
struct DiagnosticMachine;
struct Silent;
impl Route for DiagnosticMachine {
impl Route for Silent {
type Response = ();
type Error = String;
type Op = ();
type OpResult = ();
type Chunk = ();
type StreamHead = ();
type Op = Infallible;
type OpResult = Infallible;
type Chunk = Infallible;
type StreamHead = Infallible;
}
struct DiagnosticMachine;
impl Machine for DiagnosticMachine {
type Route = Self;
type Route = Silent;
type Complete = ();
fn resume(&mut self, _: Option<HostResult<Self>>) -> Step<'_, Self> {
fn resume(&mut self, _: Option<HostResult<Silent>>) -> Step<'_, Self> {
litellm_tracing::warn!("machine started");
Box::pin(async {
tokio::task::yield_now().await;