Upload Files

1
%pip install -U athena-intelligence

1
import os
2
import io
3
import asyncio
4
from athena.client import Athena
5
6
ATHENA_API_KEY = os.environ["ATHENA_API_KEY"]
7
8
athena = Athena(
9
    api_key=ATHENA_API_KEY,
10
)

1
data_frame = pd.DataFrame({"column": [1, 2, 3]})
2
result = athena.tools.save_asset(data_frame, name="My data frame")
3
print(result)

1
fig = plt.figure(figsize=(10, 6))
2
plt.plot([0, 1, 2, 3, 4, 5], [1, 2, 4, 8, 16, 32])
3
plt.show()
4
client.tools.save_asset(fig, name="Powers of two")

1
async def upload_file():
2
    # Prepare the file for upload
3
    file_bytes = io.BytesIO()
4
    with open("your_file.xlsx", "rb") as f:
5
        file_bytes.write(f.read())
6
        file_bytes.seek(0)  # Reset the cursor of the BytesIO object
7
8
    # Create the file tuple
9
    file_tuple = (
10
        "your_file.xlsx",
11
        file_bytes,
12
        "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
13
    )
14
15
    # Upload the file
16
    result = await athena.tools.save_asset(file_tuple)
17
    print(result)
18
19
# Run the async function
20
await upload_file()

1
from fastapi import FastAPI, UploadFile
2
3
app = FastAPI()
4
5
@app.post("/upload/")
6
async def upload_file(file: UploadFile):
7
    file_tuple = (file.filename, file.file, file.content_type)
8
    result = await athena.tools.save_asset(file_tuple)
9
    return {"message": "File uploaded successfully", "result": result}