macrocosmos 1.2.21 → 1.2.23

package/dist/__tests__/apex/client-chat.test.js

@@ -26,6 +26,12 @@ describe("ApexClient", () => {
         maxNewTokens: 100,
         doSample: true,
     };
+    const createChatAndCompletionParams = {
+        userPrompt: "This is a test chat, how are you?",
+        chatType: "apex",
+        completionType: "basic",
+        title: "Test Chat",
+    };
     let client;
     beforeEach(() => {
         client = new macrocosmos_1.ApexClient({
@@ -78,12 +84,7 @@ describe("ApexClient", () => {
     }, 30000);
     it("should append a chat completion", async () => {
         // create a test chat
-        const create_chat_result = await client.createChatAndCompletion({
-            userPrompt: "This is a test chat, how are you?",
-            chatType: "apex",
-            completionType: "basic",
-            title: "Test Chat",
-        });
+        const create_chat_result = await client.createChatAndCompletion(createChatAndCompletionParams);
         // Verify a chat id exists (i.e., a chat was created)
         console.log("Create chat response:", create_chat_result);
         expect(create_chat_result).toBeDefined();
@@ -114,12 +115,7 @@ describe("ApexClient", () => {
     }, 30000);
     it("should delete a chat", async () => {
         // chat ID for testing
-        const create_chat_result = await client.createChatAndCompletion({
-            userPrompt: "This is a test chat, how are you?",
-            chatType: "apex",
-            completionType: "basic",
-            title: "Test Chat",
-        });
+        const create_chat_result = await client.createChatAndCompletion(createChatAndCompletionParams);
         // Verify the response structure
         console.log("Create chat response:", create_chat_result);
         expect(create_chat_result).toBeDefined();
@@ -133,10 +129,34 @@ describe("ApexClient", () => {
         // Verify the response structure
         console.log("Stored chats:", get_chat_sessions_result);
         expect(get_chat_sessions_result).toBeDefined();
-        // should be an empty array as we just deleted the chat
+        // make sure the chat does not exist in the chats retrieved
         expect(Array.isArray(get_chat_sessions_result.chatSessions)).toBe(true);
         expect(get_chat_sessions_result.chatSessions.some(session => session.id === create_chat_result.parsedChat?.id)).toBe(false);
     }, 30000);
+    it("should delete a completion", async () => {
+        // chat ID for testing
+        const create_completion_result = await client.createChatAndCompletion(createChatAndCompletionParams);
+        console.log("Create completion response:", create_completion_result);
+        expect(create_completion_result).toBeDefined();
+        // Delete test completion
+        const delete_completion_result = await client.deleteCompletions({
+            completionIds: [create_completion_result.parsedCompletion?.id ?? ""],
+        });
+        expect(delete_completion_result).toBeDefined();
+        expect(delete_completion_result.success).toBeTruthy();
+        const get_chat_completions_result = await client.getStoredChatCompletions({
+            chatId: create_completion_result.parsedChat?.id ?? "",
+        });
+        console.log("Stored completions:", get_chat_completions_result);
+        expect(get_chat_completions_result).toBeDefined();
+        // make sure the completion does not exist in the chats retrieved
+        expect(get_chat_completions_result.chatCompletions.some(completion => completion.id === create_completion_result.parsedCompletion?.id)).toBe(false);
+        // Now delete test chat
+        const delete_chat_result = await client.deleteChats({
+            chatIds: [create_completion_result.parsedChat?.id ?? ""],
+        });
+        expect(delete_chat_result.success).toBeTruthy();
+    }, 30000);
     // Deep Researcher Tests
     it("should create a deep research job", async () => {
         // Create test parameters
@@ -192,12 +212,7 @@ describe("ApexClient", () => {
         expect(Object.keys(result.chatIds[0] || {}).length).toBe(0);
     }, 30000);
     it("should create a chat and completion for a user", async () => {
-        const result = await client.createChatAndCompletion({
-            userPrompt: "This is a test chat, how are you?",
-            chatType: "apex",
-            completionType: "basic",
-            title: "Test Chat",
-        });
+        const result = await client.createChatAndCompletion(createChatAndCompletionParams);
         // Verify the response structure
         console.log("Create chat and completion response:", result);
         expect(result).toBeDefined();
@@ -229,4 +244,25 @@ describe("ApexClient", () => {
         expect(result).toBeDefined();
         expect(Array.isArray(result.chatSessions)).toBe(true);
     }, 30000);
+    it("should update chat attribute", async () => {
+        // chat ID for testing
+        const create_chat_result = await client.createChatAndCompletion(createChatAndCompletionParams);
+        // Verify the response structure
+        console.log("Create chat response:", create_chat_result);
+        expect(create_chat_result).toBeDefined();
+        const update_chat_attributes = await client.updateChatAttributes({
+            chatId: create_chat_result.parsedChat?.id ?? "",
+            attributes: {
+                title: "Updated Test Chat",
+                chat_type: "gravity",
+            },
+        });
+        expect(update_chat_attributes.chat?.title).toBe("Updated Test Chat");
+        expect(update_chat_attributes.chat?.chatType).toBe("gravity");
+        // Delete test chat
+        const delete_chat_result = await client.deleteChats({
+            chatIds: [create_chat_result.parsedChat?.id ?? ""],
+        });
+        expect(delete_chat_result.success).toBeTruthy();
+    }, 30000);
 });

package/dist/__tests__/apex/client-web-search.test.js

@@ -34,5 +34,5 @@ describe("ApexClient WebRetrieval", () => {
         }
         // Expect to have at least one result
         expect(response.results?.length).toBeGreaterThan(0);
-    }, 30000); // Increase timeout to 30 seconds
+    }, 45000); // Increase timeout to 45 seconds
 });

package/dist/lib/apex/Client.d.ts

@@ -1,4 +1,4 @@
-import { ApexServiceClient, ChatCompletionRequest as GeneratedChatCompletionRequest, ChatCompletionResponse, ChatCompletionChunkResponse, WebRetrievalRequest as GeneratedWebRetrievalRequest, WebRetrievalResponse, ChatMessage, SubmitDeepResearcherJobResponse, GetDeepResearcherJobRequest, GetDeepResearcherJobResponse, GetStoredChatCompletionsRequest, GetStoredChatCompletionsResponse, GetChatSessionsResponse, SearchChatIdsByPromptAndCompletionTextRequest, SearchChatIdsByPromptAndCompletionTextResponse, CreateChatAndCompletionRequest, CreateChatAndCompletionResponse, CreateCompletionRequest, CreateCompletionResponse, DeleteChatsRequest, DeleteChatsResponse } from "../../generated/apex/v1/apex";
+import { ApexServiceClient, ChatCompletionRequest as GeneratedChatCompletionRequest, ChatCompletionResponse, ChatCompletionChunkResponse, WebRetrievalRequest as GeneratedWebRetrievalRequest, WebRetrievalResponse, ChatMessage, SubmitDeepResearcherJobResponse, GetDeepResearcherJobRequest, GetDeepResearcherJobResponse, GetStoredChatCompletionsRequest, GetStoredChatCompletionsResponse, GetChatSessionsResponse, SearchChatIdsByPromptAndCompletionTextRequest, SearchChatIdsByPromptAndCompletionTextResponse, CreateChatAndCompletionRequest, CreateChatAndCompletionResponse, CreateCompletionRequest, CreateCompletionResponse, DeleteChatsRequest, DeleteChatsResponse, DeleteCompletionsRequest, DeleteCompletionsResponse, UpdateChatAttributesRequest, UpdateChatAttributesResponse } from "../../generated/apex/v1/apex";
 import * as grpc from "@grpc/grpc-js";
 import { BaseClient, BaseClientOptions } from "../BaseClient";
 import { ApexStream } from "./Stream";
@@ -85,4 +85,12 @@ export declare class ApexClient extends BaseClient {
      * Delete a chat given its id
      */
     deleteChats: (params: DeleteChatsRequest) => Promise<DeleteChatsResponse>;
+    /**
+     * Delete completions given their id
+     */
+    deleteCompletions: (params: DeleteCompletionsRequest) => Promise<DeleteCompletionsResponse>;
+    /**
+     * Update chat attributes
+     */
+    updateChatAttributes: (params: UpdateChatAttributesRequest) => Promise<UpdateChatAttributesResponse>;
 }

package/dist/lib/apex/Client.js

@@ -231,6 +231,36 @@ class ApexClient extends BaseClient_1.BaseClient {
                 });
             });
         };
+        /**
+         * Delete completions given their id
+         */
+        this.deleteCompletions = async (params) => {
+            const client = this.createGrpcClient();
+            return new Promise((resolve, reject) => {
+                client.deleteCompletions(params, (error, response) => {
+                    if (error) {
+                        reject(error);
+                        return;
+                    }
+                    resolve(response);
+                });
+            });
+        };
+        /**
+         * Update chat attributes
+         */
+        this.updateChatAttributes = async (params) => {
+            const client = this.createGrpcClient();
+            return new Promise((resolve, reject) => {
+                client.updateChatAttributes(params, (error, response) => {
+                    if (error) {
+                        reject(error);
+                        return;
+                    }
+                    resolve(response);
+                });
+            });
+        };
         this.defaultTimeout = options.timeout || 60;
         this._grpcClient = grpcClient;
     }

package/dist/version.d.ts

@@ -1 +1 @@
-export declare const VERSION = "1.2.21";
+export declare const VERSION = "1.2.23";

package/dist/version.js

@@ -1,4 +1,4 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.VERSION = void 0;
-exports.VERSION = "1.2.21";
+exports.VERSION = "1.2.23";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "macrocosmos",
-  "version": "1.2.21",
+  "version": "1.2.23",
   "description": "TypeScript SDK package for Macrocosmos",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",