scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-11 03:38:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 83

feat(lens): send in-flight runs with worker progress

Browse source
This commit is contained in:
Ishaan Jaff 2026-10-03 18:01:39 -07:00
parent 1d18be780e
commit e4387d8ab2
No known key found for this signature in database
1 changed files with 20 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
litellm/proxy/lens/worker.py
View file

@ -17,7 +17,18 @@ from litellm.constants import (
)
from .analysis import AnalysisResponseError, analyze_sample, validation_details
from .models import Claim, Coverage, ExecutionContent, ModelRequest, ModelResult, Progress, Result, Review, Sample
from .models import (
Claim,
Coverage,
ExecutionContent,
InFlight,
ModelRequest,
ModelResult,
Progress,
Result,
Review,
Sample,
)
logger: Final = logging.getLogger("litellm.lens.worker")
@ -190,10 +201,16 @@ class LensWorker:
result.raise_for_status()
return ExecutionContent.model_validate(result.json())
async def progress(stage: str, coverage: Coverage, review: Review | None = None, /) -> None:
async def progress(
stage: str,
coverage: Coverage,
review: Review | None = None,
reading: tuple[InFlight, ...] | None = None,
/,
) -> None:
result: Final = await self.client.post(
prefix + "/progress",
json=Progress(stage=stage, coverage=coverage, review=review).model_dump(mode="json"),
json=Progress(stage=stage, coverage=coverage, review=review, reading=reading).model_dump(mode="json"),
)
result.raise_for_status()