UNPKG

@maximai/maxim-js

Version:

Maxim AI JS SDK. Visit https://getmaxim.ai for more info.

89 lines (88 loc) 5.09 kB
import { MaximCache } from "../cache/cache"; import { ChatCompletionMessage, CompletionRequest } from "../models/prompt"; import { ErrorConfig, Error as MaximError } from "./components"; import { ChatCompletionResult, Generation, GenerationConfig, GenerationError, TextCompletionResult } from "./components/generation"; import { Retrieval, RetrievalConfig } from "./components/retrieval"; import { Session, SessionConfig } from "./components/session"; import { Span, SpanConfig } from "./components/span"; import { ToolCall, ToolCallConfig, ToolCallError } from "./components/toolCall"; import { Trace, TraceConfig } from "./components/trace"; export type LoggerConfig = { id: string; autoFlush?: boolean; flushIntervalSeconds?: number; }; export declare class MaximLogger { private _id; private readonly writer; private readonly isDebug; constructor(params: { config: LoggerConfig; apiKey: string; baseUrl: string; isDebug?: boolean; cache: MaximCache; raiseExceptions: boolean; }); session(config: SessionConfig): Session; trace(config: TraceConfig): Trace; get id(): string; cleanup(): Promise<void>; sessionTag(sessionId: string, key: string, value: string): void; sessionEnd(sessionId: string, data?: any): void; sessionFeedback(sessionId: string, feedback: { score: number; comment?: string; }): void; sessionTrace(sessionId: string, config: TraceConfig): Trace; sessionEvaluate(sessionId: string): import("./components").EvaluateContainer; traceGeneration(traceId: string, config: GenerationConfig): Generation; traceToolCall(traceId: string, config: ToolCallConfig): ToolCall; traceRetrieval(traceId: string, config: RetrievalConfig): Retrieval; traceOutput(traceId: string, output: string): void; traceError(traceId: string, config: ErrorConfig): MaximError; traceInput(traceId: string, input: string): void; traceSpan(traceId: string, config: SpanConfig): Span; traceAddToSession(traceId: string, sessionId: string): void; traceTag(traceId: string, key: string, value: string): void; traceEvent(traceId: string, eventId: string, eventName: string, tags?: Record<string, string>, metadata?: Record<string, unknown>): void; traceEvent(traceId: string, eventName: string, tags?: Record<string, string>, metadata?: Record<string, unknown>): void; traceFeedback(traceId: string, feedback: { score: number; comment?: string; }): void; traceMetadata(traceId: string, metadata: Record<string, unknown>): void; traceEvaluate(traceId: string): import("./components").EvaluateContainer; traceEnd(traceId: string, data?: any): void; generationSetModel(generationId: string, model: string): void; generationAddTag(generationId: string, key: string, value: string): void; generationAddMessage(generationId: string, messages: (CompletionRequest | ChatCompletionMessage)[]): void; generationSetModelParameters(generationId: string, modelParameters: Record<string, any>): void; generationResult(generationId: string, result: TextCompletionResult | ChatCompletionResult): void; generationError(generationId: string, error: GenerationError): void; generationMetadata(generationId: string, metadata: Record<string, unknown>): void; generationEvaluate(generationId: string): import("./components").EvaluateContainer; generationEnd(generationId: string, data?: any): void; spanGeneration(spanId: string, config: GenerationConfig): Generation; spanRetrieval(spanId: string, config: RetrievalConfig): Retrieval; spanToolCall(spanId: string, config: ToolCallConfig): ToolCall; spanSpan(spanId: string, config: SpanConfig): Span; spanTag(spanId: string, key: string, value: string): void; spanError(spanId: string, config: ErrorConfig): MaximError; spanEvent(spanId: string, eventId: string, eventName: string, tags?: Record<string, string>, metadata?: Record<string, unknown>): void; spanEvent(spanId: string, eventName: string, tags?: Record<string, string>, metadata?: Record<string, unknown>): void; spanMetadata(spanId: string, metadata: Record<string, unknown>): void; spanEvaluate(spanId: string): import("./components").EvaluateContainer; spanEnd(spanId: string, data?: any): void; retrievalEnd(retrievalId: string): void; retrievalAddTag(retrievalId: string, key: string, value: string): void; retrievalInput(retrievalId: string, input: string): void; retrievalOutput(retrievalId: string, output: string): void; retrievalMetadata(retrievalId: string, metadata: Record<string, unknown>): void; retrievalEvaluate(retrievalId: string): import("./components").EvaluateContainer; toolCallResult(toolCallId: string, result: string): void; toolCallError(toolCallId: string, error: ToolCallError): void; toolCallAddTag(toolCallId: string, key: string, value: string): void; toolCallMetadata(toolCallId: string, metadata: Record<string, unknown>): void; flush(): void; }