mirror of
https://github.com/khoaliber/khoj.git
synced 2026-03-03 05:29:12 +00:00
405c047c0c
Currently, the personality of the agent is only included in the final response that it returns to the user. Historically, this was because models were quite bad at navigating the additional context of personality, and there was a bias towards having more control over certain operations (e.g., tool selection, question extraction). Going forward, it should be more approachable to have prompts included in the sub tasks that Khoj runs in order to response to a given query. Make this possible in this PR. This also sets us up for agent creation becoming available soon. Create custom agents in #928 Agents are useful insofar as you can personalize them to fulfill specific subtasks you need to accomplish. In this PR, we add support for using custom agents that can be configured with a custom system prompt (aka persona) and knowledge base (from your own indexed documents). Once created, private agents can be accessible only to the creator, and protected agents can be accessible via a direct link. Custom tool selection for agents in #930 Expose the functionality to select which tools a given agent has access to. By default, they have all. Can limit both information sources and output modes. Add new tools to the agent modification form
96 lines
2.4 KiB
TypeScript
96 lines
2.4 KiB
TypeScript
"use client";
|
|
|
|
import useSWR from "swr";
|
|
|
|
export interface UserProfile {
|
|
email: string;
|
|
username: string;
|
|
photo: string;
|
|
is_active: boolean;
|
|
has_documents: boolean;
|
|
detail: string;
|
|
}
|
|
|
|
const fetcher = (url: string) =>
|
|
window
|
|
.fetch(url)
|
|
.then((res) => res.json())
|
|
.catch((err) => console.warn(err));
|
|
|
|
export function useAuthenticatedData() {
|
|
const { data, error } = useSWR<UserProfile>("/api/v1/user", fetcher, {
|
|
revalidateOnFocus: false,
|
|
});
|
|
|
|
if (error || !data || data.detail === "Forbidden") return null;
|
|
|
|
return data;
|
|
}
|
|
|
|
export interface ModelOptions {
|
|
id: number;
|
|
name: string;
|
|
}
|
|
export interface SyncedContent {
|
|
computer: boolean;
|
|
github: boolean;
|
|
notion: boolean;
|
|
}
|
|
|
|
export enum SubscriptionStates {
|
|
EXPIRED = "expired",
|
|
TRIAL = "trial",
|
|
SUBSCRIBED = "subscribed",
|
|
UNSUBSCRIBED = "unsubscribed",
|
|
INVALID = "invalid",
|
|
}
|
|
|
|
export interface UserConfig {
|
|
// user info
|
|
username: string;
|
|
user_photo: string | null;
|
|
is_active: boolean;
|
|
given_name: string;
|
|
phone_number: string;
|
|
is_phone_number_verified: boolean;
|
|
// user content settings
|
|
enabled_content_source: SyncedContent;
|
|
has_documents: boolean;
|
|
notion_token: string | null;
|
|
// user model settings
|
|
search_model_options: ModelOptions[];
|
|
selected_search_model_config: number;
|
|
chat_model_options: ModelOptions[];
|
|
selected_chat_model_config: number;
|
|
paint_model_options: ModelOptions[];
|
|
selected_paint_model_config: number;
|
|
voice_model_options: ModelOptions[];
|
|
selected_voice_model_config: number;
|
|
// user billing info
|
|
subscription_state: SubscriptionStates;
|
|
subscription_renewal_date: string;
|
|
// server settings
|
|
khoj_cloud_subscription_url: string | undefined;
|
|
billing_enabled: boolean;
|
|
is_eleven_labs_enabled: boolean;
|
|
is_twilio_enabled: boolean;
|
|
khoj_version: string;
|
|
anonymous_mode: boolean;
|
|
notion_oauth_url: string;
|
|
detail: string;
|
|
}
|
|
|
|
export function useUserConfig(detailed: boolean = false) {
|
|
const url = `/api/settings?detailed=${detailed}`;
|
|
const {
|
|
data: userConfig,
|
|
error,
|
|
isLoading: isLoadingUserConfig,
|
|
} = useSWR<UserConfig>(url, fetcher, { revalidateOnFocus: false });
|
|
|
|
if (error || !userConfig || userConfig?.detail === "Forbidden")
|
|
return { userConfig: null, isLoadingUserConfig };
|
|
|
|
return { userConfig, isLoadingUserConfig };
|
|
}
|