macrocosmos 1.4.2 → 2.0.0

package/dist/generated/logger/v1/logger.js

@@ -2,7 +2,7 @@
 // Code generated by protoc-gen-ts_proto. DO NOT EDIT.
 // versions:
 //   protoc-gen-ts_proto  v2.7.0
-//   protoc               v6.32.1
+//   protoc               v6.33.0
 // source: logger/v1/logger.proto
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.LoggerServiceClient = exports.LoggerServiceService = exports.StoreRecordBatchRequest = exports.Record = exports.CreateRunRequest = exports.Ack = exports.protobufPackage = void 0;

package/dist/generated/sn13/v1/sn13_validator.js

@@ -2,7 +2,7 @@
 // Code generated by protoc-gen-ts_proto. DO NOT EDIT.
 // versions:
 //   protoc-gen-ts_proto  v2.7.0
-//   protoc               v6.32.1
+//   protoc               v6.33.0
 // source: sn13/v1/sn13_validator.proto
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Sn13ServiceClient = exports.Sn13ServiceService = exports.OnDemandDataResponse = exports.OnDemandDataRequest = exports.ValidateRedditTopicResponse = exports.ValidateRedditTopicRequest = exports.ListTopicsResponse = exports.ListTopicsResponseDetail = exports.ListTopicsRequest = exports.protobufPackage = void 0;

package/dist/index.d.ts

@@ -1,5 +1,4 @@
 export * from "./constants";
-export * from "./lib/apex";
 export * from "./lib/gravity";
 export * from "./lib/billing";
 export * from "./lib/sn13";

package/dist/index.js

@@ -16,8 +16,6 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
 Object.defineProperty(exports, "__esModule", { value: true });
 // Export constants
 __exportStar(require("./constants"), exports);
-// Export apex modules
-__exportStar(require("./lib/apex"), exports);
 // Export gravity modules
 __exportStar(require("./lib/gravity"), exports);
 // Export billing modules

package/dist/version.d.ts

@@ -1 +1 @@
-export declare const VERSION = "1.4.2";
+export declare const VERSION = "2.0.0";

package/dist/version.js

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

package/package.json

