{{ message }}
Repository navigation
Expand file tree
/
Copy pathindex.js
More file actions
301 lines (260 loc) · 9.73 KB
/
Copy pathindex.js
File metadata and controls
301 lines (260 loc) · 9.73 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
const ApiError = require('./lib/error');
const { withAutomaticRetries } = require('./lib/util');
const collections = require('./lib/collections');
const models = require('./lib/models');
const predictions = require('./lib/predictions');
const trainings = require('./lib/trainings');
const packageJSON = require('./package.json');
/**
* Replicate API client library
*
* @see https://replicate.com/docs/reference/http
* @example
* // Create a new Replicate API client instance
* const Replicate = require("replicate");
* const replicate = new Replicate({
* // get your token from https://replicate.com/account
* auth: process.env.REPLICATE_API_TOKEN,
* userAgent: "my-app/1.2.3"
* });
*
* // Run a model and await the result:
* const model = 'owner/model:version-id'
* const input = {text: 'Hello, world!'}
* const output = await replicate.run(model, { input });
*/
class Replicate {
/**
* Create a new Replicate API client instance.
*
* @param {object} options - Configuration options for the client
* @param {string} options.auth - Required. API access token
* @param {string} options.userAgent - Identifier of your app
* @param {string} [options.baseUrl] - Defaults to https://api.replicate.com/v1
* @param {Function} [options.fetch] - Fetch function to use. Defaults to `globalThis.fetch`
*/
constructor(options = {}) {
this.auth = options.auth;
this.userAgent =
options.userAgent || `replicate-javascript/${packageJSON.version}`;
this.baseUrl = options.baseUrl || 'https://api.replicate.com/v1';
this.fetch = options.fetch || globalThis.fetch;
this.collections = {
list: collections.list.bind(this),
get: collections.get.bind(this),
};
this.models = {
get: models.get.bind(this),
versions: {
list: models.versions.list.bind(this),
get: models.versions.get.bind(this),
},
};
this.predictions = {
create: predictions.create.bind(this),
get: predictions.get.bind(this),
cancel: predictions.cancel.bind(this),
list: predictions.list.bind(this),
};
this.trainings = {
create: trainings.create.bind(this),
get: trainings.get.bind(this),
cancel: trainings.cancel.bind(this),
list: trainings.list.bind(this),
};
}
/**
* Run a model and wait for its output.
*
* @param {string} identifier - Required. The model version identifier in the format "{owner}/{name}:{version}"
* @param {object} options
* @param {object} options.input - Required. An object with the model inputs
* @param {object} [options.wait] - Options for waiting for the prediction to finish
* @param {number} [options.wait.interval] - Polling interval in milliseconds. Defaults to 500
* @param {string} [options.webhook] - An HTTPS URL for receiving a webhook when the prediction has new output
* @param {string[]} [options.webhook_events_filter] - You can change which events trigger webhook requests by specifying webhook events (`start`|`output`|`logs`|`completed`)
* @param {AbortSignal} [options.signal] - AbortSignal to cancel the prediction
* @param {Function} [progress] - Callback function that receives the prediction object as it's updated. The function is called when the prediction is created, each time its updated while polling for completion, and when it's completed.
* @throws {Error} If the prediction failed
* @returns {Promise<object>} - Resolves with the output of running the model
*/
async run(identifier, options, progress) {
const { wait, ...data } = options;
// Define a pattern for owner and model names that allows
// letters, digits, and certain special characters.
// Example: "user123", "abc__123", "user.name"
const namePattern = /[a-zA-Z0-9]+(?:(?:[._]|__|[-]*)[a-zA-Z0-9]+)*/;
// Define a pattern for "owner/name:version" format with named capturing groups.
// Example: "user123/repo_a:1a2b3c"
const pattern = new RegExp(
`^(?<owner>${namePattern.source})/(?<name>${namePattern.source}):(?<version>[0-9a-fA-F]+)$`
);
const match = identifier.match(pattern);
if (!match || !match.groups) {
throw new Error(
'Invalid version. It must be in the format "owner/name:version"'
);
}
const { version } = match.groups;
let prediction = await this.predictions.create({
...data,
version,
});
// Call progress callback with the initial prediction object
if (progress) {
progress(prediction);
}
const { signal } = options;
prediction = await this.wait(prediction, wait || {}, async (updatedPrediction) => {
// Call progress callback with the updated prediction object
if (progress) {
progress(updatedPrediction);
}
if (signal && signal.aborted) {
await this.predictions.cancel(updatedPrediction.id);
return true; // stop polling
}
return false; // continue polling
});
// Call progress callback with the completed prediction object
if (progress) {
progress(prediction);
}
if (prediction.status === 'failed') {
throw new Error(`Prediction failed: ${prediction.error}`);
}
return prediction.output;
}
/**
* Make a request to the Replicate API.
*
* @param {string} route - REST API endpoint path
* @param {object} options - Request parameters
* @param {string} [options.method] - HTTP method. Defaults to GET
* @param {object} [options.params] - Query parameters
* @param {object|Headers} [options.headers] - HTTP headers
* @param {object} [options.data] - Body parameters
* @returns {Promise<Response>} - Resolves with the response object
* @throws {ApiError} If the request failed
*/
async request(route, options) {
const { auth, baseUrl, userAgent } = this;
let url;
if (route instanceof URL) {
url = route;
} else {
url = new URL(
route.startsWith('/') ? route.slice(1) : route,
baseUrl.endsWith('/') ? baseUrl : `${baseUrl}/`
);
}
const {
method = 'GET',
params = {},
data,
} = options;
Object.entries(params).forEach(([key, value]) => {
url.searchParams.append(key, value);
});
const headers = new Headers();
if (auth) {
headers.append('Authorization', `Token ${auth}`);
}
headers.append('Content-Type', 'application/json');
headers.append('User-Agent', userAgent);
if (options.headers) {
options.headers.forEach((value, key) => {
headers.append(key, value);
});
}
const init = {
method,
headers,
body: data ? JSON.stringify(data) : undefined,
};
const shouldRetry = method === 'GET' ?
(response) => (response.status === 429 || response.status >= 500) :
(response) => (response.status === 429);
const response = await withAutomaticRetries(async () => this.fetch(url, init), { shouldRetry });
if (!response.ok) {
const request = new Request(url, init);
const responseText = await response.text();
throw new ApiError(
`Request to ${url} failed with status ${response.status} ${response.statusText}: ${responseText}.`,
request,
response,
);
}
return response;
}
/**
* Paginate through a list of results.
*
* @generator
* @example
* for await (const page of replicate.paginate(replicate.predictions.list) {
* console.log(page);
* }
* @param {Function} endpoint - Function that returns a promise for the next page of results
* @yields {object[]} Each page of results
*/
async * paginate(endpoint) {
const response = await endpoint();
yield response.results;
if (response.next) {
const nextPage = () => this.request(response.next, { method: 'GET' }).then((r) => r.json());
yield* this.paginate(nextPage);
}
}
/**
* Wait for a prediction to finish.
*
* If the prediction has already finished,
* this function returns immediately.
* Otherwise, it polls the API until the prediction finishes.
*
* @async
* @param {object} prediction - Prediction object
* @param {object} options - Options
* @param {number} [options.interval] - Polling interval in milliseconds. Defaults to 500
* @param {Function} [stop] - Async callback function that is called after each polling attempt. Receives the prediction object as an argument. Return false to cancel polling.
* @throws {Error} If the prediction doesn't complete within the maximum number of attempts
* @throws {Error} If the prediction failed
* @returns {Promise<object>} Resolves with the completed prediction object
*/
async wait(prediction, options, stop) {
const { id } = prediction;
if (!id) {
throw new Error('Invalid prediction');
}
if (
prediction.status === 'succeeded' ||
prediction.status === 'failed' ||
prediction.status === 'canceled'
) {
return prediction;
}
// eslint-disable-next-line no-promise-executor-return
const sleep = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
const interval = (options && options.interval) || 500;
let updatedPrediction = await this.predictions.get(id);
while (
updatedPrediction.status !== 'succeeded' &&
updatedPrediction.status !== 'failed' &&
updatedPrediction.status !== 'canceled'
) {
/* eslint-disable no-await-in-loop */
if (stop && await stop(updatedPrediction) === true) {
break;
}
await sleep(interval);
updatedPrediction = await this.predictions.get(prediction.id);
/* eslint-enable no-await-in-loop */
}
if (updatedPrediction.status === 'failed') {
throw new Error(`Prediction failed: ${updatedPrediction.error}`);
}
return updatedPrediction;
}
}
module.exports = Replicate;
You can’t perform that action at this time.