langchain-rag-chat 0.1.1

package/dist/loaders/index.d.ts

@@ -0,0 +1,6 @@
+export { loadResources } from './load';
+export { loadPDF } from './pdfLoader';
+export { loadWebPage } from './webLoader';
+export type { Resource, LoadOptions, CustomLoder } from './load';
+export type { PDFLoader } from './pdfLoader';
+export type { WebURL } from './webLoader';

package/dist/loaders/load.d.ts

@@ -0,0 +1,31 @@
+import { Document } from '@langchain/core/documents';
+import { RecursiveCharacterTextSplitterParams } from '@langchain/classic/text_splitter';
+export interface Resource {
+    /**
+     * source assets path
+     */
+    url: string;
+    /**
+     * web site
+     */
+    selector?: string;
+    /**
+     * pdf
+     */
+    pageSplitter?: boolean;
+    text?: string;
+}
+export interface CustomLoder {
+    [K: string]: (resource: Resource) => Promise<Document<any>[]>;
+}
+export interface LoadOptions {
+    resource: Resource[];
+    /**
+     * concurrency
+     * @default 3
+     */
+    concurrencyCount?: number;
+    splitterOptions?: RecursiveCharacterTextSplitterParams;
+    loader?: CustomLoder | null;
+}
+export declare function loadResources(options: LoadOptions): Promise<Document<Record<string, any>>[][]>;

package/dist/loaders/pdfLoader.d.ts

@@ -0,0 +1,12 @@
+import { Document } from '@langchain/core/documents';
+import { LoadParameters } from 'pdf-parse';
+export interface PDFLoader extends LoadParameters {
+    /**
+     * web url
+     */
+    url: string;
+}
+export declare function loadPDF(options: PDFLoader): Promise<Document<{
+    source: string;
+    page: number;
+}>[]>;

package/dist/loaders/webLoader.d.ts

@@ -0,0 +1,16 @@
+import { Document } from '@langchain/core/documents';
+export interface WebURL {
+    /**
+     * web url
+     */
+    url: string;
+    /**
+     * css selector
+     * @default "body"
+     */
+    selector?: string;
+}
+export declare function loadWebPage(options: WebURL): Promise<Document<{
+    source: string;
+    selector: string;
+}>[]>;

package/dist/models/chatModel.d.ts

@@ -0,0 +1,2 @@
+import { ChatOpenAI, ChatOpenAIFields } from '@langchain/openai';
+export declare function createChatModel(options: ChatOpenAIFields): ChatOpenAI<import('@langchain/openai').ChatOpenAICallOptions>;

package/dist/models/embedModel.d.ts

@@ -0,0 +1,13 @@
+import { OpenAIEmbeddings, OpenAIEmbeddingsParams, OpenAIApiKey, ClientOptions } from '@langchain/openai';
+export interface EmbeddingModelOptions extends Partial<OpenAIEmbeddingsParams> {
+    verbose?: boolean;
+    /**
+     * The OpenAI API key to use.
+     * Alias for `apiKey`.
+     */
+    openAIApiKey?: OpenAIApiKey;
+    /** The OpenAI API key to use. */
+    apiKey?: OpenAIApiKey;
+    configuration?: ClientOptions;
+}
+export declare function createEmbeddingModel(options: EmbeddingModelOptions): OpenAIEmbeddings<number[]>;

package/dist/models/index.d.ts

@@ -0,0 +1,3 @@
+export { createChatModel } from './chatModel';
+export { createEmbeddingModel } from './embedModel';
+export type { EmbeddingModelOptions } from './embedModel';

package/dist/prompts/answerPrompt.d.ts

@@ -0,0 +1,6 @@
+import { ChatPromptTemplate } from '@langchain/core/prompts';
+export interface Answer {
+    question: string;
+    context: string;
+}
+export declare const answerPrompt: ChatPromptTemplate<import('@langchain/core/prompts').ParamsFromFString<"Please answer the user's question based on the content of the following document. If there is insufficient information in the document, do your best to answer the user's question within your capabilities, and then clearly indicate whether the answer is based on the content of the document.\nDocument:\n{context}\nUser question:\n{question}">, any>;

package/dist/prompts/classifyPrompt.d.ts

@@ -0,0 +1,7 @@
+import { ChatPromptTemplate } from '@langchain/core/prompts';
+export interface Classify {
+    category: string;
+    question: string;
+    content: string;
+}
+export declare const classifyPrompt: ChatPromptTemplate<import('@langchain/core/prompts').ParamsFromFString<"Classify user questions into the following categories: {category}.\n{content}\nSpecial note: Only output the classification names, do not output explanations.\nUser question: {question}">, any>;

package/dist/prompts/index.d.ts

@@ -0,0 +1,6 @@
+export { multiQueryPrompt } from './multiQueryPrompt';
+export { classifyPrompt } from './classifyPrompt';
+export { answerPrompt } from './answerPrompt';
+export type { MultiQuery } from './multiQueryPrompt';
+export type { Classify } from './classifyPrompt';
+export type { Answer } from './answerPrompt';

package/dist/prompts/multiQueryPrompt.d.ts

@@ -0,0 +1,7 @@
+import { ChatPromptTemplate } from '@langchain/core/prompts';
+export interface MultiQuery {
+    queryCount: number;
+    separator: string;
+    question: string;
+}
+export declare const multiQueryPrompt: ChatPromptTemplate<import('@langchain/core/prompts').ParamsFromFString<"You are an AI language model assistant. Your task is to generate {queryCount}\ndifferent versions of the given user question to retrieve relevant documents from a vector database.\nBy generating multiple perspectives on the user question, your goal is to help the user overcome some of\nthe limitations of the distance-based similarity search.\nProvide these alternative questions separated by '{separator}'.\nOriginal question: {question}">, any>;

package/dist/utils/concurrency.d.ts

@@ -0,0 +1,3 @@
+type Task<T> = (...args: any[]) => Promise<T>;
+export declare function concurrency<T = any>(tasks: Task<T>[], concurrency: number): Promise<T[]>;
+export {};

package/dist/utils/index.d.ts

@@ -0,0 +1 @@
+export { concurrency } from './concurrency';

package/package.json

@@ -0,0 +1,46 @@
+{
+  "name": "langchain-rag-chat",
+  "version": "0.1.1",
+  "description": "A RAG chat library built with LangChain",
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js",
+      "default": "./dist/index.js"
+    }
+  },
+  "files": [
+    "dist",
+    "README.md",
+    "LICENSE"
+  ],
+  "scripts": {
+    "dev": "vite",
+    "build": "tsc --noEmit && vite build",
+    "prepublishOnly": "npm run build",
+    "pack:check": "npm pack --dry-run"
+  },
+  "keywords": [
+    "rag",
+    "langchain",
+    "retrieval",
+    "ai"
+  ],
+  "license": "MIT",
+  "dependencies": {
+    "@langchain/classic": "^1.0.48",
+    "@langchain/core": "^1.2.11",
+    "@langchain/openai": "^1.5.13",
+    "cheerio": "^1.2.0",
+    "langchain": "^1.5.11",
+    "pdf-parse": "^2.4.5"
+  },
+  "devDependencies": {
+    "typescript": "~6.0.3",
+    "vite": "^8.2.2",
+    "vite-plugin-dts": "^5.1.0"
+  }
+}