...

RvD-Projects · Mar 4, 2024 · 29c881e · 29c881e
1 parent 36970c8
commit 29c881e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/events/messageCreateOllama.ts b/src/events/messageCreateOllama.ts
@@ -104,5 +104,6 @@ function getModelByPrefix(prefix: string): string {
 
   //TODO: Use a shortname associative listing (json)
 
-  return prefix.split(':')[1]?.split(" ")[0] ?? "llama2";
+  const model = prefix.split(':')[1]?.split(" ")[0];
+  return model?.length ? model : "llama2";
 }