shitton

src/ai/tools/utils.ts

@@ -1,5 +1,8 @@
 import {Ollama} from "ollama";
 import {z} from "zod";
+import {toolsLogger} from "./tool-logger";
+
+const logger = toolsLogger.child("utils");
 
 export function asNonEmptyString(value: unknown): string | undefined {
     return typeof value === "string" && value.trim().length > 0
@@ -79,14 +82,15 @@ export async function unloadAllOllamaModels(ollama: Ollama, exceptFor?: string[]
             );
 
         await Promise.all(unloadPromises);
-        console.log("All models have been requested to unload" + (exceptFor?.length ? ` except for [${exceptFor?.join(", ")}].` : "."));
+        logger.info("ollama.unload_all.done", {count: modelsToUnload.length, exceptFor});
     } catch (error) {
-        console.error("Error unloading models:", error);
+        logger.error("ollama.unload_all.failed", {exceptFor, error});
     }
 }
 
 export async function loadOllamaModel(model: string, ollama: Ollama, contextLength: number): Promise<boolean> {
     try {
+        logger.info("ollama.load.start", {model, contextLength});
         await ollama.generate({
             model: model,
             stream: false,
@@ -95,9 +99,10 @@ export async function loadOllamaModel(model: string, ollama: Ollama, contextLeng
                 num_ctx: contextLength
             }
         });
+        logger.info("ollama.load.done", {model, contextLength});
         return true;
     } catch (e: unknown) {
-        console.error("Error loading Ollama model:", model);
+        logger.error("ollama.load.failed", {model, contextLength, error: e});
         return false;
     }
 }