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
)
create_knowledge_base_upload_response = client.knowledge_bases.uploads.create(
    knowledge_base_id="knowledge_base_id",
    data_source_config={
        "aws_account_id": "aws_account_id",
        "aws_region": "aws_region",
        "s3_bucket": "s3_bucket",
        "source": "S3",
    },
)
print(create_knowledge_base_upload_response.upload_id)

{
  "upload_id": "<string>"
}