@ax-llm/ax 9.0.0

package/build/module/src/ai/openai/types.d.ts

@@ -0,0 +1,192 @@
+import type { TextModelConfig } from '../types.js';
+/**
+ * OpenAI: Models for text generation
+ * @export
+ */
+export declare enum OpenAIModel {
+    GPT4 = "gpt-4",
+    GPT4O = "gpt-4o",
+    GPT4Turbo = "gpt-4-turbo",
+    GPT35Turbo = "gpt-3.5-turbo",
+    GPT35TurboInstruct = "gpt-3.5-turbo-instruct",
+    GPT35TextDavinci002 = "text-davinci-002",
+    GPT3TextBabbage002 = "text-babbage-002",
+    GPT3TextAda001 = "text-ada-001"
+}
+/**
+ * OpenAI: Models for use in embeddings
+ * @export
+ */
+export declare enum OpenAIEmbedModels {
+    TextEmbeddingAda002 = "text-embedding-ada-002",
+    TextEmbedding3Small = "text-embedding-3-small",
+    TextEmbedding3Large = "text-embedding-3-large"
+}
+/**
+ * OpenAI: Models for for audio transcription
+ * @export
+ */
+export declare enum OpenAIAudioModel {
+    Whisper1 = "whisper-1"
+}
+/**
+ * OpenAI: Model options for text generation
+ * @export
+ */
+export type OpenAIConfig = Omit<TextModelConfig, 'topK'> & {
+    model: OpenAIModel | string;
+    embedModel?: OpenAIEmbedModels | string;
+    audioModel?: OpenAIAudioModel | string;
+    user?: string;
+    responseFormat?: 'json_object';
+    bestOf?: number;
+    logitBias?: Map<string, number>;
+    suffix?: string | null;
+    stop?: string[];
+    logprobs?: number;
+    echo?: boolean;
+};
+export type OpenAILogprob = {
+    tokens: string[];
+    token_logprobs: number[];
+    top_logprobs: Map<string, number>;
+    text_offset: number[];
+};
+export type OpenAIUsage = {
+    prompt_tokens: number;
+    completion_tokens: number;
+    total_tokens: number;
+};
+export interface OpenAIResponseDelta<T> {
+    id: string;
+    object: string;
+    created: number;
+    model: string;
+    choices: {
+        index: number;
+        delta: T;
+        finish_reason: 'stop' | 'length' | 'content_filter' | 'tool_calls';
+    }[];
+    usage?: OpenAIUsage;
+    system_fingerprint: string;
+}
+export type OpenAIChatRequest = {
+    model: string;
+    messages: ({
+        role: 'system';
+        content: string;
+    } | {
+        role: 'user';
+        content: string;
+        name?: string;
+    } | {
+        role: 'assistant';
+        content: string | null;
+        name?: string;
+        tool_calls?: {
+            type: 'function';
+            function: {
+                name: string;
+                arguments?: string;
+            };
+        }[];
+    } | {
+        role: 'tool';
+        content: string;
+        tool_call_id: string;
+    })[];
+    tools?: {
+        type: 'function';
+        function: {
+            name: string;
+            description: string;
+            parameters?: object;
+        };
+    }[];
+    tool_choice?: 'none' | 'auto' | 'required' | {
+        type: 'function';
+        function: {
+            name: string;
+        };
+    };
+    response_format?: {
+        type: string;
+    };
+    max_tokens: number;
+    temperature?: number;
+    top_p?: number;
+    n?: number;
+    stream?: boolean;
+    stop?: readonly string[];
+    presence_penalty?: number;
+    frequency_penalty?: number;
+    logit_bias?: Map<string, number>;
+    user?: string;
+    organization?: string;
+};
+export type OpenAIChatResponse = {
+    id: string;
+    object: 'chat.completion';
+    created: number;
+    model: string;
+    choices: {
+        index: number;
+        message: {
+            role: string;
+            content: string;
+            tool_calls?: {
+                id: string;
+                type: 'function';
+                function: {
+                    name: string;
+                    arguments: string;
+                };
+            }[];
+        };
+        finish_reason: 'stop' | 'length' | 'content_filter' | 'tool_calls';
+    }[];
+    usage?: OpenAIUsage;
+    error?: {
+        message: string;
+        type: string;
+        param: string;
+        code: number;
+    };
+    system_fingerprint: string;
+};
+export type OpenAIChatResponseDelta = OpenAIResponseDelta<{
+    content: string;
+    role?: string;
+    tool_calls?: (NonNullable<OpenAIChatResponse['choices'][0]['message']['tool_calls']>[0] & {
+        index: number;
+    })[];
+}>;
+export type OpenAIEmbedRequest = {
+    input: readonly string[];
+    model: string;
+    user?: string;
+};
+export type OpenAIEmbedResponse = {
+    model: string;
+    data: {
+        embedding: readonly number[];
+        index: number;
+    }[];
+    usage: OpenAIUsage;
+};
+export type OpenAIAudioRequest = {
+    model: string;
+    prompt?: string;
+    response_format: 'verbose_json';
+    temperature?: number;
+    language?: string;
+};
+export type OpenAIAudioResponse = {
+    duration: number;
+    segments: {
+        id: number;
+        start: number;
+        end: number;
+        text: string;
+    }[];
+};

