@ekodb/ekodb-client 0.10.0 → 0.12.0

package/README.md

@@ -299,6 +299,28 @@ const joinResults = await client.find("users", multiQuery);
 
 - `listCollections(): Promise<string[]>`
 - `deleteCollection(collection: string): Promise<void>`
+- `collectionExists(collection: string): Promise<boolean>` - Check if collection
+  exists
+- `countDocuments(collection: string): Promise<number>` - Count documents in
+  collection
+
+#### Chat Models
+
+- `getChatModels(): Promise<Record<string, string[]>>` - Get all available chat
+  models by provider
+- `getChatModel(provider: string): Promise<string[]>` - Get models for a
+  specific provider
+
+#### User Functions
+
+- `saveUserFunction(userFunction: object): Promise<string>` - Create a new user
+  function
+- `getUserFunction(label: string): Promise<object>` - Get user function by label
+- `listUserFunctions(tags?: string[]): Promise<object[]>` - List all user
+  functions (optionally filter by tags)
+- `updateUserFunction(label: string, userFunction: object): Promise<void>` -
+  Update existing user function
+- `deleteUserFunction(label: string): Promise<void>` - Delete user function
 
 #### WebSocket
 
@@ -322,6 +344,8 @@ for complete working examples:
 - `client_joins.ts` - Join operations
 - `client_batch_operations.ts` - Batch operations
 - `client_kv_operations.ts` - Key-value operations
