From 51df9bab2838a87f8367932f97d70fc75801aca8 Mon Sep 17 00:00:00 2001 From: "Schmidt, Tibor" Date: Tue, 20 Feb 2024 23:47:58 +0100 Subject: [PATCH] chore: wip --- lua/gp/config.lua | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/lua/gp/config.lua b/lua/gp/config.lua index 35de64a..b8cb596 100644 --- a/lua/gp/config.lua +++ b/lua/gp/config.lua @@ -48,6 +48,10 @@ local config = { lmsudio = { endpoint = "http://localhost:1234/v1/chat/completions", }, + googleai = { + endpoint = "https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:streamGenerateContent?key={{secret}}", + secret = os.getenv("GOOGLEAI_API_KEY"), + }, }, -- prefix for all commands @@ -118,6 +122,24 @@ local config = { .. "- Don't elide any code from your output if the answer requires coding.\n" .. "- Take a deep breath; You've got this!\n", }, + { + provider = "googleai", + name = "ChatGemini", + chat = true, + command = false, + -- string with model name or table with model name and parameters + model = { model = "gpt-4", temperature = 1.1, top_p = 1 }, + -- system prompt (use this to specify the persona/role of the AI) + system_prompt = "You are a general AI assistant.\n\n" + .. "The user provided the additional info about how they would like you to respond:\n\n" + .. "- If you're unsure don't guess and say you don't know instead.\n" + .. "- Ask question if you need clarification to provide better answer.\n" + .. "- Think deeply and carefully from first principles step by step.\n" + .. "- Zoom out first to see the big picture and then zoom in to details.\n" + .. "- Use Socratic method to improve your thinking and coding skills.\n" + .. "- Don't elide any code from your output if the answer requires coding.\n" + .. "- Take a deep breath; You've got this!\n", + }, { provider = "ollama", name = "ChatOllama",