Merge branch 'main' into add_mistral_api_support

frankroeder · Jun 26, 2024 · bf56e1b · bf56e1b
2 parents 569d5f2 + 74e757b
commit bf56e1b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/lua/parrot/init.lua b/lua/parrot/init.lua
@@ -286,7 +286,7 @@ M.refresh_state = function()
   end
 
   M._state.provider = M._state.provider or state.provider or nil
-  if M._state.provider == nil then
+  if M._state.provider == nil or M._available_providers[M._state.provider] == nil then
     M._state.provider = M._available_providers[1]
   end
 

diff --git a/lua/parrot/provider/init.lua b/lua/parrot/provider/init.lua
@@ -4,7 +4,9 @@ local Anthropic = require("parrot.provider.anthropic")
 local Perplexity = require("parrot.provider.perplexity")
 local Mistral = require("parrot.provider.mistral")
 
-local M = {}
+local M = {
+  logger = require("parrot.logger"),
+}
 
 ---@param prov_name string # name of the provider
 ---@param endpoint string # API endpoint for the provider