ai

const (
	// Posthog LLM observability properties
	AITraceID       AIProperty = "$ai_trace_id"       // The trace ID (UUID to group AI events), similar to conversation_id. Must contain only letters, numbers, and special characters: -, _, ~, ., @, (, ), !, ', :, |
	AIModel         AIProperty = "$ai_model"          // The model used (e.g., gpt-3.5-turbo)
	AIProvider      AIProperty = "$ai_provider"       // The LLM provider name
	AIInput         AIProperty = "$ai_input"          // List of messages sent to the LLM (e.g., [{"role": "user", "content": "..."}])
	AIInputTokens   AIProperty = "$ai_input_tokens"   // The number of tokens in the input (from response.usage)
	AIOutputChoices AIProperty = "$ai_output_choices" // List of choices returned by the LLM (e.g., [{"role": "assistant", "content": "..."}])
	AIOutputTokens  AIProperty = "$ai_output_tokens"  // The number of tokens in the output (from response.usage)
	AILatency       AIProperty = "$ai_latency"        // The latency of the LLM call in seconds
	AIHTTPStatus    AIProperty = "$ai_http_status"    // The HTTP status code of the LLM response
	AIBaseURL       AIProperty = "$ai_base_url"       // The base URL of the LLM provider
	AIIsError       AIProperty = "$ai_is_error"       // Boolean indicating whether the request resulted in an error
	AIError         AIProperty = "$ai_error"          // The error message or object if the request failed

	// Custom properties
	SystemPrompt    AIProperty = "$kai_system_prompt"
	ToolCallEnabled AIProperty = "$kai_tool_call_enabled"
	McpServerUrls   AIProperty = "$kai_mcp_server_urls"
	Temperature     AIProperty = "$kai_temperature"
	TopP            AIProperty = "$kai_top_p"
	TopK            AIProperty = "$kai_top_k"
	MaxTokens       AIProperty = "$kai_max_tokens"
)

