Files

from memvai import Memv

client = Memv()

# Upload a file
with open("document.pdf", "rb") as file:
    response = client.upload.batch.create(
        space_id="space_abc123",
        files=[file]
    )

print(f"Batch ID: {response.batch_id}")

# Upload multiple files
file_paths = ["doc1.pdf", "doc2.txt", "image.png"]

files = [open(path, "rb") for path in file_paths]

try:
    response = client.upload.batch.create(
        space_id="space_abc123",
        files=files
    )
    print(f"Uploaded {len(files)} files - Batch ID: {response.batch_id}")
finally:
    for file in files:
        file.close()

import time

# Upload file
with open("large_video.mp4", "rb") as file:
    response = client.upload.batch.create(
        space_id="space_abc123",
        files=[file]
    )

batch_id = response.batch_id

# Poll for completion
while True:
    status = client.upload.batch.get_status(batch_id=batch_id)

    print(f"Status: {status.status}")
    print(f"Progress: {status.processed_files}/{status.total_files}")

    if status.status == "completed":
        print("Processing complete!")
        break
    elif status.status == "failed":
        print("Processing failed")
        break

    time.sleep(2)  # Check every 2 seconds

# List all files in a space
response = client.files.list(space_id="space_abc123")

for file in response.files:
    print(f"{file.name} - {file.size} bytes")
    print(f"  Type: {file.mime_type}")
    print(f"  Status: {file.processing_status}")

# Download a file
response = client.files.download(
    file_id="file_xyz789",
    space_id="space_abc123"
)

# Save to disk
with open("downloaded_file.pdf", "wb") as f:
    f.write(response.content)

# Delete a file
response = client.files.delete(
    file_id="file_xyz789",
    space_id="space_abc123"
)

if response.success:
    print("File deleted successfully")