obsidian-plugin/logancyang_obsidian-copilot
1
0
Fork 0
mirror of https://github.com/logancyang/obsidian-copilot.git synced 2026-07-22 07:50:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add support for Cohere's Command-R and Command-R+ models (#619)

Browse source
This commit is contained in:
o-mikhailovskii 2024-09-08 23:50:38 +02:00 committed by GitHub
parent 9c770a47f3
commit c6fbe77fe5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 2351 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
esbuild.config.mjs
View file

@ -34,7 +34,7 @@ const context = await esbuild.context({
"@lezer/lr",
],
format: "cjs",
target: "es2018",
target: "es2020",
logLevel: "info",
sourcemap: prod ? false : "inline",
treeShaking: true,

2334
package-lock.json generated
View file

File diff suppressed because it is too large Load diff

4
package.json
View file

@ -64,14 +64,14 @@
"@huggingface/inference": "^2.6.4",
"@koa/cors": "^5.0.0",
"@langchain/anthropic": "^0.1.21",
"@langchain/cohere": "^0.0.2",
"@langchain/cohere": "^0.0.11",
"@langchain/community": "^0.0.16",
"@langchain/google-genai": "^0.0.7",
"@langchain/groq": "^0.0.9",
"@tabler/icons-react": "^2.14.0",
"@types/pouchdb": "^6.4.0",
"axios": "^1.3.4",
"cohere-ai": "^7.6.2",
"cohere-ai": "^7.13.0",
"crypto-js": "^4.1.1",
"esbuild-plugin-svg": "^0.1.0",
"esbuild-plugins-node-modules-polyfill": "^1.6.5",

9
src/LLMProviders/chatModelManager.ts
View file

@ -2,6 +2,7 @@ import { CustomModel, LangChainParams, ModelConfig } from "@/aiParams";
import { BUILTIN_CHAT_MODELS, ChatModelProviders } from "@/constants";
import EncryptionService from "@/encryptionService";
import { ChatAnthropicWrapped, ProxyChatOpenAI } from "@/langchainWrappers";
import { ChatCohere } from "@langchain/cohere";
import { ChatGoogleGenerativeAI } from "@langchain/google-genai";
import { ChatGroq } from "@langchain/groq";
import { ChatOllama } from "@langchain/ollama";
@ -78,6 +79,10 @@ export default class ChatModelManager {
azureOpenAIApiDeploymentName: params.azureOpenAIApiDeploymentName,
azureOpenAIApiVersion: params.azureOpenAIApiVersion,
},
[ChatModelProviders.COHEREAI]: {
apiKey: decrypt(customModel.apiKey || params.cohereApiKey),
model: customModel.name,
},
[ChatModelProviders.GOOGLE]: {
apiKey: decrypt(customModel.apiKey || params.googleApiKey),
modelName: customModel.name,
@ -146,6 +151,10 @@ export default class ChatModelManager {
constructor = ChatAnthropicWrapped;
apiKey = model.apiKey || this.getLangChainParams().anthropicApiKey;
break;
case ChatModelProviders.COHEREAI:
constructor = ChatCohere;
apiKey = model.apiKey || this.getLangChainParams().cohereApiKey;
break;
case ChatModelProviders.OPENROUTERAI:
constructor = ProxyChatOpenAI;
apiKey = model.apiKey || this.getLangChainParams().openRouterAiApiKey;

1
src/aiParams.ts
View file

@ -12,6 +12,7 @@ export interface ModelConfig {
openAIApiKey?: string;
openAIOrgId?: string;
anthropicApiKey?: string;
cohereApiKey?: string;
azureOpenAIApiKey?: string;
azureOpenAIApiInstanceName?: string;
azureOpenAIApiDeploymentName?: string;

15
src/constants.ts
View file

@ -16,6 +16,8 @@ export enum ChatModels {
AZURE_OPENAI = "azure-openai",
CLAUDE_3_5_SONNET = "claude-3-5-sonnet-20240620",
CLAUDE_3_HAIKU = "claude-3-haiku-20240307",
COMMAND_R = "command-r",
COMMAND_R_PLUS = "command-r-plus",
}
// Model Providers
@ -23,6 +25,7 @@ export enum ChatModelProviders {
OPENAI = "openai",
AZURE_OPENAI = "azure openai",
ANTHROPIC = "anthropic",
COHEREAI = "cohereai",
GOOGLE = "google",
OPENROUTERAI = "openrouterai",
GROQ = "groq",
@ -62,6 +65,18 @@ export const BUILTIN_CHAT_MODELS: CustomModel[] = [
enabled: true,
isBuiltIn: true,
},
{
name: ChatModels.COMMAND_R,
provider: ChatModelProviders.COHEREAI,
enabled: true,
isBuiltIn: true,
},
{
name: ChatModels.COMMAND_R_PLUS,
provider: ChatModelProviders.COHEREAI,
enabled: true,
isBuiltIn: true,
},
{
name: ChatModels.GEMINI_PRO,
provider: ChatModelProviders.GOOGLE,