@retrivora-ai/rag-engine 1.8.9 → 1.9.1

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 (44) hide show
  1. package/dist/{ILLMProvider-CbUtvWAW.d.mts → ILLMProvider-BQyKZLnd.d.mts} +10 -2
  2. package/dist/{ILLMProvider-CbUtvWAW.d.ts → ILLMProvider-BQyKZLnd.d.ts} +10 -2
  3. package/dist/handlers/index.d.mts +2 -2
  4. package/dist/handlers/index.d.ts +2 -2
  5. package/dist/handlers/index.js +1726 -522
  6. package/dist/handlers/index.mjs +1725 -521
  7. package/dist/{index-DgzcnqZs.d.ts → index-A0GqPsaG.d.ts} +1 -1
  8. package/dist/{index-c_y_5qdw.d.mts → index-CDftK3qs.d.mts} +1 -1
  9. package/dist/index.css +26 -0
  10. package/dist/index.d.mts +2 -2
  11. package/dist/index.d.ts +2 -2
  12. package/dist/index.js +246 -76
  13. package/dist/index.mjs +248 -76
  14. package/dist/server.d.mts +32 -5
  15. package/dist/server.d.ts +32 -5
  16. package/dist/server.js +1735 -733
  17. package/dist/server.mjs +1733 -731
  18. package/package.json +1 -1
  19. package/src/components/MarkdownComponents.tsx +3 -3
  20. package/src/components/MessageBubble.tsx +49 -2
  21. package/src/components/ProductCard.tsx +33 -6
  22. package/src/components/UIDispatcher.tsx +1 -0
  23. package/src/components/VisualizationRenderer.tsx +143 -11
  24. package/src/config/EmbeddingStrategy.ts +5 -4
  25. package/src/config/RagConfig.ts +10 -2
  26. package/src/config/serverConfig.ts +16 -1
  27. package/src/core/LLMRouter.ts +79 -0
  28. package/src/core/Pipeline.ts +262 -45
  29. package/src/core/ProviderRegistry.ts +6 -0
  30. package/src/core/QueryProcessor.ts +98 -0
  31. package/src/handlers/index.ts +8 -5
  32. package/src/hooks/useRagChat.ts +53 -17
  33. package/src/llm/providers/UniversalLLMAdapter.ts +110 -13
  34. package/src/providers/vectordb/ChromaDBProvider.ts +13 -6
  35. package/src/providers/vectordb/MilvusProvider.ts +18 -2
  36. package/src/providers/vectordb/MultiTablePostgresProvider.ts +16 -29
  37. package/src/providers/vectordb/PostgreSQLProvider.ts +4 -15
  38. package/src/providers/vectordb/QdrantProvider.ts +2 -5
  39. package/src/providers/vectordb/RedisProvider.ts +3 -4
  40. package/src/providers/vectordb/WeaviateProvider.ts +41 -3
  41. package/src/types/index.ts +26 -0
  42. package/src/utils/ProductExtractor.ts +5 -3
  43. package/src/utils/UITransformer.ts +1345 -534
  44. package/src/utils/synonyms.ts +2 -0
@@ -99,8 +99,6 @@ interface VectorDBConfig {
99
99
  * - searchFields?: string[] | string // optional override for which columns receive exact-match boosts
100
100
  */
101
101
  options: Record<string, unknown>;
102
- /** Optional distance metric (defaults to cosine if not specified and supported by provider) */
103
- distanceMetric?: 'cosine' | 'euclidean' | 'dotproduct';
104
102
  }
105
103
  type GraphDBProvider = typeof GRAPH_DB_PROVIDERS[number];
106
104
  interface GraphDBConfig {
@@ -216,6 +214,16 @@ interface RAGConfig {
216
214
  * e.g., { "name": "ProductName", "price": "SalesPrice", "image": "ThumbnailUrl" }
217
215
  */
218
216
  uiMapping?: Record<string, string>;
217
+ /**
218
+ * Custom keywords that signal graph-based retrieval should be used.
219
+ * Defaults to built-in list (relationship, connect, hierarchy, etc.).
220
+ */
221
+ graphKeywords?: string[];
222
+ /**
223
+ * Custom keywords that signal vector-based retrieval should be used.
224
+ * Defaults to built-in list (find, search, tell me about, etc.).
225
+ */
226
+ vectorKeywords?: string[];
219
227
  }
220
228
  interface RagConfig {
221
229
  /**
@@ -99,8 +99,6 @@ interface VectorDBConfig {
99
99
  * - searchFields?: string[] | string // optional override for which columns receive exact-match boosts
100
100
  */
101
101
  options: Record<string, unknown>;
102
- /** Optional distance metric (defaults to cosine if not specified and supported by provider) */
103
- distanceMetric?: 'cosine' | 'euclidean' | 'dotproduct';
104
102
  }
105
103
  type GraphDBProvider = typeof GRAPH_DB_PROVIDERS[number];
106
104
  interface GraphDBConfig {
@@ -216,6 +214,16 @@ interface RAGConfig {
216
214
  * e.g., { "name": "ProductName", "price": "SalesPrice", "image": "ThumbnailUrl" }
217
215
  */
218
216
  uiMapping?: Record<string, string>;
217
+ /**
218
+ * Custom keywords that signal graph-based retrieval should be used.
219
+ * Defaults to built-in list (relationship, connect, hierarchy, etc.).
220
+ */
221
+ graphKeywords?: string[];
222
+ /**
223
+ * Custom keywords that signal vector-based retrieval should be used.
224
+ * Defaults to built-in list (find, search, tell me about, etc.).
225
+ */
226
+ vectorKeywords?: string[];
219
227
  }
220
228
  interface RagConfig {
221
229
  /**
@@ -1,3 +1,3 @@
1
- import '../ILLMProvider-CbUtvWAW.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-c_y_5qdw.mjs';
1
+ import '../ILLMProvider-BQyKZLnd.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-CDftK3qs.mjs';
3
3
  import 'next/server';
@@ -1,3 +1,3 @@
1
- import '../ILLMProvider-CbUtvWAW.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-DgzcnqZs.js';
1
+ import '../ILLMProvider-BQyKZLnd.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-A0GqPsaG.js';
3
3
  import 'next/server';