@ax-llm/ax 9.0.0

package/build/module/src/trace/index.d.ts

@@ -0,0 +1,171 @@
+export declare const SpanAttributes: {
+    LLM_SYSTEM: string;
+    LLM_REQUEST_MODEL: string;
+    LLM_REQUEST_MAX_TOKENS: string;
+    LLM_REQUEST_TEMPERATURE: string;
+    LLM_REQUEST_TOP_K: string;
+    LLM_REQUEST_FREQUENCY_PENALTY: string;
+    LLM_REQUEST_PRESENCE_PENALTY: string;
+    LLM_REQUEST_STOP_SEQUENCES: string;
+    LLM_REQUEST_LLM_IS_STREAMING: string;
+    LLM_REQUEST_TOP_P: string;
+    LLM_USAGE_PROMPT_TOKENS: string;
+    LLM_USAGE_COMPLETION_TOKENS: string;
+    DB_SYSTEM: string;
+    DB_TABLE: string;
+    DB_NAMESPACE: string;
+    DB_ID: string;
+    DB_QUERY_TEXT: string;
+    DB_VECTOR: string;
+    DB_OPERATION_NAME: string;
+    DB_VECTOR_QUERY_TOP_K: string;
+    DB_QUERY_EMBEDDINGS: string;
+    DB_QUERY_RESULT: string;
+    DB_QUERY_EMBEDDINGS_VECTOR: string;
+    DB_QUERY_RESULT_ID: string;
+    DB_QUERY_RESULT_SCORE: string;
+    DB_QUERY_RESULT_DISTANCE: string;
+    DB_QUERY_RESULT_METADATA: string;
+    DB_QUERY_RESULT_VECTOR: string;
+    DB_QUERY_RESULT_DOCUMENT: string;
+};
+export declare const SpanEvents: {
+    LLM_PROMPT: string;
+};
+export declare enum LLMRequestTypeValues {
+    COMPLETION = "completion",
+    CHAT = "chat",
+    RERANK = "rerank",
+    UNKNOWN = "unknown"
+}
+export declare enum SpanKindValues {
+    WORKFLOW = "workflow",
+    TASK = "task",
+    AGENT = "agent",
+    TOOL = "tool",
+    UNKNOWN = "unknown"
+}
+export type AttributeValue = string | number | boolean | undefined | null;
+export interface Attributes {
+    [key: string]: AttributeValue;
+}
+export interface TimeInput {
+    timestamp: number;
+}
+export interface Exception {
+    message: string;
+    name: string;
+    stack?: string;
+}
+export interface SpanContext {
+    traceId: string;
+    spanId: string;
+}
+export declare enum SpanStatusCode {
+    OK = "OK",
+    ERROR = "ERROR"
+}
+export interface SpanStatus {
+    code: SpanStatusCode;
+    message?: string;
+}
+export interface SpanStatus {
+    code: SpanStatusCode;
+    description?: string;
+}
+export interface Span {
+    addEvent(name: string, attributesOrStartTime?: Readonly<Attributes | TimeInput>, startTime?: Readonly<TimeInput>): Span;
+    end(endTime?: Readonly<TimeInput>): void;
+    isRecording(): boolean;
+    recordException(exception: Readonly<Exception>, time?: Readonly<TimeInput>): void;
+    setAttribute(key: string, value: AttributeValue): Span;
+    setAttributes(attributes: Attributes): Span;
+    setStatus(status: Readonly<SpanStatus>): Span;
+    spanContext(): SpanContext;
+    updateName(name: string): Span;
+}
+export interface Context {
+    /**
+     * Deletes a value associated with a key from the context.
+     * Returns a new context that inherits from the current context but does not contain the value for the specified key.
+     * @param key The symbol key for which to clear the value.
+     * @returns A new Context instance without the specified key.
+     */
+    deleteValue(key: symbol): Context;
+    /**
+     * Retrieves a value from the context using a symbol as the key.
+     * @param key The symbol key which identifies a context value.
+     * @returns The value associated with the key, if any; otherwise, undefined.
+     */
+    getValue(key: symbol): unknown;
+    /**
+     * Sets a value in the context for the specified key.
+     * Returns a new context that inherits from the current context with the new key-value pair added.
+     * @param key The symbol key for which to set the value.
+     * @param value The value to set for the given key.
+     * @returns A new Context instance with the updated key-value pair.
+     */
+    setValue(key: symbol, value: unknown): Context;
+}
+export declare enum SpanKind {
+    INTERNAL = "INTERNAL",
+    SERVER = "SERVER",
+    CLIENT = "CLIENT",
+    PRODUCER = "PRODUCER",
+    CONSUMER = "CONSUMER"
+}
+export interface Link {
+    context: SpanContext;
+    attributes?: Attributes;
+}
+export interface SpanOptions {
+    /**
+     * Optional attributes that can be attached to the Span.
+     * @optional
+     */
+    attributes?: Attributes;
+    /**
+     * The kind of span, defaults to SpanKind.INTERNAL if not specified.
+     * @optional
+     */
+    kind?: SpanKind;
+    /**
+     * Links that associate this new Span with other Spans.
+     * @optional
+     */
+    links?: Link[];
+    /**
+     * Indicates whether the span should be a root span, ignoring any parent span from the context.
+     * @optional
+     */
+    root?: boolean;
+    /**
+     * A manually specified start time for the span, if required.
+     * @optional
+     */
+    startTime?: TimeInput;
+}
+export interface Span {
+    setStatus(status: Readonly<SpanStatus>): void;
+    end(): void;
+}
+export interface Tracer {
+    /**
+     * Starts a new Span with the specified options and executes the provided function. The span is automatically closed after the function executes.
+     * @param name The name of the span.
+     * @param options Span options to apply to the span.
+     * @param fn The function to execute within the span's context.
+     * @returns The return value of the function.
+     */
+    startActiveSpan<F extends (span: Span) => unknown>(name: string, options: Readonly<SpanOptions>, fn: F): ReturnType<F>;
+    /**
+     * Starts a new Span with the specified options and context, then executes the provided function. The span is automatically closed after the function executes.
+     * @param name The name of the span.
+     * @param options Span options to apply to the span.
+     * @param context Context to be used for the span.
+     * @param fn The function to execute within the span's context.
+     * @returns The return value of the function.
+     */
+    startActiveSpan<F extends (span: Span) => unknown>(name: string, options: Readonly<SpanOptions>, context: Context, fn: F): ReturnType<F>;
+    startSpan(name: string, options?: Readonly<SpanOptions>, context?: Context): Span;
+}

