@ekodb/ekodb-client 0.11.0 → 0.12.0

package/dist/client.d.ts

@@ -123,6 +123,18 @@ export interface ChatRequest {
     system_prompt?: string;
     bypass_ripple?: boolean;
 }
+export interface ToolChoice {
+    type: "auto" | "none" | "required" | "tool";
+    name?: string;
+}
+export interface ToolConfig {
+    enabled: boolean;
+    allowed_tools?: string[];
+    allowed_collections?: string[];
+    max_iterations?: number;
+    allow_write_operations?: boolean;
+    tool_choice?: ToolChoice;
+}
 export interface CreateChatSessionRequest {
     collections: CollectionConfig[];
     llm_provider: string;
@@ -132,11 +144,16 @@ export interface CreateChatSessionRequest {
     parent_id?: string;
     branch_point_idx?: number;
     max_context_messages?: number;
+    max_tokens?: number;
+    temperature?: number;
+    tool_config?: ToolConfig;
 }
 export interface ChatMessageRequest {
     message: string;
     bypass_ripple?: boolean;
     force_summarize?: boolean;
+    max_iterations?: number;
+    tool_config?: ToolConfig;
 }
 export interface TokenUsage {
     prompt_tokens: number;
@@ -195,16 +212,21 @@ export interface UpdateSessionRequest {
     llm_model?: string;
     collections?: CollectionConfig[];
     max_context_messages?: number;
+    bypass_ripple?: boolean;
+    title?: string;
+    memory?: any;
 }
 export declare enum MergeStrategy {
     Chronological = "Chronological",
     Summarized = "Summarized",
-    LatestOnly = "LatestOnly"
+    LatestOnly = "LatestOnly",
+    Interleaved = "Interleaved"
 }
 export interface MergeSessionsRequest {
     source_chat_ids: string[];
     target_chat_id: string;
     merge_strategy: MergeStrategy;
+    bypass_ripple?: boolean;
 }
 /**
  * Available chat models by provider
@@ -214,6 +236,22 @@ export interface ChatModels {
     anthropic: string[];
     perplexity: string[];
 }
+/**
+ * Request to generate embeddings
+ */
+export interface EmbedRequest {
+    text?: string;
+    texts?: string[];
+    model?: string;
+}
+/**
+ * Response from embedding generation
+ */
+export interface EmbedResponse {
+    embeddings: number[][];
+    model: string;
+    dimensions: number;
+}
 /**
  * User function definition - reusable sequence of Functions that can be called by Scripts
  */
@@ -765,13 +803,7 @@ export declare class EkoDBClient {
      */
     websocket(wsURL: string): WebSocketClient;
     /**
-     * Generate embeddings for text using ekoDB's native Functions
-     *
-     * This helper simplifies embedding generation by:
-     * 1. Creating a temporary collection with the text
-     * 2. Running a Script with FindAll + Embed Functions
-     * 3. Extracting and returning the embedding vector
-     * 4. Cleaning up temporary resources
+     * Generate embeddings for a single text
      *
      * @param text - The text to generate embeddings for
      * @param model - The embedding model to use (e.g., "text-embedding-3-small")
@@ -787,6 +819,18 @@ export declare class EkoDBClient {
      * ```
      */
     embed(text: string, model: string): Promise<number[]>;
+    /**
+     * Generate embeddings for multiple texts in a single batch request
+     *
+     * @param texts - Array of texts to generate embeddings for
+     * @param model - The embedding model to use
+     * @returns Array of embedding vectors
+     */
+    embedBatch(texts: string[], model: string): Promise<number[][]>;
+    /**
+     * Internal: make embed API request
+     */
+    private embedRequest;
     /**
      * Perform text search without embeddings
      *

package/dist/client.js

@@ -66,6 +66,7 @@ var MergeStrategy;
     MergeStrategy["Chronological"] = "Chronological";
     MergeStrategy["Summarized"] = "Summarized";
     MergeStrategy["LatestOnly"] = "LatestOnly";
+    MergeStrategy["Interleaved"] = "Interleaved";
 })(MergeStrategy || (exports.MergeStrategy = MergeStrategy = {}));
 class EkoDBClient {
     constructor(config, apiKey) {
@@ -1026,13 +1027,7 @@ class EkoDBClient {
     }
     // ========== RAG Helper Methods ==========
     /**
-     * Generate embeddings for text using ekoDB's native Functions
-     *
-     * This helper simplifies embedding generation by:
-     * 1. Creating a temporary collection with the text
-     * 2. Running a Script with FindAll + Embed Functions
-     * 3. Extracting and returning the embedding vector
-     * 4. Cleaning up temporary resources
+     * Generate embeddings for a single text
      *
      * @param text - The text to generate embeddings for
      * @param model - The embedding model to use (e.g., "text-embedding-3-small")
@@ -1048,52 +1043,28 @@ class EkoDBClient {
      * ```
      */
     async embed(text, model) {
-        const tempCollection = `embed_temp_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
-        try {
-            // Insert temporary record with the text
-            await this.insert(tempCollection, { text }, undefined);
-            // Create Script with FindAll + Embed Functions
-            const tempLabel = `embed_script_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
-            const script = {
-                label: tempLabel,
-                name: "Generate Embedding",
-                description: "Temporary script for embedding generation",
-                version: "1.0",
-                parameters: {},
-                functions: [
-                    {
-                        type: "FindAll",
-                        collection: tempCollection,
-                    },
-                    {
-                        type: "Embed",
-                        input_field: "text",
-                        output_field: "embedding",
-                        model: model,
-                    },
-                ],
-                tags: [],
-            };
-            // Save and execute the script
-            const scriptId = await this.saveScript(script);
-            const result = await this.callScript(scriptId, undefined);
-            // Clean up
-            await this.deleteScript(scriptId).catch(() => { });
-            await this.deleteCollection(tempCollection).catch(() => { });
-            // Extract embedding from result
-            if (result.records && result.records.length > 0) {
-                const record = result.records[0];
-                if (record.embedding && Array.isArray(record.embedding)) {
-                    return record.embedding;
-                }
-            }
-            throw new Error("Failed to extract embedding from result");
-        }
-        catch (error) {
-            // Ensure cleanup even on error
-            await this.deleteCollection(tempCollection).catch(() => { });
-            throw error;
+        const response = await this.embedRequest({ text, model });
+        if (response.embeddings.length === 0) {
+            throw new Error("No embedding returned");
         }
+        return response.embeddings[0];
+    }
+    /**
+     * Generate embeddings for multiple texts in a single batch request
+     *
+     * @param texts - Array of texts to generate embeddings for
+     * @param model - The embedding model to use
+     * @returns Array of embedding vectors
+     */
+    async embedBatch(texts, model) {
+        const response = await this.embedRequest({ texts, model });
+        return response.embeddings;
+    }
+    /**
+     * Internal: make embed API request
+     */
+    async embedRequest(request) {
+        return this.makeRequest("POST", "/api/embed", request, 0, true);
     }
     /**
      * Perform text search without embeddings

package/dist/index.d.ts

@@ -10,4 +10,4 @@ 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, ChatModels, UserFunction, } from "./client";
+export type { Record, Query, BatchOperationResult, ClientConfig, RateLimitInfo, CollectionConfig, ChatRequest, CreateChatSessionRequest, ChatMessageRequest, TokenUsage, ChatResponse, ChatSession, ChatSessionResponse, ListSessionsQuery, ListSessionsResponse, GetMessagesQuery, GetMessagesResponse, UpdateSessionRequest, MergeSessionsRequest, ChatModels, EmbedRequest, EmbedResponse, UserFunction, ToolChoice, ToolConfig, } from "./client";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ekodb/ekodb-client",
-  "version": "0.11.0",
+  "version": "0.12.0",
   "description": "Official TypeScript/JavaScript client for ekoDB",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -19,13 +19,13 @@
   "author": "ekoDB",
   "license": "MIT",
   "devDependencies": {
-    "@types/node": "^25.0.3",
-    "@types/ws": "^8.5.10",
-    "typescript": "^5.3.0",
-    "vitest": "^3.2.0"
+    "@types/node": "^25.3.5",
+    "@types/ws": "^8.18.1",
+    "typescript": "^5.9.3",
+    "vitest": "^4.0.18"
   },
   "dependencies": {
-    "@msgpack/msgpack": "^3.0.0",
-    "ws": "^8.16.0"
+    "@msgpack/msgpack": "^3.1.3",
+    "ws": "^8.19.0"
   }
 }

package/src/client.ts

@@ -151,6 +151,20 @@ export interface ChatRequest {
   bypass_ripple?: boolean;
 }
 
+export interface ToolChoice {
+  type: "auto" | "none" | "required" | "tool";
+  name?: string;
+}
+
+export interface ToolConfig {
+  enabled: boolean;
+  allowed_tools?: string[];
+  allowed_collections?: string[];
+  max_iterations?: number;
+  allow_write_operations?: boolean;
+  tool_choice?: ToolChoice;
+}
+
 export interface CreateChatSessionRequest {
   collections: CollectionConfig[];
   llm_provider: string;
@@ -160,12 +174,17 @@ export interface CreateChatSessionRequest {
   parent_id?: string;
   branch_point_idx?: number;
   max_context_messages?: number;
+  max_tokens?: number;
+  temperature?: number;
+  tool_config?: ToolConfig;
 }
 
 export interface ChatMessageRequest {
   message: string;
   bypass_ripple?: boolean;
   force_summarize?: boolean;
+  max_iterations?: number;
+  tool_config?: ToolConfig;
 }
 
 export interface TokenUsage {
@@ -233,18 +252,23 @@ export interface UpdateSessionRequest {
   llm_model?: string;
   collections?: CollectionConfig[];
   max_context_messages?: number;
+  bypass_ripple?: boolean;
+  title?: string;
+  memory?: any;
 }
 
 export enum MergeStrategy {
   Chronological = "Chronological",
   Summarized = "Summarized",
   LatestOnly = "LatestOnly",
+  Interleaved = "Interleaved",
 }
 
 export interface MergeSessionsRequest {
   source_chat_ids: string[];
   target_chat_id: string;
   merge_strategy: MergeStrategy;
+  bypass_ripple?: boolean;
 }
 
 /**
@@ -256,6 +280,24 @@ export interface ChatModels {
   perplexity: string[];
 }
 
+/**
+ * Request to generate embeddings
+ */
+export interface EmbedRequest {
+  text?: string;
+  texts?: string[];
+  model?: string;
+}
+
+/**
+ * Response from embedding generation
+ */
+export interface EmbedResponse {
+  embeddings: number[][];
+  model: string;
+  dimensions: number;
+}
+
 /**
  * User function definition - reusable sequence of Functions that can be called by Scripts
  */
@@ -1730,13 +1772,7 @@ export class EkoDBClient {
   // ========== RAG Helper Methods ==========
 
   /**
-   * Generate embeddings for text using ekoDB's native Functions
-   *
-   * This helper simplifies embedding generation by:
-   * 1. Creating a temporary collection with the text
-   * 2. Running a Script with FindAll + Embed Functions
-   * 3. Extracting and returning the embedding vector
-   * 4. Cleaning up temporary resources
+   * Generate embeddings for a single text
    *
    * @param text - The text to generate embeddings for
    * @param model - The embedding model to use (e.g., "text-embedding-3-small")
@@ -1752,57 +1788,36 @@ export class EkoDBClient {
    * ```
    */
   async embed(text: string, model: string): Promise<number[]> {
-    const tempCollection = `embed_temp_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
-
-    try {
-      // Insert temporary record with the text
-      await this.insert(tempCollection, { text }, undefined);
-
-      // Create Script with FindAll + Embed Functions
-      const tempLabel = `embed_script_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
-      const script: Script = {
-        label: tempLabel,
-        name: "Generate Embedding",
-        description: "Temporary script for embedding generation",
-        version: "1.0",
-        parameters: {},
-        functions: [
-          {
-            type: "FindAll",
-            collection: tempCollection,
-          },
-          {
-            type: "Embed",
-            input_field: "text",
-            output_field: "embedding",
-            model: model,
-          } as any,
-        ],
-        tags: [],
-      };
-
-      // Save and execute the script
-      const scriptId = await this.saveScript(script);
-      const result = await this.callScript(scriptId, undefined);
-
-      // Clean up
-      await this.deleteScript(scriptId).catch(() => {});
-      await this.deleteCollection(tempCollection).catch(() => {});
+    const response = await this.embedRequest({ text, model });
+    if (response.embeddings.length === 0) {
+      throw new Error("No embedding returned");
+    }
+    return response.embeddings[0];
+  }
 
-      // Extract embedding from result
-      if (result.records && result.records.length > 0) {
-        const record = result.records[0];
-        if (record.embedding && Array.isArray(record.embedding)) {
-          return record.embedding as number[];
-        }
-      }
+  /**
+   * Generate embeddings for multiple texts in a single batch request
+   *
+   * @param texts - Array of texts to generate embeddings for
+   * @param model - The embedding model to use
+   * @returns Array of embedding vectors
+   */
+  async embedBatch(texts: string[], model: string): Promise<number[][]> {
+    const response = await this.embedRequest({ texts, model });
+    return response.embeddings;
+  }
 
-      throw new Error("Failed to extract embedding from result");
-    } catch (error) {
-      // Ensure cleanup even on error
-      await this.deleteCollection(tempCollection).catch(() => {});
-      throw error;
-    }
+  /**
+   * Internal: make embed API request
+   */
+  private async embedRequest(request: EmbedRequest): Promise<EmbedResponse> {
+    return this.makeRequest<EmbedResponse>(
+      "POST",
+      "/api/embed",
+      request,
+      0,
+      true, // Force JSON
+    );
   }
 
   /**

package/src/index.ts

@@ -71,5 +71,9 @@ export type {
   UpdateSessionRequest,
   MergeSessionsRequest,
   ChatModels,
+  EmbedRequest,
+  EmbedResponse,
   UserFunction,
+  ToolChoice,
+  ToolConfig,
 } from "./client";

package/vitest.config.ts

@@ -0,0 +1,7 @@
+import { defineConfig } from "vitest/config";
+
+export default defineConfig({
+  test: {
+    exclude: ["dist/**", "node_modules/**"],
+  },
+});