package/build/module/src/ai/openai/types.js

@@ -0,0 +1,35 @@
+// Transcribe = '/v1/audio/transcriptions',
+/**
+ * OpenAI: Models for text generation
+ * @export
+ */
+export var OpenAIModel;
+(function (OpenAIModel) {
+    OpenAIModel["GPT4"] = "gpt-4";
+    OpenAIModel["GPT4O"] = "gpt-4o";
+    OpenAIModel["GPT4Turbo"] = "gpt-4-turbo";
+    OpenAIModel["GPT35Turbo"] = "gpt-3.5-turbo";
+    OpenAIModel["GPT35TurboInstruct"] = "gpt-3.5-turbo-instruct";
+    OpenAIModel["GPT35TextDavinci002"] = "text-davinci-002";
+    OpenAIModel["GPT3TextBabbage002"] = "text-babbage-002";
+    OpenAIModel["GPT3TextAda001"] = "text-ada-001";
+})(OpenAIModel || (OpenAIModel = {}));
+/**
+ * OpenAI: Models for use in embeddings
+ * @export
+ */
+export var OpenAIEmbedModels;
+(function (OpenAIEmbedModels) {
+    OpenAIEmbedModels["TextEmbeddingAda002"] = "text-embedding-ada-002";
+    OpenAIEmbedModels["TextEmbedding3Small"] = "text-embedding-3-small";
+    OpenAIEmbedModels["TextEmbedding3Large"] = "text-embedding-3-large";
+})(OpenAIEmbedModels || (OpenAIEmbedModels = {}));
+/**
+ * OpenAI: Models for for audio transcription
+ * @export
+ */
+export var OpenAIAudioModel;
+(function (OpenAIAudioModel) {
+    OpenAIAudioModel["Whisper1"] = "whisper-1";
+})(OpenAIAudioModel || (OpenAIAudioModel = {}));
+//# sourceMappingURL=types.js.map

package/build/module/src/ai/openai/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../../src/ai/openai/types.ts"],"names":[],"mappings":"AAEA,2CAA2C;AAE3C;;;GAGG;AACH,MAAM,CAAN,IAAY,WASX;AATD,WAAY,WAAW;IACrB,6BAAc,CAAA;IACd,+BAAgB,CAAA;IAChB,wCAAyB,CAAA;IACzB,2CAA4B,CAAA;IAC5B,4DAA6C,CAAA;IAC7C,uDAAwC,CAAA;IACxC,sDAAuC,CAAA;IACvC,8CAA+B,CAAA;AACjC,CAAC,EATW,WAAW,KAAX,WAAW,QAStB;AAED;;;GAGG;AACH,MAAM,CAAN,IAAY,iBAIX;AAJD,WAAY,iBAAiB;IAC3B,mEAA8C,CAAA;IAC9C,mEAA8C,CAAA;IAC9C,mEAA8C,CAAA;AAChD,CAAC,EAJW,iBAAiB,KAAjB,iBAAiB,QAI5B;AAED;;;GAGG;AACH,MAAM,CAAN,IAAY,gBAEX;AAFD,WAAY,gBAAgB;IAC1B,0CAAsB,CAAA;AACxB,CAAC,EAFW,gBAAgB,KAAhB,gBAAgB,QAE3B"}

