from metadata.sdk import configure, get_client

configure(
    host="https://your-company.getcollate.io/api",
    jwt_token="your-jwt-token"
)
client = get_client()

# Find tables with a specific column name
response = client.get("/search/fieldQuery", params={
    "fieldName": "columns.name",
    "fieldValue": "shop_id",
    "index": "table_search_index"
})

for hit in response["hits"]["hits"]:
    entity = hit["_source"]
    print(f"{entity['fullyQualifiedName']} (score: {hit['_score']})")

# Find entities with a specific tag
response = client.get("/search/fieldQuery", params={
    "fieldName": "tags.tagFQN",
    "fieldValue": "PII.Sensitive",
    "index": "table_search_index"
})

for hit in response["hits"]["hits"]:
    entity = hit["_source"]
    print(f"{entity['fullyQualifiedName']}")

{
  "took": 5,
  "timed_out": false,
  "_shards": {
    "total": 1,
    "successful": 1,
    "failed": 0,
    "skipped": 0
  },
  "hits": {
    "total": {
      "value": 12,
      "relation": "eq"
    },
    "hits": [
      {
        "_index": "table_search_index",
        "_id": "7c2191b6-81da-4c97-83c4-1a537e54a3b0",
        "_score": 1.8,
        "_source": {
          "id": "7c2191b6-81da-4c97-83c4-1a537e54a3b0",
          "name": "dim_customer",
          "fullyQualifiedName": "sample_data.ecommerce_db.shopify.dim_customer",
          "entityType": "table",
          "serviceType": "BigQuery",
          "description": "Customer dimension table with shop_id reference",
          "owners": [],
          "tags": [],
          "columns": [
            {
              "name": "shop_id",
              "dataType": "NUMERIC",
              "fullyQualifiedName": "sample_data.ecommerce_db.shopify.dim_customer.shop_id"
            }
          ]
        }
      }
    ]
  }
}