|
11 | 11 |
|
12 | 12 | (set! *warn-on-reflection* true) |
13 | 13 |
|
14 | | -;; (:models @eca.db/db*) |
15 | | - |
16 | | -(def a (models/all)) |
17 | | - |
18 | | - |
19 | | - |
20 | 14 | (defn ^:private initialize-models! [db* config] |
21 | 15 | (let [all-models (models/all) |
22 | 16 | eca-models (filter |
23 | | - (fn [[model _config]] |
24 | | - (get-in config [:models model])) |
25 | | - all-models)] |
| 17 | + (fn [[model _config]] |
| 18 | + (get-in config [:models model])) |
| 19 | + all-models)] |
26 | 20 | (swap! db* update :models merge eca-models) |
27 | 21 | (when-let [custom-providers (seq (:customProviders config))] |
28 | 22 | (let [models (reduce |
29 | | - (fn [models [provider {provider-models :models default-model :defaultModel}]] |
30 | | - (reduce |
31 | | - (fn [m model] |
32 | | - (let [known-model (get all-models model)] |
33 | | - (assoc m |
34 | | - (str (name provider) "/" model) |
35 | | - {:tools (or (:tools known-model) true) |
36 | | - :reason? (or (:reason? known-model) true) |
37 | | - :web-search (or (:web-search known-model) true) |
38 | | - :max-output-tokens (:max-output-tokens known-model) |
39 | | - :custom-provider? true |
40 | | - :default-model? (= model default-model)}))) |
41 | | - models |
42 | | - provider-models)) |
43 | | - {} |
44 | | - custom-providers)] |
| 23 | + (fn [models [provider {provider-models :models default-model :defaultModel}]] |
| 24 | + (reduce |
| 25 | + (fn [m model] |
| 26 | + (let [known-model (get all-models model)] |
| 27 | + (assoc m |
| 28 | + (str (name provider) "/" model) |
| 29 | + {:tools (or (:tools known-model) true) |
| 30 | + :reason? (or (:reason? known-model) true) |
| 31 | + :web-search (or (:web-search known-model) true) |
| 32 | + :max-output-tokens (:max-output-tokens known-model) |
| 33 | + :custom-provider? true |
| 34 | + :default-model? (= model default-model)}))) |
| 35 | + models |
| 36 | + provider-models)) |
| 37 | + {} |
| 38 | + custom-providers)] |
45 | 39 | (swap! db* update :models merge models)))) |
46 | 40 | (when-let [ollama-models (seq (llm-api/extra-models config))] |
47 | 41 | (let [models (reduce |
48 | | - (fn [models {:keys [model] :as ollama-model}] |
49 | | - (assoc models |
50 | | - (str config/ollama-model-prefix model) |
51 | | - (select-keys ollama-model [:tools :reason?]))) |
52 | | - {} |
53 | | - ollama-models)] |
| 42 | + (fn [models {:keys [model] :as ollama-model}] |
| 43 | + (assoc models |
| 44 | + (str config/ollama-model-prefix model) |
| 45 | + (select-keys ollama-model [:tools :reason?]))) |
| 46 | + {} |
| 47 | + ollama-models)] |
54 | 48 | (swap! db* update :models merge models)))) |
55 | 49 |
|
56 | 50 | (defn initialize [{:keys [db* config]} params] |
|
0 commit comments