Tests

Author: ericdallo

integration-test/integration/chat/commands_test.clj

@@ -29,7 +29,7 @@
                        {:name "config" :arguments []}
                        {:name "doctor" :arguments []}
                        {:name "repo-map-show" :arguments []}
-                       {:name "prompt-show" :arguments []}]}
+                       {:name "prompt-show" :arguments [{:name "optional-prompt"}]}]}
            resp))))
 
   (testing "We query specific commands"

src/eca/features/context.clj

@@ -75,7 +75,7 @@
          (mapcat #(parse-agents-file (str %))))))
 
 (defn ^:private file->refined-context [path lines-range]
-  (let [ext (string/lower-case (fs/extension path))]
+  (let [ext (string/lower-case (or (fs/extension path) ""))]
     (if (contains? #{"png" "jpg" "jpeg" "gif" "webp"} ext)
       {:type :image
        :media-type (case ext

test/eca/features/chat_test.clj

@@ -8,7 +8,9 @@
    [eca.llm-api :as llm-api]
    [eca.test-helper :as h]
    [matcher-combinators.matchers :as m]
-   [matcher-combinators.test :refer [match?]]))
+   [matcher-combinators.test :refer [match?]]
+   [babashka.fs :as fs]
+   [clojure.string :as string]))
 
 (h/reset-components-before-test)
 
@@ -181,6 +183,43 @@
                 :role :system}]}
              (h/messages)))))))
 
+(deftest contexts-in-prompt-test
+  (testing "When prompt contains @file we add a user message"
+    (h/reset-components!)
+    (let [{:keys [chat-id]}
+          (complete!
+            {:message "Check @/path/to/file please"}
+            {:api-mock
+             (fn [{:keys [on-first-response-received
+                          on-message-received]}]
+               (on-first-response-received {:type :text :text "On it..."})
+               (on-message-received {:type :text :text "On it..."})
+               (on-message-received {:type :finish}))})]
+      (is (match?
+            {chat-id {:id chat-id
+                      :messages [{:role "user" :content [{:type :text :text "Check @/path/to/file please"}
+                                                         {:type :text :text (m/pred #(string/includes? % "<file path"))}]}
+                                 {:role "assistant" :content [{:type :text :text "On it..."}]}]}}
+            (:chats (h/db))))
+      (is (match?
+            {:chat-content-received
+             [{:chat-id chat-id
+               :content {:type :text :text "Check @/path/to/file please\n"}
+               :role :user}
+              {:chat-id chat-id
+               :content {:type :progress :state :running :text "Waiting model"}
+               :role :system}
+              {:chat-id chat-id
+               :content {:type :progress :state :running :text "Generating"}
+               :role :system}
+              {:chat-id chat-id
+               :content {:type :text :text "On it..."}
+               :role :assistant}
+              {:chat-id chat-id
+               :content {:state :finished :type :progress}
+               :role :system}]}
+            (h/messages))))))
+
 (deftest basic-tool-calling-prompt-test
   (testing "Asking to list directories, LLM will check for allowed directories and then list files"
     (h/reset-components!)