@@ -1,14 +1,18 @@
 {
   "name": "macrocosmos",
-  "version": "1.4.2",
+  "version": "2.0.0",
   "description": "TypeScript SDK package for Macrocosmos",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
-  "files": ["dist/", "README.md", "LICENSE"],
+  "files": [
+    "dist/",
+    "README.md",
+    "LICENSE"
+  ],
   "scripts": {
     "clean": "rimraf dist",
     "gen:version": "node scripts/generate-version.js",
-    "build": "npm run clean && npm run gen:version && npm run gen:proto && tsc && cp README.md dist/ && cp LICENSE dist/",
+    "build": "npm run clean && npm run gen:version && npm run proto:gen && tsc && cp README.md dist/ && cp LICENSE dist/",
     "pack": "npm pack && tar -tf macrocosmos-$(node -p \"require('./package.json').version\").tgz && open macrocosmos-$(node -p \"require('./package.json').version\").tgz",
     "pack:rm": "rm macrocosmos-$(node -p \"require('./package.json').version\").tgz && rm -rf package",
     "ts-check": "tsc --noEmit",
@@ -18,7 +22,8 @@
     "lint:fix": "eslint . --ext .ts --fix",
     "format:check": "prettier --check \"src/**/*.{ts,js,json,md}\"",
     "format:write": "prettier --write \"src/**/*.{ts,js,json,md}\"",
-    "gen:proto": "npx protoc --plugin=protoc-gen-ts_proto=./node_modules/.bin/protoc-gen-ts_proto --ts_proto_out=./src/generated --ts_proto_opt=outputServices=grpc-js,useOptionals=messages,useExactTypes=false,esModuleInterop=true,forceLong=number -I ./protos $(find ./protos -name \"*.proto\") && prettier --write \"src/generated/**/*.{ts,js,json,md}\" ",
+    "proto:pull": "git subtree pull --prefix=protos git@github.com:macrocosm-os/macrocosmos-protos.git main --squash",
+    "proto:gen": "npx protoc --plugin=protoc-gen-ts_proto=./node_modules/.bin/protoc-gen-ts_proto --ts_proto_out=./src/generated --ts_proto_opt=outputServices=grpc-js,useOptionals=messages,useExactTypes=false,esModuleInterop=true,forceLong=number -I ./protos $(find ./protos -name \"*.proto\") && prettier --write \"src/generated/**/*.{ts,js,json,md}\" ",
     "example:sn13:list-topics": "ts-node src/__examples__/sn13/list-topics.ts",
     "prepare": "husky"
   },

package/dist/__tests__/apex/client-chat.test.d.ts

@@ -1 +0,0 @@
-export {};

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

@@ -1,324 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const macrocosmos_1 = require("macrocosmos");
-describe("ApexClient", () => {
-    const API_KEY = process.env.MACROCOSMOS_API_KEY;
-    if (!API_KEY) {
-        throw new Error("MACROCOSMOS_API_KEY environment variable is required");
-    }
-    const messages = [
-        {
-            role: "user",
-            content: "What is the capital of France?",
-        },
-    ];
-    const deepResearchMessages = [
-        {
-            role: "user",
-            content: `Can you propose a mechanism by which a decentralized network 
-      of AI agents could achieve provable alignment on abstract ethical principles 
-      without relying on human-defined ontologies or centralized arbitration?`,
-        },
-    ];
-    const samplingParameters = {
-        temperature: 0.7,
-        topP: 0.9,
-        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({
-            apiKey: API_KEY,
-            appName: "apex-client.test.ts",
-        });
-    });
-    it.skip("should make a streaming chat completion call", async () => {
-        // Create streaming completion
-        const result = await client.chat.completions.create({
-            messages,
-            stream: true,
-            samplingParameters,
-        });
-        // Handle streaming response
-        let fullResponse = "";
-        const stream = result;
-        for await (const chunk of stream) {
-            // We know the interface matches the proto definition
-            const content = chunk.choices?.[0]?.delta?.content || "";
-            fullResponse += content;
-            console.log("Received chunk:", content);
-        }
-        console.log("Full response:", fullResponse);
-        expect(fullResponse).toBeTruthy();
-        expect(fullResponse.toLowerCase()).toContain("paris");
-    }, 30000); // Increase timeout to 30 seconds for streaming
-    // TODO: This test is skipped due to a server-side issue with the non-streaming ChatCompletion endpoint.
-    // The server returns "failed to parse response" error, suggesting the response format doesn't match
-    // the expected ChatCompletionResponse protobuf format. The streaming endpoint works fine.
-    it.skip("should make a non-streaming chat completion call", async () => {
-        // Create non-streaming completion
-        const response = await client.chat.completions.create({
-            messages,
-            samplingParameters,
-        });
-        console.log("Response:", response.choices?.[0]?.message?.content);
-        expect(response.choices?.[0]?.message?.content).toBeTruthy();
-        expect(response.choices?.[0]?.message?.content?.toLowerCase()).toContain("paris");
-    }, 30000); // Increase timeout to 30 seconds
-    it("should retrieve stored chat completions", async () => {
-        // chat ID for testing
-        const chatId = "test-chat-id";
-        // Get stored chat completions
-        const result = await client.getStoredChatCompletions({ chatId });
-        // Verify the response structure
-        console.log("Stored chat completions:", result);
-        expect(result).toBeDefined();
-        expect(Array.isArray(result.chatCompletions)).toBe(true);
-        // 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 () => {
-        // create a 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();
-        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(createChatAndCompletionParams);
-        // 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({
-            chatType: "apex",
-        });
-        // Verify the response structure
-        console.log("Stored chats:", get_chat_sessions_result);
-        expect(get_chat_sessions_result).toBeDefined();
-        // 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 retrieve a completion", async () => {
-        // chat ID for testing
-        const create_chat_result = await client.createChatAndCompletion(createChatAndCompletionParams);
-        // Verify create chat was successful
-        console.log("Create chat response:", create_chat_result);
-        expect(create_chat_result).toBeDefined();
-        // Get the completion created in the chat
-        const get_chat_completion_result = await client.getChatCompletion({
-            completionId: create_chat_result.parsedCompletion?.id ?? "",
-        });
-        // Make sure the retrieved completion matches the created one
-        expect(get_chat_completion_result).toBeDefined();
-        expect(get_chat_completion_result.id).toBe(create_chat_result.parsedCompletion?.id);
-        expect(get_chat_completion_result.chatId).toBe(create_chat_result.parsedChat?.id);
-        expect(get_chat_completion_result.completionType).toBe(create_chat_result.parsedCompletion?.completionType);
-        expect(get_chat_completion_result.createdAt?.toISOString()).toBe(create_chat_result.parsedCompletion?.createdAt?.toISOString());
-        expect(get_chat_completion_result.userPromptText).toBe(create_chat_result.parsedCompletion?.userPromptText);
-        expect(get_chat_completion_result.metadata).toStrictEqual(create_chat_result.parsedCompletion?.metadata);
-        const delete_chat_result = await client.deleteChats({
-            chatIds: [create_chat_result.parsedChat?.id ?? ""],
-        });
-        expect(delete_chat_result.success).toBeTruthy();
-    }, 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 - skipped due to failure, Deep Researcher has also been retired
-    it.skip("should create a deep research job", async () => {
-        // Create test parameters
-        const params = {
-            messages: deepResearchMessages,
-            model: "Default",
-            samplingParameters: samplingParameters,
-            seed: 42,
-            uids: [1, 2, 3],
-        };
-        // Submit the job
-        const response = await client.submitDeepResearcherJob(params);
-        // Verify response structure
-        expect(response).toBeDefined();
-        expect(typeof response.jobId).toBe("string");
-        expect(typeof response.status).toBe("string");
-        expect(typeof response.createdAt).toBe("string");
-        expect(typeof response.updatedAt).toBe("string");
-        // Log response for debugging
-        console.log("Create Job Response:", response);
-    }, 30000);
-    it.skip("should get deep researcher job results", async () => {
-        // First create a job. Use simple inputs as job creation is tested in the previous test.
-        const createParams = {
-            messages: deepResearchMessages,
-        };
-        const createResponse = await client.submitDeepResearcherJob(createParams);
-        const jobId = createResponse.jobId;
-        // Then get the results
-        const response = await client.getDeepResearcherJob({ jobId: jobId });
-        // Verify response structure
-        expect(response).toBeDefined();
-        expect(response.jobId).toBe(jobId); // Job ID should match the one from the create response
-        expect(typeof response.status).toBe("string");
-        expect(typeof response.jobId).toBe("string");
-        expect(typeof response.createdAt).toBe("string");
-        expect(typeof response.updatedAt).toBe("string");
-        expect(Array.isArray(response.result)).toBe(true);
-        // Log response for debugging
-        console.log("Get Job Results Response:", response);
-    }, 60000); // Longer timeout for this test as it involves multiple API calls
-    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(createChatAndCompletionParams);
-        // 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({
-            chatType: "apex",
-        });
-        // Verify the response structure
-        console.log("Stored chats:", result);
-        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);
-    it("should update completion attribute", async () => {
-        // chat ID for testing
-        const create_chat_result = await client.createChatAndCompletion(createChatAndCompletionParams);
-        // Verify chat was created
-        console.log("Create chat and completion response:", create_chat_result);
-        expect(create_chat_result).toBeDefined();
-        const update_completion_attributes = await client.updateCompletionAttributes({
-            completionId: create_chat_result.parsedCompletion?.id ?? "",
-            completionText: "Updated completion text",
-            userPromptText: "Updated user prompt text",
-            metadata: {
-                fancy_metadata_key: "fancy_metadata_value",
-            },
-        });
-        expect(update_completion_attributes.completion?.completionText).toBe("Updated completion text");
-        expect(update_completion_attributes.completion?.userPromptText).toBe("Updated user prompt text");
-        expect(update_completion_attributes.completion?.metadata).toBeDefined();
-        expect(typeof update_completion_attributes.completion?.metadata).toBe("object");
-        expect(Object.keys(update_completion_attributes.completion?.metadata || {})
-            .length).toBe(1);
-        expect(update_completion_attributes.completion?.metadata?.fancy_metadata_key).toBe("fancy_metadata_value");
-        // Delete the test chat. This also deletes the completion.
-        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.d.ts

@@ -1 +0,0 @@
-export {};

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

@@ -1,38 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const macrocosmos_1 = require("macrocosmos");
-describe("ApexClient WebRetrieval", () => {
-    const API_KEY = process.env.MACROCOSMOS_API_KEY;
-    if (!API_KEY) {
-        throw new Error("MACROCOSMOS_API_KEY environment variable is required");
-    }
-    it.skip("should make a web retrieval request", async () => {
-        // Create ApexClient
-        const client = new macrocosmos_1.ApexClient({
-            apiKey: API_KEY,
-            appName: "apex-client-web-search.test.ts",
-        });
-        // Create web retrieval request
-        const request = {
-            searchQuery: "What is Bittensor?",
-            nResults: 5,
-            maxResponseTime: 10000, // 10 seconds
-        };
-        // Make the web retrieval call
-        const response = await client.webRetrieval(request);
-        // Validate response
-        expect(response).toBeTruthy();
-        expect(response.results).toBeDefined();
-        expect(Array.isArray(response.results)).toBe(true);
-        // Check if we have results
-        if (response.results && response.results.length > 0) {
-            console.log(`Received ${response.results.length} search results`);
-            // Log the first result
-            const firstResult = response.results[0];
-            console.log("First result URL:", firstResult.url);
-            console.log("First result content snippet:", firstResult.content?.substring(0, 100) + "...");
-        }
-        // Expect to have at least one result
-        expect(response.results?.length).toBeGreaterThan(0);
-    }, 45000); // Increase timeout to 45 seconds
-});