scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-08-28 05:25:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

test_manus_responses_api_with_file_upload

Browse source
This commit is contained in:
Ishaan Jaffer 2026-01-10 15:12:00 -08:00
parent 58bc308d47
commit 1ebe979eee
3 changed files with 3 additions and 24 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/integrations/prometheus.py
View file

@ -872,9 +872,6 @@ class PrometheusLogger(CustomLogger):
"metadata"
].get("user_api_key_auth_metadata")
# Include top-level metadata fields (excluding nested dictionaries)
# This allows accessing fields like requester_ip_address from top-level metadata
top_level_metadata = standard_logging_payload.get("metadata", {})
combined_metadata: Dict[str, Any] = {
**(_requester_metadata if _requester_metadata else {}),
**(user_api_key_auth_metadata if user_api_key_auth_metadata else {}),

2
litellm/proxy/auth/auth_checks.py
View file

@ -1830,7 +1830,7 @@ async def get_org_object(
return cached_org_obj
# else, check db
try:
query_kwargs = {"where": {"organization_id": org_id}}
query_kwargs: Dict[str, Any] = {"where": {"organization_id": org_id}}
if include_budget_table:
query_kwargs["include"] = {"litellm_budget_table": True}

22
tests/llm_responses_api_testing/test_manus_responses_api.py
View file

@ -47,10 +47,7 @@ async def test_manus_responses_api_with_agent_profile():
if got_response.status == "completed":
assert got_response.output is not None
assert len(got_response.output) > 0
else:
# Manus can return "running" or "pending" status
assert got_response.status in ["running", "pending"]
assert got_response.id is not None
@pytest.mark.asyncio
@ -113,24 +110,9 @@ async def test_manus_responses_api_with_file_upload():
response_id = getattr(response, "id", None)
assert response_id is not None, f"Response ID is None. Response: {response}"
# Step 3: Get the response status
print(f"\nStep 3: Getting response status...")
got_response = await litellm.aget_responses(
response_id=response_id,
custom_llm_provider="manus",
api_key=api_key,
)
print(f"Response status: {got_response}")
got_response_id = getattr(got_response, "id", None)
got_response_status = getattr(got_response, "status", None)
assert got_response_id == response_id
assert got_response_status in ["completed", "running", "pending"]
# Step 4: Clean up - delete the file
# Step 3: Clean up - delete the file
print(f"\nStep 4: Cleaning up - deleting file {file_id}...")
deleted_file = await litellm.afile_delete(
file_id=file_id,