@delofarag/ai-utils 1.3.4 → 1.4.0

package/dist/rag.js

@@ -1,8 +1,9 @@
 import { SupabaseVectorStore, OllamaEmbeddings, RecursiveCharacterTextSplitter, FaissStore, Document, createStuffDocumentsChain, createRetrievalChain } from "./imports";
-const baseEmbeddings = new OllamaEmbeddings({
+import { createClient } from "@supabase/supabase-js";
+export const baseEmbeddings = new OllamaEmbeddings({
     model: "nomic-embed-text"
 });
-const baseSplitter = new RecursiveCharacterTextSplitter({
+export const baseSplitter = new RecursiveCharacterTextSplitter({
     chunkSize: 500,
     chunkOverlap: 50
 });
@@ -12,7 +13,7 @@ export function turn_to_docs(docs) {
         metadata: {}
     }));
 }
-export async function createSupabaseVectoreStore({ supabase, data, table_name = "documents", RPC_function = "match_documents" }) {
+export async function createSupabaseVectoreStore(data, { supabase = createClient(process.env.NEXT_PUBLIC_SUPABASE_URL, process.env.NEXT_PUBLIC_SUPABASE_ANON_KEY || process.env.SUPABASE_SERVICE_ROLE_KEY), table_name = "documents", RPC_function = "match_documents", } = {}) {
     const docs = turn_to_docs(data);
     const splitted_docs = await baseSplitter.splitDocuments(docs);
     return await SupabaseVectorStore.fromDocuments(splitted_docs, baseEmbeddings, {
@@ -22,14 +23,14 @@ export async function createSupabaseVectoreStore({ supabase, data, table_name =
     });
 }
 // Bestehenden Supabase Store holen (ohne neue Docs)
-export function getSupabaseVectorStore({ supabase, table_name = "documents", RPC_function = "match_documents" }) {
+export function getSupabaseVectorStore({ supabase = createClient(process.env.NEXT_PUBLIC_SUPABASE_URL, process.env.NEXT_PUBLIC_SUPABASE_ANON_KEY || process.env.SUPABASE_SERVICE_ROLE_KEY), table_name = "documents", RPC_function = "match_documents" } = {}) {
     return new SupabaseVectorStore(baseEmbeddings, {
         client: supabase,
         tableName: table_name,
         queryName: RPC_function
     });
 }
-export async function createFaissStore({ data, save_path = "faiss_rag", embeddings = baseEmbeddings }) {
+export async function createFaissStore(data, { save_path = "faiss_rag", embeddings = baseEmbeddings } = {}) {
     const docs = turn_to_docs(data);
     const splitted_docs = await baseSplitter.splitDocuments(docs);
     const vectore_store = await FaissStore.fromDocuments(splitted_docs, embeddings);
@@ -56,17 +57,4 @@ export async function createRAGChain({ vectorStore, llm, prompt, num_of_results_
     });
     return retrievalChain;
 }
-// Erstellt eine Retrieval Chain direkt aus einem Retriever
-export async function createRAGChainFromRetriever({ retriever, llm, prompt }) {
-    const model = llm;
-    const documentChain = await createStuffDocumentsChain({
-        llm: model,
-        prompt: prompt
-    });
-    const retrievalChain = await createRetrievalChain({
-        retriever: retriever,
-        combineDocsChain: documentChain
-    });
-    return retrievalChain;
-}
 //# sourceMappingURL=rag.js.map

package/dist/rag.js.map

@@ -1 +1 @@
-{"version":3,"file":"rag.js","sourceRoot":"","sources":["../src/rag.ts"],"names":[],"mappings":"AAAA,OAAO,EACH,mBAAmB,EACnB,gBAAgB,EAChB,8BAA8B,EAC9B,UAAU,EACV,QAAQ,EAER,yBAAyB,EACzB,oBAAoB,EAKvB,MAAM,WAAW,CAAA;AAUlB,MAAM,cAAc,GAAG,IAAI,gBAAgB,CAAC;IACxC,KAAK,EAAE,kBAAkB;CAC5B,CAAC,CAAA;AAEF,MAAM,YAAY,GAAG,IAAI,8BAA8B,CAAC;IACpD,SAAS,EAAE,GAAG;IACd,YAAY,EAAE,EAAE;CACnB,CAAC,CAAA;AAEF,MAAM,UAAU,YAAY,CAAI,IAAS;IACrC,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,QAAQ,CAAC;QAChC,WAAW,EAAE,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,EAAC,IAAI,EAAC,CAAC,CAAC;QACvE,QAAQ,EAAE,EAAE;KACf,CAAC,CAAC,CAAA;AACP,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAAC,EAAC,QAAQ,EAAE,IAAI,EAAE,UAAU,GAAG,WAAW,EAAE,YAAY,GAAG,iBAAiB,EAAmD;IAC3K,MAAM,IAAI,GAAG,YAAY,CAAC,IAAI,CAAC,CAAA;IAC/B,MAAM,aAAa,GAAG,MAAM,YAAY,CAAC,cAAc,CAAC,IAAI,CAAC,CAAA;IAC7D,OAAO,MAAM,mBAAmB,CAAC,aAAa,CAC1C,aAAa,EACb,cAAc,EACd;QACI,MAAM,EAAE,QAAQ;QAChB,SAAS,EAAE,UAAU;QACrB,SAAS,EAAE,YAAY;KAC1B,CACJ,CAAA;AAEL,CAAC;AAED,oDAAoD;AACpD,MAAM,UAAU,sBAAsB,CAAC,EAAC,QAAQ,EAAE,UAAU,GAAG,WAAW,EAAE,YAAY,GAAG,iBAAiB,EAAyE;IACjL,OAAO,IAAI,mBAAmB,CAAC,cAAc,EAAE;QAC3C,MAAM,EAAE,QAAQ;QAChB,SAAS,EAAE,UAAU;QACrB,SAAS,EAAE,YAAY;KAC1B,CAAC,CAAA;AACN,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,EAAC,IAAI,EAAE,SAAS,GAAG,WAAW,EAAE,UAAU,GAAG,cAAc,EAAgE;IAC9J,MAAM,IAAI,GAAG,YAAY,CAAC,IAAI,CAAC,CAAA;IAC/B,MAAM,aAAa,GAAG,MAAM,YAAY,CAAC,cAAc,CAAC,IAAI,CAAC,CAAA;IAC7D,MAAM,aAAa,GAAG,MAAM,UAAU,CAAC,aAAa,CAChD,aAAa,EACb,UAAU,CACb,CAAA;IACD,IAAI,SAAS,EAAE,CAAC;QACZ,MAAM,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;IACvC,CAAC;IACD,OAAO,aAAa,CAAA;AACxB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,EAAC,IAAI,EAAE,UAAU,GAAG,cAAc,EAA0C;IAC7G,OAAO,MAAM,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,UAAU,CAAC,CAAA;AAClD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,EACjC,WAAW,EACX,GAAG,EACH,MAAM,EACN,uBAAuB,GAAG,CAAC,EAM9B;IACG,MAAM,KAAK,GAAG,GAAG,CAAA;IACjB,MAAM,SAAS,GAAG,WAAW,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,uBAAuB,EAAE,CAAC,CAAA;IAEzE,gDAAgD;IAChD,MAAM,aAAa,GAAG,MAAM,yBAAyB,CAAC;QAClD,GAAG,EAAE,KAAY;QACjB,MAAM,EAAE,MAAa;KACxB,CAAC,CAAA;IAEF,0DAA0D;IAC1D,MAAM,cAAc,GAAG,MAAM,oBAAoB,CAAC;QAC9C,SAAS,EAAE,SAAgB;QAC3B,gBAAgB,EAAE,aAAa;KAClC,CAAC,CAAA;IAEF,OAAO,cAAc,CAAA;AACzB,CAAC;AAED,2DAA2D;AAC3D,MAAM,CAAC,KAAK,UAAU,2BAA2B,CAAC,EAC9C,SAAS,EACT,GAAG,EACH,MAAM,EAKT;IACG,MAAM,KAAK,GAAG,GAAG,CAAA;IAEjB,MAAM,aAAa,GAAG,MAAM,yBAAyB,CAAC;QAClD,GAAG,EAAE,KAAY;QACjB,MAAM,EAAE,MAAa;KACxB,CAAC,CAAA;IAEF,MAAM,cAAc,GAAG,MAAM,oBAAoB,CAAC;QAC9C,SAAS,EAAE,SAAgB;QAC3B,gBAAgB,EAAE,aAAa;KAClC,CAAC,CAAA;IAEF,OAAO,cAAc,CAAA;AACzB,CAAC"}
+{"version":3,"file":"rag.js","sourceRoot":"","sources":["../src/rag.ts"],"names":[],"mappings":"AAAA,OAAO,EACH,mBAAmB,EACnB,gBAAgB,EAChB,8BAA8B,EAC9B,UAAU,EACV,QAAQ,EAER,yBAAyB,EACzB,oBAAoB,EAKvB,MAAM,WAAW,CAAA;AAElB,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAA;AAQpD,MAAM,CAAC,MAAM,cAAc,GAAG,IAAI,gBAAgB,CAAC;IAC/C,KAAK,EAAE,kBAAkB;CAC5B,CAAC,CAAA;AAEF,MAAM,CAAC,MAAM,YAAY,GAAG,IAAI,8BAA8B,CAAC;IAC3D,SAAS,EAAE,GAAG;IACd,YAAY,EAAE,EAAE;CACnB,CAAC,CAAA;AAEF,MAAM,UAAU,YAAY,CAAI,IAAS;IACrC,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,QAAQ,CAAC;QAChC,WAAW,EAAE,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,EAAC,IAAI,EAAC,CAAC,CAAC;QACvE,QAAQ,EAAE,EAAE;KACf,CAAC,CAAC,CAAA;AACP,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC5C,IAAa,EACb,EACI,QAAQ,GAAG,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,wBAAyB,EAAE,OAAO,CAAC,GAAG,CAAC,6BAA6B,IAAI,OAAO,CAAC,GAAG,CAAC,yBAA0B,CAAC,EACnJ,UAAU,GAAG,WAAW,EACxB,YAAY,GAAG,iBAAiB,MACZ,EAAE;IAC1B,MAAM,IAAI,GAAG,YAAY,CAAC,IAAI,CAAC,CAAA;IAC/B,MAAM,aAAa,GAAG,MAAM,YAAY,CAAC,cAAc,CAAC,IAAI,CAAC,CAAA;IAC7D,OAAO,MAAM,mBAAmB,CAAC,aAAa,CAC1C,aAAa,EACb,cAAc,EACd;QACI,MAAM,EAAE,QAAQ;QAChB,SAAS,EAAE,UAAU;QACrB,SAAS,EAAE,YAAY;KAC1B,CACJ,CAAA;AACL,CAAC;AAED,oDAAoD;AACpD,MAAM,UAAU,sBAAsB,CAAC,EACnC,QAAQ,GAAG,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,wBAAyB,EAAE,OAAO,CAAC,GAAG,CAAC,6BAA6B,IAAI,OAAO,CAAC,GAAG,CAAC,yBAA0B,CAAC,EACnJ,UAAU,GAAG,WAAW,EACxB,YAAY,GAAG,iBAAiB,KACX,EAAE;IACvB,OAAO,IAAI,mBAAmB,CAAC,cAAc,EAAE;QAC3C,MAAM,EAAE,QAAQ;QAChB,SAAS,EAAE,UAAU;QACrB,SAAS,EAAE,YAAY;KAC1B,CAAC,CAAA;AACN,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAClC,IAAa,EACb,EACI,SAAS,GAAG,WAAW,EACvB,UAAU,GAAG,cAAc,KACqB,EAAE;IACtD,MAAM,IAAI,GAAG,YAAY,CAAC,IAAI,CAAC,CAAA;IAC/B,MAAM,aAAa,GAAG,MAAM,YAAY,CAAC,cAAc,CAAC,IAAI,CAAC,CAAA;IAC7D,MAAM,aAAa,GAAG,MAAM,UAAU,CAAC,aAAa,CAChD,aAAa,EACb,UAAU,CACb,CAAA;IACD,IAAI,SAAS,EAAE,CAAC;QACZ,MAAM,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;IACvC,CAAC;IACD,OAAO,aAAa,CAAA;AACxB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,EAAC,IAAI,EAAE,UAAU,GAAG,cAAc,EAA0C;IAC7G,OAAO,MAAM,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,UAAU,CAAC,CAAA;AAClD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,EACjC,WAAW,EACX,GAAG,EACH,MAAM,EACN,uBAAuB,GAAG,CAAC,EAM9B;IACG,MAAM,KAAK,GAAG,GAAG,CAAA;IACjB,MAAM,SAAS,GAAG,WAAW,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,uBAAuB,EAAE,CAAC,CAAA;IAEzE,gDAAgD;IAChD,MAAM,aAAa,GAAG,MAAM,yBAAyB,CAAC;QAClD,GAAG,EAAE,KAAY;QACjB,MAAM,EAAE,MAAa;KACxB,CAAC,CAAA;IAEF,0DAA0D;IAC1D,MAAM,cAAc,GAAG,MAAM,oBAAoB,CAAC;QAC9C,SAAS,EAAE,SAAgB;QAC3B,gBAAgB,EAAE,aAAa;KAClC,CAAC,CAAA;IAEF,OAAO,cAAc,CAAA;AACzB,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@delofarag/ai-utils",
-  "version": "1.3.4",
+  "version": "1.4.0",
   "type": "module",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",