package/build/module/src/trace/index.js

@@ -0,0 +1,67 @@
+export const SpanAttributes = {
+    // LLM
+    LLM_SYSTEM: 'gen_ai.system',
+    LLM_REQUEST_MODEL: 'gen_ai.request.model',
+    LLM_REQUEST_MAX_TOKENS: 'gen_ai.request.max_tokens',
+    LLM_REQUEST_TEMPERATURE: 'gen_ai.request.temperature',
+    LLM_REQUEST_TOP_K: 'gen_ai.request.top_k',
+    LLM_REQUEST_FREQUENCY_PENALTY: 'gen_ai.request.frequency_penalty',
+    LLM_REQUEST_PRESENCE_PENALTY: 'gen_ai.request.presence_penalty',
+    LLM_REQUEST_STOP_SEQUENCES: 'gen_ai.request.stop_sequences',
+    LLM_REQUEST_LLM_IS_STREAMING: 'gen_ai.request.llm_is_streaming',
+    LLM_REQUEST_TOP_P: 'gen_ai.request.top_p',
+    LLM_USAGE_PROMPT_TOKENS: 'gen_ai.usage.prompt_tokens',
+    LLM_USAGE_COMPLETION_TOKENS: 'gen_ai.usage.completion_tokens',
+    // Vector DB
+    DB_SYSTEM: 'db.system',
+    DB_TABLE: 'db.table',
+    DB_NAMESPACE: 'db.namespace',
+    DB_ID: 'db.id',
+    DB_QUERY_TEXT: 'db.query.text',
+    DB_VECTOR: 'db.vector',
+    DB_OPERATION_NAME: 'db.operation.name',
+    DB_VECTOR_QUERY_TOP_K: 'db.vector.query.top_k',
+    DB_QUERY_EMBEDDINGS: 'db.query.embeddings',
+    DB_QUERY_RESULT: 'db.query.result',
+    // Query Embeddings
+    DB_QUERY_EMBEDDINGS_VECTOR: 'db.query.embeddings.vector',
+    // Query Result (canonical format)
+    DB_QUERY_RESULT_ID: 'db.query.result.id',
+    DB_QUERY_RESULT_SCORE: 'db.query.result.score',
+    DB_QUERY_RESULT_DISTANCE: 'db.query.result.distance',
+    DB_QUERY_RESULT_METADATA: 'db.query.result.metadata',
+    DB_QUERY_RESULT_VECTOR: 'db.query.result.vector',
+    DB_QUERY_RESULT_DOCUMENT: 'db.query.result.document'
+};
+export const SpanEvents = {
+    LLM_PROMPT: 'gen_ai.prompt'
+};
+export var LLMRequestTypeValues;
+(function (LLMRequestTypeValues) {
+    LLMRequestTypeValues["COMPLETION"] = "completion";
+    LLMRequestTypeValues["CHAT"] = "chat";
+    LLMRequestTypeValues["RERANK"] = "rerank";
+    LLMRequestTypeValues["UNKNOWN"] = "unknown";
+})(LLMRequestTypeValues || (LLMRequestTypeValues = {}));
+export var SpanKindValues;
+(function (SpanKindValues) {
+    SpanKindValues["WORKFLOW"] = "workflow";
+    SpanKindValues["TASK"] = "task";
+    SpanKindValues["AGENT"] = "agent";
+    SpanKindValues["TOOL"] = "tool";
+    SpanKindValues["UNKNOWN"] = "unknown";
+})(SpanKindValues || (SpanKindValues = {}));
+export var SpanStatusCode;
+(function (SpanStatusCode) {
+    SpanStatusCode["OK"] = "OK";
+    SpanStatusCode["ERROR"] = "ERROR";
+})(SpanStatusCode || (SpanStatusCode = {}));
+export var SpanKind;
+(function (SpanKind) {
+    SpanKind["INTERNAL"] = "INTERNAL";
+    SpanKind["SERVER"] = "SERVER";
+    SpanKind["CLIENT"] = "CLIENT";
+    SpanKind["PRODUCER"] = "PRODUCER";
+    SpanKind["CONSUMER"] = "CONSUMER";
+})(SpanKind || (SpanKind = {}));
+//# sourceMappingURL=index.js.map