package/build/module/src/ai/together/api.d.ts

@@ -0,0 +1,22 @@
+import type { AIServiceOptions } from '../../text/types.js';
+import { OpenAI } from '../openai/api.js';
+import type { OpenAIConfig } from '../openai/types.js';
+type TogetherAIConfig = OpenAIConfig;
+/**
+ * TogetherAI: Default Model options for text generation
+ * @export
+ */
+export declare const TogetherDefaultConfig: () => TogetherAIConfig;
+export interface TogetherArgs {
+    apiKey: string;
+    config: Readonly<TogetherAIConfig>;
+    options?: Readonly<AIServiceOptions>;
+}
+/**
+ * TogetherAI: AI Service
+ * @export
+ */
+export declare class Together extends OpenAI {
+    constructor({ apiKey, config, options }: Readonly<TogetherArgs>);
+}
+export {};

package/build/module/src/ai/together/api.js

@@ -0,0 +1,29 @@
+import { BaseAIDefaultConfig } from '../base.js';
+import { OpenAI } from '../openai/api.js';
+/**
+ * TogetherAI: Default Model options for text generation
+ * @export
+ */
+export const TogetherDefaultConfig = () => structuredClone({
+    model: 'llama2-70b-4096',
+    ...BaseAIDefaultConfig()
+});
+/**
+ * TogetherAI: AI Service
+ * @export
+ */
+export class Together extends OpenAI {
+    constructor({ apiKey, config = TogetherDefaultConfig(), options }) {
+        if (!apiKey || apiKey === '') {
+            throw new Error('Together API key not set');
+        }
+        super({
+            apiKey,
+            config,
+            options,
+            apiURL: 'https://api.together.xyz/v1'
+        });
+        super.setName('Together');
+    }
+}
+//# sourceMappingURL=api.js.map

package/build/module/src/ai/together/api.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../../src/ai/together/api.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AACjD,OAAO,EAAE,MAAM,EAAE,MAAM,kBAAkB,CAAC;AAK1C;;;GAGG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAG,GAAqB,EAAE,CAC1D,eAAe,CAAC;IACd,KAAK,EAAE,iBAAiB;IACxB,GAAG,mBAAmB,EAAE;CACzB,CAAC,CAAC;AAQL;;;GAGG;AACH,MAAM,OAAO,QAAS,SAAQ,MAAM;IAClC,YAAY,EACV,MAAM,EACN,MAAM,GAAG,qBAAqB,EAAE,EAChC,OAAO,EACgB;QACvB,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAC9C,CAAC;QACD,KAAK,CAAC;YACJ,MAAM;YACN,MAAM;YACN,OAAO;YACP,MAAM,EAAE,6BAA6B;SACtC,CAAC,CAAC;QAEH,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAC5B,CAAC;CACF"}

package/build/module/src/ai/together/index.d.ts

@@ -0,0 +1 @@
+export * from './api.js';

package/build/module/src/ai/together/index.js

@@ -0,0 +1,2 @@
+export * from './api.js';
+//# sourceMappingURL=index.js.map

package/build/module/src/ai/together/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/ai/together/index.ts"],"names":[],"mappings":"AAAA,cAAc,UAAU,CAAC"}

package/build/module/src/ai/together/info.d.ts

@@ -0,0 +1,2 @@
+import type { TextModelInfo } from '../types.js';
+export declare const modelInfo: TextModelInfo[];

package/build/module/src/ai/together/info.js

@@ -0,0 +1,2 @@
+export const modelInfo = [];
+//# sourceMappingURL=info.js.map

package/build/module/src/ai/together/info.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"info.js","sourceRoot":"","sources":["../../../../../src/ai/together/info.ts"],"names":[],"mappings":"AAEA,MAAM,CAAC,MAAM,SAAS,GAAoB,EAAE,CAAC"}

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

