@ekodb/ekodb-client 0.2.1 → 0.4.0

package/dist/functions.js

@@ -0,0 +1,154 @@
+"use strict";
+/**
+ * Scripts API for ekoDB TypeScript client
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Stage = exports.ChatMessage = void 0;
+exports.ChatMessage = {
+    system: (content) => ({
+        role: "system",
+        content: content,
+    }),
+    user: (content) => ({
+        role: "user",
+        content: content,
+    }),
+    assistant: (content) => ({
+        role: "assistant",
+        content: content,
+    }),
+};
+// Stage builder functions
+exports.Stage = {
+    findAll: (collection) => ({
+        type: "FindAll",
+        collection,
+    }),
+    query: (collection, filter, sort, limit, skip) => ({
+        type: "Query",
+        collection,
+        filter,
+        sort,
+        limit,
+        skip,
+    }),
+    project: (fields, exclude = false) => ({
+        type: "Project",
+        fields,
+        exclude,
+    }),
+    group: (by_fields, functions) => ({
+        type: "Group",
+        by_fields,
+        functions,
+    }),
+    count: (output_field = "count") => ({
+        type: "Count",
+        output_field,
+    }),
+    insert: (collection, record, bypassRipple = false, ttl) => ({
+        type: "Insert",
+        collection,
+        record,
+        bypass_ripple: bypassRipple,
+        ttl,
+    }),
+    update: (collection, filter, updates, bypassRipple = false, ttl) => ({
+        type: "Update",
+        collection,
+        filter,
+        updates,
+        bypass_ripple: bypassRipple,
+        ttl,
+    }),
+    updateById: (collection, record_id, updates, bypassRipple = false, ttl) => ({
+        type: "UpdateById",
+        collection,
+        record_id,
+        updates,
+        bypass_ripple: bypassRipple,
+        ttl,
+    }),
+    delete: (collection, filter, bypassRipple = false) => ({
+        type: "Delete",
+        collection,
+        filter,
+        bypass_ripple: bypassRipple,
+    }),
+    deleteById: (collection, record_id, bypassRipple = false) => ({
+        type: "DeleteById",
+        collection,
+        record_id,
+        bypass_ripple: bypassRipple,
+    }),
+    batchInsert: (collection, records, bypassRipple = false) => ({
+        type: "BatchInsert",
+        collection,
+        records,
+        bypass_ripple: bypassRipple,
+    }),
+    batchDelete: (collection, record_ids, bypassRipple = false) => ({
+        type: "BatchDelete",
+        collection,
+        record_ids,
+        bypass_ripple: bypassRipple,
+    }),
+    filter: (filter) => ({
+        type: "Filter",
+        filter,
+    }),
+    sort: (sort) => ({
+        type: "Sort",
+        sort,
+    }),
+    limit: (limit) => ({
+        type: "Limit",
+        limit,
+    }),
+    skip: (skip) => ({
+        type: "Skip",
+        skip,
+    }),
+    httpRequest: (url, method = "GET", headers, body) => ({
+        type: "HttpRequest",
+        url,
+        method,
+        headers,
+        body,
+    }),
+    vectorSearch: (collection, query_vector, limit, threshold) => ({
+        type: "VectorSearch",
+        collection,
+        query_vector,
+        limit,
+        threshold,
+    }),
+    textSearch: (collection, query_text, options) => ({
+        type: "TextSearch",
+        collection,
+        query_text,
+        fields: options?.fields,
+        limit: options?.limit,
+        fuzzy: options?.fuzzy,
+    }),
+    hybridSearch: (collection, query_text, query_vector, limit) => ({
+        type: "HybridSearch",
+        collection,
+        query_text,
+        query_vector,
+        limit,
+    }),
+    chat: (messages, model, temperature, max_tokens) => ({
+        type: "Chat",
+        messages,
+        model,
+        temperature,
+        max_tokens,
+    }),
+    embed: (input_field, output_field, model) => ({
+        type: "Embed",
+        input_field,
+        output_field,
+        model,
+    }),
+};

package/dist/index.d.ts

@@ -1,9 +1,11 @@
-export { EkoDBClient, WebSocketClient, MergeStrategy, RateLimitError, } from "./client";
+export { EkoDBClient, WebSocketClient, SerializationFormat, MergeStrategy, RateLimitError, } from "./client";
 export { QueryBuilder, SortOrder } from "./query-builder";
 export { SearchQueryBuilder } from "./search";
 export { SchemaBuilder, FieldTypeSchemaBuilder, VectorIndexAlgorithm, DistanceMetric, } from "./schema";
 export { JoinBuilder } from "./join";
+export { Stage, ChatMessage } from "./functions";
 export type { SearchQuery, SearchResult, SearchResponse } from "./search";
 export type { Schema, FieldTypeSchema, IndexConfig, CollectionMetadata, } from "./schema";
 export type { JoinConfig } from "./join";
+export type { Script, ParameterDefinition, FunctionStageConfig, GroupFunctionConfig, SortFieldConfig, FunctionResult, FunctionStats, StageStats, } from "./functions";
 export type { Record, Query, BatchOperationResult, ClientConfig, RateLimitInfo, CollectionConfig, ChatRequest, CreateChatSessionRequest, ChatMessageRequest, TokenUsage, ChatResponse, ChatSession, ChatSessionResponse, ListSessionsQuery, ListSessionsResponse, GetMessagesQuery, GetMessagesResponse, UpdateSessionRequest, MergeSessionsRequest, } from "./client";

package/dist/index.js

@@ -1,9 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.JoinBuilder = exports.DistanceMetric = exports.VectorIndexAlgorithm = exports.FieldTypeSchemaBuilder = exports.SchemaBuilder = exports.SearchQueryBuilder = exports.SortOrder = exports.QueryBuilder = exports.RateLimitError = exports.MergeStrategy = exports.WebSocketClient = exports.EkoDBClient = void 0;
+exports.ChatMessage = exports.Stage = exports.JoinBuilder = exports.DistanceMetric = exports.VectorIndexAlgorithm = exports.FieldTypeSchemaBuilder = exports.SchemaBuilder = exports.SearchQueryBuilder = exports.SortOrder = exports.QueryBuilder = exports.RateLimitError = exports.MergeStrategy = exports.SerializationFormat = exports.WebSocketClient = exports.EkoDBClient = void 0;
 var client_1 = require("./client");
 Object.defineProperty(exports, "EkoDBClient", { enumerable: true, get: function () { return client_1.EkoDBClient; } });
 Object.defineProperty(exports, "WebSocketClient", { enumerable: true, get: function () { return client_1.WebSocketClient; } });
+Object.defineProperty(exports, "SerializationFormat", { enumerable: true, get: function () { return client_1.SerializationFormat; } });
 Object.defineProperty(exports, "MergeStrategy", { enumerable: true, get: function () { return client_1.MergeStrategy; } });
 Object.defineProperty(exports, "RateLimitError", { enumerable: true, get: function () { return client_1.RateLimitError; } });
 var query_builder_1 = require("./query-builder");
@@ -18,3 +19,6 @@ Object.defineProperty(exports, "VectorIndexAlgorithm", { enumerable: true, get:
 Object.defineProperty(exports, "DistanceMetric", { enumerable: true, get: function () { return schema_1.DistanceMetric; } });
 var join_1 = require("./join");
 Object.defineProperty(exports, "JoinBuilder", { enumerable: true, get: function () { return join_1.JoinBuilder; } });
+var functions_1 = require("./functions");
+Object.defineProperty(exports, "Stage", { enumerable: true, get: function () { return functions_1.Stage; } });
+Object.defineProperty(exports, "ChatMessage", { enumerable: true, get: function () { return functions_1.ChatMessage; } });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ekodb/ekodb-client",
-  "version": "0.2.1",
+  "version": "0.4.0",
   "description": "Official TypeScript/JavaScript client for ekoDB",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -22,6 +22,7 @@
     "typescript": "^5.3.0"
   },
   "dependencies": {
+    "@msgpack/msgpack": "^3.0.0",
     "ws": "^8.16.0"
   }
 }