Flows

flows = client.flows.list()

for flow in flows.items:
    print(f"Flow: {flow.flow_id}")
    print(f"Created: {flow.metadata.time_created}")

for page in client.flows.list():
    for flow in page.items:
        print(f"Flow: {flow.flow_id}")
        print(f"  Name: {flow.spec.name}")
        print(f"  Created: {flow.metadata.time_created}")
        print(f"  Blocks: {len(flow.spec.blocks)}")

from datetime import datetime, timedelta

cutoff = (datetime.now() - timedelta(days=7)).timestamp() * 1000

recent_flows = []
for page in client.flows.list():
    for flow in page.items:
        if flow.metadata.time_created > cutoff:
            recent_flows.append(flow)

print(f"Found {len(recent_flows)} recent flows")

all_flows = list(client.flows.list().all_items())
print(f"Total flows: {len(all_flows)}")

flow = client.flows.get("flow_abc123")

print(f"Flow ID: {flow.flow_id}")
print(f"Name: {flow.spec.name}")
print(f"Blocks: {len(flow.spec.blocks)}")

flow = client.flows.get("flow_abc123")
flow.info()  # Prints formatted flow summary

flow = client.flows.get("flow_abc123")

print(f"Flow: {flow.spec.name}")
print(f"Blocks ({len(flow.spec.blocks)}):")

for i, block in enumerate(flow.spec.blocks, 1):
    print(f"  {i}. {block.block}")
    if block.settings:
        for key, value in list(block.settings.items())[:3]:
            print(f"      {key}: {value}")

flow = client.flows.get("flow_abc123")

print("Global Settings:")
for key, value in flow.spec.settings.items():
    print(f"  {key}: {value}")

print("\nTolerations:")
for key, values in flow.spec.tolerations.items():
    print(f"  {key}: {values}")

flow = client.flows.get("flow_abc123")

print(f"Organization: {flow.organization_id}")
print(f"Schema: {flow.spec.schema}")
print(f"ETag: {flow.etag}")

if flow.spec.training_project:
    print(f"Training Project: {flow.spec.training_project}")

deleted = client.flows.delete("flow_abc123")

flow = client.flows.get("flow_abc123")
print(f"About to delete flow: {flow.spec.name}")
print(f"  Blocks: {len(flow.spec.blocks)}")

confirm = input("Delete? (yes/no): ")
if confirm.lower() == "yes":
    client.flows.delete("flow_abc123")
    print("Deleted.")

flow = client.flows.create(
    spec={
        "name": "My Training Flow",
        "schema": "v1",
        "blocks": [],
        "settings": {},
        "tolerations": {}
    }
)

print(f"Created flow: {flow.flow_id}")
print(f"Name: {flow.spec.name}")

flow = client.flows.create(
    spec={
        "name": "Production Flow",
        "schema": "v1",
        "blocks": [
            {
                "block": "data_loader",
                "settings": {"batch_size": 32},
                "style": {}
            }
        ],
        "settings": {"learning_rate": 0.001},
        "tolerations": {}
    },
    metadata={"environment": "production", "version": "1.0"}
)

flow = client.flows.update(
    flow_id="flow_abc123",
    spec={"name": "Updated Flow Name"}
)

print(f"Updated: {flow.spec.name}")

flow = client.flows.update(
    flow_id="flow_abc123",
    spec={"training_project": "project_xyz789"},
    metadata={"status": "active"}
)

from vi.api.resources.flows.responses import Flow

from vi.api.resources.flows.responses import FlowSpec

from vi.api.resources.flows.responses import FlowBlock