forked from replicate/replicate-javascript
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathidentifier.js
More file actions
39 lines (35 loc) · 1003 Bytes
/
identifier.js
File metadata and controls
39 lines (35 loc) · 1003 Bytes
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
/*
* A reference to a model version in the format `owner/name` or `owner/name:version`.
*/
class ModelVersionIdentifier {
/*
* @param {string} Required. The model owner.
* @param {string} Required. The model name.
* @param {string} The model version.
*/
constructor(owner, name, version = null) {
this.owner = owner;
this.name = name;
this.version = version;
}
/*
* Parse a reference to a model version
*
* @param {string}
* @returns {ModelVersionIdentifier}
* @throws {Error} If the reference is invalid.
*/
static parse(ref) {
const match = ref.match(
/^(?<owner>[^/]+)\/(?<name>[^/:]+)(:(?<version>.+))?$/
);
if (!match) {
throw new Error(
`Invalid reference to model version: ${ref}. Expected format: owner/name or owner/name:version`
);
}
const { owner, name, version } = match.groups;
return new ModelVersionIdentifier(owner, name, version);
}
}
module.exports = ModelVersionIdentifier;