@reverbia/sdk 1.0.0-next.20251203130707 → 1.0.0-next.20251205064608

package/dist/react/index.d.mts

@@ -688,6 +688,17 @@ type UseMemoryResult = {
  */
 declare function useMemory(options?: UseMemoryOptions): UseMemoryResult;
 
+interface PdfFile {
+    url: string;
+    mediaType?: string;
+    filename?: string;
+}
+declare function usePdf(): {
+    extractPdfContext: (files: PdfFile[]) => Promise<string | null>;
+    isProcessing: boolean;
+    error: Error | null;
+};
+
 type UseModelsOptions = {
     /**
      * Custom function to get auth token for API calls
@@ -846,4 +857,4 @@ declare function executeTool(tool: ClientTool, params: Record<string, unknown>):
     error?: string;
 }>;
 
-export { type ClientTool, DEFAULT_TOOL_SELECTOR_MODEL, type ToolExecutionResult, type ToolParameter, type ToolSelectionResult, createMemoryContextSystemMessage, decryptData, decryptDataBytes, encryptData, executeTool, extractConversationContext, formatMemoriesForChat, selectTool, useChat, useEncryption, useImageGeneration, useMemory, useModels, useSearch };
+export { type ClientTool, DEFAULT_TOOL_SELECTOR_MODEL, type PdfFile, type ToolExecutionResult, type ToolParameter, type ToolSelectionResult, createMemoryContextSystemMessage, decryptData, decryptDataBytes, encryptData, executeTool, extractConversationContext, formatMemoriesForChat, selectTool, useChat, useEncryption, useImageGeneration, useMemory, useModels, usePdf, useSearch };

package/dist/react/index.d.ts

@@ -688,6 +688,17 @@ type UseMemoryResult = {
  */
 declare function useMemory(options?: UseMemoryOptions): UseMemoryResult;
 
+interface PdfFile {
+    url: string;
+    mediaType?: string;
+    filename?: string;
+}
+declare function usePdf(): {
+    extractPdfContext: (files: PdfFile[]) => Promise<string | null>;
+    isProcessing: boolean;
+    error: Error | null;
+};
+
 type UseModelsOptions = {
     /**
      * Custom function to get auth token for API calls
@@ -846,4 +857,4 @@ declare function executeTool(tool: ClientTool, params: Record<string, unknown>):
     error?: string;
 }>;
 
-export { type ClientTool, DEFAULT_TOOL_SELECTOR_MODEL, type ToolExecutionResult, type ToolParameter, type ToolSelectionResult, createMemoryContextSystemMessage, decryptData, decryptDataBytes, encryptData, executeTool, extractConversationContext, formatMemoriesForChat, selectTool, useChat, useEncryption, useImageGeneration, useMemory, useModels, useSearch };
+export { type ClientTool, DEFAULT_TOOL_SELECTOR_MODEL, type PdfFile, type ToolExecutionResult, type ToolParameter, type ToolSelectionResult, createMemoryContextSystemMessage, decryptData, decryptDataBytes, encryptData, executeTool, extractConversationContext, formatMemoriesForChat, selectTool, useChat, useEncryption, useImageGeneration, useMemory, useModels, usePdf, useSearch };

package/dist/react/index.mjs

@@ -830,7 +830,7 @@ async function getTextGenerationPipeline(options) {
   if (sharedPipeline && currentModel === model && currentDevice === device) {
     return sharedPipeline;
   }
-  const { pipeline, env } = await import("./transformers.node-BSHUG7OY.mjs");
+  const { pipeline, env } = await import("./transformers.node-LUTOZWVQ.mjs");
   env.allowLocalModels = false;
   if (env.backends?.onnx) {
     env.backends.onnx.logLevel = "fatal";
@@ -856,7 +856,7 @@ async function generateLocalChatCompletion(messages, options = {}) {
     onToken,
     signal
   } = options;
-  const { TextStreamer } = await import("./transformers.node-BSHUG7OY.mjs");
+  const { TextStreamer } = await import("./transformers.node-LUTOZWVQ.mjs");
   const chatPipeline = await getTextGenerationPipeline({
     model,
     device: "wasm",
@@ -1816,7 +1816,7 @@ var generateEmbeddingForText = async (text, options = {}) => {
   }
   try {
     if (!embeddingPipeline) {
-      const { pipeline } = await import("./transformers.node-BSHUG7OY.mjs");
+      const { pipeline } = await import("./transformers.node-LUTOZWVQ.mjs");
       embeddingPipeline = await pipeline("feature-extraction", model);
     }
     const output = await embeddingPipeline(text, {
@@ -2144,13 +2144,90 @@ function useMemory(options = {}) {
   };
 }
 
+// src/react/usePdf.ts
+import { useCallback as useCallback3, useState as useState2 } from "react";
+
+// src/lib/pdf.ts
+import * as pdfjs from "pdfjs-dist";
+pdfjs.GlobalWorkerOptions.workerSrc = `https://unpkg.com/pdfjs-dist@${pdfjs.version}/build/pdf.worker.min.mjs`;
+async function extractTextFromPdf(pdfDataUrl) {
+  try {
+    const loadingTask = pdfjs.getDocument(pdfDataUrl);
+    const pdf = await loadingTask.promise;
+    const textParts = [];
+    for (let i = 1; i <= pdf.numPages; i++) {
+      const page = await pdf.getPage(i);
+      const textContent = await page.getTextContent();
+      const pageText = textContent.items.map((item) => item.str).join(" ");
+      if (pageText.trim()) {
+        textParts.push(pageText);
+      }
+    }
+    return textParts.join("\n\n");
+  } catch (error) {
+    console.error("Error extracting text from PDF:", error);
+    throw error;
+  }
+}
+
+// src/react/usePdf.ts
+var PDF_MIME_TYPE = "application/pdf";
+function usePdf() {
+  const [isProcessing, setIsProcessing] = useState2(false);
+  const [error, setError] = useState2(null);
+  const extractPdfContext = useCallback3(
+    async (files) => {
+      setIsProcessing(true);
+      setError(null);
+      try {
+        const pdfFiles = files.filter(
+          (file) => file.mediaType === PDF_MIME_TYPE || file.filename?.toLowerCase().endsWith(".pdf")
+        );
+        if (pdfFiles.length === 0) {
+          return null;
+        }
+        const contexts = await Promise.all(
+          pdfFiles.map(async (file) => {
+            try {
+              const text = await extractTextFromPdf(file.url);
+              if (!text.trim()) {
+                console.warn(`No text found in PDF ${file.filename}`);
+                return null;
+              }
+              return `[Context from PDF attachment ${file.filename}]:
+${text}`;
+            } catch (err) {
+              console.error(`Failed to process PDF ${file.filename}:`, err);
+              return null;
+            }
+          })
+        );
+        const mergedContext = contexts.filter(Boolean).join("\n\n");
+        return mergedContext || null;
+      } catch (err) {
+        const processedError = err instanceof Error ? err : new Error(String(err));
+        setError(processedError);
+        throw processedError;
+      } finally {
+        setIsProcessing(false);
+      }
+    },
+    []
+  );
+  return {
+    extractPdfContext,
+    isProcessing,
+    error
+  };
+}
+
 // src/react/useModels.ts
-import { useCallback as useCallback3, useEffect as useEffect3, useRef as useRef4, useState as useState2 } from "react";
+import { useCallback as useCallback4, useEffect as useEffect3, useRef as useRef4, useState as useState3 } from "react";
 function useModels(options = {}) {
   const { getToken, baseUrl = BASE_URL, provider, autoFetch = true } = options;
-  const [models, setModels] = useState2([]);
-  const [isLoading, setIsLoading] = useState2(false);
-  const [error, setError] = useState2(null);
+  const [models, setModels] = useState3([]);
+  const [isLoading, setIsLoading] = useState3(false);
+  const [error, setError] = useState3(null);
   const getTokenRef = useRef4(getToken);
   const baseUrlRef = useRef4(baseUrl);
   const providerRef = useRef4(provider);
@@ -2168,7 +2245,7 @@ function useModels(options = {}) {
       }
     };
   }, []);
-  const fetchModels = useCallback3(async () => {
+  const fetchModels = useCallback4(async () => {
     if (abortControllerRef.current) {
       abortControllerRef.current.abort();
     }
@@ -2226,7 +2303,7 @@ function useModels(options = {}) {
       }
     }
   }, []);
-  const refetch = useCallback3(async () => {
+  const refetch = useCallback4(async () => {
     setModels([]);
     await fetchModels();
   }, [fetchModels]);
@@ -2249,13 +2326,13 @@ function useModels(options = {}) {
 }
 
 // src/react/useSearch.ts
-import { useCallback as useCallback4, useEffect as useEffect4, useRef as useRef5, useState as useState3 } from "react";
+import { useCallback as useCallback5, useEffect as useEffect4, useRef as useRef5, useState as useState4 } from "react";
 function useSearch(options = {}) {
   const { getToken, baseUrl = BASE_URL, onError } = options;
-  const [isLoading, setIsLoading] = useState3(false);
-  const [results, setResults] = useState3(null);
-  const [response, setResponse] = useState3(null);
-  const [error, setError] = useState3(null);
+  const [isLoading, setIsLoading] = useState4(false);
+  const [results, setResults] = useState4(null);
+  const [response, setResponse] = useState4(null);
+  const [error, setError] = useState4(null);
   const abortControllerRef = useRef5(null);
   useEffect4(() => {
     return () => {
@@ -2265,7 +2342,7 @@ function useSearch(options = {}) {
       }
     };
   }, []);
-  const search = useCallback4(
+  const search = useCallback5(
     async (query, searchOptions = {}) => {
       if (abortControllerRef.current) {
         abortControllerRef.current.abort();
@@ -2333,10 +2410,10 @@ function useSearch(options = {}) {
 }
 
 // src/react/useImageGeneration.ts
-import { useCallback as useCallback5, useEffect as useEffect5, useRef as useRef6, useState as useState4 } from "react";
+import { useCallback as useCallback6, useEffect as useEffect5, useRef as useRef6, useState as useState5 } from "react";
 function useImageGeneration(options = {}) {
   const { getToken, baseUrl = BASE_URL, onFinish, onError } = options;
-  const [isLoading, setIsLoading] = useState4(false);
+  const [isLoading, setIsLoading] = useState5(false);
   const abortControllerRef = useRef6(null);
   useEffect5(() => {
     return () => {
@@ -2346,13 +2423,13 @@ function useImageGeneration(options = {}) {
       }
     };
   }, []);
-  const stop = useCallback5(() => {
+  const stop = useCallback6(() => {
     if (abortControllerRef.current) {
       abortControllerRef.current.abort();
       abortControllerRef.current = null;
     }
   }, []);
-  const generateImage = useCallback5(
+  const generateImage = useCallback6(
     async (args) => {
       if (abortControllerRef.current) {
         abortControllerRef.current.abort();
@@ -2483,5 +2560,6 @@ export {
   useImageGeneration,
   useMemory,
   useModels,
+  usePdf,
   useSearch
 };