package/build/module/src/trace/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/trace/index.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,cAAc,GAAG;IAC5B,MAAM;IACN,UAAU,EAAE,eAAe;IAC3B,iBAAiB,EAAE,sBAAsB;IACzC,sBAAsB,EAAE,2BAA2B;IACnD,uBAAuB,EAAE,4BAA4B;IACrD,iBAAiB,EAAE,sBAAsB;IACzC,6BAA6B,EAAE,kCAAkC;IACjE,4BAA4B,EAAE,iCAAiC;IAC/D,0BAA0B,EAAE,+BAA+B;IAC3D,4BAA4B,EAAE,iCAAiC;IAC/D,iBAAiB,EAAE,sBAAsB;IAEzC,uBAAuB,EAAE,4BAA4B;IACrD,2BAA2B,EAAE,gCAAgC;IAE7D,YAAY;IACZ,SAAS,EAAE,WAAW;IACtB,QAAQ,EAAE,UAAU;IACpB,YAAY,EAAE,cAAc;IAC5B,KAAK,EAAE,OAAO;IACd,aAAa,EAAE,eAAe;IAC9B,SAAS,EAAE,WAAW;IACtB,iBAAiB,EAAE,mBAAmB;IACtC,qBAAqB,EAAE,uBAAuB;IAE9C,mBAAmB,EAAE,qBAAqB;IAC1C,eAAe,EAAE,iBAAiB;IAElC,mBAAmB;IACnB,0BAA0B,EAAE,4BAA4B;IAExD,kCAAkC;IAClC,kBAAkB,EAAE,oBAAoB;IACxC,qBAAqB,EAAE,uBAAuB;IAC9C,wBAAwB,EAAE,0BAA0B;IACpD,wBAAwB,EAAE,0BAA0B;IACpD,sBAAsB,EAAE,wBAAwB;IAChD,wBAAwB,EAAE,0BAA0B;CACrD,CAAC;AAEF,MAAM,CAAC,MAAM,UAAU,GAAG;IACxB,UAAU,EAAE,eAAe;CAC5B,CAAC;AAEF,MAAM,CAAN,IAAY,oBAKX;AALD,WAAY,oBAAoB;IAC9B,iDAAyB,CAAA;IACzB,qCAAa,CAAA;IACb,yCAAiB,CAAA;IACjB,2CAAmB,CAAA;AACrB,CAAC,EALW,oBAAoB,KAApB,oBAAoB,QAK/B;AAED,MAAM,CAAN,IAAY,cAMX;AAND,WAAY,cAAc;IACxB,uCAAqB,CAAA;IACrB,+BAAa,CAAA;IACb,iCAAe,CAAA;IACf,+BAAa,CAAA;IACb,qCAAmB,CAAA;AACrB,CAAC,EANW,cAAc,KAAd,cAAc,QAMzB;AAuBD,MAAM,CAAN,IAAY,cAGX;AAHD,WAAY,cAAc;IACxB,2BAAS,CAAA;IACT,iCAAe,CAAA;AACjB,CAAC,EAHW,cAAc,KAAd,cAAc,QAGzB;AA2DD,MAAM,CAAN,IAAY,QAMX;AAND,WAAY,QAAQ;IAClB,iCAAqB,CAAA;IACrB,6BAAiB,CAAA;IACjB,6BAAiB,CAAA;IACjB,iCAAqB,CAAA;IACrB,iCAAqB,CAAA;AACvB,CAAC,EANW,QAAQ,KAAR,QAAQ,QAMnB"}

