Switch query check sub-graph to use first and last messages

Author: slimslenderslacks

prompts/sql/query-check.md

@@ -14,7 +14,7 @@ tools:
     container:
       image: vonwig/sqlite:latest
       command:
-        - "./Chinook.db"
+        - "{{database}}"
         - "{{sql}}"
 tool_choice: required
 ---

src/graphs/sql.clj

@@ -102,6 +102,7 @@
                                  {:init-state
                                   [(state/messages-reset)
                                    (state/messages-from-prompt "prompts/sql/query-check.md")
+                                   (state/messages-take 1)
                                    (state/messages-take-last 1)]
                                   :construct-graph graph/generate-one-tool-call
                                   :next-state state/append-new-messages})]

src/state.clj

@@ -93,6 +93,11 @@
     (-> state
         (update-in [:messages] (fnil concat []) (take-last n (:messages orig))))))
 
+(defn messages-take [n]
+  (fn [orig state]
+    (-> state
+        (update-in [:messages] (fnil concat []) (take n (:messages orig))))))
+
 (defn messages-append [coll]
   (fn [_ state]
     (-> state