import { ConversationMessage } from '../config/types.js';
import { FunctionDefinition } from '../services/function-bridge.js';
import type {
  ChatCompletion,
  ChatCompletionMessageParam,
  ChatCompletionMessageToolCall,
  ChatCompletionCreateParamsNonStreaming,
} from 'openai/resources/chat/completions';

export interface ModelInfo {
  id: string;
  created?: number;
  owned_by?: string;
  object?: string;
  context_window?: number;
  description?: string;
}

export interface ProviderOptions {
  apiKey?: string;
  baseURL: string;
  model: string;
  availableModels?: string[];
  temperature?: number;
  timeout?: number;
  maxRetries?: number;
  systemPrompt?: string;
}

export interface ToolCall {
  id: string;
  type: 'function';
  function: {
    name: string;
    arguments: string; // JSON string
  };
}

export interface ChatOptions {
  messages: ConversationMessage[];
  model?: string;
  stream?: boolean;
  temperature?: number;
  systemPrompt?: string;
  tools?: FunctionDefinition[];
  toolChoice?: 'auto' | 'none' | { type: 'function'; function: { name: string } };
}

export interface ChatResponse {
  content: string;
  usage?: {
    promptTokens: number;
    completionTokens: number;
    totalTokens: number;
  };
  model: string;
  finishReason?: string;
  toolCalls?: ToolCall[];
}

// OpenAI API parameter types
export type OpenAIChatParams = ChatCompletionCreateParamsNonStreaming;
export type OpenAIMessage = ChatCompletionMessageParam;
export type OpenAIChatResponse = ChatCompletion;
export type OpenAIToolCall = ChatCompletionMessageToolCall;

// Common interface for all duck providers (HTTP and CLI)
export interface IDuckProvider {
  name: string;
  nickname: string;
  chat(options: ChatOptions): Promise<ChatResponse>;
  healthCheck(): Promise<boolean>;
  listModels(): Promise<ModelInfo[]>;
  getInfo(): {
    name: string;
    nickname: string;
    model: string;
    type: 'http' | 'cli';
    availableModels?: string[];
    baseURL?: string;
    hasApiKey?: boolean;
    cliCommand?: string;
    cliType?: string;
  };
}

// Enhanced response type with MCP functionality
export interface MCPResult {
  id: string;
  success: boolean;
  data?: unknown;
  error?: string;
}