package/build/module/src/types/index.d.ts

@@ -0,0 +1,81 @@
+import type { TextModelConfig, TextModelInfo, TextResponse } from '../ai/types.js';
+import { type FunctionJSONSchema } from '../text/functions.js';
+export type APIError = {
+    pathname: string;
+    statusCode: number;
+    statusMessage?: string;
+    headers?: any;
+    request: any;
+    response: any;
+};
+export type ParsingError = {
+    message: string;
+    value: string;
+};
+export type TextModelInfoWithProvider = TextModelInfo & {
+    provider: string;
+};
+export type AITextRequestIdentity = {
+    user?: string;
+    organization?: string;
+};
+export type AITextBaseRequest = {
+    identity?: Readonly<AITextRequestIdentity>;
+};
+export type AITextChatRequest = {
+    chatPrompt: Readonly<{
+        role: 'system';
+        content: string;
+    } | {
+        role: 'user';
+        content: string;
+        name?: string;
+    } | {
+        role: 'assistant';
+        content: string | null;
+        name?: string;
+        functionCalls?: {
+            id: string;
+            type: 'function';
+            function: {
+                name: string;
+                arguments?: string | object;
+            };
+        }[];
+    } | {
+        role: 'function';
+        content: string;
+        functionId: string;
+    }>[];
+    functions?: Readonly<{
+        name: string;
+        description: string;
+        parameters?: FunctionJSONSchema;
+    }>[];
+    functionCall?: 'none' | 'auto' | 'required' | {
+        type: 'function';
+        function: {
+            name: string;
+        };
+    };
+    modelConfig?: Readonly<TextModelConfig>;
+    modelInfo?: Readonly<TextModelInfoWithProvider>;
+} & AITextBaseRequest;
+export type AITextEmbedRequest = {
+    texts?: readonly string[];
+    embedModelInfo?: Readonly<TextModelInfoWithProvider>;
+} & AITextBaseRequest;
+export type AITextTraceStepRequest = AITextChatRequest | AITextEmbedRequest;
+export type AITextTraceStepResponse = Omit<TextResponse, 'sessionId'> & {
+    modelResponseTime?: number;
+    embedModelResponseTime?: number;
+    parsingError?: Readonly<ParsingError>;
+    apiError?: Readonly<APIError>;
+};
+export type AITextTraceStep = {
+    traceId: string;
+    sessionId?: string;
+    request: AITextTraceStepRequest;
+    response: AITextTraceStepResponse;
+    createdAt: string;
+};

package/build/module/src/types/index.js

@@ -0,0 +1,2 @@
+import {} from '../text/functions.js';
+//# sourceMappingURL=index.js.map

package/build/module/src/types/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/types/index.ts"],"names":[],"mappings":"AAKA,OAAO,EAA2B,MAAM,sBAAsB,CAAC"}

package/build/module/src/util/apicall.d.ts

