Content Tags API: Categorize Wolfia Content

curl -X GET 'https://api.wolfia.com/v1/business-segments/' \
  -H "X-API-Key: wolfia-api-YOUR_KEY_HERE"

[
  {
    "id": "a1b2c3d4-e5f6-7890-abcd-ef1234567890",
    "name": "SOC 2",
    "description": "SOC 2 compliance questionnaires",
    "is_ai_generated": false,
    "status": "ACTIVE"
  },
  {
    "id": "b2c3d4e5-f6a7-8901-bcde-f12345678901",
    "name": "ISO 27001",
    "description": "ISO 27001 security questionnaires",
    "is_ai_generated": true,
    "status": "DRAFT"
  }
]

import os

import requests

API_KEY = os.environ['WOLFIA_API_KEY']
BASE_URL = 'https://api.wolfia.com/v1'


def upload_with_business_segments(file_path, company_name, segment_names):
    """Upload a questionnaire with matching business segments applied."""
    headers = {'X-API-Key': API_KEY}

    # Fetch available business segments
    segments_resp = requests.get(
        f'{BASE_URL}/business-segments/',
        headers=headers,
        params={'status': 'ACTIVE'},
        timeout=10,
    )
    segments_resp.raise_for_status()
    all_segments = segments_resp.json()

    # Match requested segment names (case-insensitive)
    tag_ids = [
        s['id'] for s in all_segments
        if s['name'].lower() in [n.lower() for n in segment_names]
    ]

    # Upload the questionnaire with business segments
    with open(file_path, 'rb') as f:
        upload_resp = requests.post(
            f'{BASE_URL}/questionnaire/upload/file',
            headers=headers,
            files={'file': (os.path.basename(file_path), f, 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')},
            data={
                'company_name': company_name,
                'tag_ids': tag_ids,
            },
            timeout=60,
        )
    upload_resp.raise_for_status()
    return upload_resp.json()


if __name__ == '__main__':
    result = upload_with_business_segments('questionnaire.xlsx', 'Acme Corp', ['SOC 2', 'ISO 27001'])
    print(f"Uploaded: {result['questionnaire_id']}")