Repository navigation
Expand file tree
/
Copy pathindex.js
More file actions
73 lines (61 loc) · 2.29 KB
/
Copy pathindex.js
File metadata and controls
73 lines (61 loc) · 2.29 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
import { body_request, url_request } from "./request.js";
const base_url = "http://localhost:6333/";
const QdrantResponse = function(response) {
this.err = response[0];
this.response = response[1];
}
export const Qdrant = function(url){
this.url = url||base_url;
};
//DELETE http://localhost:6333/collections/{collection_name}
Qdrant.prototype.delete_collection = async function (name) {
let qdrant_url = this.url;
let url = `${qdrant_url}collections/${name}`;
return new QdrantResponse(await body_request(url,null,'DELETE'));
}
//PUT http://localhost:6333/collections/{collection_name}
Qdrant.prototype.create_collection = async function (name,body) {
let qdrant_url = this.url;
let url = `${qdrant_url}collections/${name}`;
return new QdrantResponse(await body_request(url,body,'PUT'));
}
//GET http://localhost:6333/collections/{collection_name}
Qdrant.prototype.get_collection = async function (name) {
let qdrant_url = this.url;
let url = `${qdrant_url}collections/${name}`;
return new QdrantResponse(await url_request(url));
}
//PUT http://localhost:6333/collections/{collection_name}/points
Qdrant.prototype.upload_points = async function (name,points) {
let qdrant_url = this.url;
let url = `${qdrant_url}collections/${name}/points`;
return new QdrantResponse(await body_request(url,{points:points},'PUT'));
}
//POST http://localhost:6333/collections/{collection_name}/points/search
Qdrant.prototype.search_collection = async function (name,vector,k,ef,filter) {
k = k || 5;
ef = ef || 128;
let qdrant_url = this.url;
let url = `${qdrant_url}collections/${name}/points/search`;
let query = {
"params": {
"hnsw_ef": ef
},
"vector": vector,
"top": k
};
if (filter) query.filter = filter;
return new QdrantResponse(await body_request(url,query,'POST'));
}
//Same as search_collection but allows free-form query by the client
Qdrant.prototype.query_collection = async function (name,query) {
let qdrant_url = this.url;
let url = `${qdrant_url}collections/${name}/points/search`;
return new QdrantResponse(await body_request(url,query,'POST'));
}
//Get the specific points by ids
Qdrant.prototype.retrieve_points = async function (name,query) {
let qdrant_url = this.url;
let url = `${qdrant_url}collections/${name}/points`;
return new QdrantResponse(await body_request(url,query,'POST'));
}