import os
from scale_gp import SGPClient

client = SGPClient(
    api_key=os.environ.get("SGP_API_KEY"),  # This is the default and can be omitted
)
knowledge_base_data_source = client.knowledge_base_data_sources.create(
    account_id="account_id",
    data_source_config={
        "aws_account_id": "aws_account_id",
        "aws_region": "aws_region",
        "s3_bucket": "s3_bucket",
        "source": "S3",
    },
    name="name",
)
print(knowledge_base_data_source.id)

{
  "name": "<string>",
  "data_source_config": {
    "source": "<string>",
    "s3_bucket": "<string>",
    "aws_region": "<string>",
    "aws_account_id": "<string>",
    "s3_prefix": ""
  },
  "id": "<string>",
  "created_at": "2023-11-07T05:31:56Z",
  "updated_at": "2023-11-07T05:31:56Z",
  "created_by_user_id": "<string>",
  "created_by_identity_type": "user",
  "account_id": "<string>",
  "description": "<string>",
  "tagging_information": {
    "type": "all",
    "tags_to_apply": {}
  }
}