@gitlab/gitlab-ai-provider 3.1.2 → 3.2.0

package/dist/index.d.ts

@@ -1,7 +1,7 @@
 import { LanguageModelV2, LanguageModelV2CallOptions, LanguageModelV2Content, LanguageModelV2FinishReason, LanguageModelV2Usage, LanguageModelV2CallWarning, LanguageModelV2StreamPart } from '@ai-sdk/provider';
 import { z } from 'zod';
 
-interface GitLabAgenticConfig {
+interface GitLabAnthropicConfig {
     provider: string;
     instanceUrl: string;
     getHeaders: () => Record<string, string>;
@@ -36,20 +36,20 @@ interface GitLabAgenticConfig {
     aiGatewayUrl?: string;
 }
 /**
- * GitLab Agentic Language Model
+ * GitLab Anthropic Language Model
  *
  * This model uses GitLab's Anthropic proxy to provide native tool calling support
  * for the duo-chat model. It connects to Claude through GitLab's cloud proxy
  * at https://cloud.gitlab.com/ai/v1/proxy/anthropic/
  */
-declare class GitLabAgenticLanguageModel implements LanguageModelV2 {
+declare class GitLabAnthropicLanguageModel implements LanguageModelV2 {
     readonly specificationVersion: "v2";
     readonly modelId: string;
     readonly supportedUrls: Record<string, RegExp[]>;
     private readonly config;
     private readonly directAccessClient;
     private anthropicClient;
-    constructor(modelId: string, config: GitLabAgenticConfig);
+    constructor(modelId: string, config: GitLabAnthropicConfig);
     get provider(): string;
     /**
      * Get or create an Anthropic client with valid credentials
@@ -118,33 +118,33 @@ interface GitLabProvider {
      *   anthropicModel: 'claude-sonnet-4-5-20250929'
      * });
      */
-    agenticChat(modelId: string, options?: GitLabAgenticOptions): GitLabAgenticLanguageModel;
+    agenticChat(modelId: string, options?: GitLabAgenticOptions): GitLabAnthropicLanguageModel;
     textEmbeddingModel(modelId: string): never;
     imageModel(modelId: string): never;
 }
 interface GitLabAgenticOptions {
     /**
-     * The Anthropic model to use
+     * Override the provider-specific model (optional).
+     * Must be a valid model for the detected provider.
      *
-     * If not specified, automatically maps from the model ID:
-     * - 'duo-chat-opus-4-5' → 'claude-opus-4-5-20251101'
-     * - 'duo-chat-sonnet-4-5' → 'claude-sonnet-4-5-20250929'
-     * - 'duo-chat-haiku-4-5' → 'claude-haiku-4-5-20251001'
+     * For Anthropic models:
+     * - 'claude-opus-4-5-20251101'
+     * - 'claude-sonnet-4-5-20250929'
+     * - 'claude-haiku-4-5-20251001'
      *
-     * For unmapped model IDs, defaults to 'claude-sonnet-4-5-20250929'
-     *
-     * @default Automatically mapped from model ID, or 'claude-sonnet-4-5-20250929'
-     * @example
-     * // Use automatic mapping
-     * const model = gitlab.agenticChat('duo-chat-opus-4-5');
+     * For OpenAI models:
+     * - 'gpt-5.1-2025-11-13'
+     * - 'gpt-5-mini-2025-08-07'
+     * - 'gpt-5-codex'
+     * - 'gpt-5.2-codex'
      *
      * @example
      * // Override with explicit model
      * const model = gitlab.agenticChat('duo-chat-opus-4-5', {
-     *   anthropicModel: 'claude-sonnet-4-5-20250929'
+     *   providerModel: 'claude-sonnet-4-5-20250929'
      * });
      */
-    anthropicModel?: string;
+    providerModel?: string;
     /**
      * Maximum tokens to generate
      * @default 8192
@@ -214,32 +214,93 @@ declare function createGitLab(options?: GitLabProviderSettings): GitLabProvider;
  */
 declare const gitlab: GitLabProvider;
 
-/**
- * Maps GitLab model IDs to their corresponding Anthropic model identifiers.
- *
- * This mapping allows users to specify model variants by model ID without
- * needing to manually configure the anthropicModel option.
- *
- * @example
- * const model = gitlab.agenticChat('duo-chat-opus-4-5');
- * // Automatically uses 'claude-opus-4-5-20251101'
- */
-declare const MODEL_ID_TO_ANTHROPIC_MODEL: Record<string, string>;
-/**
- * Gets the Anthropic model identifier for a given GitLab model ID.
- *
- * @param modelId - The GitLab model ID (e.g., 'duo-chat-opus-4-5')
- * @returns The Anthropic model identifier, or undefined if no mapping exists
- *
- * @example
- * getAnthropicModelForModelId('duo-chat-opus-4-5')
- * // Returns: 'claude-opus-4-5-20251101'
- *
- * @example
- * getAnthropicModelForModelId('duo-chat')
- * // Returns: undefined (uses default)
- */
+interface GitLabOpenAIConfig {
+    provider: string;
+    instanceUrl: string;
+    getHeaders: () => Record<string, string>;
+    fetch?: typeof fetch;
+    refreshApiKey?: () => Promise<void>;
+    openaiModel?: string;
+    maxTokens?: number;
+    featureFlags?: {
+        DuoAgentPlatformNext: true;
+    } & Record<string, boolean>;
+    aiGatewayUrl?: string;
+    /** Whether to use the Responses API instead of Chat Completions API */
+    useResponsesApi?: boolean;
+}
+declare class GitLabOpenAILanguageModel implements LanguageModelV2 {
+    readonly specificationVersion: "v2";
+    readonly modelId: string;
+    readonly supportedUrls: Record<string, RegExp[]>;
+    private readonly config;
+    private readonly directAccessClient;
+    private readonly useResponsesApi;
+    private openaiClient;
+    constructor(modelId: string, config: GitLabOpenAIConfig);
+    get provider(): string;
+    private getOpenAIClient;
+    private isTokenError;
+    private convertTools;
+    private convertToolChoice;
+    private convertPrompt;
+    private convertFinishReason;
+    /**
+     * Convert tools to Responses API format
+     */
+    private convertToolsForResponses;
+    /**
+     * Convert prompt to Responses API input format
+     */
+    private convertPromptForResponses;
+    /**
+     * Extract system instructions from prompt
+     */
+    private extractSystemInstructions;
+    /**
+     * Convert Responses API status to finish reason
+     * Note: Responses API returns 'completed' even when making tool calls,
+     * so we need to check the content for tool calls separately.
+     */
+    private convertResponsesStatus;
+    doGenerate(options: LanguageModelV2CallOptions): Promise<{
+        content: LanguageModelV2Content[];
+        finishReason: LanguageModelV2FinishReason;
+        usage: LanguageModelV2Usage;
+        warnings: LanguageModelV2CallWarning[];
+    }>;
+    private doGenerateWithChatApi;
+    private doGenerateWithResponsesApi;
+    doStream(options: LanguageModelV2CallOptions): Promise<{
+        stream: ReadableStream<LanguageModelV2StreamPart>;
+        request?: {
+            body?: unknown;
+        };
+        response?: {
+            headers?: Record<string, string>;
+        };
+    }>;
+    private doStreamWithChatApi;
+    private doStreamWithResponsesApi;
+}
+
+type ModelProvider = 'anthropic' | 'openai';
+type OpenAIApiType = 'chat' | 'responses';
+interface ModelMapping {
+    provider: ModelProvider;
+    model: string;
+    /** For OpenAI models, which API to use: 'chat' for /v1/chat/completions, 'responses' for /v1/responses */
+    openaiApiType?: OpenAIApiType;
+}
+declare const MODEL_MAPPINGS: Record<string, ModelMapping>;
+declare function getModelMapping(modelId: string): ModelMapping | undefined;
+declare function getProviderForModelId(modelId: string): ModelProvider | undefined;
+declare function getValidModelsForProvider(provider: ModelProvider): string[];
 declare function getAnthropicModelForModelId(modelId: string): string | undefined;
+declare function getOpenAIModelForModelId(modelId: string): string | undefined;
+declare function getOpenAIApiType(modelId: string): OpenAIApiType;
+declare function isResponsesApiModel(modelId: string): boolean;
+declare const MODEL_ID_TO_ANTHROPIC_MODEL: Record<string, string>;
 
 interface GitLabErrorOptions {
     message: string;
@@ -265,15 +326,15 @@ declare const gitlabOAuthTokenResponseSchema: z.ZodObject<{
     expires_in: z.ZodNumber;
     created_at: z.ZodNumber;
 }, "strip", z.ZodTypeAny, {
+    created_at?: number;
     access_token?: string;
     refresh_token?: string;
     expires_in?: number;
-    created_at?: number;
 }, {
+    created_at?: number;
     access_token?: string;
     refresh_token?: string;
     expires_in?: number;
-    created_at?: number;
 }>;
 type GitLabOAuthTokenResponse = z.infer<typeof gitlabOAuthTokenResponseSchema>;
 
@@ -568,10 +629,16 @@ declare class GitLabDirectAccessClient {
      * Get the Anthropic proxy base URL
      */
     getAnthropicProxyUrl(): string;
+    /**
+     * Get the OpenAI proxy base URL
+     * Note: The OpenAI SDK expects a base URL like https://api.openai.com/v1
+     * and appends paths like /chat/completions. So we need /v1 at the end.
+     */
+    getOpenAIProxyUrl(): string;
     /**
      * Invalidate the cached token
      */
     invalidateToken(): void;
 }
 
-export { BUNDLED_CLIENT_ID, DEFAULT_AI_GATEWAY_URL, type DirectAccessToken, GITLAB_COM_URL, type GitLabAgenticConfig, GitLabAgenticLanguageModel, type GitLabAgenticOptions, GitLabDirectAccessClient, type GitLabDirectAccessConfig, GitLabError, type GitLabErrorOptions, GitLabOAuthManager, type GitLabOAuthTokenResponse, type GitLabOAuthTokens, type GitLabProject, GitLabProjectCache, GitLabProjectDetector, type GitLabProjectDetectorConfig, type GitLabProvider, type GitLabProviderSettings, MODEL_ID_TO_ANTHROPIC_MODEL, OAUTH_SCOPES, type OpenCodeAuth, type OpenCodeAuthApi, type OpenCodeAuthOAuth, TOKEN_EXPIRY_SKEW_MS, createGitLab, getAnthropicModelForModelId, gitlab };
+export { BUNDLED_CLIENT_ID, DEFAULT_AI_GATEWAY_URL, type DirectAccessToken, GITLAB_COM_URL, type GitLabAgenticOptions, type GitLabAnthropicConfig, GitLabAnthropicLanguageModel, GitLabDirectAccessClient, type GitLabDirectAccessConfig, GitLabError, type GitLabErrorOptions, GitLabOAuthManager, type GitLabOAuthTokenResponse, type GitLabOAuthTokens, type GitLabOpenAIConfig, GitLabOpenAILanguageModel, type GitLabProject, GitLabProjectCache, GitLabProjectDetector, type GitLabProjectDetectorConfig, type GitLabProvider, type GitLabProviderSettings, MODEL_ID_TO_ANTHROPIC_MODEL, MODEL_MAPPINGS, type ModelMapping, type ModelProvider, OAUTH_SCOPES, type OpenAIApiType, type OpenCodeAuth, type OpenCodeAuthApi, type OpenCodeAuthOAuth, TOKEN_EXPIRY_SKEW_MS, createGitLab, getAnthropicModelForModelId, getModelMapping, getOpenAIApiType, getOpenAIModelForModelId, getProviderForModelId, getValidModelsForProvider, gitlab, isResponsesApiModel };