Thomas
45d8bc9443
* Add Homarr provider proxy foundation * Complete Homarr provider integration * Harden Homarr provider integration * Address Homarr provider review feedback * Address CodeRabbit provider feedback * Localize provider errors * Handle provider activity load failures * docs: improve workshop and homepage * Harden Homarr provider limits * fix: harden the Homarr provider * fix: address provider review feedback * fix: clarify provider request accounting * feat: complete the Homarr provider endpoint * fix: terminate invalid provider streams
68 lines
2.4 KiB
TypeScript
68 lines
2.4 KiB
TypeScript
"use client";
|
|
|
|
import { createContext, useContext } from "react";
|
|
|
|
import type { AssistantProvider } from "@homarr/definitions";
|
|
import type { WorkshopAssistantUsage, WorkshopUser } from "@homarr/workshop/schema";
|
|
|
|
import type { AssistantReasoningMode, AssistantRuntimeModelOption } from "./assistant-preferences";
|
|
|
|
/**
|
|
* Context and hooks live in their own module so that consumers such as the user menu can read the
|
|
* assistant state without pulling in `assistant-provider`, which statically imports the
|
|
* assistant-ui runtime, the Lexical composer, the Markdown renderer and Mermaid. Importing those
|
|
* from a component that renders on every page would ship them to every page.
|
|
*/
|
|
export interface AssistantContextValue {
|
|
enabled: boolean;
|
|
unavailableDescription: string | null;
|
|
opened: boolean;
|
|
isRunning: boolean;
|
|
isRefreshing: boolean;
|
|
unreadCount: number;
|
|
open: () => void;
|
|
close: () => void;
|
|
toggle: () => void;
|
|
sendPrompt: (prompt: string) => boolean;
|
|
refreshCurrentView: () => Promise<void>;
|
|
}
|
|
|
|
export const AssistantContext = createContext<AssistantContextValue | null>(null);
|
|
|
|
export interface AssistantPreferencesContextValue {
|
|
provider: AssistantProvider | null;
|
|
defaultModelId: string | null;
|
|
modelId: string | null;
|
|
models: AssistantRuntimeModelOption[];
|
|
reasoning: AssistantReasoningMode;
|
|
isLoading: boolean;
|
|
providerUser: WorkshopUser | null;
|
|
quota: WorkshopAssistantUsage | null;
|
|
quotaLoading: boolean;
|
|
quotaError: string | null;
|
|
setModelId: (modelId: string) => void;
|
|
setReasoning: (reasoning: AssistantReasoningMode) => void;
|
|
getRequestBody: () => { modelId?: string; reasoning: AssistantReasoningMode };
|
|
getRequestHeaders: () => Record<string, string>;
|
|
refreshQuota: () => Promise<void>;
|
|
signInToProvider: () => Promise<void>;
|
|
}
|
|
|
|
export const AssistantPreferencesContext = createContext<AssistantPreferencesContextValue | null>(null);
|
|
|
|
export const useHomarrAssistant = () => {
|
|
const value = useContext(AssistantContext);
|
|
if (!value) {
|
|
throw new Error("useHomarrAssistant must be used within AssistantProvider");
|
|
}
|
|
return value;
|
|
};
|
|
|
|
export const useOptionalHomarrAssistant = () => useContext(AssistantContext);
|
|
|
|
export const useAssistantPreferences = () => {
|
|
const value = useContext(AssistantPreferencesContext);
|
|
if (!value) throw new Error("useAssistantPreferences must be used within AssistantPreferencesProvider");
|
|
return value;
|
|
};
|