import OpenAI from "openai";

const openai = new OpenAI();

// Set the assistant ID here or pass it as an argument

const ASSISTANT_ID = "";

const assistantId = process.argv[2] ?? ASSISTANT_ID;

// Create a thread

const thread = await openai.beta.threads.create();

console.log("Thread created: ", thread);

// Create a thread and add a message to it

const USER_INPUT = "What is the square root of 25?";

await openai.beta.threads.messages.create(thread.id, {

role: "user",

content: USER_INPUT,

});

console.log(`Message added to thread: '${USER_INPUT}'`);

// Run the assistant on the thread

let run = await openai.beta.threads.runs.create(thread.id, {

assistant_id: assistantId,

});

console.log("Run created: ", run);

// Poll the run results every 10 seconds until the run is completed with a timeout of 5 minutes

const TIMEOUT = 300000;

const startTime = Date.now();

while (

run.status !== "completed" &&

run.status !== "failed" &&

Date.now() - startTime < TIMEOUT

) {

run = await openai.beta.threads.runs.retrieve(thread.id, run.id);

console.log("Run status: ", run.status);

await new Promise((resolve) => setTimeout(resolve, 10000));

}

// Get the updated thread

const threadMessages = await openai.beta.threads.messages.list(thread.id);

for (const message of threadMessages.data) {

console.log(

`\n\nRole: ${message.role}\nContent: ${JSON.stringify(

message.content,

null,

2

)}`

);

}