mirror of
https://github.com/aljazceru/goose.git
synced 2025-12-18 14:44:21 +01:00
c99d5beb07
Co-authored-by: Michael Paquette <mpaquette@pax8.com> Co-authored-by: Alice Hau <ahau@squareup.com>
124 lines
3.7 KiB
TypeScript
124 lines
3.7 KiB
TypeScript
export const openai_models = ['gpt-4o-mini', 'gpt-4o', 'gpt-4-turbo', 'o1'];
|
|
|
|
export const anthropic_models = [
|
|
'claude-3-5-sonnet-latest',
|
|
'claude-3-5-sonnet-2',
|
|
'claude-3-5-haiku-latest',
|
|
'claude-3-opus-latest',
|
|
];
|
|
|
|
export const google_models = [
|
|
'gemini-1.5-pro',
|
|
'gemini-1.5-flash',
|
|
'gemini-2.0-flash',
|
|
'gemini-2.0-flash-lite-preview-02-05',
|
|
'gemini-2.0-flash-thinking-exp-01-21',
|
|
'gemini-2.0-pro-exp-02-05',
|
|
'gemini-2.5-pro-exp-03-25',
|
|
];
|
|
|
|
export const groq_models = ['llama-3.3-70b-versatile'];
|
|
|
|
export const ollama_mdoels = ['qwen2.5'];
|
|
|
|
export const openrouter_models = ['anthropic/claude-3.5-sonnet'];
|
|
|
|
export const azure_openai_models = ['gpt-4o'];
|
|
|
|
export const gcp_vertex_ai_models = [
|
|
'claude-3-7-sonnet@20250219',
|
|
'claude-3-5-sonnet-v2@20241022',
|
|
'claude-3-5-sonnet@20240620',
|
|
'claude-3-5-haiku@20241022',
|
|
'gemini-1.5-pro-002',
|
|
'gemini-2.0-flash-001',
|
|
'gemini-2.0-pro-exp-02-05',
|
|
'gemini-2.5-pro-exp-03-25',
|
|
];
|
|
|
|
export const default_models = {
|
|
openai: 'gpt-4o',
|
|
anthropic: 'claude-3-5-sonnet-latest',
|
|
databricks: 'goose',
|
|
google: 'gemini-2.0-flash-exp',
|
|
groq: 'llama-3.3-70b-versatile',
|
|
openrouter: 'anthropic/claude-3.5-sonnet',
|
|
ollama: 'qwen2.5',
|
|
azure_openai: 'gpt-4o',
|
|
gcp_vertex_ai: 'gemini-2.0-flash-001',
|
|
aws_bedrock: 'us.anthropic.claude-3-7-sonnet-20250219-v1:0',
|
|
};
|
|
|
|
export function getDefaultModel(key: string): string | undefined {
|
|
return default_models[key as keyof typeof default_models] || undefined;
|
|
}
|
|
|
|
export const short_list = ['gpt-4o', 'claude-3-5-sonnet-latest'];
|
|
|
|
export const required_keys = {
|
|
OpenAI: ['OPENAI_API_KEY', 'OPENAI_HOST', 'OPENAI_BASE_PATH'],
|
|
Anthropic: ['ANTHROPIC_API_KEY', 'ANTHROPIC_HOST'],
|
|
Databricks: ['DATABRICKS_HOST'],
|
|
Groq: ['GROQ_API_KEY'],
|
|
Ollama: ['OLLAMA_HOST'],
|
|
Google: ['GOOGLE_API_KEY'],
|
|
OpenRouter: ['OPENROUTER_API_KEY'],
|
|
'Azure OpenAI': [
|
|
'AZURE_OPENAI_ENDPOINT',
|
|
'AZURE_OPENAI_DEPLOYMENT_NAME',
|
|
'AZURE_OPENAI_API_VERSION',
|
|
],
|
|
'GCP Vertex AI': ['GCP_PROJECT_ID', 'GCP_LOCATION'],
|
|
'AWS Bedrock': ['AWS_PROFILE'],
|
|
};
|
|
|
|
export const default_key_value = {
|
|
ANTHROPIC_HOST: 'https://api.anthropic.com',
|
|
OPENAI_HOST: 'https://api.openai.com',
|
|
OPENAI_BASE_PATH: 'v1/chat/completions',
|
|
OLLAMA_HOST: 'localhost',
|
|
GCP_LOCATION: 'us-central1',
|
|
AZURE_OPENAI_API_VERSION: '2024-10-21',
|
|
};
|
|
|
|
export const supported_providers = [
|
|
'OpenAI',
|
|
'Anthropic',
|
|
'Databricks',
|
|
'Groq',
|
|
'Google',
|
|
'Ollama',
|
|
'OpenRouter',
|
|
'Azure OpenAI',
|
|
'GCP Vertex AI',
|
|
'AWS Bedrock',
|
|
];
|
|
|
|
export const model_docs_link = [
|
|
{ name: 'OpenAI', href: 'https://platform.openai.com/docs/models' },
|
|
{ name: 'Anthropic', href: 'https://docs.anthropic.com/en/docs/about-claude/models' },
|
|
{ name: 'Google', href: 'https://ai.google/get-started/our-models/' },
|
|
{ name: 'Groq', href: 'https://console.groq.com/docs/models' },
|
|
{
|
|
name: 'Databricks',
|
|
href: 'https://docs.databricks.com/en/generative-ai/external-models/index.html',
|
|
},
|
|
{ name: 'OpenRouter', href: 'https://openrouter.ai/models' },
|
|
{ name: 'Ollama', href: 'https://ollama.com/library' },
|
|
{ name: 'GCP Vertex AI', href: 'https://cloud.google.com/vertex-ai' },
|
|
{ name: 'AWS Bedrock', href: 'https://console.aws.amazon.com/bedrock/home#/model-catalog' },
|
|
];
|
|
|
|
export const provider_aliases = [
|
|
{ provider: 'OpenAI', alias: 'openai' },
|
|
{ provider: 'Anthropic', alias: 'anthropic' },
|
|
{ provider: 'Ollama', alias: 'ollama' },
|
|
{ provider: 'Groq', alias: 'groq' },
|
|
{ provider: 'Databricks', alias: 'databricks' },
|
|
{ provider: 'OpenRouter', alias: 'openrouter' },
|
|
{ provider: 'Google', alias: 'google' },
|
|
{ provider: 'Azure OpenAI', alias: 'azure_openai' },
|
|
{ provider: 'GCP Vertex AI', alias: 'gcp_vertex_ai' },
|
|
{ provider: 'AWS Bedrock', alias: 'aws_bedrock' },
|
|
];
|