Add basic claude support

This commit is contained in:
maver
2023-06-25 18:20:53 +02:00
parent c920b6c511
commit aaa2c48e7d
5 changed files with 80 additions and 60 deletions

View File

@ -90,6 +90,7 @@ import {
openai_messages_count,
getTokenCountOpenAI,
chat_completion_sources,
setupClaudePromptManager,
getTokenizerModel,
getChatCompletionModel,
} from "./scripts/openai.js";
@ -4529,6 +4530,18 @@ function changeMainAPI() {
getStatus();
getHordeModels();
}
console.log(oai_settings.chat_completion_source)
switch (oai_settings.chat_completion_source) {
case chat_completion_sources.OPENAI:
console.log('Setting up OpenAI prompt manager');
setupOpenAIPromptManager(oai_settings);
break;
case chat_completion_sources.CLAUDE:
console.log('Setting up Claude prompt manager');
setupClaudePromptManager(oai_settings);
break;
}
}
////////////////////////////////////////////////////
@ -5121,7 +5134,6 @@ async function getSettings(type) {
// OpenAI
loadOpenAISettings(data, settings);
setupOpenAIPromptManager(settings);
// Horde
loadHordeSettings(settings);