190 lines
7.6 KiB
JavaScript
190 lines
7.6 KiB
JavaScript
"use strict";
|
|
// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
|
|
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
if (k2 === undefined) k2 = k;
|
|
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
}
|
|
Object.defineProperty(o, k2, desc);
|
|
}) : (function(o, m, k, k2) {
|
|
if (k2 === undefined) k2 = k;
|
|
o[k2] = m[k];
|
|
}));
|
|
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
|
|
Object.defineProperty(o, "default", { enumerable: true, value: v });
|
|
}) : function(o, v) {
|
|
o["default"] = v;
|
|
});
|
|
var __importStar = (this && this.__importStar) || function (mod) {
|
|
if (mod && mod.__esModule) return mod;
|
|
var result = {};
|
|
if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
|
|
__setModuleDefault(result, mod);
|
|
return result;
|
|
};
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
exports.RunsPage = exports.Runs = void 0;
|
|
const resource_1 = require("../../../../resource.js");
|
|
const core_1 = require("../../../../core.js");
|
|
const AssistantStream_1 = require("../../../../lib/AssistantStream.js");
|
|
const core_2 = require("../../../../core.js");
|
|
const RunsAPI = __importStar(require("./runs.js"));
|
|
const StepsAPI = __importStar(require("./steps.js"));
|
|
const pagination_1 = require("../../../../pagination.js");
|
|
class Runs extends resource_1.APIResource {
|
|
constructor() {
|
|
super(...arguments);
|
|
this.steps = new StepsAPI.Steps(this._client);
|
|
}
|
|
create(threadId, body, options) {
|
|
return this._client.post(`/threads/${threadId}/runs`, {
|
|
body,
|
|
...options,
|
|
headers: { 'OpenAI-Beta': 'assistants=v2', ...options?.headers },
|
|
stream: body.stream ?? false,
|
|
});
|
|
}
|
|
/**
|
|
* Retrieves a run.
|
|
*/
|
|
retrieve(threadId, runId, options) {
|
|
return this._client.get(`/threads/${threadId}/runs/${runId}`, {
|
|
...options,
|
|
headers: { 'OpenAI-Beta': 'assistants=v2', ...options?.headers },
|
|
});
|
|
}
|
|
/**
|
|
* Modifies a run.
|
|
*/
|
|
update(threadId, runId, body, options) {
|
|
return this._client.post(`/threads/${threadId}/runs/${runId}`, {
|
|
body,
|
|
...options,
|
|
headers: { 'OpenAI-Beta': 'assistants=v2', ...options?.headers },
|
|
});
|
|
}
|
|
list(threadId, query = {}, options) {
|
|
if ((0, core_1.isRequestOptions)(query)) {
|
|
return this.list(threadId, {}, query);
|
|
}
|
|
return this._client.getAPIList(`/threads/${threadId}/runs`, RunsPage, {
|
|
query,
|
|
...options,
|
|
headers: { 'OpenAI-Beta': 'assistants=v2', ...options?.headers },
|
|
});
|
|
}
|
|
/**
|
|
* Cancels a run that is `in_progress`.
|
|
*/
|
|
cancel(threadId, runId, options) {
|
|
return this._client.post(`/threads/${threadId}/runs/${runId}/cancel`, {
|
|
...options,
|
|
headers: { 'OpenAI-Beta': 'assistants=v2', ...options?.headers },
|
|
});
|
|
}
|
|
/**
|
|
* A helper to create a run an poll for a terminal state. More information on Run
|
|
* lifecycles can be found here:
|
|
* https://platform.openai.com/docs/assistants/how-it-works/runs-and-run-steps
|
|
*/
|
|
async createAndPoll(threadId, body, options) {
|
|
const run = await this.create(threadId, body, options);
|
|
return await this.poll(threadId, run.id, options);
|
|
}
|
|
/**
|
|
* Create a Run stream
|
|
*
|
|
* @deprecated use `stream` instead
|
|
*/
|
|
createAndStream(threadId, body, options) {
|
|
return AssistantStream_1.AssistantStream.createAssistantStream(threadId, this._client.beta.threads.runs, body, options);
|
|
}
|
|
/**
|
|
* A helper to poll a run status until it reaches a terminal state. More
|
|
* information on Run lifecycles can be found here:
|
|
* https://platform.openai.com/docs/assistants/how-it-works/runs-and-run-steps
|
|
*/
|
|
async poll(threadId, runId, options) {
|
|
const headers = { ...options?.headers, 'X-Stainless-Poll-Helper': 'true' };
|
|
if (options?.pollIntervalMs) {
|
|
headers['X-Stainless-Custom-Poll-Interval'] = options.pollIntervalMs.toString();
|
|
}
|
|
while (true) {
|
|
const { data: run, response } = await this.retrieve(threadId, runId, {
|
|
...options,
|
|
headers: { ...options?.headers, ...headers },
|
|
}).withResponse();
|
|
switch (run.status) {
|
|
//If we are in any sort of intermediate state we poll
|
|
case 'queued':
|
|
case 'in_progress':
|
|
case 'cancelling':
|
|
let sleepInterval = 5000;
|
|
if (options?.pollIntervalMs) {
|
|
sleepInterval = options.pollIntervalMs;
|
|
}
|
|
else {
|
|
const headerInterval = response.headers.get('openai-poll-after-ms');
|
|
if (headerInterval) {
|
|
const headerIntervalMs = parseInt(headerInterval);
|
|
if (!isNaN(headerIntervalMs)) {
|
|
sleepInterval = headerIntervalMs;
|
|
}
|
|
}
|
|
}
|
|
await (0, core_2.sleep)(sleepInterval);
|
|
break;
|
|
//We return the run in any terminal state.
|
|
case 'requires_action':
|
|
case 'incomplete':
|
|
case 'cancelled':
|
|
case 'completed':
|
|
case 'failed':
|
|
case 'expired':
|
|
return run;
|
|
}
|
|
}
|
|
}
|
|
/**
|
|
* Create a Run stream
|
|
*/
|
|
stream(threadId, body, options) {
|
|
return AssistantStream_1.AssistantStream.createAssistantStream(threadId, this._client.beta.threads.runs, body, options);
|
|
}
|
|
submitToolOutputs(threadId, runId, body, options) {
|
|
return this._client.post(`/threads/${threadId}/runs/${runId}/submit_tool_outputs`, {
|
|
body,
|
|
...options,
|
|
headers: { 'OpenAI-Beta': 'assistants=v2', ...options?.headers },
|
|
stream: body.stream ?? false,
|
|
});
|
|
}
|
|
/**
|
|
* A helper to submit a tool output to a run and poll for a terminal run state.
|
|
* More information on Run lifecycles can be found here:
|
|
* https://platform.openai.com/docs/assistants/how-it-works/runs-and-run-steps
|
|
*/
|
|
async submitToolOutputsAndPoll(threadId, runId, body, options) {
|
|
const run = await this.submitToolOutputs(threadId, runId, body, options);
|
|
return await this.poll(threadId, run.id, options);
|
|
}
|
|
/**
|
|
* Submit the tool outputs from a previous run and stream the run to a terminal
|
|
* state. More information on Run lifecycles can be found here:
|
|
* https://platform.openai.com/docs/assistants/how-it-works/runs-and-run-steps
|
|
*/
|
|
submitToolOutputsStream(threadId, runId, body, options) {
|
|
return AssistantStream_1.AssistantStream.createToolAssistantStream(threadId, runId, this._client.beta.threads.runs, body, options);
|
|
}
|
|
}
|
|
exports.Runs = Runs;
|
|
class RunsPage extends pagination_1.CursorPage {
|
|
}
|
|
exports.RunsPage = RunsPage;
|
|
(function (Runs) {
|
|
Runs.RunsPage = RunsAPI.RunsPage;
|
|
Runs.Steps = StepsAPI.Steps;
|
|
Runs.RunStepsPage = StepsAPI.RunStepsPage;
|
|
})(Runs = exports.Runs || (exports.Runs = {}));
|
|
//# sourceMappingURL=runs.js.map
|