@maximai/maxim-js/src/lib/logger/openai/client.js

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

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.MaximOpenAIClient = void 0;
const chat_1 = require("./chat");
const responses_1 = require("./responses");
const realtime_1 = require("./realtime");
/**
 * A wrapped OpenAI client that automatically logs all chat completions and responses to Maxim.
 *
 * This class provides the same interface as the OpenAI client but with automatic
 * logging of traces and generations for observability purposes.
 *
 * @example
 * ```typescript
 * import OpenAI from 'openai';
 * import { Maxim } from '@maximai/maxim-js';
 * import { MaximOpenAIClient } from '@maximai/maxim-js/openai';
 *
 * const maxim = new Maxim({ apiKey: process.env.MAXIM_API_KEY });
 * const logger = await maxim.logger({ id: 'my-app' });
 *
 * const openai = new OpenAI();
 * const client = new MaximOpenAIClient(openai, logger);
 *
 * // Automatically logged to Maxim
 * const response = await client.chat.completions.create({
 *   model: 'gpt-4',
 *   messages: [{ role: 'user', content: 'Hello!' }]
 * });
 * ```
 *
 * @example
 * ```typescript
 * // With custom trace ID
 * const response = await client.chat.completions.create(
 *   {
 *     model: 'gpt-4',
 *     messages: [{ role: 'user', content: 'Hello!' }]
 *   },
 *   {
 *     headers: {
 *       'x-maxim-trace-id': 'my-custom-trace-id',
 *       'x-maxim-generation-name': 'greeting-generation'
 *     }
 *   }
 * );
 * ```
 *
 * @example
 * ```typescript
 * // Streaming with automatic logging
 * const stream = await client.chat.completions.create({
 *   model: 'gpt-4',
 *   messages: [{ role: 'user', content: 'Tell me a story' }],
 *   stream: true
 * });
 *
 * for await (const chunk of stream) {
 *   process.stdout.write(chunk.choices[0]?.delta?.content || '');
 * }
 * // Logging happens automatically when stream completes
 * ```
 *
 * @example
 * ```typescript
 * // Using Responses API (non-streaming)
 * const response = await client.responses.create({
 *   model: 'gpt-4.1',
 *   input: 'What is the meaning of life?'
 * });
 *
 * // Using Responses API (streaming)
 * const stream = await client.responses.stream({
 *   model: 'gpt-4.1',
 *   input: 'Tell me a story'
 * });
 * for await (const event of stream) {
 *   // process events
 * }
 * // Logging happens automatically when stream completes
 * ```
 */
class MaximOpenAIClient {
    /**
     * Creates a new MaximOpenAIClient.
     *
     * @param client - The OpenAI client instance to wrap.
     * @param logger - The MaximLogger instance to use for logging.
     */
    constructor(client, logger) {
        this.client = client;
        this.logger = logger;
        this._chat = new chat_1.MaximOpenAIChat(client, logger);
        this._responses = new responses_1.MaximOpenAIResponses(client, logger);
    }
    /**
     * Access the chat resource with automatic Maxim logging.
     */
    get chat() {
        return this._chat;
    }
    /**
     * Access the responses resource with automatic Maxim logging.
     */
    get responses() {
        return this._responses;
    }
    /**
     * Wraps an OpenAI Realtime client with automatic Maxim logging.
     *
     * The Realtime API uses WebSocket connections which are created separately
     * from the standard OpenAI client. This method wraps an existing realtime
     * client instance to enable automatic logging of all realtime events.
     *
     * @param realtimeClient - The OpenAI Realtime client (OpenAIRealtimeWS or OpenAIRealtimeWebSocket)
     * @param headers - Optional headers for session/generation metadata
     * @returns A wrapper that logs realtime events to Maxim
     *
     * @example
     * ```typescript
     * import { OpenAIRealtimeWS } from 'openai/realtime/ws';
     * import { Maxim, MaximOpenAIClient } from '@maximai/maxim-js';
     *
     * const maxim = new Maxim({ apiKey: process.env.MAXIM_API_KEY });
     * const logger = await maxim.logger({ id: 'my-app' });
     *
     * const openai = new OpenAI();
     * const client = new MaximOpenAIClient(openai, logger);
     *
     * // Create the realtime client separately
     * const rt = new OpenAIRealtimeWS({ model: 'gpt-4o-realtime-preview' });
     *
     * // Wrap it with Maxim logging
     * const wrapper = client.wrapRealtime(rt, {
     *   'maxim-session-name': 'Voice Assistant Session'
     * });
     *
     * // Use rt normally - all events are automatically logged
     * rt.socket.on('open', () => {
     *   rt.send({
     *     type: 'session.update',
     *     session: { modalities: ['text', 'audio'] }
     *   });
     * });
     *
     * // Remember to cleanup when done
     * // wrapper.cleanup();
     * ```
     */
    wrapRealtime(realtimeClient, headers) {
        return new realtime_1.MaximOpenAIRealtimeWrapper(realtimeClient, this.logger, headers);
    }
}
exports.MaximOpenAIClient = MaximOpenAIClient;
//# sourceMappingURL=client.js.map