diff --git a/litellm/integrations/prometheus.py b/litellm/integrations/prometheus.py index bb9d6166b01..2385edc5297 100644 --- a/litellm/integrations/prometheus.py +++ b/litellm/integrations/prometheus.py @@ -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 {}), diff --git a/litellm/proxy/auth/auth_checks.py b/litellm/proxy/auth/auth_checks.py index 843da0e3a0d..a741869e5fc 100644 --- a/litellm/proxy/auth/auth_checks.py +++ b/litellm/proxy/auth/auth_checks.py @@ -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} diff --git a/tests/llm_responses_api_testing/test_manus_responses_api.py b/tests/llm_responses_api_testing/test_manus_responses_api.py index 6256363f407..6a2aed3812d 100644 --- a/tests/llm_responses_api_testing/test_manus_responses_api.py +++ b/tests/llm_responses_api_testing/test_manus_responses_api.py @@ -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,