macrocosmos 1.4.2 → 2.1.0

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

@@ -1,104 +0,0 @@
-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, UpdateCompletionAttributesRequest, UpdateCompletionAttributesResponse, GetChatSessionsRequest, GetChatCompletionRequest, StoredChatCompletion } from "../../generated/apex/v1/apex";
-import * as grpc from "@grpc/grpc-js";
-import { BaseClient, BaseClientOptions } from "../BaseClient";
-import { ApexStream } from "./Stream";
-import { MarkFieldsOptional } from "../util.types";
-type ChatCompletionRequest = MarkFieldsOptional<GeneratedChatCompletionRequest, "uids">;
-type WebRetrievalRequest = MarkFieldsOptional<GeneratedWebRetrievalRequest, "uids">;
-export { ApexStream, ChatCompletionChunkResponse, WebRetrievalRequest, WebRetrievalResponse, ChatCompletionRequest, ChatMessage, SubmitDeepResearcherJobResponse, GetDeepResearcherJobRequest, GetDeepResearcherJobResponse, GetStoredChatCompletionsRequest, GetStoredChatCompletionsResponse, GetChatSessionsResponse, };
-interface ApexClientOptions extends BaseClientOptions {
-    timeout?: number;
-}
-interface ApexService extends grpc.ServiceClientConstructor, ApexServiceClient {
-}
-export interface ApexProtoClient {
-    ApexService: {
-        new (address: string, credentials: grpc.ChannelCredentials): ApexService;
-    };
-}
-export interface ChatCompletionsCreate {
-    (params: ChatCompletionRequest & {
-        stream: true;
-    }, 
-    /** options are not used, but are accepted for compatibility with the OpenAI API */
-    _options?: unknown): Promise<ApexStream<ChatCompletionChunkResponse>>;
-    (params: ChatCompletionRequest & {
-        stream?: false | undefined;
-    }, 
-    /** options are not used, but are accepted for compatibility with the OpenAI API */
-    _options?: unknown): Promise<ChatCompletionResponse>;
-}
-/**
- * Client for interacting with the Apex API
- * Provides OpenAI-compatible interface over gRPC
- */
-export declare class ApexClient extends BaseClient {
-    private _grpcClient?;
-    private defaultTimeout;
-    constructor(options: ApexClientOptions, grpcClient?: ApexServiceClient);
-    protected createGrpcClient(): ApexServiceClient;
-    /**
-     * Get the default timeout for chat completions
-     */
-    protected getDefaultTimeout(): number;
-    /**
-     * OpenAI-compatible chat completions API
-     */
-    chat: {
-        completions: {
-            create: ChatCompletionsCreate;
-        };
-    };
-    /**
-     * Web retrieval API for searching the internet
-     */
-    webRetrieval: (params: WebRetrievalRequest) => Promise<WebRetrievalResponse>;
-    /**
-     * Submit a deep researcher job with proper defaults
-     */
-    submitDeepResearcherJob: (params: ChatCompletionRequest) => Promise<SubmitDeepResearcherJobResponse>;
-    /**
-     * Get the status and results of a deep research job
-     */
-    getDeepResearcherJob: (params: GetDeepResearcherJobRequest) => Promise<GetDeepResearcherJobResponse>;
-    /**
-     * Get completions of a chat
-     */
-    getStoredChatCompletions: (params: GetStoredChatCompletionsRequest) => Promise<GetStoredChatCompletionsResponse>;
-    /**
-     * Get the user's stored chats
-     */
-    getChatSessions: (params: GetChatSessionsRequest) => Promise<GetChatSessionsResponse>;
-    /**
-     * Search a user's prompt and completion text and return associated chat ids
-     */
-    searchChatIdsByPromptAndCompletionText: (params: SearchChatIdsByPromptAndCompletionTextRequest) => Promise<SearchChatIdsByPromptAndCompletionTextResponse>;
-    /**
-     * Create a chat and completion for a user
-     */
-    createChatAndCompletion: (params: CreateChatAndCompletionRequest) => Promise<CreateChatAndCompletionResponse>;
-    /**
-     * Create completion for a chat
-     */
-    createCompletion: (params: CreateCompletionRequest) => Promise<CreateCompletionResponse>;
-    /**
-     * 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>;
-    /**
-     * Update completion attributes
-     */
-    updateCompletionAttributes: (params: UpdateCompletionAttributesRequest) => Promise<UpdateCompletionAttributesResponse>;
-    /**
-     * GetCompletion by ID client
-     */
-    getChatCompletion: (params: GetChatCompletionRequest) => Promise<StoredChatCompletion>;
-}

