数据分析工作坊

ws.upsert(
    "db-snippet",
    {
        "object_id": "snippet-customer-count",
        "database_id": database_id,
        "title": "Count customers",
        "sql": "SELECT COUNT(*) AS n FROM customers",
        "readonly": True,
        "card_text": "Count all customers in the warehouse.",
    },
)

result = ws.database.run_snippet("snippet-customer-count")
print(result["rows"])

suggestions = ws.database.suggest_sql("How many customers are there?")
print([row["object_id"] for row in suggestions["snippets"]])

ws.upsert(
    "db-predicate",
    {
        "object_id": "predicate-customer-id",
        "column_id": columns[0]["object_id"],
        "expression": "customer_id = :customer_id",
        "desc": "Select one customer by primary key.",
    },
)

toolkit = ws.to_mcp(profile="database")

suggested = toolkit.run(
    "suggest_sql",
    question="How many customers are there?",
)
counted = toolkit.run(
    "run_snippet",
    snippet_id="snippet-customer-count",
)