@@ -0,0 +1,21 @@
+/// <reference types="node" />
+/// <reference types="node" />
+/// <reference types="node" />
+import { type ReadableStream } from 'stream/web';
+import type { Span } from '../trace/index.js';
+/**
+ * Util: API details
+ * @export
+ */
+export type API = {
+    name?: string;
+    headers?: Record<string, string>;
+    put?: boolean;
+};
+export declare const apiCall: <TRequest = unknown, TResponse = unknown>(api: Readonly<API & {
+    url: string | URL;
+    stream?: boolean;
+    debug?: boolean;
+    fetch?: typeof fetch;
+    span?: Span;
+}>, json: TRequest) => Promise<TResponse | ReadableStream<TResponse>>;

package/build/module/src/util/apicall.js

@@ -0,0 +1,137 @@
+import path from 'path';
+import { TextDecoderStream as TextDecoderStreamNative } from 'stream/web';
+import { TextDecoderStreamPolyfill } from './stream.js';
+import { JSONStringifyStream } from './transform.js';
+const TextDecoderStream = TextDecoderStreamNative ?? TextDecoderStreamPolyfill;
+export const apiCall = async (api, json) => {
+    const baseUrl = new URL(process.env.PROXY ?? api.url);
+    const apiPath = path.join(baseUrl.pathname, api.name ?? '/', baseUrl.search);
+    const apiUrl = new URL(apiPath, baseUrl);
+    if (api.span?.isRecording()) {
+        api.span.setAttributes({
+            'http.request.method': api.put ? 'PUT' : 'POST',
+            'url.full': apiUrl.href
+        });
+    }
+    let res;
+    try {
+        res = await (api.fetch ?? fetch)(apiUrl, {
+            method: api.put ? 'PUT' : 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+                ...api.headers
+            },
+            body: JSON.stringify(json)
+        });
+        if (res.status >= 400) {
+            const body = JSON.stringify(await res.json(), null, 2);
+            throw new Error(`API Error: ${apiUrl.href}, ${res.status}, ${res.statusText}\n${body}`);
+        }
+        if (!res.body) {
+            throw new Error('Response body is null');
+        }
+    }
+    catch (e) {
+        if (api.span?.isRecording()) {
+            api.span.recordException(e);
+        }
+        throw new Error(`API Error: ${apiUrl.href}, ${e}`);
+    }
+    if (!api.stream) {
+        return (await res.json());
+    }
+    const st = res.body
+        .pipeThrough(new TextDecoderStream())
+        .pipeThrough(new JSONStringifyStream());
+    return st;
+};
+// for await (const chunk of res.body as unknown as AsyncIterable<Uint8Array>) {
+//   console.log('CHUNK', decoder.decode(chunk));
+// }
+// const res = await axios.post(apiUrl, json, {
+//   headers: api.headers,
+//   responseType: 'stream'
+//   // responseType: api.stream ? 'stream' : 'json'
+// });
+// const res = await request
+//   .send(json as object)
+//   .set(headers ?? {})
+//   .type('json')
+//   .accept('json')
+//   .retry(3);
+// return res.body;
+// } catch (e) {
+//   const err = e as SuperAgentError;
+//   throw httpError(`apiCall:`, apiUrl, json, err);
+// }
+/*
+export const apiCallWithUpload = async <
+  Request,
+  Response,
+  APIType extends API & { url: string }
+>(
+  api: Readonly<APIType>,
+  json: Request,
+  file: string
+): Promise<Response> => {
+  if (!file) {
+    throw new Error('File is required');
+  }
+
+  const headers = api.headers;
+  const baseUrl = api.url;
+  const apiPath = api.name ?? '/';
+  const apiUrl = new URL(apiPath, baseUrl).toString();
+
+  try {
+    const data = await superagent
+      .post(apiUrl)
+      .retry(3)
+      .attach('file', file)
+      .set(headers ?? {})
+      .field(json as { [fieldName: string]: string });
+
+    return data.body;
+  } catch (e) {
+    throw httpError('apiCallWithUpload', apiUrl, null, e as SuperAgentError);
+  }
+};
+*/
+// export type SuperAgentError = {
+//   response: superagent.Response;
+//   code: unknown;
+//   syscall: unknown;
+//   address: unknown;
+//   port: unknown;
+//   request: unknown;
+// };
+// export const httpError = (
+//   message: string,
+//   apiUrl: string,
+//   json: unknown,
+//   { response, code, syscall, address, port }: Readonly<SuperAgentError>
+// ) => {
+//   const err = new Error(message) as Error & { data: unknown };
+//   if (!response) {
+//     err.data = {
+//       apiUrl,
+//       code,
+//       syscall,
+//       address,
+//       port,
+//       request: json
+//     };
+//     return err;
+//   }
+//   const { headers, status, body } = response;
+//   err.data = {
+//     apiUrl,
+//     statusCode: status,
+//     headers,
+//     request: json,
+//     response: body,
+//     error: body.error
+//   };
+//   return err;
+// };
+//# sourceMappingURL=apicall.js.map

