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
)
identity = client.identity.retrieve()
print(identity.current_teamspace_id)

{
  "object": "identity",
  "user_id": "<string>",
  "current_teamspace_id": "<string>",
  "teamspaces": [
    {
      "teamspace_id": "<string>",
      "permissions": {
        "role": "owner",
        "agent_ids": [
          "<string>"
        ]
      }
    }
  ]
}