+- `client_chat_models.ts` - Chat models API
+- `client_user_functions.ts` - User functions API
 - And more...
 
 ## License

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,77 @@ 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
+ */
+export interface ChatModels {
+    openai: string[];
+    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
+ */
+export interface UserFunction {
+    label: string;
+    name: string;
+    description?: string;
+    version?: string;
+    parameters: {
+        [key: string]: ParameterDefinition;
+    };
+    functions: FunctionStageConfig[];
+    tags?: string[];
+    id?: string;
+    created_at?: string;
+    updated_at?: string;
+}
+/**
+ * Parameter definition for functions
+ */
+export interface ParameterDefinition {
+    required: boolean;
+    default?: any;
+    description?: string;
+}
+/**
+ * Function stage configuration for pipelines
+ */
+export interface FunctionStageConfig {
+    type: string;
+    [key: string]: any;
 }
 export declare class EkoDBClient {
     private baseURL;
@@ -635,6 +713,24 @@ export declare class EkoDBClient {
      * Merge multiple chat sessions into one
      */
     mergeChatSessions(request: MergeSessionsRequest): Promise<ChatSessionResponse>;
+    /**
+     * Get all available chat models from all providers
+     * @returns ChatModels object with models organized by provider
+     */
+    getChatModels(): Promise<ChatModels>;
+    /**
+     * Get available models for a specific provider
+     * @param provider - Provider name (e.g., "openai", "anthropic", "perplexity")
+     * @returns Array of model names for the provider
+     */
+    getChatModel(provider: string): Promise<string[]>;
+    /**
+     * Get a specific chat message by ID
+     * @param sessionId - Chat session ID
+     * @param messageId - Message ID
+     * @returns The chat message record
+     */
+    getChatMessage(sessionId: string, messageId: string): Promise<Record>;
     /**
      * Save a new script definition
      */
@@ -661,18 +757,53 @@ export declare class EkoDBClient {
     callScript(idOrLabel: string, params?: {
         [key: string]: any;
     }): Promise<FunctionResult>;
+    /**
+     * Save a new user function
+     * @param userFunction - The user function definition
+     * @returns The ID of the created user function
+     */
+    saveUserFunction(userFunction: UserFunction): Promise<string>;
+    /**
+     * Get a user function by label
+     * @param label - The user function label
+     * @returns The user function definition
+     */
+    getUserFunction(label: string): Promise<UserFunction>;
+    /**
+     * List all user functions, optionally filtered by tags
+     * @param tags - Optional array of tags to filter by
+     * @returns Array of user functions
+     */
+    listUserFunctions(tags?: string[]): Promise<UserFunction[]>;
+    /**
+     * Update an existing user function by label
+     * @param label - The user function label
+     * @param userFunction - The updated user function definition
+     */
+    updateUserFunction(label: string, userFunction: UserFunction): Promise<void>;
+    /**
+     * Delete a user function by label
+     * @param label - The user function label
+     */
+    deleteUserFunction(label: string): Promise<void>;
+    /**
+     * Check if a collection exists
+     * @param collection - Collection name to check
+     * @returns true if the collection exists, false otherwise
+     */
+    collectionExists(collection: string): Promise<boolean>;
+    /**
+     * Count documents in a collection
+     * @param collection - Collection name
+     * @returns Number of documents in the collection
+     */
+    countDocuments(collection: string): Promise<number>;
     /**
      * Create a WebSocket client
      */
     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")
@@ -688,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) {
@@ -882,6 +883,30 @@ class EkoDBClient {
     async mergeChatSessions(request) {
         return this.makeRequest("POST", "/api/chat/merge", request, 0, true);
     }
+    /**
+     * Get all available chat models from all providers
+     * @returns ChatModels object with models organized by provider
+     */
+    async getChatModels() {
+        return this.makeRequest("GET", "/api/chat_models", undefined, 0, true);
+    }
+    /**
+     * Get available models for a specific provider
+     * @param provider - Provider name (e.g., "openai", "anthropic", "perplexity")
+     * @returns Array of model names for the provider
+     */
+    async getChatModel(provider) {
+        return this.makeRequest("GET", `/api/chat_models/${encodeURIComponent(provider)}`, undefined, 0, true);
+    }
+    /**
+     * Get a specific chat message by ID
+     * @param sessionId - Chat session ID
+     * @param messageId - Message ID
+     * @returns The chat message record
+     */
+    async getChatMessage(sessionId, messageId) {
+        return this.makeRequest("GET", `/api/chat/${sessionId}/messages/${messageId}`, undefined, 0, true);
+    }
     // ========================================================================
     // SCRIPTS API
     // ========================================================================
@@ -923,6 +948,77 @@ class EkoDBClient {
     async callScript(idOrLabel, params) {
         return this.makeRequest("POST", `/api/functions/${idOrLabel}`, params || {});
     }
+    // ========================================================================
+    // USER FUNCTIONS API
+    // ========================================================================
+    /**
+     * Save a new user function
+     * @param userFunction - The user function definition
+     * @returns The ID of the created user function
+     */
+    async saveUserFunction(userFunction) {
+        const result = await this.makeRequest("POST", "/api/functions", userFunction, 0, true);
+        return result.id;
+    }
+    /**
+     * Get a user function by label
+     * @param label - The user function label
+     * @returns The user function definition
+     */
+    async getUserFunction(label) {
+        return this.makeRequest("GET", `/api/functions/${encodeURIComponent(label)}`, undefined, 0, true);
+    }
+    /**
+     * List all user functions, optionally filtered by tags
+     * @param tags - Optional array of tags to filter by
+     * @returns Array of user functions
+     */
+    async listUserFunctions(tags) {
+        const params = tags ? `?tags=${tags.join(",")}` : "";
+        return this.makeRequest("GET", `/api/functions${params}`, undefined, 0, true);
+    }
+    /**
+     * Update an existing user function by label
+     * @param label - The user function label
+     * @param userFunction - The updated user function definition
+     */
+    async updateUserFunction(label, userFunction) {
+        await this.makeRequest("PUT", `/api/functions/${encodeURIComponent(label)}`, userFunction, 0, true);
+    }
+    /**
+     * Delete a user function by label
+     * @param label - The user function label
+     */
+    async deleteUserFunction(label) {
+        await this.makeRequest("DELETE", `/api/functions/${encodeURIComponent(label)}`, undefined, 0, true);
+    }
+    // ========================================================================
+    // COLLECTION UTILITIES
+    // ========================================================================
+    /**
+     * Check if a collection exists
+     * @param collection - Collection name to check
+     * @returns true if the collection exists, false otherwise
+     */
+    async collectionExists(collection) {
+        try {
+            const collections = await this.listCollections();
+            return collections.includes(collection);
+        }
+        catch {
+            return false;
+        }
+    }
+    /**
+     * Count documents in a collection
+     * @param collection - Collection name
+     * @returns Number of documents in the collection
+     */
+    async countDocuments(collection) {
+        const query = new query_builder_1.QueryBuilder().limit(100000).build();
+        const records = await this.find(collection, query);
+        return records.length;
+    }
     /**
      * Create a WebSocket client
      */
@@ -931,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")
@@ -953,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/client.test.js

@@ -765,3 +765,154 @@ function mockErrorResponse(status, message) {
         });
     });
 });
+// ============================================================================
+// Chat Models Tests
+// ============================================================================
+(0, vitest_1.describe)("EkoDBClient chat models", () => {
+    (0, vitest_1.it)("gets all chat models", async () => {
+        const client = createTestClient();
+        mockTokenResponse();
+        mockJsonResponse({
+            openai: ["gpt-4", "gpt-3.5-turbo"],
+            anthropic: ["claude-3-opus", "claude-3-sonnet"],
+            perplexity: ["llama-3.1-sonar-small"],
+        });
+        const result = await client.getChatModels();
+        (0, vitest_1.expect)(result.openai).toHaveLength(2);
+        (0, vitest_1.expect)(result.anthropic).toHaveLength(2);
+        (0, vitest_1.expect)(result.perplexity).toHaveLength(1);
+    });
+    (0, vitest_1.it)("gets models for specific provider", async () => {
+        const client = createTestClient();
+        mockTokenResponse();
+        mockJsonResponse(["gpt-4", "gpt-3.5-turbo", "gpt-4-turbo"]);
+        const result = await client.getChatModel("openai");
+        (0, vitest_1.expect)(result).toHaveLength(3);
+        (0, vitest_1.expect)(result).toContain("gpt-4");
+    });
+    (0, vitest_1.it)("gets specific chat message", async () => {
+        const client = createTestClient();
+        mockTokenResponse();
+        mockJsonResponse({
+            id: "msg_123",
+            role: "user",
+            content: "Hello there",
+            created_at: "2024-01-01T00:00:00Z",
+        });
+        const result = await client.getChatMessage("chat_123", "msg_123");
+        (0, vitest_1.expect)(result).toHaveProperty("id", "msg_123");
+        (0, vitest_1.expect)(result).toHaveProperty("content", "Hello there");
+    });
+});
+// ============================================================================
+// User Functions Tests
+// ============================================================================
+(0, vitest_1.describe)("EkoDBClient user functions", () => {
+    (0, vitest_1.it)("saves user function", async () => {
+        const client = createTestClient();
+        mockTokenResponse();
+        mockJsonResponse({ id: "uf_123" });
+        const userFunction = {
+            label: "my_function",
+            name: "My Function",
+            parameters: {},
+            functions: [],
+        };
+        const result = await client.saveUserFunction(userFunction);
+        (0, vitest_1.expect)(result).toBe("uf_123");
+    });
+    (0, vitest_1.it)("gets user function by label", async () => {
+        const client = createTestClient();
+        mockTokenResponse();
+        mockJsonResponse({
+            label: "my_function",
+            name: "My Function",
+            parameters: {},
+            functions: [],
+            id: "uf_123",
+        });
+        const result = await client.getUserFunction("my_function");
+        (0, vitest_1.expect)(result).toHaveProperty("label", "my_function");
+        (0, vitest_1.expect)(result).toHaveProperty("id", "uf_123");
+    });
+    (0, vitest_1.it)("lists user functions", async () => {
+        const client = createTestClient();
+        mockTokenResponse();
+        mockJsonResponse([
+            { label: "func_1", name: "Function 1" },
+            { label: "func_2", name: "Function 2" },
+        ]);
+        const result = await client.listUserFunctions();
+        (0, vitest_1.expect)(result).toHaveLength(2);
+        (0, vitest_1.expect)(result[0]).toHaveProperty("label", "func_1");
+    });
+    (0, vitest_1.it)("lists user functions filtered by tags", async () => {
+        const client = createTestClient();
+        mockTokenResponse();
+        mockJsonResponse([{ label: "func_1", name: "Function 1", tags: ["data"] }]);
+        const result = await client.listUserFunctions(["data"]);
+        (0, vitest_1.expect)(result).toHaveLength(1);
+    });
+    (0, vitest_1.it)("updates user function", async () => {
+        const client = createTestClient();
+        mockTokenResponse();
+        mockJsonResponse({ status: "updated" });
+        const userFunction = {
+            label: "my_function",
+            name: "Updated Function",
+            parameters: {},
+            functions: [],
+        };
+        await (0, vitest_1.expect)(client.updateUserFunction("my_function", userFunction)).resolves.not.toThrow();
+    });
+    (0, vitest_1.it)("deletes user function", async () => {
+        const client = createTestClient();
+        mockTokenResponse();
+        mockJsonResponse({ status: "deleted" });
+        await (0, vitest_1.expect)(client.deleteUserFunction("my_function")).resolves.not.toThrow();
+    });
+});
+// ============================================================================
+// Collection Utility Tests
+// ============================================================================
+(0, vitest_1.describe)("EkoDBClient collection utilities", () => {
+    (0, vitest_1.it)("collectionExists returns true for existing collection", async () => {
+        const client = createTestClient();
+        mockTokenResponse();
+        mockJsonResponse({ collections: ["users", "posts", "comments"] });
+        const result = await client.collectionExists("users");
+        (0, vitest_1.expect)(result).toBe(true);
+    });
+    (0, vitest_1.it)("collectionExists returns false for non-existing collection", async () => {
+        const client = createTestClient();
+        mockTokenResponse();
+        mockJsonResponse({ collections: ["users", "posts", "comments"] });
+        const result = await client.collectionExists("nonexistent");
+        (0, vitest_1.expect)(result).toBe(false);
+    });
+    (0, vitest_1.it)("collectionExists returns false on error", async () => {
+        const client = createTestClient();
+        mockTokenResponse();
+        mockErrorResponse(500, "Server error");
+        const result = await client.collectionExists("users");
+        (0, vitest_1.expect)(result).toBe(false);
+    });
+    (0, vitest_1.it)("countDocuments returns correct count", async () => {
+        const client = createTestClient();
+        mockTokenResponse();
+        mockJsonResponse([
+            { id: "1", name: "A" },
+            { id: "2", name: "B" },
+            { id: "3", name: "C" },
+        ]);
+        const result = await client.countDocuments("users");
+        (0, vitest_1.expect)(result).toBe(3);
+    });
+    (0, vitest_1.it)("countDocuments returns zero for empty collection", async () => {
+        const client = createTestClient();
+        mockTokenResponse();
+        mockJsonResponse([]);
+        const result = await client.countDocuments("empty_collection");
+        (0, vitest_1.expect)(result).toBe(0);
+    });
+});

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, } 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.10.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.test.ts

