import os
from datagrid_ai import Datagrid

client = Datagrid(
    api_key=os.environ.get("DATAGRID_API_KEY"),  # This is the default and can be omitted
)
search_ai_result = client.search.search_ai(
    query="query",
)
print(search_ai_result.answer)

{
  "object": "search_ai_result",
  "answer": "<string>",
  "sources": [
    {
      "index": 123,
      "node_id": "<string>",
      "type": "file",
      "title": "<string>",
      "dataset_name": "<string>",
      "table_name": "<string>",
      "emoji": "<string>",
      "url": "<string>",
      "thumbnail_uri": "<string>",
      "page_numbers": [
        123
      ]
    }
  ],
  "search_results": {
    "object": "search_tree",
    "query": "<string>",
    "data": [
      {}
    ],
    "nav_items": [
      {}
    ],
    "has_more": true,
    "credits": {
      "consumed": 123
    },
    "next": "<string>"
  },
  "credits": {
    "consumed": 123
  }
}