package/build/module/src/util/apicall.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"apicall.js","sourceRoot":"","sources":["../../../../src/util/apicall.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAEL,iBAAiB,IAAI,uBAAuB,EAC7C,MAAM,YAAY,CAAC;AAIpB,OAAO,EAAE,yBAAyB,EAAE,MAAM,aAAa,CAAC;AACxD,OAAO,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AAYrD,MAAM,iBAAiB,GAAG,uBAAuB,IAAI,yBAAyB,CAAC;AAE/E,MAAM,CAAC,MAAM,OAAO,GAAG,KAAK,EAC1B,GAQC,EACD,IAAc,EACkC,EAAE;IAClD,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;IACtD,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,IAAI,IAAI,GAAG,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;IAC7E,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAEzC,IAAI,GAAG,CAAC,IAAI,EAAE,WAAW,EAAE,EAAE,CAAC;QAC5B,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC;YACrB,qBAAqB,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM;YAC/C,UAAU,EAAE,MAAM,CAAC,IAAI;SACxB,CAAC,CAAC;IACL,CAAC;IAED,IAAI,GAAqD,CAAC;IAE1D,IAAI,CAAC;QACH,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC,KAAK,IAAI,KAAK,CAAC,CAAC,MAAM,EAAE;YACvC,MAAM,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM;YAChC,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,GAAG,GAAG,CAAC,OAAO;aACf;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CAAC,CAAC;QAEH,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,EAAE,CAAC;YACtB,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YACvD,MAAM,IAAI,KAAK,CACb,cAAc,MAAM,CAAC,IAAI,KAAK,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,UAAU,KAAK,IAAI,EAAE,CACvE,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,IAAI,GAAG,CAAC,IAAI,EAAE,WAAW,EAAE,EAAE,CAAC;YAC5B,GAAG,CAAC,IAAI,CAAC,eAAe,CAAC,CAAU,CAAC,CAAC;QACvC,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,cAAc,MAAM,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC,CAAC;IACrD,CAAC;IAED,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC;QAChB,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAc,CAAC;IACzC,CAAC;IAED,MAAM,EAAE,GAAG,GAAG,CAAC,IAAI;SAChB,WAAW,CAAC,IAAI,iBAAiB,EAAE,CAAC;SACpC,WAAW,CAAC,IAAI,mBAAmB,EAAa,CAAC,CAAC;IAErD,OAAO,EAAE,CAAC;AACZ,CAAC,CAAC;AAEF,gFAAgF;AAChF,iDAAiD;AACjD,IAAI;AAEJ,+CAA+C;AAC/C,0BAA0B;AAC1B,2BAA2B;AAC3B,oDAAoD;AACpD,MAAM;AAEN,4BAA4B;AAC5B,0BAA0B;AAC1B,wBAAwB;AACxB,kBAAkB;AAClB,oBAAoB;AACpB,eAAe;AACf,mBAAmB;AAEnB,gBAAgB;AAChB,sCAAsC;AACtC,oDAAoD;AACpD,IAAI;AAEJ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAgCE;AAEF,kCAAkC;AAClC,mCAAmC;AACnC,mBAAmB;AACnB,sBAAsB;AACtB,sBAAsB;AACtB,mBAAmB;AACnB,sBAAsB;AACtB,KAAK;AAEL,6BAA6B;AAC7B,qBAAqB;AACrB,oBAAoB;AACpB,mBAAmB;AACnB,0EAA0E;AAC1E,SAAS;AACT,iEAAiE;AAEjE,qBAAqB;AACrB,mBAAmB;AACnB,gBAAgB;AAChB,cAAc;AACd,iBAAiB;AACjB,iBAAiB;AACjB,cAAc;AACd,sBAAsB;AACtB,SAAS;AACT,kBAAkB;AAClB,MAAM;AAEN,gDAAgD;AAChD,iBAAiB;AACjB,cAAc;AACd,0BAA0B;AAC1B,eAAe;AACf,qBAAqB;AACrB,sBAAsB;AACtB,wBAAwB;AACxB,OAAO;AACP,gBAAgB;AAChB,KAAK"}