const (
	// OpenAI Models
	GPT4       BaseModel = "gpt-4"
	GPT4o      BaseModel = "gpt-4o"
	GPT4oMini  BaseModel = "gpt-4o-mini"
	GPT4Turbo  BaseModel = "gpt-4-turbo"
	GPT35Turbo BaseModel = "gpt-3.5-turbo"
	GPT5       BaseModel = "gpt-5"
	GPT5Nano   BaseModel = "gpt-5-nano"
	GPT5Mini   BaseModel = "gpt-5-mini"
	GPT5_1     BaseModel = "gpt-5.1"
	GPT5_2     BaseModel = "gpt-5.2"
	GPT5_2_Pro BaseModel = "gpt-5.2-pro"

	GPT5_1Codex    BaseModel = "gpt-5.1-codex"
	GPT5_1CodexMax BaseModel = "gpt-5.1-codex-max"
	GPT5_2Codex    BaseModel = "gpt-5.2-codex"
	GPT5_2CodexMax BaseModel = "gpt-5.2-codex-max"
	O1             BaseModel = "o1"
	O1Mini         BaseModel = "o1-mini"
	O1Preview      BaseModel = "o1-preview"
	GPTOSS_20B     BaseModel = "gpt-oss-20b"
	GPTOSS_120B    BaseModel = "gpt-oss-120b"

	// Text Embedding Models
	TextEmbeddingAda002 BaseModel = "text-embedding-ada-002"
	TextEmbedding3Large BaseModel = "text-embedding-3-large"
	TextEmbedding3Small BaseModel = "text-embedding-3-small"

	// Claude Models
	Claude35Sonnet  BaseModel = "claude-3.5-sonnet"
	Claude35Haiku   BaseModel = "claude-3.5-haiku"
	Claude3Sonnet   BaseModel = "claude-3-sonnet"
	Claude3Haiku    BaseModel = "claude-3-haiku"
	Claude3Opus     BaseModel = "claude-3-opus"
	Claude37Sonnet  BaseModel = "claude-3.7-sonnet"
	Claude4Sonnet   BaseModel = "claude-4-sonnet"
	Claude4_5Sonnet BaseModel = "claude-4.5-sonnet"
	Claude4Opus     BaseModel = "claude-4-opus"
	Claude4_5Opus   BaseModel = "claude-4.5-opus"
	ClaudeInstant   BaseModel = "claude-instant"
	ClaudeV2        BaseModel = "claude-v2"

	// Llama Models
	Llama3_8B        BaseModel = "llama-3-8b"
	Llama3_70B       BaseModel = "llama-3-70b"
	Llama31_8B       BaseModel = "llama-3.1-8b"
	Llama31_70B      BaseModel = "llama-3.1-70b"
	Llama31_405B     BaseModel = "llama-3.1-405b"
	Llama32_1B       BaseModel = "llama-3.2-1b"
	Llama32_3B       BaseModel = "llama-3.2-3b"
	Llama32_11B      BaseModel = "llama-3.2-11b"
	Llama32_90B      BaseModel = "llama-3.2-90b"
	Llama33_70B      BaseModel = "llama-3.3-70b"
	Llama4_Guard_12B BaseModel = "llama-4-guard-12b"
	Llama4_Scout_17B BaseModel = "llama-4-scout-17b"

	// Mistral Models
	Mistral7B    BaseModel = "mistral-7b"
	Mixtral8x7B  BaseModel = "mixtral-8x7b"
	MistralLarge BaseModel = "mistral-large"
	MistralSmall BaseModel = "mistral-small"

	// Quew Models
	Quew3_32B                 BaseModel = "quew3-32b"
	Quew3_235B_VL_Thinking    BaseModel = "quew3-235b-vl"
	Quew3_235B_VL_Instruct    BaseModel = "quew3-235b-vl-instruct"
	Quew3_235B_Thinking       BaseModel = "quew3-235b-thinking"
	Quew3_235B_Instruct       BaseModel = "quew3-235b-instruct"
	Qwen3_Coder_480B_Instruct BaseModel = "qwen3-coder-480b-instruct"

	// Moonshot Models
	KimiK2Thinking BaseModel = "kimi-k2-thinking"

	MiniMaxM2   BaseModel = "minimax-m2"
	MiniMaxM2P1 BaseModel = "minimax-m2p1"

	GLM4_7 BaseModel = "glm4-7"
	GLM4_6 BaseModel = "glm4-6"

	// Deepseek
	DeepSeekV3P2 BaseModel = "deepseek-v3p2"

	// Amazon Titan Models
	TitanTextG1Large BaseModel = "titan-text-g1-large"
	TitanTextPremier BaseModel = "titan-text-premier"
	TitanTextLite    BaseModel = "titan-text-lite"
	TitanTextExpress BaseModel = "titan-text-express"
	TitanEmbedText   BaseModel = "titan-embed-text"
	TitanEmbedImage  BaseModel = "titan-embed-image"

	// Amazon Nova Models
	NovaPro    BaseModel = "nova-pro"
	NovaLite   BaseModel = "nova-lite"
	NovaCanvas BaseModel = "nova-canvas"
	NovaReel   BaseModel = "nova-reel"
	NovaMicro  BaseModel = "nova-micro"

	// Google Models
	Gemini3FlashPreview BaseModel = "gemini-3-flash-preview"
	Gemini3ProPreview   BaseModel = "gemini-3-pro-preview"
	Gemini25Flash       BaseModel = "gemini-2.5-flash"
	Gemini25Pro         BaseModel = "gemini-2.5-pro"
	Gemini20Flash       BaseModel = "gemini-2.0-flash"
	Gemini20FlashLite   BaseModel = "gemini-2.0-flash-lite"
	Gemini15Flash       BaseModel = "gemini-1.5-flash"
	Gemini15Flash8B     BaseModel = "gemini-1.5-flash-8b"
	Gemini15Pro         BaseModel = "gemini-1.5-pro"
	GeminiEmbedding     BaseModel = "gemini-embedding"
	PaLM2               BaseModel = "palm-2"

	// xAI Models
	Grok4              BaseModel = "grok-4"
	GrokCodeFast       BaseModel = "grok-code-fast-1"
	Grok4ReasoningFast BaseModel = "grok-4-fast-reasoning"
	Grok4Fast          BaseModel = "grok-4-fast-non-reasoning"
	Grok3              BaseModel = "grok-3"
	Grok3Mini          BaseModel = "grok-3-mini"

	// Sarvam AI Models
	SarvamM BaseModel = "sarvam-m"
)

const (
	GROK_2_IMAGE       ImageModels = "grok-2-image"
	GPT_1_IMAGE        ImageModels = "gpt-image-1"
	DALL_E_3           ImageModels = "dall-e-3"
	DALL_E_2           ImageModels = "dall-e-2"
	GEMINI_NANO_BANANA ImageModels = "gemini-2.5-flash-image-preview"
	// Google/Gemini Image Models
	GEMINI_3_PRO_IMAGE ImageModels = "gemini-3-pro-image-preview"
	// Segmind Models
	SEGMIND_SD          ImageModels = "segmind-sd-3.5-large"
	SEGMIND_PROTOVIS    ImageModels = "segmind-protovis-lightning"
	SEGMIND_SAMARITAN   ImageModels = "segmind-samaritan-3d"
	SEGMIND_DREAMSHAPER ImageModels = "segmind-dreamshaper-lightning"
	SEGMIND_NANO_BANANA ImageModels = "segmind-nano-banana"
	SEGMIND_FLUX        ImageModels = "segmind-flux-schnell"
	SEGMIND_MIDJOURNEY  ImageModels = "segmind-midjourney"
	SEGMIND_SDXL        ImageModels = "segmind-sdxl-txt2img"
	SEGMIND_SD15        ImageModels = "segmind-sd15-txt2img"
)

const (
	OpenAI      Provider = "openai"
	Anthropic   Provider = "anthropic"
	Bedrock     Provider = "bedrock"
	Google      Provider = "google"
	XAI         Provider = "xai"
	Groq        Provider = "groq"
	FireworksAI Provider = "fireworksai"
	OpenRouter  Provider = "openrouter"
	Sarvam      Provider = "sarvam"
)

const (
	GoogleProjectID SpecialConfig = "google_project_id"
	GoogleLocation  SpecialConfig = "google_location"
	GoogleAPIKey    SpecialConfig = "google_api_key"
)