From 1557eaa2ea941dcc5fec98cc359579b742f1e4d8 Mon Sep 17 00:00:00 2001
From: "Dung Duc Huynh (Kaka)" <870029+jellydn@users.noreply.github.com>
Date: Wed, 11 Oct 2023 14:05:23 +0000
Subject: [PATCH] fix: limit to top 3 sources for TabNine

---
 lua/lazyvim/plugins/extras/coding/tabnine.lua | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/lua/lazyvim/plugins/extras/coding/tabnine.lua b/lua/lazyvim/plugins/extras/coding/tabnine.lua
index 063b24f3..da688a7f 100644
--- a/lua/lazyvim/plugins/extras/coding/tabnine.lua
+++ b/lua/lazyvim/plugins/extras/coding/tabnine.lua
@@ -1,14 +1,22 @@
 return {
+  -- Tabnine cmp source
   {
     "nvim-cmp",
     dependencies = {
-      -- Add TabNine source, make sure you run :CmpTabnineHub after installation.
+      -- Add TabNine support, make sure you run :CmpTabnineHub after installation.
       {
         "tzachar/cmp-tabnine",
         build = "./install.sh",
         dependencies = "hrsh7th/nvim-cmp",
-        -- Only limit top 3 sources, refer https://github.com/tzachar/cmp-tabnine?tab=readme-ov-file#setup
-        opts = { max_lines = 1000, max_num_results = 3, sort = true },
+        -- Only limit top 3 suggestions from Tabnine
+        config = function()
+          local tabnine = require("cmp_tabnine.config")
+          tabnine:setup({
+            max_lines = 1000,
+            max_num_results = 3,
+            sort = true,
+          })
+        end,
       },
     },
     ---@param opts cmp.ConfigSchema
@@ -20,7 +28,7 @@ return {
       })
     end,
   },
-
+  -- Show TabNine status in lualine
   {
     "nvim-lualine/lualine.nvim",
     optional = true,