Skip to content
Snippets Groups Projects
Unverified Commit 8743be67 authored by Timothy Carambat's avatar Timothy Carambat Committed by GitHub
Browse files

assume default model where appropriate (#366)

* assume default model where appropriate

* merge with master and fix other model refs
parent c22c50cc
No related branches found
No related tags found
No related merge requests found
......@@ -12,7 +12,7 @@ class AnthropicLLM {
apiKey: process.env.ANTHROPIC_API_KEY,
});
this.anthropic = anthropic;
this.model = process.env.ANTHROPIC_MODEL_PREF;
this.model = process.env.ANTHROPIC_MODEL_PREF || "claude-2";
this.limits = {
history: this.promptWindowLimit() * 0.15,
system: this.promptWindowLimit() * 0.15,
......
......@@ -73,7 +73,7 @@ Context:
async sendChat(chatHistory = [], prompt, workspace = {}, rawHistory = []) {
if (!this.model)
throw new Error(
`LMStudio chat: ${model} is not valid or defined for chat completion!`
`LMStudio chat: ${this.model} is not valid or defined for chat completion!`
);
const textResponse = await this.lmstudio
......@@ -110,7 +110,7 @@ Context:
async streamChat(chatHistory = [], prompt, workspace = {}, rawHistory = []) {
if (!this.model)
throw new Error(
`LMStudio chat: ${model} is not valid or defined for chat completion!`
`LMStudio chat: ${this.model} is not valid or defined for chat completion!`
);
const streamRequest = await this.lmstudio.createChatCompletion(
......
......@@ -11,7 +11,7 @@ class OpenAiLLM extends OpenAiEmbedder {
apiKey: process.env.OPEN_AI_KEY,
});
this.openai = new OpenAIApi(config);
this.model = process.env.OPEN_MODEL_PREF;
this.model = process.env.OPEN_MODEL_PREF || "gpt-3.5-turbo";
this.limits = {
history: this.promptWindowLimit() * 0.15,
system: this.promptWindowLimit() * 0.15,
......@@ -107,15 +107,14 @@ Context:
}
async sendChat(chatHistory = [], prompt, workspace = {}, rawHistory = []) {
const model = process.env.OPEN_MODEL_PREF;
if (!(await this.isValidChatCompletionModel(model)))
if (!(await this.isValidChatCompletionModel(this.model)))
throw new Error(
`OpenAI chat: ${model} is not valid for chat completion!`
`OpenAI chat: ${this.model} is not valid for chat completion!`
);
const textResponse = await this.openai
.createChatCompletion({
model,
model: this.model,
temperature: Number(workspace?.openAiTemp ?? 0.7),
n: 1,
messages: await this.compressMessages(
......@@ -145,15 +144,14 @@ Context:
}
async streamChat(chatHistory = [], prompt, workspace = {}, rawHistory = []) {
const model = process.env.OPEN_MODEL_PREF;
if (!(await this.isValidChatCompletionModel(model)))
if (!(await this.isValidChatCompletionModel(this.model)))
throw new Error(
`OpenAI chat: ${model} is not valid for chat completion!`
`OpenAI chat: ${this.model} is not valid for chat completion!`
);
const streamRequest = await this.openai.createChatCompletion(
{
model,
model: this.model,
stream: true,
temperature: Number(workspace?.openAiTemp ?? 0.7),
n: 1,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment