Add --log-level option

Author: ericdallo

CHANGELOG.md

@@ -6,6 +6,7 @@
 - Add web_search capability
 - Add `o3` model support.
 - Support custom API urls for OpenAI and Anthropic
+- Add `--log-level <level>` option for better debugging.
 
 ## 0.0.3
 

src/eca/llm_providers/anthropic.clj

@@ -72,6 +72,7 @@
         content-block* (atom nil)
         on-response-fn
         (fn handle-response [event data]
+          (llm-util/log-response logger-tag event data)
           (case event
             "content_block_delta" (case (-> data :delta :type)
                                     "text_delta" (on-message-received {:type :text

src/eca/llm_providers/ollama.clj

@@ -2,6 +2,7 @@
   (:require
    [cheshire.core :as json]
    [clojure.java.io :as io]
+   [eca.llm-util :as llm-util]
    [eca.logger :as logger]
    [hato.client :as http]))
 
@@ -23,7 +24,9 @@
                                  {:throw-exceptions? false
                                   :as :json})]
       (if (= 200 status)
-        (:models body)
+        (do
+          (llm-util/log-response logger-tag "api_ps" body)
+          (:models body))
         (do
           (logger/warn logger-tag "Unknown status code:" status)
           [])))
@@ -72,6 +75,7 @@
               :stream true}
         url (format chat-url (base-url host port))
         on-response-fn (fn handle-response [data]
+                         (llm-util/log-response logger-tag "chat" data)
                          (let [{:keys [message done_reason]} data]
                            (on-message-received
                             (cond-> {}

src/eca/llm_providers/openai.clj

@@ -57,6 +57,7 @@
                        web-search (conj {:type "web_search_preview"}))
               :stream true}
         on-response-fn (fn handle-response [event data]
+                         (llm-util/log-response logger-tag event data)
                          (case event
                            "response.output_text.delta" (on-message-received {:type :text
                                                                               :text (:delta data)})

src/eca/llm_util.clj

@@ -1,7 +1,8 @@
 (ns eca.llm-util
   (:require
    [cheshire.core :as json]
-   [clojure.string :as string])
+   [clojure.string :as string]
+   [eca.logger :as logger])
   (:import
    [java.io BufferedReader]))
 
@@ -14,3 +15,6 @@
           (cons [(subs event 7)
                  (json/parse-string (subs data 6) true)]
                 (lazy-seq (event-data-seq rdr))))))))
+
+(defn log-response [tag event data]
+  (logger/debug tag event data))

src/eca/logger.clj

@@ -2,16 +2,27 @@
 
 (set! *warn-on-reflection* true)
 
-(defn ^:private stderr-print [& args]
-  (binding [*out* *err*]
-    (apply println args)))
+(def ^:dynamic *level* :info)
+
+(def ^:private level->value
+  {:error 1
+   :warn 2
+   :info 3
+   :debug 4})
+
+(defn ^:private stderr-print [level & args]
+  (when (<= (level->value level) (level->value *level*))
+    (binding [*out* *err*]
+      (apply println args))))
 
-;; TODO create better logger capability.
 (defn info [& args]
-  (apply stderr-print args))
+  (apply stderr-print :info args))
 
 (defn warn [& args]
-  (apply stderr-print args))
+  (apply stderr-print :warn args))
+
+(defn debug [& args]
+  (apply stderr-print :debug args))
 
 (defn format-time-delta-ms [start-time end-time]
   (format "%.0fms" (float (/ (- end-time start-time) 1000000))))

src/eca/main.clj

@@ -6,6 +6,7 @@
    [borkdude.dynaload]
    [clojure.string :as string]
    [eca.config :as config]
+   [eca.logger :as logger]
    [eca.server :as server]))
 
 (set! *warn-on-reflection* true)
@@ -37,11 +38,19 @@
   (str "The following errors occurred while parsing your command:\n\n"
        (string/join \newline errors)))
 
+(def log-levels #{"error" "warn" "info" "debug"})
+
 (def cli-spec
   {:order [:help :version :verbose]
    :spec {:help {:alias :h
                  :desc "Print the available commands and its options"}
           :version {:desc "Print eca version"}
+          :log-level {:ref "<LEVEL>"
+                      :desc "The log level of eca logs, accepts. Defaults to 'info'."
+                      :default "info"
+                      :validate {:pred log-levels
+                                 :ex-msg (fn [{:keys [_option _value]}]
+                                           (format "Must be in %s" log-levels))}}
           :verbose {:desc "Use stdout for eca logs instead of default log settings"}}})
 
 (defn ^:private parse-opts
@@ -81,6 +90,7 @@
 (defn ^:private handle-action!
   [action options]
   (when (= "server" action)
+    (alter-var-root #'logger/*level* (constantly (keyword (:log-level options))))
     (let [finished @(server/run-io-server! (:verbose options))]
       {:result-code (if (= :done finished) 0 1)})))