@retrivora-ai/rag-engine 1.9.0 → 1.9.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (47) hide show
  1. package/dist/{ILLMProvider-BOJFz3Na.d.mts → ILLMProvider-Bw2A28nU.d.mts} +12 -0
  2. package/dist/{ILLMProvider-BOJFz3Na.d.ts → ILLMProvider-Bw2A28nU.d.ts} +12 -0
  3. package/dist/handlers/index.d.mts +2 -2
  4. package/dist/handlers/index.d.ts +2 -2
  5. package/dist/handlers/index.js +1874 -542
  6. package/dist/handlers/index.mjs +1873 -541
  7. package/dist/{index-D3V9Et2M.d.mts → index-B70ZLkfG.d.mts} +1 -1
  8. package/dist/{index-BwpcaziY.d.ts → index-DVu-mkAM.d.ts} +1 -1
  9. package/dist/index.css +83 -0
  10. package/dist/index.d.mts +2 -2
  11. package/dist/index.d.ts +2 -2
  12. package/dist/index.js +330 -106
  13. package/dist/index.mjs +333 -107
  14. package/dist/server.d.mts +32 -5
  15. package/dist/server.d.ts +32 -5
  16. package/dist/server.js +1871 -736
  17. package/dist/server.mjs +1870 -735
  18. package/package.json +1 -1
  19. package/src/components/ChatWindow.tsx +24 -14
  20. package/src/components/MarkdownComponents.tsx +3 -3
  21. package/src/components/MessageBubble.tsx +89 -7
  22. package/src/components/ProductCard.tsx +29 -2
  23. package/src/components/UIDispatcher.tsx +1 -0
  24. package/src/components/VisualizationRenderer.tsx +143 -11
  25. package/src/config/EmbeddingStrategy.ts +5 -4
  26. package/src/config/RagConfig.ts +10 -0
  27. package/src/config/serverConfig.ts +16 -1
  28. package/src/core/LLMRouter.ts +79 -0
  29. package/src/core/Pipeline.ts +295 -51
  30. package/src/core/ProviderRegistry.ts +6 -0
  31. package/src/core/QueryProcessor.ts +108 -9
  32. package/src/handlers/index.ts +37 -11
  33. package/src/hooks/useRagChat.ts +77 -17
  34. package/src/llm/providers/UniversalLLMAdapter.ts +110 -13
  35. package/src/providers/vectordb/ChromaDBProvider.ts +13 -2
  36. package/src/providers/vectordb/MilvusProvider.ts +18 -2
  37. package/src/providers/vectordb/MultiTablePostgresProvider.ts +48 -16
  38. package/src/providers/vectordb/PostgreSQLProvider.ts +1 -1
  39. package/src/providers/vectordb/QdrantProvider.ts +1 -1
  40. package/src/providers/vectordb/RedisProvider.ts +3 -4
  41. package/src/providers/vectordb/WeaviateProvider.ts +41 -3
  42. package/src/types/chat.ts +2 -0
  43. package/src/types/index.ts +26 -0
  44. package/src/utils/ProductExtractor.ts +5 -3
  45. package/src/utils/SchemaMapper.ts +6 -4
  46. package/src/utils/UITransformer.ts +1350 -490
  47. package/src/utils/synonyms.ts +6 -4
@@ -52,6 +52,8 @@ interface UseRagChatReturn {
52
52
  retry: () => Promise<void>;
53
53
  /** Programmatically set the conversation (e.g. to restore from a DB) */
54
54
  setMessages: React.Dispatch<React.SetStateAction<RagMessage[]>>;
55
+ /** Abort the current active stream request */
56
+ stop: () => void;
55
57
  }
56
58
 
57
59
  /**
@@ -214,6 +216,16 @@ interface RAGConfig {
214
216
  * e.g., { "name": "ProductName", "price": "SalesPrice", "image": "ThumbnailUrl" }
215
217
  */
216
218
  uiMapping?: Record<string, string>;
219
+ /**
220
+ * Custom keywords that signal graph-based retrieval should be used.
221
+ * Defaults to built-in list (relationship, connect, hierarchy, etc.).
222
+ */
223
+ graphKeywords?: string[];
224
+ /**
225
+ * Custom keywords that signal vector-based retrieval should be used.
226
+ * Defaults to built-in list (find, search, tell me about, etc.).
227
+ */
228
+ vectorKeywords?: string[];
217
229
  }
218
230
  interface RagConfig {
219
231
  /**
@@ -52,6 +52,8 @@ interface UseRagChatReturn {
52
52
  retry: () => Promise<void>;
53
53
  /** Programmatically set the conversation (e.g. to restore from a DB) */
54
54
  setMessages: React.Dispatch<React.SetStateAction<RagMessage[]>>;
55
+ /** Abort the current active stream request */
56
+ stop: () => void;
55
57
  }
56
58
 
57
59
  /**
@@ -214,6 +216,16 @@ interface RAGConfig {
214
216
  * e.g., { "name": "ProductName", "price": "SalesPrice", "image": "ThumbnailUrl" }
215
217
  */
216
218
  uiMapping?: Record<string, string>;
219
+ /**
220
+ * Custom keywords that signal graph-based retrieval should be used.
221
+ * Defaults to built-in list (relationship, connect, hierarchy, etc.).
222
+ */
223
+ graphKeywords?: string[];
224
+ /**
225
+ * Custom keywords that signal vector-based retrieval should be used.
226
+ * Defaults to built-in list (find, search, tell me about, etc.).
227
+ */
228
+ vectorKeywords?: string[];
217
229
  }
218
230
  interface RagConfig {
219
231
  /**
@@ -1,3 +1,3 @@
1
- import '../ILLMProvider-BOJFz3Na.mjs';
2
- export { c as createChatHandler, d as createHealthHandler, e as createIngestHandler, f as createStreamHandler, k as createSuggestionsHandler, g as createUploadHandler, s as sseErrorFrame, h as sseFrame, i as sseMetaFrame, l as sseObservabilityFrame, j as sseTextFrame, m as sseUIFrame } from '../index-D3V9Et2M.mjs';
1
+ import '../ILLMProvider-Bw2A28nU.mjs';
2
+ export { c as createChatHandler, d as createHealthHandler, e as createIngestHandler, f as createStreamHandler, k as createSuggestionsHandler, g as createUploadHandler, s as sseErrorFrame, h as sseFrame, i as sseMetaFrame, l as sseObservabilityFrame, j as sseTextFrame, m as sseUIFrame } from '../index-B70ZLkfG.mjs';
3
3
  import 'next/server';
@@ -1,3 +1,3 @@
1
- import '../ILLMProvider-BOJFz3Na.js';
2
- export { c as createChatHandler, d as createHealthHandler, e as createIngestHandler, f as createStreamHandler, k as createSuggestionsHandler, g as createUploadHandler, s as sseErrorFrame, h as sseFrame, i as sseMetaFrame, l as sseObservabilityFrame, j as sseTextFrame, m as sseUIFrame } from '../index-BwpcaziY.js';
1
+ import '../ILLMProvider-Bw2A28nU.js';
2
+ export { c as createChatHandler, d as createHealthHandler, e as createIngestHandler, f as createStreamHandler, k as createSuggestionsHandler, g as createUploadHandler, s as sseErrorFrame, h as sseFrame, i as sseMetaFrame, l as sseObservabilityFrame, j as sseTextFrame, m as sseUIFrame } from '../index-DVu-mkAM.js';
3
3
  import 'next/server';