78 lines
2.3 KiB
Python
78 lines
2.3 KiB
Python
#!/usr/bin/env python3
|
|
# /// script
|
|
# dependencies = [
|
|
# "weaviate-client==4.19.2",
|
|
# "typer==0.21.0",
|
|
# ]
|
|
# ///
|
|
"""
|
|
List all Weaviate collections.
|
|
|
|
Usage:
|
|
uv run list_collections.py [--json]
|
|
|
|
Environment Variables:
|
|
WEAVIATE_URL: Weaviate Cloud cluster URL
|
|
WEAVIATE_API_KEY: API key for authentication
|
|
"""
|
|
|
|
import json
|
|
import sys
|
|
|
|
import typer
|
|
import weaviate
|
|
|
|
# Import shared connection utilities (local to this skill)
|
|
from weaviate_conn import get_client
|
|
|
|
app = typer.Typer()
|
|
|
|
|
|
@app.command()
|
|
def main(
|
|
json_output: bool = typer.Option(False, "--json", help="Output in JSON format"),
|
|
):
|
|
"""List all Weaviate collections."""
|
|
try:
|
|
with get_client() as client:
|
|
print("Fetching collections...", file=sys.stderr)
|
|
collections = client.collections.list_all(simple=False)
|
|
print(f"Found {len(collections)} collections.", file=sys.stderr)
|
|
|
|
if json_output:
|
|
result = []
|
|
for name, config in collections.items():
|
|
result.append(
|
|
{
|
|
"name": name,
|
|
"description": config.description,
|
|
"properties": [
|
|
{"name": p.name, "data_type": str(p.data_type)}
|
|
for p in config.properties
|
|
],
|
|
}
|
|
)
|
|
print(json.dumps(result, indent=2, default=str))
|
|
else:
|
|
if not collections:
|
|
print("No collections found.")
|
|
else:
|
|
print("## Collections\n")
|
|
print("| Name | Description | Properties |")
|
|
print("|------|-------------|------------|")
|
|
for name, config in collections.items():
|
|
props = ", ".join([p.name for p in config.properties])
|
|
desc = config.description or "N/A"
|
|
print(f"| {name} | {desc} | {props} |")
|
|
|
|
except weaviate.exceptions.WeaviateConnectionError as e:
|
|
print(f"Error: Connection failed - {e}", file=sys.stderr)
|
|
raise typer.Exit(1)
|
|
except Exception as e:
|
|
print(f"Error: {e}", file=sys.stderr)
|
|
raise typer.Exit(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app()
|