mirrors/mindcraft
1
0
Fork 0
mirror of https://github.com/kolbytn/mindcraft.git synced 2025-09-03 21:03:10 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

this absolutely will not work by the time im done with it

Browse source
This commit is contained in:
Copper 2024-06-18 19:09:09 -07:00 • committed by GitHub
parent 1504805f70
commit 1de106be19
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
src/agent/prompter.js
View file

@ -10,6 +10,7 @@ import { GPT } from '../models/gpt.js';
import { Claude } from '../models/claude.js'; import { Claude } from '../models/claude.js';
import { ReplicateAPI } from '../models/replicate.js'; import { ReplicateAPI } from '../models/replicate.js';
import { Local } from '../models/local.js'; import { Local } from '../models/local.js';
import { Mixtral } from '../models/groq.js';
export class Prompter { export class Prompter {
@ -31,6 +32,8 @@ export class Prompter {
chat.api = 'anthropic'; chat.api = 'anthropic';
else if (chat.model.includes('meta/') || chat.model.includes('mistralai/') || chat.model.includes('replicate/')) else if (chat.model.includes('meta/') || chat.model.includes('mistralai/') || chat.model.includes('replicate/'))
chat.api = 'replicate'; chat.api = 'replicate';
else if (chat.model.includes('mixtral'))
chat.api = 'groq';
else else
chat.api = 'ollama'; chat.api = 'ollama';
} }
@ -47,6 +50,8 @@ export class Prompter {
this.chat_model = new ReplicateAPI(chat.model, chat.url); this.chat_model = new ReplicateAPI(chat.model, chat.url);
else if (chat.api == 'ollama') else if (chat.api == 'ollama')
this.chat_model = new Local(chat.model, chat.url); this.chat_model = new Local(chat.model, chat.url);
else if (chat.api == 'groq')
this.chat_model = new Mixtral(chat.model, chat.url)
else else
throw new Error('Unknown API:', api); throw new Error('Unknown API:', api);