wip

olimorris · Feb 13, 2025 · 861dc80 · 861dc80
1 parent 04b72b3
commit 861dc80
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/lua/codecompanion/strategies/inline/init.lua b/lua/codecompanion/strategies/inline/init.lua
@@ -415,7 +415,7 @@ function Inline:submit(prompt)
     })
 end
 
----Once the requset has been completed, we can process the output
+---Once the request has been completed, we can process the output
 ---@param output string The output from the LLM
 ---@return nil
 function Inline:done(output)

diff --git a/tests/strategies/inline/test_inline.lua b/tests/strategies/inline/test_inline.lua
@@ -135,8 +135,9 @@ T["Inline"]["forms correct prompts"] = function()
 end
 
 T["Inline"]["generates correct prompt structure"] = function()
-  -- Capture the submitted prompts
   local submitted_prompts = {}
+
+  -- Mock the submit function
   function inline:submit(prompts)
     submitted_prompts = prompts
   end
@@ -147,11 +148,8 @@ T["Inline"]["generates correct prompt structure"] = function()
   }
   inline:prompt(user_prompt)
 
-  -- Should be a system prompt and the user prompt
-  h.eq(#submitted_prompts, 2)
-
+  h.eq(#submitted_prompts, 2) -- Should be a system prompt and the user prompt
   h.eq(submitted_prompts[1].role, "system")
-
   h.eq(submitted_prompts[2].role, "user")
   h.eq(submitted_prompts[2].content, "<user_prompt>Test prompt</user_prompt>")
 end