klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-09 05:39:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Extract isUserSubscribed check from Agents page to make it resusable

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2024-10-20 22:57:34 -07:00
parent bdbe8f003e
commit 54b92eaf73
2 changed files with 14 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/interface/web/app/agents/page.tsx
View File

@@ -11,7 +11,7 @@ import {
UserProfile, UserProfile,
ModelOptions, ModelOptions,
useUserConfig, useUserConfig,
SubscriptionStates, isUserSubscribed,
} from "../common/auth"; } from "../common/auth";
import { Lightning, Plus } from "@phosphor-icons/react"; import { Lightning, Plus } from "@phosphor-icons/react";
@@ -267,14 +267,7 @@ export default function Agents() {
const modelOptions: ModelOptions[] = userConfig?.chat_model_options || []; const modelOptions: ModelOptions[] = userConfig?.chat_model_options || [];
const selectedChatModelOption: number = userConfig?.selected_chat_model_config || 0; const selectedChatModelOption: number = userConfig?.selected_chat_model_config || 0;
const isSubscribed: boolean = const isSubscribed: boolean = isUserSubscribed(userConfig);
(userConfig?.subscription_state &&
[
SubscriptionStates.SUBSCRIBED.valueOf(),
SubscriptionStates.TRIAL.valueOf(),
SubscriptionStates.UNSUBSCRIBED.valueOf(),
].includes(userConfig.subscription_state)) ||
false;
// The default model option should map to the item in the modelOptions array that has the same id as the selectedChatModelOption // The default model option should map to the item in the modelOptions array that has the same id as the selectedChatModelOption
const defaultModelOption = modelOptions.find( const defaultModelOption = modelOptions.find(

12
src/interface/web/app/common/auth.ts
View File

@@ -93,3 +93,15 @@ export function useUserConfig(detailed: boolean = false) {
return { userConfig, isLoadingUserConfig }; return { userConfig, isLoadingUserConfig };
} }
export function isUserSubscribed(userConfig: UserConfig | null): boolean {
return (
(userConfig?.subscription_state &&
[
SubscriptionStates.SUBSCRIBED.valueOf(),
SubscriptionStates.TRIAL.valueOf(),
SubscriptionStates.UNSUBSCRIBED.valueOf(),
].includes(userConfig.subscription_state)) ||
false
);
}