mirror of
https://github.com/tiennm99/litellm.git
synced 2026-06-28 13:11:20 +00:00
Ishaan Jaff
3631e8fa1d
* test_router_acreate_container_without_model * _init_containers_api_endpoints * test_init_containers_api_endpoints * init container files endpoints * init files api * init container files API * add containers api file content * add code interpreter output UI * add code interpreter input ui * refactor code interpreter ui * fix: require model selection * cleaner container provision * fix ContainerFileObject * def container_file_content_handler( add * add retrieve_container_file_content * aretrieve_container_file_content * UI fix model * fix linting errors
120 lines
3.7 KiB
Python
120 lines
3.7 KiB
Python
"""
|
|
E2E Test for Container Files API.
|
|
|
|
Tests the container files endpoints using LiteLLM SDK methods.
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import time
|
|
|
|
import pytest
|
|
|
|
sys.path.insert(0, os.path.abspath("../.."))
|
|
|
|
from litellm.containers import (
|
|
create_container,
|
|
delete_container,
|
|
)
|
|
from litellm.containers.endpoint_factory import (
|
|
list_container_files,
|
|
retrieve_container_file,
|
|
retrieve_container_file_content,
|
|
delete_container_file,
|
|
)
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
not os.getenv("OPENAI_API_KEY"),
|
|
reason="OPENAI_API_KEY not set"
|
|
)
|
|
def test_container_files_api():
|
|
"""
|
|
Test container files API: list, retrieve, delete.
|
|
|
|
Flow:
|
|
1. Create a container
|
|
2. List files (should be empty)
|
|
3. Try retrieve file (should error - no files)
|
|
4. Try delete file (should error - no files)
|
|
5. Cleanup: delete container
|
|
"""
|
|
api_key = os.getenv("OPENAI_API_KEY")
|
|
|
|
# 1. Create container
|
|
print("\n1. Creating container...")
|
|
container = create_container(
|
|
name=f"test-files-api-{int(time.time())}",
|
|
custom_llm_provider="openai",
|
|
api_key=api_key,
|
|
expires_after={"anchor": "last_active_at", "minutes": 5},
|
|
)
|
|
print(f" Created: {container.id}")
|
|
|
|
try:
|
|
# 2. List files
|
|
print("2. Listing container files...")
|
|
files = list_container_files(
|
|
container_id=container.id,
|
|
custom_llm_provider="openai",
|
|
api_key=api_key,
|
|
)
|
|
assert files.object == "list"
|
|
assert isinstance(files.data, list)
|
|
assert len(files.data) == 0 # New container has no files
|
|
print(f" Files found: {len(files.data)} ✓")
|
|
|
|
# 3. Try retrieve non-existent file metadata (should raise error)
|
|
print("3. Testing retrieve_container_file (expect error)...")
|
|
try:
|
|
retrieve_container_file(
|
|
container_id=container.id,
|
|
file_id="cfile_nonexistent",
|
|
custom_llm_provider="openai",
|
|
api_key=api_key,
|
|
)
|
|
assert False, "Should have raised error for non-existent file"
|
|
except Exception as e:
|
|
assert "not found" in str(e).lower() or "invalid" in str(e).lower()
|
|
print(f" Got expected error ✓")
|
|
|
|
# 3b. Try retrieve non-existent file content (should raise error)
|
|
print("3b. Testing retrieve_container_file_content (expect error)...")
|
|
try:
|
|
retrieve_container_file_content(
|
|
container_id=container.id,
|
|
file_id="cfile_nonexistent",
|
|
custom_llm_provider="openai",
|
|
api_key=api_key,
|
|
)
|
|
assert False, "Should have raised error for non-existent file content"
|
|
except Exception as e:
|
|
print(f" Got expected error ✓")
|
|
|
|
# 4. Try delete non-existent file (should raise error)
|
|
print("4. Testing delete_container_file (expect error)...")
|
|
try:
|
|
delete_container_file(
|
|
container_id=container.id,
|
|
file_id="cfile_nonexistent",
|
|
custom_llm_provider="openai",
|
|
api_key=api_key,
|
|
)
|
|
assert False, "Should have raised error for non-existent file"
|
|
except Exception as e:
|
|
# Delete returns 400 for non-existent files
|
|
print(f" Got expected error ✓")
|
|
|
|
finally:
|
|
# 5. Cleanup
|
|
print("5. Deleting container...")
|
|
result = delete_container(
|
|
container_id=container.id,
|
|
custom_llm_provider="openai",
|
|
api_key=api_key,
|
|
)
|
|
assert result.deleted is True
|
|
print(f" Deleted ✓")
|
|
|
|
print("\nAll container files API tests passed! ✓")
|