From 15ea0d5ae7c393493407ffb0c151656cc27ee01b Mon Sep 17 00:00:00 2001
From: Elvis Claros Castro <elvis@claros.ar>
Date: Fri, 30 Aug 2024 15:03:49 -0300
Subject: Feat: suport for cmp-ai

---
 lua/pi/completition/init.lua | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

(limited to 'lua/pi/completition/init.lua')

diff --git a/lua/pi/completition/init.lua b/lua/pi/completition/init.lua
index ec128a0..5b9fd18 100644
--- a/lua/pi/completition/init.lua
+++ b/lua/pi/completition/init.lua
@@ -55,13 +55,14 @@ cmp.setup {
     ["<c-space>"] = cmp.mapping.complete(),
   },
   sources = {
-    { name = "dictionary", keyword_length = 2},
+    -- { name = "dictionary", keyword_length = 2},
     { name = "nvim_lua" },
     { name = "nvim_lsp" },
     { name = "path" },
     { name = "luasnip" },
     { name = "copilot" },
     {name = "codeium"},
+    { name = 'cmp_ai' },
     { name = 'html-css',
         option = {
           max_count = {}, -- not ready yet
@@ -109,6 +110,7 @@ cmp.setup {
       local KIND_ICONS = {
         Tailwind = '󰹞󰹞󰹞󰹞󰹞󰹞󰹞󰹞',
         Copilot = "",
+        Ollama = "",
         Text = "󰉿",
         Method = "󰆧",
         Function = "󰊕",
@@ -135,6 +137,9 @@ cmp.setup {
         Operator = "󰆕",
         Codeium = "",
         TypeParameter = "󰜅",
+        OpenAI = "",
+        TinyLlama = "󱝊",
+        LlamaCPP = ""
       }
 
       if vim_item.kind == 'Color' and entry.completion_item.documentation then
@@ -179,6 +184,10 @@ cmp.setup {
 
 vim.api.nvim_set_hl(0, "CmpItemKindCopilot", { fg = "#3426C3" })
 vim.api.nvim_set_hl(0, "CmpItemKindCodeium", { fg = "#2ACDBA" })
+vim.api.nvim_set_hl(0, "CmpItemKindOllama", { fg = "#FDFEFF" })
+vim.api.nvim_set_hl(0, "CmpItemKindOpenAI", { fg = "#080808" })
+vim.api.nvim_set_hl(0, "CmpItemKindTinyLlama", { fg = "#DF468C"})
+vim.api.nvim_set_hl(0, "CmpItemKindLlamaCPP", { fg = "#FF8235"})
 
 -- Set configuration for specific filetype.
 cmp.setup.filetype("gitcommit", {
-- 
cgit v1.2.3