@@ -0,0 +1,70 @@
+import type { AITextTraceStep } from '../types/index.js';
+export type TextModelInfo = {
+    name: string;
+    currency?: string;
+    characterIsToken?: boolean;
+    promptTokenCostPer1M?: number;
+    completionTokenCostPer1M?: number;
+    aliases?: string[];
+};
+export type TokenUsage = {
+    promptTokens: number;
+    completionTokens: number;
+    totalTokens: number;
+};
+export type TextModelConfig = {
+    maxTokens?: number;
+    temperature?: number;
+    topP?: number;
+    topK?: number;
+    presencePenalty?: number;
+    frequencyPenalty?: number;
+    stopSequences?: string[];
+    endSequences?: string[];
+    stream?: boolean;
+    n?: number;
+};
+export type TextResponseFunctionCall = {
+    id?: string;
+    name: string;
+    args: string;
+};
+export type TextResponseResult = {
+    content: string | null;
+    name?: string;
+    id?: string;
+    functionCalls?: {
+        id: string;
+        type: 'function';
+        function: {
+            name: string;
+            arguments?: string | object;
+        };
+    }[];
+    finishReason?: 'stop' | 'length' | 'function_call' | 'content_filter' | 'error';
+};
+export type TextResponse = {
+    sessionId?: string;
+    remoteId?: string;
+    results: readonly TextResponseResult[];
+    modelUsage?: TokenUsage;
+    embedModelUsage?: TokenUsage;
+};
+export type EmbedResponse = {
+    remoteId?: string;
+    sessionId?: string;
+    embeddings: readonly (readonly number[])[];
+    modelUsage?: TokenUsage;
+};
+export type TranscriptResponse = {
+    sessionId?: string;
+    duration: number;
+    segments: {
+        id: number;
+        start: number;
+        end: number;
+        text: string;
+    }[];
+};
+export type LoggerFunction = (traceStep: Readonly<AITextTraceStep>) => void;
+export type RateLimiterFunction = <T>(func: unknown) => T;

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

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../src/ai/types.ts"],"names":[],"mappings":""}

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

@@ -0,0 +1,13 @@
+import type { TextModelInfo, TextResponse, TextResponseResult } from './types.js';
+export declare const findItemByNameOrAlias: (list: readonly TextModelInfo[], name: string) => TextModelInfo | undefined;
+export declare const uniqBy: <T>(array: readonly T[], uniqueField: (value: T) => unknown) => T[];
+export declare const parseFunction: (value: string) => {
+    name: string;
+    args?: string;
+} | undefined;
+export interface mergeFunctionsState {
+    lastId: string;
+}
+export declare function mergeFunctionCalls(functionCalls: NonNullable<TextResponseResult['functionCalls']>, functionCallDeltas: Readonly<NonNullable<TextResponseResult['functionCalls']>>, state?: mergeFunctionsState): NonNullable<TextResponseResult['functionCalls']>[0] | undefined;
+export declare function mergeTextResponses(responses: readonly TextResponse[]): TextResponse;
+export declare const hashObject: (obj: object) => string;

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