package/build/module/src/util/log.d.ts

@@ -0,0 +1,14 @@
+export declare class ColorLog {
+    private readonly ANSI_WHITE_BRIGHT;
+    private readonly ANSI_GREEN_BRIGHT;
+    private readonly ANSI_BLUE_BRIGHT;
+    private readonly ANSI_YELLOW;
+    private readonly ANSI_RED;
+    private readonly ANSI_RESET;
+    private colorize;
+    whiteBright(text: string): string;
+    greenBright(text: string): string;
+    blueBright(text: string): string;
+    yellow(text: string): string;
+    red(text: string): string;
+}

package/build/module/src/util/log.js

@@ -0,0 +1,30 @@
+export class ColorLog {
+    // ANSI escape codes for different colors
+    ANSI_WHITE_BRIGHT = '\x1b[97m';
+    ANSI_GREEN_BRIGHT = '\x1b[92m';
+    ANSI_BLUE_BRIGHT = '\x1b[94m';
+    ANSI_YELLOW = '\x1b[93m';
+    ANSI_RED = '\x1b[91m';
+    ANSI_RESET = '\x1b[0m';
+    // Method to wrap text with the specified ANSI color code
+    colorize(text, colorCode) {
+        return `${colorCode}${text}${this.ANSI_RESET}`;
+    }
+    // Public methods to colorize text in various colors
+    whiteBright(text) {
+        return this.colorize(text, this.ANSI_WHITE_BRIGHT);
+    }
+    greenBright(text) {
+        return this.colorize(text, this.ANSI_GREEN_BRIGHT);
+    }
+    blueBright(text) {
+        return this.colorize(text, this.ANSI_BLUE_BRIGHT);
+    }
+    yellow(text) {
+        return this.colorize(text, this.ANSI_YELLOW);
+    }
+    red(text) {
+        return this.colorize(text, this.ANSI_RED);
+    }
+}
+//# sourceMappingURL=log.js.map

package/build/module/src/util/log.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"log.js","sourceRoot":"","sources":["../../../../src/util/log.ts"],"names":[],"mappings":"AAAA,MAAM,OAAO,QAAQ;IACnB,yCAAyC;IACxB,iBAAiB,GAAG,UAAU,CAAC;IAC/B,iBAAiB,GAAG,UAAU,CAAC;IAC/B,gBAAgB,GAAG,UAAU,CAAC;IAC9B,WAAW,GAAG,UAAU,CAAC;IACzB,QAAQ,GAAG,UAAU,CAAC;IACtB,UAAU,GAAG,SAAS,CAAC;IAExC,yDAAyD;IACjD,QAAQ,CAAC,IAAY,EAAE,SAAiB;QAC9C,OAAO,GAAG,SAAS,GAAG,IAAI,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;IACjD,CAAC;IAED,oDAAoD;IAC7C,WAAW,CAAC,IAAY;QAC7B,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACrD,CAAC;IAEM,WAAW,CAAC,IAAY;QAC7B,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACrD,CAAC;IAEM,UAAU,CAAC,IAAY;QAC5B,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;IACpD,CAAC;IAEM,MAAM,CAAC,IAAY;QACxB,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;IAC/C,CAAC;IAEM,GAAG,CAAC,IAAY;QACrB,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC5C,CAAC;CACF"}

package/build/module/src/util/other.d.ts

@@ -0,0 +1 @@
+export declare function sleep(ms: number): Promise<unknown>;

package/build/module/src/util/other.js

@@ -0,0 +1,4 @@
+export async function sleep(ms) {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+}
+//# sourceMappingURL=other.js.map

package/build/module/src/util/other.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"other.js","sourceRoot":"","sources":["../../../../src/util/other.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,KAAK,UAAU,KAAK,CAAC,EAAU;IACpC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3D,CAAC"}

package/build/module/src/util/stream.d.ts

@@ -0,0 +1,10 @@
+/// <reference types="node" />
+import { TransformStream } from 'stream/web';
+export interface TextDecoderCommon {
+    readonly encoding: string;
+    readonly fatal: boolean;
+    readonly ignoreBOM: boolean;
+}
+export declare class TextDecoderStreamPolyfill extends TransformStream<ArrayBuffer | Uint8Array, string> {
+    constructor();
+}