@@ -1031,3 +1031,215 @@ describe("Convenience methods", () => {
     });
   });
 });
+
+// ============================================================================
+// Chat Models Tests
+// ============================================================================
+
+describe("EkoDBClient chat models", () => {
+  it("gets all chat models", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse({
+      openai: ["gpt-4", "gpt-3.5-turbo"],
+      anthropic: ["claude-3-opus", "claude-3-sonnet"],
+      perplexity: ["llama-3.1-sonar-small"],
+    });
+
+    const result = await client.getChatModels();
+
+    expect(result.openai).toHaveLength(2);
+    expect(result.anthropic).toHaveLength(2);
+    expect(result.perplexity).toHaveLength(1);
+  });
+
+  it("gets models for specific provider", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse(["gpt-4", "gpt-3.5-turbo", "gpt-4-turbo"]);
+
+    const result = await client.getChatModel("openai");
+
+    expect(result).toHaveLength(3);
+    expect(result).toContain("gpt-4");
+  });
+
+  it("gets specific chat message", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse({
+      id: "msg_123",
+      role: "user",
+      content: "Hello there",
+      created_at: "2024-01-01T00:00:00Z",
+    });
+
+    const result = await client.getChatMessage("chat_123", "msg_123");
+
+    expect(result).toHaveProperty("id", "msg_123");
+    expect(result).toHaveProperty("content", "Hello there");
+  });
+});
+
+// ============================================================================
+// User Functions Tests
+// ============================================================================
+
+describe("EkoDBClient user functions", () => {
+  it("saves user function", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse({ id: "uf_123" });
+
+    const userFunction = {
+      label: "my_function",
+      name: "My Function",
+      parameters: {},
+      functions: [],
+    };
+    const result = await client.saveUserFunction(userFunction);
+
+    expect(result).toBe("uf_123");
+  });
+
+  it("gets user function by label", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse({
+      label: "my_function",
+      name: "My Function",
+      parameters: {},
+      functions: [],
+      id: "uf_123",
+    });
+
+    const result = await client.getUserFunction("my_function");
+
+    expect(result).toHaveProperty("label", "my_function");
+    expect(result).toHaveProperty("id", "uf_123");
+  });
+
+  it("lists user functions", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse([
+      { label: "func_1", name: "Function 1" },
+      { label: "func_2", name: "Function 2" },
+    ]);
+
+    const result = await client.listUserFunctions();
+
+    expect(result).toHaveLength(2);
+    expect(result[0]).toHaveProperty("label", "func_1");
+  });
+
+  it("lists user functions filtered by tags", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse([{ label: "func_1", name: "Function 1", tags: ["data"] }]);
+
+    const result = await client.listUserFunctions(["data"]);
+
+    expect(result).toHaveLength(1);
+  });
+
+  it("updates user function", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse({ status: "updated" });
+
+    const userFunction = {
+      label: "my_function",
+      name: "Updated Function",
+      parameters: {},
+      functions: [],
+    };
+    await expect(
+      client.updateUserFunction("my_function", userFunction),
+    ).resolves.not.toThrow();
+  });
+
+  it("deletes user function", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse({ status: "deleted" });
+
+    await expect(
+      client.deleteUserFunction("my_function"),
+    ).resolves.not.toThrow();
+  });
+});
+
+// ============================================================================
+// Collection Utility Tests
+// ============================================================================
+
+describe("EkoDBClient collection utilities", () => {
+  it("collectionExists returns true for existing collection", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse({ collections: ["users", "posts", "comments"] });
+
+    const result = await client.collectionExists("users");
+
+    expect(result).toBe(true);
+  });
+
+  it("collectionExists returns false for non-existing collection", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse({ collections: ["users", "posts", "comments"] });
+
+    const result = await client.collectionExists("nonexistent");
+
+    expect(result).toBe(false);
+  });
+
+  it("collectionExists returns false on error", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockErrorResponse(500, "Server error");
+
+    const result = await client.collectionExists("users");
+
+    expect(result).toBe(false);
+  });
+
+  it("countDocuments returns correct count", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse([
+      { id: "1", name: "A" },
+      { id: "2", name: "B" },
+      { id: "3", name: "C" },
+    ]);
+
+    const result = await client.countDocuments("users");
+
+    expect(result).toBe(3);
+  });
+
+  it("countDocuments returns zero for empty collection", async () => {
+    const client = createTestClient();
+
+    mockTokenResponse();
+    mockJsonResponse([]);
+
+    const result = await client.countDocuments("empty_collection");
+
+    expect(result).toBe(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,83 @@ 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;
+}
+
+/**
+ * Available chat models by provider
+ */
+export interface ChatModels {
+  openai: string[];
+  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
+ */
+export interface UserFunction {
+  label: string;
+  name: string;
+  description?: string;
+  version?: string;
+  parameters: { [key: string]: ParameterDefinition };
+  functions: FunctionStageConfig[];
+  tags?: string[];
+  id?: string;
+  created_at?: string;
+  updated_at?: string;
+}
+
+/**
+ * Parameter definition for functions
+ */
+export interface ParameterDefinition {
+  required: boolean;
+  default?: any;
+  description?: string;
+}
+
+/**
+ * Function stage configuration for pipelines
+ */
+export interface FunctionStageConfig {
+  type: string;
+  [key: string]: any;
 }
 
 export class EkoDBClient {
@@ -1462,6 +1546,51 @@ export class EkoDBClient {
     );
   }
 
+  /**
+   * Get all available chat models from all providers
+   * @returns ChatModels object with models organized by provider
+   */
+  async getChatModels(): Promise<ChatModels> {
+    return this.makeRequest<ChatModels>(
+      "GET",
+      "/api/chat_models",
+      undefined,
+      0,
+      true, // Force JSON for chat operations
+    );
+  }
+
+  /**
+   * Get available models for a specific provider
+   * @param provider - Provider name (e.g., "openai", "anthropic", "perplexity")
+   * @returns Array of model names for the provider
+   */
+  async getChatModel(provider: string): Promise<string[]> {
+    return this.makeRequest<string[]>(
+      "GET",
+      `/api/chat_models/${encodeURIComponent(provider)}`,
+      undefined,
+      0,
+      true, // Force JSON for chat operations
+    );
+  }
+
+  /**
+   * Get a specific chat message by ID
+   * @param sessionId - Chat session ID
+   * @param messageId - Message ID
+   * @returns The chat message record
+   */
+  async getChatMessage(sessionId: string, messageId: string): Promise<Record> {
+    return this.makeRequest<Record>(
+      "GET",
+      `/api/chat/${sessionId}/messages/${messageId}`,
+      undefined,
+      0,
+      true, // Force JSON for chat operations
+    );
+  }
+
   // ========================================================================
   // SCRIPTS API
   // ========================================================================
@@ -1521,6 +1650,118 @@ export class EkoDBClient {
     );
   }
 
+  // ========================================================================
+  // USER FUNCTIONS API
+  // ========================================================================
+
+  /**
+   * Save a new user function
+   * @param userFunction - The user function definition
+   * @returns The ID of the created user function
+   */
+  async saveUserFunction(userFunction: UserFunction): Promise<string> {
+    const result = await this.makeRequest<{ id: string }>(
+      "POST",
+      "/api/functions",
+      userFunction,
+      0,
+      true, // Force JSON
+    );
+    return result.id;
+  }
+
+  /**
+   * Get a user function by label
+   * @param label - The user function label
+   * @returns The user function definition
+   */
+  async getUserFunction(label: string): Promise<UserFunction> {
+    return this.makeRequest<UserFunction>(
+      "GET",
+      `/api/functions/${encodeURIComponent(label)}`,
+      undefined,
+      0,
+      true, // Force JSON
+    );
+  }
+
+  /**
+   * List all user functions, optionally filtered by tags
+   * @param tags - Optional array of tags to filter by
+   * @returns Array of user functions
+   */
+  async listUserFunctions(tags?: string[]): Promise<UserFunction[]> {
+    const params = tags ? `?tags=${tags.join(",")}` : "";
+    return this.makeRequest<UserFunction[]>(
+      "GET",
+      `/api/functions${params}`,
+      undefined,
+      0,
+      true, // Force JSON
+    );
+  }
+
+  /**
+   * Update an existing user function by label
+   * @param label - The user function label
+   * @param userFunction - The updated user function definition
+   */
+  async updateUserFunction(
+    label: string,
+    userFunction: UserFunction,
+  ): Promise<void> {
+    await this.makeRequest<void>(
+      "PUT",
+      `/api/functions/${encodeURIComponent(label)}`,
+      userFunction,
+      0,
+      true, // Force JSON
+    );
+  }
+
+  /**
+   * Delete a user function by label
+   * @param label - The user function label
+   */
+  async deleteUserFunction(label: string): Promise<void> {
+    await this.makeRequest<void>(
+      "DELETE",
+      `/api/functions/${encodeURIComponent(label)}`,
+      undefined,
+      0,
+      true, // Force JSON
+    );
+  }
+
+  // ========================================================================
+  // COLLECTION UTILITIES
+  // ========================================================================
+
+  /**
+   * Check if a collection exists
+   * @param collection - Collection name to check
+   * @returns true if the collection exists, false otherwise
+   */
+  async collectionExists(collection: string): Promise<boolean> {
+    try {
+      const collections = await this.listCollections();
+      return collections.includes(collection);
+    } catch {
+      return false;
+    }
+  }
+
+  /**
+   * Count documents in a collection
+   * @param collection - Collection name
+   * @returns Number of documents in the collection
+   */
+  async countDocuments(collection: string): Promise<number> {
+    const query = new QueryBuilder().limit(100000).build();
+    const records = await this.find(collection, query);
+    return records.length;
+  }
+
   /**
    * Create a WebSocket client
    */
@@ -1531,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")
@@ -1553,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

@@ -70,4 +70,10 @@ export type {
   GetMessagesResponse,
   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/**"],
+  },
+});