Send files to an agent

interface RunFileInput {
  name: string;
  contentType: string;
  dataBase64: string;
}

import { readFile } from "node:fs/promises";
import { OaoClient } from "@oao/sdk-js";

const client = new OaoClient({
  baseUrl: process.env.OAO_API_URL!,
  apiKey: process.env.OAO_API_KEY!,
});

const source = await readFile("src/index.ts");
const created = await client.createSession(
  process.env.OAO_PROJECT_ID!,
  {
    agentId: process.env.OAO_AGENT_ID!,
    title: "Review the entry point",
    initialMessage: "Review the attached entry point for unsafe boundaries.",
    files: [
      {
        name: "index.ts",
        contentType: "application/typescript",
        dataBase64: source.toString("base64"),
      },
    ],
  },
  { idempotencyKey: "entry-point-review-1" },
);

FILE_BASE64="$(base64 < src/index.ts | tr -d '\n')"

curl --fail-with-body \
  --request POST \
  --header "Authorization: Bearer $OAO_API_KEY" \
  --header 'Content-Type: application/json' \
  --header 'Idempotency-Key: entry-point-review-1' \
  --data "{
    \"agentId\": \"$OAO_AGENT_ID\",
    \"title\": \"Review the entry point\",
    \"initialMessage\": \"Review the attached entry point.\",
    \"files\": [{
      \"name\": \"index.ts\",
      \"contentType\": \"application/typescript\",
      \"dataBase64\": \"$FILE_BASE64\"
    }]
  }" \
  "$OAO_API_URL/v1/projects/$OAO_PROJECT_ID/sessions"

await client.submitRun(
  projectId,
  sessionId,
  {
    message: "Compare this revision with the file from the previous turn.",
    files: [
      {
        name: "index-v2.ts",
        contentType: "application/typescript",
        dataBase64: revisedSource.toString("base64"),
      },
    ],
  },
  { idempotencyKey: "entry-point-review-2" },
);

import pandas as pd

path = ".oao/attachments/<run-id>/tasks.xlsx"
workbook = pd.ExcelFile(path)
print(workbook.sheet_names)
print(pd.read_excel(path, sheet_name=workbook.sheet_names[0]).head())

import { writeFile } from "node:fs/promises";

const files = await client.listSessionFiles(projectId, sessionId);
const result = files.data.find((file) => file.path === "output/result.csv");
if (!result) throw new Error("The file was not persisted");

const download = await client.downloadSessionFile(
  projectId,
  sessionId,
  result.path,
);
await writeFile(result.name, download.bytes);