fix: omg file generation

marco-souza · Sep 26, 2024 · 4f1c612 · 4f1c612
1 parent 90dc531
commit 4f1c612
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lua/ollama/ollama.lua b/lua/ollama/ollama.lua
@@ -122,7 +122,7 @@ local function generate_modelfile(prompt)
   logger.debug("Prompt is not empty, generating Modelfile using `omg`")
 
   -- gnerate file using omg
-  local modelfile = vim.fn.system("omg " .. prompt)
+  local modelfile = vim.fn.system("omg '" .. prompt .. "'")
   return modelfile
 end