@aristech-org/nlp-client
Version:
A Node.js client library for the Aristech NLP Service
419 lines (418 loc) • 16.4 kB
JavaScript
// Code generated by protoc-gen-ts_proto. DO NOT EDIT.
// versions:
// protoc-gen-ts_proto v2.6.1
// protoc v3.21.12
// source: nlp_server.proto
/* eslint-disable */
import { BinaryReader, BinaryWriter } from "@bufbuild/protobuf/wire";
import { makeGenericClientConstructor, } from "@grpc/grpc-js";
import { GetContentRequest, GetContentResponse, GetIntentsRequest, GetIntentsResponse, GetScoreLimitsRequest, GetScoreLimitsResponse, RemoveContentRequest, RemoveContentResponse, UpdateContentRequest, UpdateContentResponse, } from "./intents.js";
import { AddProjectRequest, AddProjectResponse, EmbeddingModel, GetEmbeddingModelsRequest, GetProjectsRequest, Project, RemoveProjectRequest, RemoveProjectResponse, UpdateProjectRequest, UpdateProjectResponse, } from "./projects.js";
export const protobufPackage = "aristech.nlp";
function createBaseFunctionRequest() {
return {};
}
export const FunctionRequest = {
encode(_, writer = new BinaryWriter()) {
return writer;
},
decode(input, length) {
const reader = input instanceof BinaryReader ? input : new BinaryReader(input);
let end = length === undefined ? reader.len : reader.pos + length;
const message = createBaseFunctionRequest();
while (reader.pos < end) {
const tag = reader.uint32();
switch (tag >>> 3) {
}
if ((tag & 7) === 4 || tag === 0) {
break;
}
reader.skip(tag & 7);
}
return message;
},
fromJSON(_) {
return {};
},
toJSON(_) {
const obj = {};
return obj;
},
create(base) {
return FunctionRequest.fromPartial(base ?? {});
},
fromPartial(_) {
const message = createBaseFunctionRequest();
return message;
},
};
function createBaseFunctionMessage() {
return { id: "", name: "", description: "", arguments: [] };
}
export const FunctionMessage = {
encode(message, writer = new BinaryWriter()) {
if (message.id !== "") {
writer.uint32(10).string(message.id);
}
if (message.name !== "") {
writer.uint32(18).string(message.name);
}
if (message.description !== "") {
writer.uint32(26).string(message.description);
}
for (const v of message.arguments) {
writer.uint32(34).string(v);
}
return writer;
},
decode(input, length) {
const reader = input instanceof BinaryReader ? input : new BinaryReader(input);
let end = length === undefined ? reader.len : reader.pos + length;
const message = createBaseFunctionMessage();
while (reader.pos < end) {
const tag = reader.uint32();
switch (tag >>> 3) {
case 1: {
if (tag !== 10) {
break;
}
message.id = reader.string();
continue;
}
case 2: {
if (tag !== 18) {
break;
}
message.name = reader.string();
continue;
}
case 3: {
if (tag !== 26) {
break;
}
message.description = reader.string();
continue;
}
case 4: {
if (tag !== 34) {
break;
}
message.arguments.push(reader.string());
continue;
}
}
if ((tag & 7) === 4 || tag === 0) {
break;
}
reader.skip(tag & 7);
}
return message;
},
fromJSON(object) {
return {
id: isSet(object.id) ? globalThis.String(object.id) : "",
name: isSet(object.name) ? globalThis.String(object.name) : "",
description: isSet(object.description) ? globalThis.String(object.description) : "",
arguments: globalThis.Array.isArray(object?.arguments)
? object.arguments.map((e) => globalThis.String(e))
: [],
};
},
toJSON(message) {
const obj = {};
if (message.id !== "") {
obj.id = message.id;
}
if (message.name !== "") {
obj.name = message.name;
}
if (message.description !== "") {
obj.description = message.description;
}
if (message.arguments?.length) {
obj.arguments = message.arguments;
}
return obj;
},
create(base) {
return FunctionMessage.fromPartial(base ?? {});
},
fromPartial(object) {
const message = createBaseFunctionMessage();
message.id = object.id ?? "";
message.name = object.name ?? "";
message.description = object.description ?? "";
message.arguments = object.arguments?.map((e) => e) || [];
return message;
},
};
function createBaseRunFunctionsRequest() {
return { functions: [], input: "", arguments: [] };
}
export const RunFunctionsRequest = {
encode(message, writer = new BinaryWriter()) {
for (const v of message.functions) {
FunctionMessage.encode(v, writer.uint32(10).fork()).join();
}
if (message.input !== "") {
writer.uint32(18).string(message.input);
}
for (const v of message.arguments) {
writer.uint32(26).string(v);
}
return writer;
},
decode(input, length) {
const reader = input instanceof BinaryReader ? input : new BinaryReader(input);
let end = length === undefined ? reader.len : reader.pos + length;
const message = createBaseRunFunctionsRequest();
while (reader.pos < end) {
const tag = reader.uint32();
switch (tag >>> 3) {
case 1: {
if (tag !== 10) {
break;
}
message.functions.push(FunctionMessage.decode(reader, reader.uint32()));
continue;
}
case 2: {
if (tag !== 18) {
break;
}
message.input = reader.string();
continue;
}
case 3: {
if (tag !== 26) {
break;
}
message.arguments.push(reader.string());
continue;
}
}
if ((tag & 7) === 4 || tag === 0) {
break;
}
reader.skip(tag & 7);
}
return message;
},
fromJSON(object) {
return {
functions: globalThis.Array.isArray(object?.functions)
? object.functions.map((e) => FunctionMessage.fromJSON(e))
: [],
input: isSet(object.input) ? globalThis.String(object.input) : "",
arguments: globalThis.Array.isArray(object?.arguments)
? object.arguments.map((e) => globalThis.String(e))
: [],
};
},
toJSON(message) {
const obj = {};
if (message.functions?.length) {
obj.functions = message.functions.map((e) => FunctionMessage.toJSON(e));
}
if (message.input !== "") {
obj.input = message.input;
}
if (message.arguments?.length) {
obj.arguments = message.arguments;
}
return obj;
},
create(base) {
return RunFunctionsRequest.fromPartial(base ?? {});
},
fromPartial(object) {
const message = createBaseRunFunctionsRequest();
message.functions = object.functions?.map((e) => FunctionMessage.fromPartial(e)) || [];
message.input = object.input ?? "";
message.arguments = object.arguments?.map((e) => e) || [];
return message;
},
};
function createBaseRunFunctionsResponse() {
return { output: "" };
}
export const RunFunctionsResponse = {
encode(message, writer = new BinaryWriter()) {
if (message.output !== "") {
writer.uint32(10).string(message.output);
}
return writer;
},
decode(input, length) {
const reader = input instanceof BinaryReader ? input : new BinaryReader(input);
let end = length === undefined ? reader.len : reader.pos + length;
const message = createBaseRunFunctionsResponse();
while (reader.pos < end) {
const tag = reader.uint32();
switch (tag >>> 3) {
case 1: {
if (tag !== 10) {
break;
}
message.output = reader.string();
continue;
}
}
if ((tag & 7) === 4 || tag === 0) {
break;
}
reader.skip(tag & 7);
}
return message;
},
fromJSON(object) {
return { output: isSet(object.output) ? globalThis.String(object.output) : "" };
},
toJSON(message) {
const obj = {};
if (message.output !== "") {
obj.output = message.output;
}
return obj;
},
create(base) {
return RunFunctionsResponse.fromPartial(base ?? {});
},
fromPartial(object) {
const message = createBaseRunFunctionsResponse();
message.output = object.output ?? "";
return message;
},
};
export const NLPServerService = {
/**
* A simple RPC.
* returns all available processing Models.
*/
getFunctions: {
path: "/aristech.nlp.NLPServer/GetFunctions",
requestStream: false,
responseStream: true,
requestSerialize: (value) => Buffer.from(FunctionRequest.encode(value).finish()),
requestDeserialize: (value) => FunctionRequest.decode(value),
responseSerialize: (value) => Buffer.from(FunctionMessage.encode(value).finish()),
responseDeserialize: (value) => FunctionMessage.decode(value),
},
/**
* A server-to-client streaming method
* Returns the processed text
*/
runFunctions: {
path: "/aristech.nlp.NLPServer/RunFunctions",
requestStream: false,
responseStream: false,
requestSerialize: (value) => Buffer.from(RunFunctionsRequest.encode(value).finish()),
requestDeserialize: (value) => RunFunctionsRequest.decode(value),
responseSerialize: (value) => Buffer.from(RunFunctionsResponse.encode(value).finish()),
responseDeserialize: (value) => RunFunctionsResponse.decode(value),
},
/** Method to add content to a vector database. */
updateContent: {
path: "/aristech.nlp.NLPServer/UpdateContent",
requestStream: false,
responseStream: false,
requestSerialize: (value) => Buffer.from(UpdateContentRequest.encode(value).finish()),
requestDeserialize: (value) => UpdateContentRequest.decode(value),
responseSerialize: (value) => Buffer.from(UpdateContentResponse.encode(value).finish()),
responseDeserialize: (value) => UpdateContentResponse.decode(value),
},
/** Method to remove content from a vector database. */
removeContent: {
path: "/aristech.nlp.NLPServer/RemoveContent",
requestStream: false,
responseStream: false,
requestSerialize: (value) => Buffer.from(RemoveContentRequest.encode(value).finish()),
requestDeserialize: (value) => RemoveContentRequest.decode(value),
responseSerialize: (value) => Buffer.from(RemoveContentResponse.encode(value).finish()),
responseDeserialize: (value) => RemoveContentResponse.decode(value),
},
/** Method to get content from a vector database. */
getContent: {
path: "/aristech.nlp.NLPServer/GetContent",
requestStream: false,
responseStream: true,
requestSerialize: (value) => Buffer.from(GetContentRequest.encode(value).finish()),
requestDeserialize: (value) => GetContentRequest.decode(value),
responseSerialize: (value) => Buffer.from(GetContentResponse.encode(value).finish()),
responseDeserialize: (value) => GetContentResponse.decode(value),
},
/** Method to initialize a project */
addProject: {
path: "/aristech.nlp.NLPServer/AddProject",
requestStream: false,
responseStream: false,
requestSerialize: (value) => Buffer.from(AddProjectRequest.encode(value).finish()),
requestDeserialize: (value) => AddProjectRequest.decode(value),
responseSerialize: (value) => Buffer.from(AddProjectResponse.encode(value).finish()),
responseDeserialize: (value) => AddProjectResponse.decode(value),
},
/** Method to remove Project */
removeProject: {
path: "/aristech.nlp.NLPServer/RemoveProject",
requestStream: false,
responseStream: false,
requestSerialize: (value) => Buffer.from(RemoveProjectRequest.encode(value).finish()),
requestDeserialize: (value) => RemoveProjectRequest.decode(value),
responseSerialize: (value) => Buffer.from(RemoveProjectResponse.encode(value).finish()),
responseDeserialize: (value) => RemoveProjectResponse.decode(value),
},
/** update the settings of a project */
updateProject: {
path: "/aristech.nlp.NLPServer/UpdateProject",
requestStream: false,
responseStream: false,
requestSerialize: (value) => Buffer.from(UpdateProjectRequest.encode(value).finish()),
requestDeserialize: (value) => UpdateProjectRequest.decode(value),
responseSerialize: (value) => Buffer.from(UpdateProjectResponse.encode(value).finish()),
responseDeserialize: (value) => UpdateProjectResponse.decode(value),
},
/** Method to get projects from a vector database */
getProjects: {
path: "/aristech.nlp.NLPServer/GetProjects",
requestStream: false,
responseStream: true,
requestSerialize: (value) => Buffer.from(GetProjectsRequest.encode(value).finish()),
requestDeserialize: (value) => GetProjectsRequest.decode(value),
responseSerialize: (value) => Buffer.from(Project.encode(value).finish()),
responseDeserialize: (value) => Project.decode(value),
},
/** Method to get intents from a project */
getIntents: {
path: "/aristech.nlp.NLPServer/GetIntents",
requestStream: false,
responseStream: true,
requestSerialize: (value) => Buffer.from(GetIntentsRequest.encode(value).finish()),
requestDeserialize: (value) => GetIntentsRequest.decode(value),
responseSerialize: (value) => Buffer.from(GetIntentsResponse.encode(value).finish()),
responseDeserialize: (value) => GetIntentsResponse.decode(value),
},
/** Method to get score limits for a project */
getScoreLimits: {
path: "/aristech.nlp.NLPServer/GetScoreLimits",
requestStream: false,
responseStream: false,
requestSerialize: (value) => Buffer.from(GetScoreLimitsRequest.encode(value).finish()),
requestDeserialize: (value) => GetScoreLimitsRequest.decode(value),
responseSerialize: (value) => Buffer.from(GetScoreLimitsResponse.encode(value).finish()),
responseDeserialize: (value) => GetScoreLimitsResponse.decode(value),
},
/** get the available embedding models to be used for similarity search */
getEmbeddingModels: {
path: "/aristech.nlp.NLPServer/GetEmbeddingModels",
requestStream: false,
responseStream: true,
requestSerialize: (value) => Buffer.from(GetEmbeddingModelsRequest.encode(value).finish()),
requestDeserialize: (value) => GetEmbeddingModelsRequest.decode(value),
responseSerialize: (value) => Buffer.from(EmbeddingModel.encode(value).finish()),
responseDeserialize: (value) => EmbeddingModel.decode(value),
},
};
export const NLPServerClient = makeGenericClientConstructor(NLPServerService, "aristech.nlp.NLPServer");
function isSet(value) {
return value !== null && value !== undefined;
}