feat(llm): cohere support (#167)

should be good set of defaults now, one in US, one in canada, and
microsoft :/

Signed-off-by: Aaron Pham <contact@aarnphm.xyz>
This commit is contained in:
Aaron Pham
2024-08-23 09:36:40 -04:00
committed by GitHub
parent 12d7cd8ec7
commit d2775135a3
5 changed files with 142 additions and 3 deletions

View File

@@ -42,14 +42,24 @@ M.defaults = {
claude = {
endpoint = "https://api.anthropic.com",
model = "claude-3-5-sonnet-20240620",
["local"] = false,
temperature = 0,
max_tokens = 4096,
["local"] = false,
},
---@type AvanteGeminiProvider
gemini = {
endpoint = "https://generativelanguage.googleapis.com/v1beta/models",
model = "gemini-1.5-pro",
temperature = 0,
max_tokens = 4096,
["local"] = false,
},
---@type AvanteGeminiProvider
cohere = {
endpoint = "https://api.cohere.com",
model = "command-r-plus",
temperature = 0,
max_tokens = 3072,
["local"] = false,
},
---To add support for custom provider, follow the format below