...

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