the final parts of tutorial

1 files changed, 7 insertions, 2 deletions

diff --git a/lua/99/request/init.lua b/lua/99/request/init.lua
index 327aa0f..3cff512 100644
--- a/lua/99/request/init.lua
+++ b/lua/99/request/init.lua
@@ -81,8 +81,10 @@ end
 --- @param obs _99.Providers.Observer | nil
 local function observer_from_request(r, obs)
   return {
+    on_start = obs and obs.on_start or function() end,
     on_complete = function(status, res)
       r.state = status
+      r.context._99:finish_request(r.context, status)
       if obs then
         obs.on_complete(status, res)
       end
@@ -108,7 +110,6 @@ function Request:start(observer)
   )
   self.state = "requesting"
 
-  self.context._99:track_request(self.context)
   self.context:finalize()
   for _, content in ipairs(self.context.ai_context) do
     self:add_prompt_content(content)
@@ -117,7 +118,11 @@ function Request:start(observer)
   local prompt = table.concat(self._content, "\n")
   self.context:save_prompt(prompt)
   self.logger:debug("start", "prompt", prompt)
-  self.provider:make_request(prompt, self, observer_from_request(self, observer))
+  self.provider:make_request(
+    prompt,
+    self,
+    observer_from_request(self, observer)
+  )
 end
 
 return Request