from openapi import SDK


with SDK(
    bearer_auth="<YOUR_BEARER_TOKEN_HERE>",
) as sdk:

    res = sdk.search.graphrag.ingest_graph_rag_documents(store_name="<value>", x_api_key_id="<id>", x_user_id="<id>", x_exchange_jwt_external_user_id="<id>", documents=[
        {
            "document_id": "doc_001",
            "text": "Atlas Infrastructure Group owns Rivera Logistics.",
            "metadata": {
                "source": "benchmark",
            },
        },
    ])

    # Handle response
    print(res)

{
  "chunk_count": 1,
  "document_count": 1,
  "edge_count": 1,
  "entity_count": 2,
  "processing_time_ms": 42
}