cactus-react-native 1.0.1 → 1.1.0

package/src/types/CactusLM.ts

@@ -1,6 +1,7 @@
 export interface CactusLMParams {
   model?: string;
   contextSize?: number;
+  corpusDir?: string;
 }
 
 export interface CactusLMDownloadParams {
@@ -9,10 +10,11 @@ export interface CactusLMDownloadParams {
 
 export interface Message {
   role: 'user' | 'assistant' | 'system';
-  content: string;
+  content?: string;
+  images?: string[];
 }
 
-export interface Options {
+export interface CompleteOptions {
   temperature?: number;
   topP?: number;
   topK?: number;
@@ -21,7 +23,6 @@ export interface Options {
 }
 
 export interface Tool {
-  type: 'function';
   name: string;
   description: string;
   parameters: {
@@ -38,9 +39,10 @@ export interface Tool {
 
 export interface CactusLMCompleteParams {
   messages: Message[];
-  options?: Options;
+  options?: CompleteOptions;
   tools?: Tool[];
   onToken?: (token: string) => void;
+  mode?: 'local' | 'hybrid';
 }
 
 export interface CactusLMCompleteResult {
@@ -66,6 +68,10 @@ export interface CactusLMEmbedResult {
   embedding: number[];
 }
 
-export interface CactusLMGetModelsParams {
-  forceRefresh?: boolean;
+export interface CactusLMImageEmbedParams {
+  imagePath: string;
+}
+
+export interface CactusLMImageEmbedResult {
+  embedding: number[];
 }

package/src/types/CactusSTT.ts

@@ -0,0 +1,42 @@
+export interface CactusSTTParams {
+  model?: string;
+  contextSize?: number;
+}
+
+export interface CactusSTTDownloadParams {
+  onProgress?: (progress: number) => void;
+}
+
+export interface TranscribeOptions {
+  temperature?: number;
+  topP?: number;
+  topK?: number;
+  maxTokens?: number;
+  stopSequences?: string[];
+}
+
+export interface CactusSTTTranscribeParams {
+  audioFilePath: string;
+  prompt?: string;
+  options?: TranscribeOptions;
+  onToken?: (token: string) => void;
+}
+
+export interface CactusSTTTranscribeResult {
+  success: boolean;
+  response: string;
+  timeToFirstTokenMs: number;
+  totalTimeMs: number;
+  tokensPerSecond: number;
+  prefillTokens: number;
+  decodeTokens: number;
+  totalTokens: number;
+}
+
+export interface CactusSTTAudioEmbedParams {
+  audioPath: string;
+}
+
+export interface CactusSTTAudioEmbedResult {
+  embedding: number[];
+}