From 2b38cb2ebe82e5d434bd3644438c589121e0ac00 Mon Sep 17 00:00:00 2001
From: Dominik Polakovics <dominik.polakovics@cloonar.com>
Date: Thu, 12 Dec 2024 21:05:55 +0100
Subject: [PATCH] fix: fix the yaml loading

---
 lua/chatgpt_nvim/config.lua | 2 +-
 lua/chatgpt_nvim/init.lua   | 1 -
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/lua/chatgpt_nvim/config.lua b/lua/chatgpt_nvim/config.lua
index c41b500..ba71ef9 100644
--- a/lua/chatgpt_nvim/config.lua
+++ b/lua/chatgpt_nvim/config.lua
@@ -60,7 +60,7 @@ function M.load()
   if data and ok_yaml then
     local ok, result = pcall(lyaml.load, data)
     if ok and type(result) == "table" then
-      local config = result[1]
+      local config = result
       if type(config) == "table" then
         return {
           initial_prompt = config.initial_prompt or "",
diff --git a/lua/chatgpt_nvim/init.lua b/lua/chatgpt_nvim/init.lua
index 1fe3943..98c2014 100644
--- a/lua/chatgpt_nvim/init.lua
+++ b/lua/chatgpt_nvim/init.lua
@@ -21,7 +21,6 @@ local function parse_response(raw)
     vim.api.nvim_err_writeln("Failed to parse YAML response.")
     return nil
   end
-  data = data[1]
   return data
 end