import { EventEmitter } from "events";

export type StructureProperty = {
  type?: "string" | "integer" | "number" | "boolean" | "object";
  description?: string;
  enum?: string[];
  properties?: StructureSchema;
};

export type StructureSchema = {
  [key: string]: StructureProperty;
};

export interface ToolParameter {
  type: "string" | "integer" | "number" | "boolean" | "object";
  description: string;
  name: string;
  enum?: string[];
}
// this describes the tool
export interface ToolSchema {
  name: string;
  description: string;
  parameters: { [key: string]: ToolParameter };
}
// tool description + the actual function
export interface ToolDefinition {
  handler: (parameters: any) => Promise<any>; // parameter should always look like a json
  schema: ToolSchema;
}

export interface LLMConfigOptions {
  instructions?: string;
}

export interface LLMGenerateOptions {
  structure: StructureSchema;
  prompt: string;
}

export interface LLMGenerateTextOptions {
  prompt: string;
  shouldStream?: boolean;
  stream?: EventEmitter;
}

export interface LLMGenerateTextStreamOptions {
  prompt: string;
}

export interface LLMGenerateWithToolsOptions {
  prompt: string;
  maxTokens?: number;
  temperature?: number;
  toolTimeout?: number; // in milliseconds
  stream?: EventEmitter;
}

export interface ToolCall {
  name: string;
  parameters: any;
  id: string;
}

export type FoundationModelsAvailability =
  | "available"
  | "appleIntelligenceNotEnabled"
  | "modelNotReady"
  | "unavailable";