package/dist/lib/apex/Client.js

@@ -1,444 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.ApexClient = exports.GetChatSessionsResponse = exports.GetStoredChatCompletionsResponse = exports.GetStoredChatCompletionsRequest = exports.GetDeepResearcherJobResponse = exports.GetDeepResearcherJobRequest = exports.SubmitDeepResearcherJobResponse = exports.ChatMessage = exports.WebRetrievalResponse = exports.ChatCompletionChunkResponse = exports.ApexStream = void 0;
-const apex_1 = require("../../generated/apex/v1/apex");
-Object.defineProperty(exports, "ChatCompletionChunkResponse", { enumerable: true, get: function () { return apex_1.ChatCompletionChunkResponse; } });
-Object.defineProperty(exports, "WebRetrievalResponse", { enumerable: true, get: function () { return apex_1.WebRetrievalResponse; } });
-Object.defineProperty(exports, "ChatMessage", { enumerable: true, get: function () { return apex_1.ChatMessage; } });
-Object.defineProperty(exports, "SubmitDeepResearcherJobResponse", { enumerable: true, get: function () { return apex_1.SubmitDeepResearcherJobResponse; } });
-Object.defineProperty(exports, "GetDeepResearcherJobRequest", { enumerable: true, get: function () { return apex_1.GetDeepResearcherJobRequest; } });
-Object.defineProperty(exports, "GetDeepResearcherJobResponse", { enumerable: true, get: function () { return apex_1.GetDeepResearcherJobResponse; } });
-Object.defineProperty(exports, "GetStoredChatCompletionsRequest", { enumerable: true, get: function () { return apex_1.GetStoredChatCompletionsRequest; } });
-Object.defineProperty(exports, "GetStoredChatCompletionsResponse", { enumerable: true, get: function () { return apex_1.GetStoredChatCompletionsResponse; } });
-Object.defineProperty(exports, "GetChatSessionsResponse", { enumerable: true, get: function () { return apex_1.GetChatSessionsResponse; } });
-const BaseClient_1 = require("../BaseClient");
-const Stream_1 = require("./Stream");
-Object.defineProperty(exports, "ApexStream", { enumerable: true, get: function () { return Stream_1.ApexStream; } });
-function chatCompletionsCreate(params, _options) {
-    const client = this.createGrpcClient();
-    const requestParams = {
-        ...params,
-        uids: params.uids ?? [],
-        timeout: params.timeout || this.getDefaultTimeout(),
-    };
-    if (requestParams.stream) {
-        const stream = client.chatCompletionStream(requestParams);
-        const controller = new AbortController();
-        return Promise.resolve(Stream_1.ApexStream.fromGrpcStream(stream, controller));
-    }
-    else {
-        if (this.isSecure()) {
-            return new Promise((resolve, reject) => {
-                client.chatCompletion(requestParams, (error, response) => {
-                    if (error) {
-                        reject(error);
-                        return;
-                    }
-                    resolve(response);
-                });
-            });
-        }
-        const metadata = this.createAuthMetadata();
-        return new Promise((resolve, reject) => {
-            client.chatCompletion(requestParams, metadata, (error, response) => {
-                if (error) {
-                    reject(error);
-                    return;
-                }
-                resolve(response);
-            });
-        });
-    }
-}
-/**
- * Client for interacting with the Apex API
- * Provides OpenAI-compatible interface over gRPC
- */
-class ApexClient extends BaseClient_1.BaseClient {
-    constructor(options, grpcClient) {
-        super(options);
-        /**
-         * OpenAI-compatible chat completions API
-         */
-        this.chat = {
-            completions: {
-                create: chatCompletionsCreate.bind(this),
-            },
-        };
-        /**
-         * Web retrieval API for searching the internet
-         */
-        this.webRetrieval = (params) => {
-            const client = this.createGrpcClient();
-            if (this.isSecure()) {
-                return new Promise((resolve, reject) => {
-                    client.webRetrieval({ ...params, uids: params.uids ?? [] }, (error, response) => {
-                        if (error) {
-                            reject(error);
-                            return;
-                        }
-                        resolve(response);
-                    });
-                });
-            }
-            const metadata = this.createAuthMetadata();
-            return new Promise((resolve, reject) => {
-                client.webRetrieval({ ...params, uids: params.uids ?? [] }, metadata, (error, response) => {
-                    if (error) {
-                        reject(error);
-                        return;
-                    }
-                    resolve(response);
-                });
-            });
-        };
-        /**
-         * Submit a deep researcher job with proper defaults
-         */
-        this.submitDeepResearcherJob = (params) => {
-            const client = this.createGrpcClient();
-            const request = {
-                ...params,
-                uids: params.uids ?? [],
-                // Required internal fields for Deep Researcher
-                task: "InferenceTask",
-                mixture: false,
-                inferenceMode: "Chain-of-Thought",
-                stream: true,
-            };
-            if (this.isSecure()) {
-                return new Promise((resolve, reject) => {
-                    client.submitDeepResearcherJob(request, (error, response) => {
-                        if (error) {
-                            reject(error);
-                            return;
-                        }
-                        resolve(response);
-                    });
-                });
-            }
-            const metadata = this.createAuthMetadata();
-            return new Promise((resolve, reject) => {
-                client.submitDeepResearcherJob(request, metadata, (error, response) => {
-                    if (error) {
-                        reject(error);
-                        return;
-                    }
-                    resolve(response);
-                });
-            });
-        };
-        /**
-         * Get the status and results of a deep research job
-         */
-        this.getDeepResearcherJob = (params) => {
-            const client = this.createGrpcClient();
-            if (this.isSecure()) {
-                return new Promise((resolve, reject) => {
-                    client.getDeepResearcherJob(params, (error, response) => {
-                        if (error) {
-                            reject(error);
-                            return;
-                        }
-                        resolve(response);
-                    });
-                });
-            }
-            const metadata = this.createAuthMetadata();
-            return new Promise((resolve, reject) => {
-                client.getDeepResearcherJob(params, metadata, (error, response) => {
-                    if (error) {
-                        reject(error);
-                        return;
-                    }
-                    resolve(response);
-                });
-            });
-        };
-        /**
-         * Get completions of a chat
-         */
-        this.getStoredChatCompletions = (params) => {
-            const client = this.createGrpcClient();
-            if (this.isSecure()) {
-                return new Promise((resolve, reject) => {
-                    client.getStoredChatCompletions(params, (error, response) => {
-                        if (error) {
-                            reject(error);
-                            return;
-                        }
-                        resolve(response);
-                    });
-                });
-            }
-            const metadata = this.createAuthMetadata();
-            return new Promise((resolve, reject) => {
-                client.getStoredChatCompletions(params, metadata, (error, response) => {
-                    if (error) {
-                        reject(error);
-                        return;
-                    }
-                    resolve(response);
-                });
-            });
-        };
-        /**
-         * Get the user's stored chats
-         */
-        this.getChatSessions = (params) => {
-            const client = this.createGrpcClient();
-            if (this.isSecure()) {
-                return new Promise((resolve, reject) => {
-                    client.getChatSessions(params, (error, response) => {
-                        if (error) {
-                            reject(error);
-                            return;
-                        }
-                        resolve(response);
-                    });
-                });
-            }
-            const metadata = this.createAuthMetadata();
-            return new Promise((resolve, reject) => {
-                client.getChatSessions(params, metadata, (error, response) => {
-                    if (error) {
-                        reject(error);
-                        return;
-                    }
-                    resolve(response);
-                });
-            });
-        };
-        /**
-         * Search a user's prompt and completion text and return associated chat ids
-         */
-        this.searchChatIdsByPromptAndCompletionText = (params) => {
-            const client = this.createGrpcClient();
-            if (this.isSecure()) {
-                return new Promise((resolve, reject) => {
-                    client.searchChatIdsByPromptAndCompletionText(params, (error, response) => {
-                        if (error) {
-                            reject(error);
-                            return;
-                        }
-                        resolve(response);
-                    });
-                });
-            }
-            const metadata = this.createAuthMetadata();
-            return new Promise((resolve, reject) => {
-                client.searchChatIdsByPromptAndCompletionText(params, metadata, (error, response) => {
-                    if (error) {
-                        reject(error);
-                        return;
-                    }
-                    resolve(response);
-                });
-            });
-        };
-        /**
-         * Create a chat and completion for a user
-         */
-        this.createChatAndCompletion = (params) => {
-            const client = this.createGrpcClient();
-            if (this.isSecure()) {
-                return new Promise((resolve, reject) => {
-                    client.createChatAndCompletion(params, (error, response) => {
-                        if (error) {
-                            reject(error);
-                            return;
-                        }
-                        resolve(response);
-                    });
-                });
-            }
-            const metadata = this.createAuthMetadata();
-            return new Promise((resolve, reject) => {
-                client.createChatAndCompletion(params, metadata, (error, response) => {
-                    if (error) {
-                        reject(error);
-                        return;
-                    }
-                    resolve(response);
-                });
-            });
-        };
-        /**
-         * Create completion for a chat
-         */
-        this.createCompletion = (params) => {
-            const client = this.createGrpcClient();
-            if (this.isSecure()) {
-                return new Promise((resolve, reject) => {
-                    client.createCompletion(params, (error, response) => {
-                        if (error) {
-                            reject(error);
-                            return;
-                        }
-                        resolve(response);
-                    });
-                });
-            }
-            const metadata = this.createAuthMetadata();
-            return new Promise((resolve, reject) => {
-                client.createCompletion(params, metadata, (error, response) => {
-                    if (error) {
-                        reject(error);
-                        return;
-                    }
-                    resolve(response);
-                });
-            });
-        };
-        /**
-         * Delete a chat given its id
-         */
-        this.deleteChats = (params) => {
-            const client = this.createGrpcClient();
-            if (this.isSecure()) {
-                return new Promise((resolve, reject) => {
-                    client.deleteChats(params, (error, response) => {
-                        if (error) {
-                            reject(error);
-                            return;
-                        }
-                        resolve(response);
-                    });
-                });
-            }
-            const metadata = this.createAuthMetadata();
-            return new Promise((resolve, reject) => {
-                client.deleteChats(params, metadata, (error, response) => {
-                    if (error) {
-                        reject(error);
-                        return;
-                    }
-                    resolve(response);
-                });
-            });
-        };
-        /**
-         * Delete completions given their id
-         */
-        this.deleteCompletions = (params) => {
-            const client = this.createGrpcClient();
-            if (this.isSecure()) {
-                return new Promise((resolve, reject) => {
-                    client.deleteCompletions(params, (error, response) => {
-                        if (error) {
-                            reject(error);
-                            return;
-                        }
-                        resolve(response);
-                    });
-                });
-            }
-            const metadata = this.createAuthMetadata();
-            return new Promise((resolve, reject) => {
-                client.deleteCompletions(params, metadata, (error, response) => {
-                    if (error) {
-                        reject(error);
-                        return;
-                    }
-                    resolve(response);
-                });
-            });
-        };
-        /**
-         * Update chat attributes
-         */
-        this.updateChatAttributes = (params) => {
-            const client = this.createGrpcClient();
-            if (this.isSecure()) {
-                return new Promise((resolve, reject) => {
-                    client.updateChatAttributes(params, (error, response) => {
-                        if (error) {
-                            reject(error);
-                            return;
-                        }
-                        resolve(response);
-                    });
-                });
-            }
-            const metadata = this.createAuthMetadata();
-            return new Promise((resolve, reject) => {
-                client.updateChatAttributes(params, metadata, (error, response) => {
-                    if (error) {
-                        reject(error);
-                        return;
-                    }
-                    resolve(response);
-                });
-            });
-        };
-        /**
-         * Update completion attributes
-         */
-        this.updateCompletionAttributes = (params) => {
-            const client = this.createGrpcClient();
-            if (this.isSecure()) {
-                return new Promise((resolve, reject) => {
-                    client.updateCompletionAttributes(params, (error, response) => {
-                        if (error) {
-                            reject(error);
-                            return;
-                        }
-                        resolve(response);
-                    });
-                });
-            }
-            const metadata = this.createAuthMetadata();
-            return new Promise((resolve, reject) => {
-                client.updateCompletionAttributes(params, metadata, (error, response) => {
-                    if (error) {
-                        reject(error);
-                        return;
-                    }
-                    resolve(response);
-                });
-            });
-        };
-        /**
-         * GetCompletion by ID client
-         */
-        this.getChatCompletion = (params) => {
-            const client = this.createGrpcClient();
-            if (this.isSecure()) {
-                return new Promise((resolve, reject) => {
-                    client.getChatCompletion(params, (error, response) => {
-                        if (error) {
-                            reject(error);
-                            return;
-                        }
-                        resolve(response);
-                    });
-                });
-            }
-            const metadata = this.createAuthMetadata();
-            return new Promise((resolve, reject) => {
-                client.getChatCompletion(params, metadata, (error, response) => {
-                    if (error) {
-                        reject(error);
-                        return;
-                    }
-                    resolve(response);
-                });
-            });
-        };
-        this.defaultTimeout = options.timeout || 60;
-        this._grpcClient = grpcClient;
-    }
-    createGrpcClient() {
-        if (this._grpcClient)
-            return this._grpcClient;
-        // Create gRPC client with proper credentials
-        return new apex_1.ApexServiceClient(this.getBaseURL(), this.createChannelCredentials());
-    }
-    /**
-     * Get the default timeout for chat completions
-     */
-    getDefaultTimeout() {
-        return this.defaultTimeout;
-    }
-}
-exports.ApexClient = ApexClient;

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

@@ -1,31 +0,0 @@
-export type ReadableStreamInterface = ReadableStream<Uint8Array>;
-/**
- * A Stream implementation for gRPC streaming responses
- * Similar to OpenAI's Stream class but specific to Apex API
- */
-export declare class ApexStream<Item> implements AsyncIterable<Item> {
-    private iterator;
-    /**
-     * The AbortController for this stream. Call `abort()` to cancel the stream.
-     */
-    controller: AbortController;
-    constructor(iterator: () => AsyncIterator<Item>, controller: AbortController);
-    /**
-     * Creates a Stream from a gRPC streaming response
-     */
-    static fromGrpcStream<Item>(grpcStream: {
-        on: (event: string, listener: (...args: unknown[]) => void) => void;
-        cancel: () => void;
-        removeAllListeners: (event: string) => void;
-    }, controller: AbortController): ApexStream<Item>;
-    [Symbol.asyncIterator](): AsyncIterator<Item>;
-    /**
-     * Splits the stream into two streams which can be
-     * independently read from at different speeds.
-     */
-    tee(): [ApexStream<Item>, ApexStream<Item>];
-    /**
-     * Converts this stream to a ReadableStream
-     */
-    toReadableStream(): ReadableStreamInterface;
-}