wip

lua/tabnine/chat/codelens.lua

@@ -1,6 +1,7 @@
 local chat = require("tabnine.chat")
 local config = require("tabnine.config")
 local consts = require("tabnine.consts")
+local state = require("tabnine.state")
 local utils = require("tabnine.utils")
 local api = vim.api
 local fn = vim.fn
@@ -26,7 +27,9 @@ function M.reload_buf_supports_symbols()
 end
 
 function M.should_display()
-	return vim.lsp.buf.server_ready()
+	return config.get_config().codelens_enabled
+		and state.active
+		and vim.lsp.buf.server_ready()
 		and not vim.tbl_contains(config.get_config().exclude_filetypes, vim.bo.filetype)
 		and buf_supports_symbols
 end

lua/tabnine/chat/init.lua

@@ -196,7 +196,7 @@ end
 
 function M.open(on_ready)
 	if not M.enabled then
-		vim.notify("Tabnine Chat is available only for preview users")
+		vim.notify("Tabnine Chat is available only for Pro users")
 		return
 	end
 

lua/tabnine/config.lua

@@ -9,6 +9,7 @@ function M.set_config(o)
 		debounce_ms = 800,
 		suggestion_color = { gui = "#808080", cterm = 244 },
 		codelens_color = { gui = "#808080", cterm = 244 },
+		codelens_enabled = true,
 		exclude_filetypes = { "TelescopePrompt", "NvimTree" },
 		log_file_path = nil,
 		tabnine_enterprise_host = nil,