iframe-proxy | Sunbelt Computer Software

Repository navigation

History

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

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] - Whether to wait for the prediction to finish. Defaults to false

* @param {number} [options.wait.interval] - Polling interval in milliseconds. Defaults to 250

* @param {number} [options.wait.maxAttempts] - Maximum number of polling attempts. Defaults to no limit

* @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`)

* @throws {Error} If the prediction failed

* @returns {Promise<object>} - Resolves with the output of running the model

async run(identifier, 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;

const prediction = await this.predictions.create({

wait: true,

...options,

version,

});

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} parameters - Request parameters

* @param {string} [parameters.method] - HTTP method. Defaults to GET

* @param {object} [parameters.params] - Query parameters

* @param {object} [parameters.data] - Body parameters

* @returns {Promise<object>} - Resolves with the API response data

* @throws {Error} If the request failed

async request(route, parameters) {

const { auth, baseUrl, userAgent } = this;

const url = new URL(

route.startsWith('/') ? route.slice(1) : route,

baseUrl.endsWith('/') ? baseUrl : `${baseUrl}/`

);

const { method = 'GET', params = {}, data } = parameters;

Object.entries(params).forEach(([key, value]) => {

url.searchParams.append(key, value);

});

const headers = {

Authorization: `Token ${auth}`,

'Content-Type': 'application/json',

'User-Agent': userAgent,

};

const response = await this.fetch(url, {

method,

headers,

body: data ? JSON.stringify(data) : undefined,

});

if (!response.ok) {

throw new Error(`API request failed: ${response.statusText}`);

}

return response.json();

}

/**

* 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' });

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 250

* @param {number} [options.maxAttempts] - Maximum number of polling attempts. Defaults to no limit

* @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) {

const { id } = prediction;

if (!id) {

throw new Error('Invalid prediction');

}

if (

prediction.status === 'succeeded' ||

prediction.status === 'failed' ||

prediction.status === 'canceled'

) {

return prediction;

}

let updatedPrediction = await this.predictions.get(id);

// eslint-disable-next-line no-promise-executor-return

const sleep = (ms) => new Promise((resolve) => setTimeout(resolve, ms));

let attempts = 0;

const interval = options.interval || 250;

const maxAttempts = options.maxAttempts || null;

while (

updatedPrediction.status !== 'succeeded' &&

updatedPrediction.status !== 'failed' &&

updatedPrediction.status !== 'canceled'

) {

attempts += 1;

if (maxAttempts && attempts > maxAttempts) {

throw new Error(

`Prediction ${id} did not finish after ${maxAttempts} attempts`

);

}

/* eslint-disable no-await-in-loop */

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;

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Repository navigation

index.js

index.js

Sunbelt Computer Software

PL/B Language Development and Support

FilesExpand file tree

index.js

Latest commit

History

index.js

File metadata and controls