chore(provider): update models, parameters, and link to API documenta…

…tion
frankroeder · Sep 23, 2024 · 1dc77f1 · 1dc77f1
1 parent b347a1c
commit 1dc77f1
Show file tree

Hide file tree

Showing 4 changed files with 22 additions and 14 deletions.
diff --git a/lua/parrot/provider/anthropic.lua b/lua/parrot/provider/anthropic.lua
@@ -14,8 +14,8 @@ local AVAILABLE_API_PARAMETERS = {
   -- required
   model = true,
   messages = true,
-  -- optional
   max_tokens = true,
+  -- optional
   metadata = true,
   stop_sequences = true,
   stream = true,

diff --git a/lua/parrot/provider/mistral.lua b/lua/parrot/provider/mistral.lua
@@ -9,6 +9,7 @@ local Mistral = {}
 Mistral.__index = Mistral
 
 -- Available API parameters for Mistral
+-- https://docs.mistral.ai/api/
 local AVAILABLE_API_PARAMETERS = {
   -- required
   model = true,
@@ -17,9 +18,11 @@ local AVAILABLE_API_PARAMETERS = {
   temperature = true,
   top_p = true,
   max_tokens = true,
+  min_tokens = true,
   stream = true,
-  safe_prompt = true,
+  stop = true,
   random_seed = true,
+  safe_prompt = true,
 }
 
 -- Creates a new Mistral instance

diff --git a/lua/parrot/provider/openai.lua b/lua/parrot/provider/openai.lua
@@ -151,24 +151,25 @@ function OpenAI:get_available_models(online)
     }):start()
   end
   return {
-    "gpt-4o-mini",
+    "gpt-4o",
     "gpt-3.5-turbo",
     "gpt-3.5-turbo-0125",
-    "gpt-3.5-turbo-1106",
-    "gpt-3.5-turbo-16k",
-    "gpt-3.5-turbo-instruct",
-    "gpt-3.5-turbo-instruct-0914",
-    "gpt-4",
-    "gpt-4-0125-preview",
-    "gpt-4-0613",
     "gpt-4-1106-preview",
     "gpt-4-turbo",
+    "gpt-4o-2024-05-13",
+    "gpt-4-0125-preview",
     "gpt-4-turbo-2024-04-09",
     "gpt-4-turbo-preview",
-    "gpt-4o",
-    "gpt-4o-2024-05-13",
+    "gpt-3.5-turbo-16k",
     "gpt-4o-2024-08-06",
+    "gpt-3.5-turbo-1106",
+    "gpt-3.5-turbo-instruct-0914",
     "gpt-4o-mini-2024-07-18",
+    "gpt-4o-mini",
+    "gpt-4",
+    "gpt-4-0613",
+    "gpt-3.5-turbo-instruct",
+    "chatgpt-4o-latest",
   }
 end
 

diff --git a/lua/parrot/provider/perplexity.lua b/lua/parrot/provider/perplexity.lua
@@ -9,16 +9,20 @@ local Perplexity = {}
 Perplexity.__index = Perplexity
 
 -- Available API parameters for Perplexity
--- https://docs.perplexity.ai/reference/post_chat_completions
+-- https://docs.perplexity.ai/api-reference/chat-completions
 local AVAILABLE_API_PARAMETERS = {
   -- required
-  messages = true,
   model = true,
+  messages = true,
   -- optional
   max_tokens = true,
   temperature = true,
   top_p = true,
   return_citations = true,
+  search_domain_filter = true,
+  return_images = true,
+  return_related_questions = true,
+  search_recency_filter = true,
   top_k = true,
   stream = true,
   presence_penalty = true,