@@ -0,0 +1,112 @@
+import { createHash } from 'crypto';
+export const findItemByNameOrAlias = (list, name) => {
+    for (const item of list) {
+        if (item.name === name || item.aliases?.includes(name)) {
+            return item;
+        }
+    }
+    return undefined;
+};
+export const uniqBy = (array, uniqueField) => {
+    const uniqueValues = new Map();
+    array.forEach((value) => {
+        const field = uniqueField(value);
+        if (!uniqueValues.has(field)) {
+            uniqueValues.set(field, value);
+        }
+    });
+    return Array.from(uniqueValues.values());
+};
+const functionCallRe = /(\w+)\((.*)\)/s;
+export const parseFunction = (value) => {
+    let v;
+    // extract function calls
+    if ((v = functionCallRe.exec(value)) !== null) {
+        const name = v.at(1)?.trim();
+        const args = v.at(2)?.trim();
+        if (!name || name) {
+            throw new Error(`Invalid function format: ${value}`);
+        }
+        return { name, args };
+    }
+    return;
+};
+export function mergeFunctionCalls(
+// eslint-disable-next-line functional/prefer-immutable-types
+functionCalls, functionCallDeltas, 
+// eslint-disable-next-line functional/prefer-immutable-types
+state) {
+    for (const _fc of functionCallDeltas) {
+        const fc = functionCalls.find((fc) => fc.id === _fc.id);
+        if (fc) {
+            if (typeof _fc.function.name == 'string') {
+                fc.function.name += _fc.function.name;
+            }
+            if (typeof _fc.function.arguments == 'string') {
+                fc.function.arguments += _fc.function.arguments;
+            }
+            if (typeof _fc.function.arguments == 'object') {
+                fc.function.arguments = _fc.function.arguments;
+            }
+        }
+        else {
+            functionCalls.push(_fc);
+        }
+        if (!state) {
+            continue;
+        }
+        let retFunc;
+        if (state.lastId !== _fc.id) {
+            retFunc = functionCalls.find((fc) => fc.id === state.lastId);
+        }
+        state.lastId = _fc.id;
+        if (retFunc) {
+            return retFunc;
+        }
+    }
+}
+export function mergeTextResponses(responses) {
+    const functionCalls = [];
+    let content = '';
+    // Variables to store the other overwritten values
+    let lastSessionId;
+    let lastRemoteId;
+    let lastModelUsage;
+    let lastEmbedModelUsage;
+    let lastResults = [];
+    for (const response of responses) {
+        for (const result of response.results ?? []) {
+            if (result.content) {
+                content += result.content;
+            }
+            if (result.functionCalls) {
+                mergeFunctionCalls(functionCalls, result.functionCalls);
+            }
+        }
+        // Overwrite other values
+        lastSessionId = response.sessionId;
+        lastRemoteId = response.remoteId;
+        lastModelUsage = response.modelUsage;
+        lastEmbedModelUsage = response.embedModelUsage;
+        lastResults = response.results ?? [];
+    }
+    return {
+        sessionId: lastSessionId,
+        remoteId: lastRemoteId,
+        results: [
+            {
+                ...lastResults[0],
+                content,
+                functionCalls
+            }
+        ],
+        modelUsage: lastModelUsage,
+        embedModelUsage: lastEmbedModelUsage
+    };
+}
+export const hashObject = (obj) => {
+    const hash = createHash('sha256');
+    hash.update(JSON.stringify(obj));
+    return hash.digest('hex');
+};
+//# sourceMappingURL=util.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"util.js","sourceRoot":"","sources":["../../../../src/ai/util.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AASpC,MAAM,CAAC,MAAM,qBAAqB,GAAG,CACnC,IAA8B,EAC9B,IAAY,EACe,EAAE;IAC7B,KAAK,MAAM,IAAI,IAAI,IAAI,EAAE,CAAC;QACxB,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,IAAI,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YACvD,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,MAAM,GAAG,CACpB,KAAmB,EACnB,WAAkC,EAC7B,EAAE;IACP,MAAM,YAAY,GAAG,IAAI,GAAG,EAAE,CAAC;IAE/B,KAAK,CAAC,OAAO,CAAC,CAAC,KAAQ,EAAE,EAAE;QACzB,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC;QAEjC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;YAC7B,YAAY,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QACjC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC;AAC3C,CAAC,CAAC;AAEF,MAAM,cAAc,GAAG,gBAAgB,CAAC;AAExC,MAAM,CAAC,MAAM,aAAa,GAAG,CAC3B,KAAa,EACgC,EAAE;IAC/C,IAAI,CAAkB,CAAC;IAEvB,yBAAyB;IACzB,IAAI,CAAC,CAAC,GAAG,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QAC9C,MAAM,IAAI,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC;QAC7B,MAAM,IAAI,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC;QAC7B,IAAI,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,4BAA4B,KAAK,EAAE,CAAC,CAAC;QACvD,CAAC;QACD,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;IACxB,CAAC;IACD,OAAO;AACT,CAAC,CAAC;AAMF,MAAM,UAAU,kBAAkB;AAChC,6DAA6D;AAC7D,aAA+D,EAC/D,kBAEC;AACD,6DAA6D;AAC7D,KAA2B;IAE3B,KAAK,MAAM,GAAG,IAAI,kBAAkB,EAAE,CAAC;QACrC,MAAM,EAAE,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,KAAK,GAAG,CAAC,EAAE,CAAC,CAAC;QAExD,IAAI,EAAE,EAAE,CAAC;YACP,IAAI,OAAO,GAAG,CAAC,QAAQ,CAAC,IAAI,IAAI,QAAQ,EAAE,CAAC;gBACzC,EAAE,CAAC,QAAQ,CAAC,IAAI,IAAI,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC;YACxC,CAAC;YAED,IAAI,OAAO,GAAG,CAAC,QAAQ,CAAC,SAAS,IAAI,QAAQ,EAAE,CAAC;gBAC9C,EAAE,CAAC,QAAQ,CAAC,SAAS,IAAI,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC;YAClD,CAAC;YAED,IAAI,OAAO,GAAG,CAAC,QAAQ,CAAC,SAAS,IAAI,QAAQ,EAAE,CAAC;gBAC9C,EAAE,CAAC,QAAQ,CAAC,SAAS,GAAG,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC;YACjD,CAAC;QACH,CAAC;aAAM,CAAC;YACN,aAAa,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;QAED,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,SAAS;QACX,CAAC;QAED,IAAI,OAAO,CAAC;QACZ,IAAI,KAAK,CAAC,MAAM,KAAK,GAAG,CAAC,EAAE,EAAE,CAAC;YAC5B,OAAO,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,KAAK,KAAK,CAAC,MAAM,CAAC,CAAC;QAC/D,CAAC;QAED,KAAK,CAAC,MAAM,GAAG,GAAG,CAAC,EAAE,CAAC;QACtB,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,OAAO,CAAC;QACjB,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,SAAkC;IAElC,MAAM,aAAa,GAAqD,EAAE,CAAC;IAC3E,IAAI,OAAO,GAAG,EAAE,CAAC;IAEjB,kDAAkD;IAClD,IAAI,aAAiC,CAAC;IACtC,IAAI,YAAgC,CAAC;IACrC,IAAI,cAAsC,CAAC;IAC3C,IAAI,mBAA2C,CAAC;IAChD,IAAI,WAAW,GAAkC,EAAE,CAAC;IAEpD,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QACjC,KAAK,MAAM,MAAM,IAAI,QAAQ,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;YAC5C,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,OAAO,IAAI,MAAM,CAAC,OAAO,CAAC;YAC5B,CAAC;YACD,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;gBACzB,kBAAkB,CAAC,aAAa,EAAE,MAAM,CAAC,aAAa,CAAC,CAAC;YAC1D,CAAC;QACH,CAAC;QAED,yBAAyB;QACzB,aAAa,GAAG,QAAQ,CAAC,SAAS,CAAC;QACnC,YAAY,GAAG,QAAQ,CAAC,QAAQ,CAAC;QACjC,cAAc,GAAG,QAAQ,CAAC,UAAU,CAAC;QACrC,mBAAmB,GAAG,QAAQ,CAAC,eAAe,CAAC;QAC/C,WAAW,GAAG,QAAQ,CAAC,OAAO,IAAI,EAAE,CAAC;IACvC,CAAC;IAED,OAAO;QACL,SAAS,EAAE,aAAa;QACxB,QAAQ,EAAE,YAAY;QACtB,OAAO,EAAE;YACP;gBACE,GAAG,WAAW,CAAC,CAAC,CAAC;gBACjB,OAAO;gBACP,aAAa;aACd;SACF;QACD,UAAU,EAAE,cAAc;QAC1B,eAAe,EAAE,mBAAmB;KACrC,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,MAAM,UAAU,GAAG,CAAC,GAAW,EAAE,EAAE;IACxC,MAAM,IAAI,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC;IAClC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;IACjC,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC5B,CAAC,CAAC"}

