mirror of
https://github.com/logancyang/obsidian-copilot.git
synced 2026-07-22 07:50:24 +00:00
a5300eb707
* Improve plus user onboarding * Update embedding switch logic * Fix reset settings and new user * make isplususer default to false * update default state
82 lines
2.8 KiB
TypeScript
82 lines
2.8 KiB
TypeScript
import { ChainType } from "@/chainFactory";
|
|
import {
|
|
ChatModelProviders,
|
|
ChatModels,
|
|
EmbeddingModelProviders,
|
|
EmbeddingModels,
|
|
PlusUtmMedium,
|
|
} from "@/constants";
|
|
import { BrevilabsClient } from "@/LLMProviders/brevilabsClient";
|
|
import { getSettings, setSettings, updateSetting, useSettingsValue } from "@/settings/model";
|
|
import { setChainType, setModelKey } from "@/aiParams";
|
|
import { CopilotPlusExpiredModal } from "@/components/modals/CopilotPlusExpiredModal";
|
|
import VectorStoreManager from "@/search/vectorStoreManager";
|
|
|
|
export const DEFAULT_COPILOT_PLUS_CHAT_MODEL = ChatModels.COPILOT_PLUS_FLASH;
|
|
export const DEFAULT_COPILOT_PLUS_CHAT_MODEL_KEY =
|
|
DEFAULT_COPILOT_PLUS_CHAT_MODEL + "|" + ChatModelProviders.COPILOT_PLUS;
|
|
export const DEFAULT_COPILOT_PLUS_EMBEDDING_MODEL = EmbeddingModels.COPILOT_PLUS_SMALL;
|
|
export const DEFAULT_COPILOT_PLUS_EMBEDDING_MODEL_KEY =
|
|
DEFAULT_COPILOT_PLUS_EMBEDDING_MODEL + "|" + EmbeddingModelProviders.COPILOT_PLUS;
|
|
|
|
/** Check if the model key is a Copilot Plus model. */
|
|
export function isPlusModel(modelKey: string): boolean {
|
|
return modelKey.split("|")[1] === EmbeddingModelProviders.COPILOT_PLUS;
|
|
}
|
|
|
|
/** Hook to get the isPlusUser setting. */
|
|
export function useIsPlusUser(): boolean | undefined {
|
|
const settings = useSettingsValue();
|
|
return settings.isPlusUser;
|
|
}
|
|
|
|
/** Check if the user is a Plus user. */
|
|
export async function checkIsPlusUser(): Promise<boolean | undefined> {
|
|
if (!getSettings().plusLicenseKey) {
|
|
turnOffPlus();
|
|
return false;
|
|
}
|
|
const brevilabsClient = BrevilabsClient.getInstance();
|
|
return await brevilabsClient.validateLicenseKey();
|
|
}
|
|
|
|
/**
|
|
* Apply the Copilot Plus settings.
|
|
* WARNING! If the embedding model is changed, the vault will be indexed. Use it
|
|
* with caution.
|
|
*/
|
|
export function applyPlusSettings(): void {
|
|
const defaultModelKey = DEFAULT_COPILOT_PLUS_CHAT_MODEL_KEY;
|
|
const embeddingModelKey = DEFAULT_COPILOT_PLUS_EMBEDDING_MODEL_KEY;
|
|
const previousEmbeddingModelKey = getSettings().embeddingModelKey;
|
|
setModelKey(defaultModelKey);
|
|
setChainType(ChainType.COPILOT_PLUS_CHAIN);
|
|
setSettings({
|
|
defaultModelKey,
|
|
embeddingModelKey,
|
|
defaultChainType: ChainType.COPILOT_PLUS_CHAIN,
|
|
});
|
|
if (previousEmbeddingModelKey !== embeddingModelKey) {
|
|
VectorStoreManager.getInstance().indexVaultToVectorStore(true);
|
|
}
|
|
}
|
|
|
|
export function createPlusPageUrl(medium: PlusUtmMedium): string {
|
|
return `https://www.obsidiancopilot.com?utm_source=obsidian&utm_medium=${medium}`;
|
|
}
|
|
|
|
export function navigateToPlusPage(medium: PlusUtmMedium): void {
|
|
window.open(createPlusPageUrl(medium), "_blank");
|
|
}
|
|
|
|
export function turnOnPlus(): void {
|
|
updateSetting("isPlusUser", true);
|
|
}
|
|
|
|
export function turnOffPlus(): void {
|
|
const previousIsPlusUser = getSettings().isPlusUser;
|
|
updateSetting("isPlusUser", false);
|
|
if (previousIsPlusUser) {
|
|
new CopilotPlusExpiredModal(app).open();
|
|
}
|
|
}
|