package/build/module/src/util/stream.js

@@ -0,0 +1,28 @@
+import { TransformStream } from 'stream/web';
+class TextDecodeTransformer {
+    decoder;
+    constructor() {
+        this.decoder = new TextDecoder();
+    }
+    transform(chunk, controller) {
+        if (!(chunk instanceof ArrayBuffer || ArrayBuffer.isView(chunk))) {
+            throw new TypeError('Input data must be a BufferSource');
+        }
+        const text = this.decoder.decode(chunk, { stream: true });
+        if (text.length !== 0) {
+            controller.enqueue(text);
+        }
+    }
+    flush(controller) {
+        const text = this.decoder.decode();
+        if (text.length !== 0) {
+            controller.enqueue(text);
+        }
+    }
+}
+export class TextDecoderStreamPolyfill extends TransformStream {
+    constructor() {
+        super(new TextDecodeTransformer());
+    }
+}
+//# sourceMappingURL=stream.js.map

package/build/module/src/util/stream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream.js","sourceRoot":"","sources":["../../../../src/util/stream.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,eAAe,EAEhB,MAAM,YAAY,CAAC;AAQpB,MAAM,qBAAqB;IAGjB,OAAO,CAAC;IAEhB;QACE,IAAI,CAAC,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;IACnC,CAAC;IAED,SAAS,CACP,KAA+B,EAC/B,UAAoD;QAEpD,IAAI,CAAC,CAAC,KAAK,YAAY,WAAW,IAAI,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;YACjE,MAAM,IAAI,SAAS,CAAC,mCAAmC,CAAC,CAAC;QAC3D,CAAC;QACD,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1D,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtB,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAoD;QACxD,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;QACnC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtB,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;CACF;AAED,MAAM,OAAO,yBAA0B,SAAQ,eAG9C;IACC;QACE,KAAK,CAAC,IAAI,qBAAqB,EAAE,CAAC,CAAC;IACrC,CAAC;CACF"}

package/build/module/src/util/transform.d.ts

@@ -0,0 +1,8 @@
+/// <reference types="node" />
+import { TransformStream } from 'stream/web';
+export declare class JSONStringifyStream<O> extends TransformStream<string, O> {
+    constructor();
+}
+export declare class RespTransformStream<I, O> extends TransformStream<I, O> {
+    constructor(transformFn: (arg0: I) => O, doneCallback?: (args0: readonly O[]) => Promise<void>);
+}

package/build/module/src/util/transform.js

@@ -0,0 +1,57 @@
+import { TransformStream } from 'stream/web';
+class JSONTransformer {
+    async transform(obj, controller) {
+        (obj.split('\n') ?? [])
+            .map(extractJson)
+            .map((v) => {
+            try {
+                return v && v.length > 0 ? JSON.parse(v) : null;
+            }
+            catch (e) {
+                return null;
+            }
+        })
+            .filter((v) => v)
+            .forEach((v) => v && controller.enqueue(v));
+    }
+}
+export class JSONStringifyStream extends TransformStream {
+    constructor() {
+        super(new JSONTransformer());
+    }
+}
+class TypeTransformer {
+    buffer;
+    doneCallback;
+    transformFn;
+    constructor(transformFn, doneCallback) {
+        this.transformFn = transformFn;
+        this.doneCallback = doneCallback;
+        this.buffer = doneCallback ? [] : undefined;
+    }
+    async transform(obj, controller) {
+        const val = this.transformFn(obj);
+        if (val) {
+            controller.enqueue(val);
+            this.buffer?.push(val);
+        }
+    }
+    async flush(controller) {
+        await this.doneCallback?.(this.buffer ?? []);
+        controller.terminate();
+    }
+}
+export class RespTransformStream extends TransformStream {
+    constructor(transformFn, doneCallback) {
+        super(new TypeTransformer(transformFn, doneCallback));
+    }
+}
+const extractJson = (str) => {
+    const startIndex = str.indexOf('{');
+    const endIndex = str.lastIndexOf('}');
+    if (startIndex !== -1 && endIndex !== -1) {
+        return str.substring(startIndex, endIndex + 1);
+    }
+    return null;
+};
+//# sourceMappingURL=transform.js.map