package/build/module/src/db/base.d.ts

@@ -0,0 +1,24 @@
+/// <reference types="node" />
+import { type Span, type Tracer } from '../trace/index.js';
+import type { DBQueryRequest, DBQueryResponse, DBService, DBUpsertRequest, DBUpsertResponse } from './types.js';
+export interface BaseArgs {
+    fetch?: typeof fetch;
+    tracer?: Tracer;
+}
+export interface BaseOpOptions {
+    span?: Span;
+}
+export declare class BaseDB implements DBService {
+    protected name: string;
+    protected fetch?: typeof fetch;
+    private tracer?;
+    _upsert?: (req: Readonly<DBUpsertRequest>, update?: boolean, options?: Readonly<BaseOpOptions>) => Promise<DBUpsertResponse>;
+    _batchUpsert?: (batchReq: Readonly<DBUpsertRequest[]>, update?: boolean, options?: Readonly<BaseOpOptions>) => Promise<DBUpsertResponse>;
+    _query?: (req: Readonly<DBQueryRequest>, options?: Readonly<BaseOpOptions>) => Promise<DBQueryResponse>;
+    constructor({ name, fetch, tracer }: Readonly<BaseArgs & {
+        name: string;
+    }>);
+    upsert(req: Readonly<DBUpsertRequest>, update?: boolean): Promise<DBUpsertResponse>;
+    batchUpsert(req: Readonly<DBUpsertRequest[]>, update?: boolean): Promise<DBUpsertResponse>;
+    query(req: Readonly<DBQueryRequest>): Promise<DBQueryResponse>;
+}

