macrocosmos 1.2.20 → 1.2.21

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

@@ -76,42 +76,67 @@ describe("ApexClient", () => {
         // chat id doesn't exist so check first element is an empty object
         expect(Object.keys(result.chatCompletions[0] || {}).length).toBe(0);
     }, 30000);
-    // it("should append a chat completion", 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",
-    //   });
-    //
-    //   // Verify the response structure
-    //   console.log("Create chat and completion response:", create_chat_result);
-    //   expect(create_chat_result).toBeDefined();
-    //   // Get stored chat completions
-    //   const result = await client.createCompletion({
-    //     chatId: create_chat_result.parsedChat?.id ?? "",
-    //     userPrompt: "This is a test completion, how are you?",
-    //     completionType: "basic",
-    //   });
-    //
-    //   // Verify the response structure
-    //   console.log("Create completion response:", result);
-    //   expect(result).toBeDefined();
-    //
-    //   // Check ParsedCompletion
-    //   expect(result.parsedCompletion).toBeDefined();
-    //   expect(result.parsedCompletion!.id).toBeDefined();
-    //   expect(result.parsedCompletion!.chatId).toBe(
-    //     create_chat_result.parsedChat!.id,
-    //   );
-    //   expect(result.parsedCompletion!.completionType).toBe("basic");
-    //   expect(result.parsedCompletion!.userPromptText).toBe(
-    //     "This is a test completion, how are you?",
-    //   );
-    //
-    // }, 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",
+        });
+        // Verify a chat id exists (i.e., a chat was created)
+        console.log("Create chat response:", create_chat_result);
+        expect(create_chat_result).toBeDefined();
+        expect(create_chat_result.parsedChat?.id).toBeDefined();
+        // Create Completion under this chat
+        const result = await client.createCompletion({
+            chatId: create_chat_result.parsedChat?.id ?? "",
+            userPrompt: "This is a test completion, how are you?",
+            completionType: "basic",
+        });
+        // Verify the response structure
+        console.log("Create completion response:", result);
+        expect(result).toBeDefined();
+        // Check ParsedCompletion
+        expect(result.parsedCompletion).toBeDefined();
+        expect(result.parsedCompletion.id).toBeDefined();
+        expect(result.parsedCompletion.chatId).toBe(create_chat_result.parsedChat.id);
+        // make sure the completion type is basic
+        expect(result.parsedCompletion.completionType).toBe("basic");
+        // make sure the completion text is what we sent
+        expect(result.parsedCompletion.userPromptText).toBe("This is a test completion, how are you?");
+        // Delete test chat
+        const delete_chat_result = await client.deleteChats({
+            chatIds: [create_chat_result.parsedChat?.id ?? ""],
+        });
+        expect(delete_chat_result).toBeDefined();
+        expect(delete_chat_result.success).toBeTruthy();
+    }, 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",
+        });
+        // Verify the response structure
+        console.log("Create chat response:", create_chat_result);
+        expect(create_chat_result).toBeDefined();
+        // Delete test chat
+        const delete_chat_result = await client.deleteChats({
+            chatIds: [create_chat_result.parsedChat?.id ?? ""],
+        });
+        expect(delete_chat_result).toBeDefined();
+        expect(delete_chat_result.success).toBeTruthy();
+        const get_chat_sessions_result = await client.getChatSessions();
+        // 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
+        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);
     // Deep Researcher Tests
     it("should create a deep research job", async () => {
         // Create test parameters
@@ -153,55 +178,55 @@ describe("ApexClient", () => {
         // Log response for debugging
         console.log("Get Job Results Response:", response);
     }, 60000); // Longer timeout for this test as it involves multiple API calls
-    // TODO: Uncomment this when delete chat is available, so we can remove all test chats first
-    // it("should retrieve a user's completions based on the search term", async () => {
-    //   const searchTerm = "France?";
-    //   // Get stored chat completions
-    //   const result = await client.searchChatIdsByPromptAndCompletionText({
-    //     searchTerm: searchTerm,
-    //   });
-    //   // Verify the response structure
-    //   console.log("Stored chats:", result);
-    //   expect(result).toBeDefined();
-    //   expect(Array.isArray(result.chatIds)).toBe(true);
-    //   // chat id doesn't exist so check first element is an empty object
-    //   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",
-    //   });
-    //   // Verify the response structure
-    //   console.log("Create chat and completion response:", result);
-    //   expect(result).toBeDefined();
-    //   // Check ParsedChat
-    //   expect(result.parsedChat).toBeDefined();
-    //   expect(result.parsedChat!.id).toBeDefined();
-    //   expect(result.parsedChat!.title).toBe("Test Chat");
-    //   expect(result.parsedChat!.chatType).toBe("apex");
-    //   expect(result.parsedChat!.createdAt).toBeDefined();
-    //   // Check ParsedCompletion
-    //   expect(result.parsedCompletion).toBeDefined();
-    //   expect(result.parsedCompletion!.id).toBeDefined();
-    //   expect(result.parsedCompletion!.chatId).toBe(result.parsedChat!.id);
-    //   expect(result.parsedCompletion!.completionType).toBe("basic");
-    //   expect(result.parsedCompletion!.userPromptText).toBe(
-    //     "This is a test chat, how are you?",
-    //   );
-    //   expect(result.parsedCompletion!.completionText).toBeDefined();
-    //   expect(result.parsedCompletion!.createdAt).toBeDefined();
-    // }, 30000);
-    // it("should retrieve a user's stored chats", async () => {
-    //   // Get stored chat completions
-    //   const result = await client.getChatSessions();
-    //   // Verify the response structure
-    //   console.log("Stored chats:", result);
-    //   expect(result).toBeDefined();
-    //   expect(Array.isArray(result.chatSessions)).toBe(true);
-    //   // chat id doesn't exist so check first element is an empty object
-    //   expect(Object.keys(result.chatSessions[0] || {}).length).toBe(0);
-    // }, 30000);
+    it("should retrieve a user's completions based on the search term", async () => {
+        const searchTerm = "France?";
+        // Get stored chat completions
+        const result = await client.searchChatIdsByPromptAndCompletionText({
+            searchTerm: searchTerm,
+        });
+        // Verify the response structure
+        console.log("Stored chats:", result);
+        expect(result).toBeDefined();
+        expect(Array.isArray(result.chatIds)).toBe(true);
+        // chat id doesn't exist so check first element is an empty object
+        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",
+        });
+        // Verify the response structure
+        console.log("Create chat and completion response:", result);
+        expect(result).toBeDefined();
+        // Check ParsedChat
+        expect(result.parsedChat).toBeDefined();
+        expect(result.parsedChat.id).toBeDefined();
+        expect(result.parsedChat.title).toBe("Test Chat");
+        expect(result.parsedChat.chatType).toBe("apex");
+        expect(result.parsedChat.createdAt).toBeDefined();
+        // Check ParsedCompletion
+        expect(result.parsedCompletion).toBeDefined();
+        expect(result.parsedCompletion.id).toBeDefined();
+        expect(result.parsedCompletion.chatId).toBe(result.parsedChat.id);
+        expect(result.parsedCompletion.completionType).toBe("basic");
+        expect(result.parsedCompletion.userPromptText).toBe("This is a test chat, how are you?");
+        expect(result.parsedCompletion.completionText).toBeDefined();
+        expect(result.parsedCompletion.createdAt).toBeDefined();
+        // Delete test chat
+        const delete_chat_result = await client.deleteChats({
+            chatIds: [result.parsedChat?.id ?? ""],
+        });
+        expect(delete_chat_result.success).toBeTruthy();
+    }, 30000);
+    it("should retrieve a user's stored chats", async () => {
+        // Get stored chat completions
+        const result = await client.getChatSessions();
+        // Verify the response structure
+        console.log("Stored chats:", result);
+        expect(result).toBeDefined();
+        expect(Array.isArray(result.chatSessions)).toBe(true);
+    }, 30000);
 });

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 } 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 } from "../../generated/apex/v1/apex";
 import * as grpc from "@grpc/grpc-js";
 import { BaseClient, BaseClientOptions } from "../BaseClient";
 import { ApexStream } from "./Stream";
@@ -81,4 +81,8 @@ export declare class ApexClient extends BaseClient {
      * Create completion for a chat
      */
     createCompletion: (params: CreateCompletionRequest) => Promise<CreateCompletionResponse>;
+    /**
+     * Delete a chat given its id
+     */
+    deleteChats: (params: DeleteChatsRequest) => Promise<DeleteChatsResponse>;
 }

package/dist/lib/apex/Client.js

@@ -216,6 +216,21 @@ class ApexClient extends BaseClient_1.BaseClient {
                 });
             });
         };
+        /**
+         * Delete a chat given its id
+         */
+        this.deleteChats = async (params) => {
+            const client = this.createGrpcClient();
+            return new Promise((resolve, reject) => {
+                client.deleteChats(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.20";
+export declare const VERSION = "1.2.21";

package/dist/version.js

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

package/package.json

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