@subhajit-gorai/react-native-mediapipe-llm
Version:
React Native binding for Google AI Edge Gallery's MediaPipe on-device LLM inference engine
27 lines (24 loc) • 584 B
text/typescript
export interface LlmOptions {
modelPath: string;
maxTokens?: number;
temperature?: number;
topK?: number;
topP?: number;
}
export interface ChatMessage {
id: string;
text: string;
isUser: boolean;
timestamp: Date;
}
export interface LlmInferenceHook {
generateResponse: (prompt: string, partialCallback?: (partial: string) => void) => Promise<string>;
initialize: (options: LlmOptions) => Promise<boolean>;
isInitialized: boolean;
isLoading: boolean;
}
export type RootStackParamList = {
Welcome: undefined;
Chat: undefined;
Settings: undefined;
};