package/build/module/src/db/base.js

@@ -0,0 +1,90 @@
+import { SpanAttributes, SpanKind } from '../trace/index.js';
+export class BaseDB {
+    name;
+    fetch;
+    tracer;
+    _upsert;
+    _batchUpsert;
+    _query;
+    constructor({ name, fetch, tracer }) {
+        this.name = name;
+        this.fetch = fetch;
+        this.tracer = tracer;
+    }
+    async upsert(req, update) {
+        if (!this._upsert) {
+            throw new Error('upsert() not implemented');
+        }
+        const _upsert = this._upsert;
+        if (!this.tracer) {
+            return await _upsert(req, update);
+        }
+        return await this.tracer?.startActiveSpan('DB Upsert Request', {
+            kind: SpanKind.SERVER,
+            attributes: {
+                [SpanAttributes.DB_SYSTEM]: this.name,
+                [SpanAttributes.DB_OPERATION_NAME]: 'upsert',
+                [SpanAttributes.DB_TABLE]: req.table,
+                [SpanAttributes.DB_NAMESPACE]: req.namespace,
+                [SpanAttributes.DB_OPERATION_NAME]: update ? 'update' : 'insert'
+            }
+        }, async (span) => {
+            const res = await _upsert(req, update, { span });
+            span.end();
+            return res;
+        });
+    }
+    async batchUpsert(req, update) {
+        if (!this._batchUpsert) {
+            throw new Error('batchUpsert() not implemented');
+        }
+        if (req.length == 0) {
+            throw new Error('Batch request is empty');
+        }
+        if (!req[0]) {
+            throw new Error('Batch request is invalid first element is undefined');
+        }
+        const _batchUpsert = this._batchUpsert;
+        if (!this.tracer) {
+            return await _batchUpsert(req, update);
+        }
+        return await this.tracer?.startActiveSpan('DB Batch Upsert Request', {
+            kind: SpanKind.SERVER,
+            attributes: {
+                [SpanAttributes.DB_SYSTEM]: this.name,
+                [SpanAttributes.DB_OPERATION_NAME]: 'upsert',
+                [SpanAttributes.DB_TABLE]: req[0].table,
+                [SpanAttributes.DB_NAMESPACE]: req[0].namespace,
+                [SpanAttributes.DB_OPERATION_NAME]: update ? 'update' : 'insert'
+            }
+        }, async (span) => {
+            const res = await _batchUpsert(req, update, { span });
+            span.end();
+            return res;
+        });
+    }
+    async query(req) {
+        if (!this._query) {
+            throw new Error('query() not implemented');
+        }
+        const _query = this._query;
+        if (!this.tracer) {
+            return await _query(req);
+        }
+        return await this.tracer?.startActiveSpan('DB Query Request', {
+            kind: SpanKind.SERVER,
+            attributes: {
+                [SpanAttributes.DB_SYSTEM]: this.name,
+                [SpanAttributes.DB_OPERATION_NAME]: 'upsert',
+                [SpanAttributes.DB_TABLE]: req.table,
+                [SpanAttributes.DB_NAMESPACE]: req.namespace,
+                [SpanAttributes.DB_OPERATION_NAME]: 'query'
+            }
+        }, async (span) => {
+            const res = await _query(req, { span });
+            span.end();
+            return res;
+        });
+    }
+}
+//# sourceMappingURL=base.js.map

package/build/module/src/db/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../../../src/db/base.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,cAAc,EACd,QAAQ,EAET,MAAM,mBAAmB,CAAC;AAmB3B,MAAM,OAAO,MAAM;IACP,IAAI,CAAS;IACb,KAAK,CAAgB;IACvB,MAAM,CAAU;IAExB,OAAO,CAIwB;IAE/B,YAAY,CAImB;IAE/B,MAAM,CAGwB;IAE9B,YAAY,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAyC;QACxE,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,MAAM,CACV,GAA8B,EAC9B,MAAgB;QAEhB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAC9C,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QAE7B,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,OAAO,MAAM,OAAO,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACpC,CAAC;QAED,OAAO,MAAM,IAAI,CAAC,MAAM,EAAE,eAAe,CACvC,mBAAmB,EACnB;YACE,IAAI,EAAE,QAAQ,CAAC,MAAM;YACrB,UAAU,EAAE;gBACV,CAAC,cAAc,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,IAAI;gBACrC,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,QAAQ;gBAC5C,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,GAAG,CAAC,KAAK;gBACpC,CAAC,cAAc,CAAC,YAAY,CAAC,EAAE,GAAG,CAAC,SAAS;gBAC5C,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ;aACjE;SACF,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;YACjD,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,OAAO,GAAG,CAAC;QACb,CAAC,CACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,WAAW,CACf,GAAgC,EAChC,MAAgB;QAEhB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;QACnD,CAAC;QACD,IAAI,GAAG,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAC5C,CAAC;QACD,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;QACzE,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC;QAEvC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,OAAO,MAAM,YAAY,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACzC,CAAC;QAED,OAAO,MAAM,IAAI,CAAC,MAAM,EAAE,eAAe,CACvC,yBAAyB,EACzB;YACE,IAAI,EAAE,QAAQ,CAAC,MAAM;YACrB,UAAU,EAAE;gBACV,CAAC,cAAc,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,IAAI;gBACrC,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,QAAQ;gBAC5C,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK;gBACvC,CAAC,cAAc,CAAC,YAAY,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS;gBAC/C,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ;aACjE;SACF,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,MAAM,GAAG,GAAG,MAAM,YAAY,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;YACtD,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,OAAO,GAAG,CAAC;QACb,CAAC,CACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,GAA6B;QACvC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QAC7C,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAE3B,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,OAAO,MAAM,MAAM,CAAC,GAAG,CAAC,CAAC;QAC3B,CAAC;QAED,OAAO,MAAM,IAAI,CAAC,MAAM,EAAE,eAAe,CACvC,kBAAkB,EAClB;YACE,IAAI,EAAE,QAAQ,CAAC,MAAM;YACrB,UAAU,EAAE;gBACV,CAAC,cAAc,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,IAAI;gBACrC,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,QAAQ;gBAC5C,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,GAAG,CAAC,KAAK;gBACpC,CAAC,cAAc,CAAC,YAAY,CAAC,EAAE,GAAG,CAAC,SAAS;gBAC5C,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,OAAO;aAC5C;SACF,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,GAAG,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;YACxC,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,OAAO,GAAG,CAAC;QACb,CAAC,CACF,CAAC;IACJ,CAAC;CACF"}

package/build/module/src/db/cloudflare.d.ts

@@ -0,0 +1,20 @@
+/// <reference types="node" />
+import { type BaseArgs, BaseDB, type BaseOpOptions } from './base.js';
+import type { DBQueryRequest, DBQueryResponse, DBUpsertRequest, DBUpsertResponse } from './types.js';
+export interface CloudflareArgs {
+    apiKey: string;
+    accountId: string;
+    fetch?: typeof fetch;
+}
+/**
+ * Cloudflare: DB Service
+ * @export
+ */
+export declare class Cloudflare extends BaseDB {
+    private apiKey;
+    private accountId;
+    constructor({ apiKey, accountId, fetch, tracer }: Readonly<CloudflareArgs & BaseArgs>);
+    _upsert: (req: Readonly<DBUpsertRequest>, update?: boolean, options?: Readonly<BaseOpOptions>) => Promise<DBUpsertResponse>;
+    batchUpsert: (batchReq: Readonly<DBUpsertRequest[]>, update?: boolean, options?: Readonly<BaseOpOptions>) => Promise<DBUpsertResponse>;
+    query: (req: Readonly<DBQueryRequest>, options?: Readonly<BaseOpOptions>) => Promise<DBQueryResponse>;
+}