@flutchai/flutch-sdk 0.1.0
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.
- package/README.md +287 -0
- package/dist/api/callback-token.guard.d.ts +17 -0
- package/dist/api/callback-token.guard.js +80 -0
- package/dist/api/callback-token.guard.js.map +1 -0
- package/dist/api/callback.controller.d.ts +7 -0
- package/dist/api/callback.controller.js +44 -0
- package/dist/api/callback.controller.js.map +1 -0
- package/dist/api/graph.controller.d.ts +32 -0
- package/dist/api/graph.controller.js +187 -0
- package/dist/api/graph.controller.js.map +1 -0
- package/dist/api/ui-dispatch.controller.d.ts +34 -0
- package/dist/api/ui-dispatch.controller.js +178 -0
- package/dist/api/ui-dispatch.controller.js.map +1 -0
- package/dist/bootstrap.d.ts +4 -0
- package/dist/bootstrap.js +177 -0
- package/dist/bootstrap.js.map +1 -0
- package/dist/callbacks/callback-acl.service.d.ts +24 -0
- package/dist/callbacks/callback-acl.service.js +144 -0
- package/dist/callbacks/callback-acl.service.js.map +1 -0
- package/dist/callbacks/callback-auditor.service.d.ts +47 -0
- package/dist/callbacks/callback-auditor.service.js +286 -0
- package/dist/callbacks/callback-auditor.service.js.map +1 -0
- package/dist/callbacks/callback-metrics.service.d.ts +34 -0
- package/dist/callbacks/callback-metrics.service.js +216 -0
- package/dist/callbacks/callback-metrics.service.js.map +1 -0
- package/dist/callbacks/callback-patch.service.d.ts +13 -0
- package/dist/callbacks/callback-patch.service.js +51 -0
- package/dist/callbacks/callback-patch.service.js.map +1 -0
- package/dist/callbacks/callback-rate-limiter.d.ts +27 -0
- package/dist/callbacks/callback-rate-limiter.js +129 -0
- package/dist/callbacks/callback-rate-limiter.js.map +1 -0
- package/dist/callbacks/callback-registry.d.ts +7 -0
- package/dist/callbacks/callback-registry.js +29 -0
- package/dist/callbacks/callback-registry.js.map +1 -0
- package/dist/callbacks/callback-store.d.ts +19 -0
- package/dist/callbacks/callback-store.js +145 -0
- package/dist/callbacks/callback-store.js.map +1 -0
- package/dist/callbacks/example.callback.d.ts +2 -0
- package/dist/callbacks/example.callback.js +10 -0
- package/dist/callbacks/example.callback.js.map +1 -0
- package/dist/callbacks/idempotency-manager.d.ts +54 -0
- package/dist/callbacks/idempotency-manager.js +230 -0
- package/dist/callbacks/idempotency-manager.js.map +1 -0
- package/dist/callbacks/index.d.ts +13 -0
- package/dist/callbacks/index.js +30 -0
- package/dist/callbacks/index.js.map +1 -0
- package/dist/callbacks/smart-callback.router.d.ts +52 -0
- package/dist/callbacks/smart-callback.router.js +213 -0
- package/dist/callbacks/smart-callback.router.js.map +1 -0
- package/dist/callbacks/telegram-patch.handler.d.ts +6 -0
- package/dist/callbacks/telegram-patch.handler.js +24 -0
- package/dist/callbacks/telegram-patch.handler.js.map +1 -0
- package/dist/callbacks/universal-callback.service.d.ts +14 -0
- package/dist/callbacks/universal-callback.service.js +20 -0
- package/dist/callbacks/universal-callback.service.js.map +1 -0
- package/dist/callbacks/web-patch.handler.d.ts +6 -0
- package/dist/callbacks/web-patch.handler.js +24 -0
- package/dist/callbacks/web-patch.handler.js.map +1 -0
- package/dist/core/abstract-graph.builder.d.ts +128 -0
- package/dist/core/abstract-graph.builder.js +488 -0
- package/dist/core/abstract-graph.builder.js.map +1 -0
- package/dist/core/builder-registry.service.d.ts +6 -0
- package/dist/core/builder-registry.service.js +29 -0
- package/dist/core/builder-registry.service.js.map +1 -0
- package/dist/core/universal-graph.module.d.ts +10 -0
- package/dist/core/universal-graph.module.js +294 -0
- package/dist/core/universal-graph.module.js.map +1 -0
- package/dist/decorators/callback.decorators.d.ts +21 -0
- package/dist/decorators/callback.decorators.js +73 -0
- package/dist/decorators/callback.decorators.js.map +1 -0
- package/dist/endpoint-registry/endpoint.decorators.d.ts +35 -0
- package/dist/endpoint-registry/endpoint.decorators.js +115 -0
- package/dist/endpoint-registry/endpoint.decorators.js.map +1 -0
- package/dist/endpoint-registry/endpoint.registry.d.ts +44 -0
- package/dist/endpoint-registry/endpoint.registry.js +90 -0
- package/dist/endpoint-registry/endpoint.registry.js.map +1 -0
- package/dist/endpoint-registry/index.d.ts +3 -0
- package/dist/endpoint-registry/index.js +20 -0
- package/dist/endpoint-registry/index.js.map +1 -0
- package/dist/endpoint-registry/ui-endpoints.discovery.d.ts +11 -0
- package/dist/endpoint-registry/ui-endpoints.discovery.js +72 -0
- package/dist/endpoint-registry/ui-endpoints.discovery.js.map +1 -0
- package/dist/engine/api-call-tracer.utils.d.ts +8 -0
- package/dist/engine/api-call-tracer.utils.js +184 -0
- package/dist/engine/api-call-tracer.utils.js.map +1 -0
- package/dist/engine/event-processor.utils.d.ts +49 -0
- package/dist/engine/event-processor.utils.js +314 -0
- package/dist/engine/event-processor.utils.js.map +1 -0
- package/dist/engine/graph-engine.factory.d.ts +12 -0
- package/dist/engine/graph-engine.factory.js +43 -0
- package/dist/engine/graph-engine.factory.js.map +1 -0
- package/dist/engine/langgraph-engine.d.ts +14 -0
- package/dist/engine/langgraph-engine.js +261 -0
- package/dist/engine/langgraph-engine.js.map +1 -0
- package/dist/index.d.ts +27 -0
- package/dist/index.js +48 -0
- package/dist/index.js.map +1 -0
- package/dist/interfaces/callback.interface.d.ts +46 -0
- package/dist/interfaces/callback.interface.js +3 -0
- package/dist/interfaces/callback.interface.js.map +1 -0
- package/dist/interfaces/graph-registry.interface.d.ts +8 -0
- package/dist/interfaces/graph-registry.interface.js +3 -0
- package/dist/interfaces/graph-registry.interface.js.map +1 -0
- package/dist/interfaces/graph-service.interface.d.ts +43 -0
- package/dist/interfaces/graph-service.interface.js +9 -0
- package/dist/interfaces/graph-service.interface.js.map +1 -0
- package/dist/interfaces/index.d.ts +3 -0
- package/dist/interfaces/index.js +20 -0
- package/dist/interfaces/index.js.map +1 -0
- package/dist/llm/index.d.ts +6 -0
- package/dist/llm/index.js +23 -0
- package/dist/llm/index.js.map +1 -0
- package/dist/llm/llm.initializer.d.ts +14 -0
- package/dist/llm/llm.initializer.js +167 -0
- package/dist/llm/llm.initializer.js.map +1 -0
- package/dist/llm/llm.interface.d.ts +6 -0
- package/dist/llm/llm.interface.js +3 -0
- package/dist/llm/llm.interface.js.map +1 -0
- package/dist/llm/llm.types.d.ts +32 -0
- package/dist/llm/llm.types.js +3 -0
- package/dist/llm/llm.types.js.map +1 -0
- package/dist/llm/model.initializer.d.ts +45 -0
- package/dist/llm/model.initializer.js +630 -0
- package/dist/llm/model.initializer.js.map +1 -0
- package/dist/llm/model.interface.d.ts +38 -0
- package/dist/llm/model.interface.js +4 -0
- package/dist/llm/model.interface.js.map +1 -0
- package/dist/llm/rerankers/voyageai-rerank.d.ts +17 -0
- package/dist/llm/rerankers/voyageai-rerank.js +56 -0
- package/dist/llm/rerankers/voyageai-rerank.js.map +1 -0
- package/dist/retriever/database-factory.d.ts +15 -0
- package/dist/retriever/database-factory.js +83 -0
- package/dist/retriever/database-factory.js.map +1 -0
- package/dist/retriever/index.d.ts +2 -0
- package/dist/retriever/index.js +19 -0
- package/dist/retriever/index.js.map +1 -0
- package/dist/retriever/retriever.client.d.ts +28 -0
- package/dist/retriever/retriever.client.js +97 -0
- package/dist/retriever/retriever.client.js.map +1 -0
- package/dist/retriever/retriever.service.d.ts +25 -0
- package/dist/retriever/retriever.service.js +265 -0
- package/dist/retriever/retriever.service.js.map +1 -0
- package/dist/retriever/types.d.ts +22 -0
- package/dist/retriever/types.js +3 -0
- package/dist/retriever/types.js.map +1 -0
- package/dist/schemas/graph-manifest.schema.d.ts +134 -0
- package/dist/schemas/graph-manifest.schema.js +180 -0
- package/dist/schemas/graph-manifest.schema.js.map +1 -0
- package/dist/schemas/index.d.ts +1 -0
- package/dist/schemas/index.js +18 -0
- package/dist/schemas/index.js.map +1 -0
- package/dist/shared-types/index.d.ts +199 -0
- package/dist/shared-types/index.js +58 -0
- package/dist/shared-types/index.js.map +1 -0
- package/dist/tools/index.d.ts +3 -0
- package/dist/tools/index.js +20 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/tools/mcp-converter.d.ts +15 -0
- package/dist/tools/mcp-converter.js +133 -0
- package/dist/tools/mcp-converter.js.map +1 -0
- package/dist/tools/mcp-tool-filter.d.ts +11 -0
- package/dist/tools/mcp-tool-filter.js +123 -0
- package/dist/tools/mcp-tool-filter.js.map +1 -0
- package/dist/tools/mcp.interfaces.d.ts +19 -0
- package/dist/tools/mcp.interfaces.js +3 -0
- package/dist/tools/mcp.interfaces.js.map +1 -0
- package/dist/types/graph-types.d.ts +82 -0
- package/dist/types/graph-types.js +6 -0
- package/dist/types/graph-types.js.map +1 -0
- package/dist/utils/error.utils.d.ts +18 -0
- package/dist/utils/error.utils.js +78 -0
- package/dist/utils/error.utils.js.map +1 -0
- package/dist/utils/graph-type.utils.d.ts +14 -0
- package/dist/utils/graph-type.utils.js +71 -0
- package/dist/utils/graph-type.utils.js.map +1 -0
- package/dist/utils/index.d.ts +1 -0
- package/dist/utils/index.js +18 -0
- package/dist/utils/index.js.map +1 -0
- package/dist/utils/usage-recorder.d.ts +23 -0
- package/dist/utils/usage-recorder.js +29 -0
- package/dist/utils/usage-recorder.js.map +1 -0
- package/dist/versioning/index.d.ts +2 -0
- package/dist/versioning/index.js +19 -0
- package/dist/versioning/index.js.map +1 -0
- package/dist/versioning/versioned-graph.service.d.ts +16 -0
- package/dist/versioning/versioned-graph.service.js +132 -0
- package/dist/versioning/versioned-graph.service.js.map +1 -0
- package/dist/versioning/versioning.types.d.ts +21 -0
- package/dist/versioning/versioning.types.js +3 -0
- package/dist/versioning/versioning.types.js.map +1 -0
- package/package.json +70 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model.initializer.js","sourceRoot":"","sources":["../../src/llm/model.initializer.ts"],"names":[],"mappings":";;;AAokCA,sDA+CC;AAhnCD,8CAA+C;AAC/C,0DAA0D;AAC1D,2CAAwC;AAYxC,SAAS,sBAAsB;IAC7B,MAAM,MAAM,GAAG,IAAI,eAAM,CAAC,wBAAwB,CAAC,CAAC;IAEpD,MAAM,CAAC,IAAI,CACT,mNAAmN,CACpN,CAAC;IAGF,MAAM,UAAU,GAAG;QACjB,mBAAU,CAAC,SAAgB;QAC3B,8BAAe,CAAC,SAAgB;KACjC,CAAC;IAEF,UAAU,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,KAAK,EAAE,EAAE;QACtC,MAAM,SAAS,GAAG,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,iBAAiB,CAAC;QACjE,MAAM,CAAC,IAAI,CAAC,YAAY,SAAS,oBAAoB,CAAC,CAAC;QAGvD,MAAM,wBAAwB,GAAG,SAAS,CAAC,gBAAgB,CAAC;QAE5D,IAAI,wBAAwB,EAAE,CAAC;YAC7B,SAAS,CAAC,gBAAgB,GAAG,UAAU,OAAY;gBACjD,MAAM,MAAM,GAAG,wBAAwB,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;gBAK5D,IACE,MAAM,CAAC,KAAK;oBACZ,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC;wBAC7B,kBAAkB,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EACxC,CAAC;oBAED,IAAI,MAAM,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;wBACpC,MAAM,CAAC,qBAAqB,GAAG,MAAM,CAAC,UAAU,CAAC;wBACjD,OAAO,MAAM,CAAC,UAAU,CAAC;oBAC3B,CAAC;oBAGD,IACE,MAAM,CAAC,iBAAiB,KAAK,SAAS;wBACtC,CAAC,MAAM,CAAC,qBAAqB,EAC7B,CAAC;wBACD,MAAM,CAAC,qBAAqB,GAAG,MAAM,CAAC,iBAAiB,CAAC;wBACxD,OAAO,MAAM,CAAC,iBAAiB,CAAC;oBAClC,CAAC;oBAID,MAAM,mBAAmB,GAAG,MAAM,CAAC,WAAW,CAAC;oBAC/C,IAAI,MAAM,CAAC,WAAW,KAAK,SAAS,IAAI,MAAM,CAAC,WAAW,KAAK,CAAC,EAAE,CAAC;wBACjE,MAAM,CAAC,WAAW,GAAG,CAAC,CAAC;wBACvB,MAAM,CAAC,KAAK,CACV,yBAAyB,MAAM,CAAC,KAAK,KAAK,mBAAmB,iDAAiD,CAC/G,CAAC;oBACJ,CAAC;gBAGH,CAAC;gBAGD,IACE,MAAM,CAAC,KAAK;oBACZ,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC;wBAC7B,kBAAkB,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EACxC,CAAC;oBAED,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;wBAC3B,MAAM,CAAC,cAAc,GAAG,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;wBAChD,MAAM,CAAC,IAAI,CACT,6DAA6D,MAAM,CAAC,KAAK,EAAE,CAC5E,CAAC;oBACJ,CAAC;yBAAM,IAAI,MAAM,CAAC,cAAc,CAAC,aAAa,KAAK,IAAI,EAAE,CAAC;wBACxD,MAAM,CAAC,cAAc,CAAC,aAAa,GAAG,IAAI,CAAC;wBAC3C,MAAM,CAAC,IAAI,CACT,+DAA+D,MAAM,CAAC,KAAK,EAAE,CAC9E,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAED,OAAO,MAAM,CAAC;YAChB,CAAC,CAAC;YAEF,MAAM,CAAC,IAAI,CACT,wBAAwB,SAAS,4DAA4D,CAC9F,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,IAAI,CACT,sDAAsD,SAAS,EAAE,CAClE,CAAC;QACJ,CAAC;QAGD,MAAM,2BAA2B,GAAG,SAAS,CAAC,mBAAmB,CAAC;QAElE,IAAI,2BAA2B,EAAE,CAAC;YAChC,SAAS,CAAC,mBAAmB,GAAG,KAAK,WACnC,OAAY,EACZ,OAAa;gBAGb,IACE,CAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,KAAK;oBACd,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC;wBAC9B,kBAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EACzC,CAAC;oBACD,IAAI,UAAU,GAAG,KAAK,CAAC;oBAGvB,IAAI,OAAO,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;wBACrC,OAAO,CAAC,qBAAqB,GAAG,OAAO,CAAC,UAAU,CAAC;wBACnD,OAAO,OAAO,CAAC,UAAU,CAAC;wBAC1B,UAAU,GAAG,IAAI,CAAC;oBACpB,CAAC;oBAGD,IAAI,OAAO,CAAC,WAAW,KAAK,SAAS,IAAI,OAAO,CAAC,WAAW,KAAK,CAAC,EAAE,CAAC;wBACnE,MAAM,YAAY,GAAG,OAAO,CAAC,WAAW,CAAC;wBACzC,OAAO,CAAC,WAAW,GAAG,CAAC,CAAC;wBACxB,MAAM,CAAC,KAAK,CACV,gDAAgD,OAAO,CAAC,KAAK,KAAK,YAAY,OAAO,CACtF,CAAC;wBACF,UAAU,GAAG,IAAI,CAAC;oBACpB,CAAC;oBAGD,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;wBAC5B,OAAO,CAAC,cAAc,GAAG,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;wBACjD,MAAM,CAAC,KAAK,CACV,sEAAsE,OAAO,CAAC,KAAK,EAAE,CACtF,CAAC;wBACF,UAAU,GAAG,IAAI,CAAC;oBACpB,CAAC;yBAAM,IAAI,OAAO,CAAC,cAAc,CAAC,aAAa,KAAK,IAAI,EAAE,CAAC;wBACzD,OAAO,CAAC,cAAc,CAAC,aAAa,GAAG,IAAI,CAAC;wBAC5C,MAAM,CAAC,KAAK,CACV,wEAAwE,OAAO,CAAC,KAAK,EAAE,CACxF,CAAC;wBACF,UAAU,GAAG,IAAI,CAAC;oBACpB,CAAC;oBAED,IAAI,UAAU,EAAE,CAAC;wBACf,MAAM,CAAC,KAAK,CACV,mDAAmD,OAAO,CAAC,KAAK,EAAE,CACnE,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAED,MAAM,MAAM,GAAG,MAAM,2BAA2B,CAAC,IAAI,CACnD,IAAI,EACJ,OAAO,EACP,OAAO,CACR,CAAC;gBAGF,IACE,CAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,KAAK;oBACd,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC;wBAC9B,kBAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EACzC,CAAC;oBACD,MAAM,CAAC,IAAI,CACT,2CAA2C,OAAO,CAAC,KAAK,GAAG,CAC5D,CAAC;oBACF,MAAM,CAAC,IAAI,CAAC,kBAAkB,MAAM,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBAEtE,IAAI,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,KAAK,EAAE,CAAC;wBAClB,MAAM,CAAC,IAAI,CACT,gBAAgB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CACxD,CAAC;oBACJ,CAAC;yBAAM,CAAC;wBACN,MAAM,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;oBAC5C,CAAC;oBAED,IAAI,CAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,OAAO,KAAI,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;wBACzC,MAAM,CAAC,IAAI,CACT,sBAAsB,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAClE,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAED,OAAO,MAAM,CAAC;YAChB,CAAC,CAAC;YAEF,MAAM,CAAC,IAAI,CACT,wBAAwB,SAAS,+DAA+D,CACjG,CAAC;QACJ,CAAC;QAGD,MAAM,wBAAwB,GAAG,SAAS,CAAC,gBAAgB,CAAC;QAE5D,IAAI,wBAAwB,EAAE,CAAC;YAC7B,SAAS,CAAC,gBAAgB,GAAG;;gBAC3B,MAAM,KAAK,GACT,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,KAAK,KAAI,MAAC,IAAY,CAAC,SAAS,0CAAE,SAAS,CAAA,CAAC;gBAKrE,MAAM,WAAW,GACf,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC;oBAClB,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC;oBACvB,gBAAgB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBAE/B,MAAM,cAAc,GAAG,wBAAwB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAE3D,MAAM,CAAC,IAAI,CACT,6CAA6C,KAAK,cAAc,WAAW,cAAc,cAAc,eAAe,IAAI,CAAC,SAAS,WAAY,IAAY,CAAC,KAAK,eAAe,IAAI,CAAC,SAAS,CAAC,MAAC,IAAY,CAAC,SAAS,0CAAE,SAAS,CAAC,EAAE,CACtO,CAAC;gBAEF,OAAO,WAAW,CAAC;YACrB,CAAC,CAAC;YAEF,MAAM,CAAC,IAAI,CACT,wBAAwB,SAAS,sEAAsE,CACxG,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,IAAI,CACT,sDAAsD,SAAS,EAAE,CAClE,CAAC;QACJ,CAAC;QAOD,MAAM,cAAc,GAAG,SAAS,CAAC,MAAM,CAAC;QAExC,IAAI,cAAc,EAAE,CAAC;YACnB,SAAS,CAAC,MAAM,GAAG,KAAK,WAAW,GAAG,IAAW;;gBAC/C,MAAM,KAAK,GACT,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,KAAK,KAAI,MAAC,IAAY,CAAC,SAAS,0CAAE,SAAS,CAAA,CAAC;gBAErE,IAAI,KAAK,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;oBACrC,MAAM,CAAC,IAAI,CAAC,qCAAqC,KAAK,EAAE,CAAC,CAAC;oBAG1D,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;wBACZ,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;wBACvB,MAAM,CAAC,IAAI,CACT,qCAAqC,MAAM,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAC5E,CAAC;wBACF,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;4BACjB,MAAM,CAAC,IAAI,CACT,kCAAkC,MAAM,CAAC,KAAK,CAAC,MAAM,QAAQ,CAC9D,CAAC;wBACJ,CAAC;oBACH,CAAC;oBAGD,MAAM,UAAU,GACb,IAAY,CAAC,KAAK;wBAClB,IAAY,CAAC,UAAU;wBACvB,IAAY,CAAC,KAAK,CAAC;oBACtB,IAAI,UAAU,EAAE,CAAC;wBACf,MAAM,CAAC,IAAI,CACT,wCAAwC,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,EAAE,CAChG,CAAC;oBACJ,CAAC;yBAAM,CAAC;wBACN,MAAM,CAAC,IAAI,CAAC,wCAAwC,CAAC,CAAC;oBACxD,CAAC;gBACH,CAAC;gBAED,IAAI,MAAM,CAAC;gBACX,IAAI,CAAC;oBACH,MAAM,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBAClD,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,KAAK,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;wBACrC,MAAM,CAAC,KAAK,CACV,gDAAgD,KAAK,GAAG,EACxD;4BACE,YAAY,EACV,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;4BACxD,UAAU,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;4BAC5D,SAAS,EAAE,MAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,WAAW,0CAAE,IAAI;4BACnC,IAAI,EAAE,IAAI,CAAC,MAAM;4BACjB,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;4BACpB,UAAU,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE;4BACrD,KAAK,EAAE,CAAA,MAAA,MAAA,IAAI,CAAC,CAAC,CAAC,0CAAE,KAAK,0CAAE,MAAM,KAAI,CAAC;yBACnC,CACF,CAAC;oBACJ,CAAC;oBACD,MAAM,KAAK,CAAC;gBACd,CAAC;gBAED,IAAI,KAAK,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;oBACrC,MAAM,CAAC,IAAI,CAAC,gDAAgD,KAAK,GAAG,CAAC,CAAC;oBACtE,MAAM,CAAC,IAAI,CAAC,gBAAgB,MAAM,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBAEpE,IAAI,CAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,cAAc,MAAI,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,aAAa,CAAA,EAAE,CAAC;wBACpD,MAAM,KAAK,GAAG,MAAM,CAAC,cAAc,IAAI,MAAM,CAAC,aAAa,CAAC;wBAC5D,MAAM,CAAC,IAAI,CACT,yBAAyB,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAC1D,CAAC;oBACJ,CAAC;oBAED,IAAI,CAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,iBAAiB,MAAI,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,gBAAgB,CAAA,EAAE,CAAC;wBAC1D,MAAM,gBAAgB,GACpB,MAAM,CAAC,iBAAiB,IAAI,MAAM,CAAC,gBAAgB,CAAC;wBACtD,MAAM,CAAC,IAAI,CACT,4BAA4B,IAAI,CAAC,SAAS,CAAC,gBAAgB,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CACxE,CAAC;wBAGF,IACE,CAAC,MAAM,CAAC,cAAc;6BACtB,gBAAgB,aAAhB,gBAAgB,uBAAhB,gBAAgB,CAAE,mBAAmB,CAAA,EACrC,CAAC;4BACD,MAAM,cAAc,GAAG,gBAAgB,CAAC,mBAAmB,CAAC;4BAC5D,MAAM,CAAC,cAAc,GAAG;gCACtB,YAAY,EAAE,cAAc,CAAC,YAAY,IAAI,CAAC;gCAC9C,aAAa,EAAE,cAAc,CAAC,gBAAgB,IAAI,CAAC;gCACnD,YAAY,EAAE,cAAc,CAAC,WAAW,IAAI,CAAC;6BAC9C,CAAC;4BAEF,MAAM,CAAC,IAAI,CACT,kEAAkE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CACnH,CAAC;wBACJ,CAAC;oBACH,CAAC;oBAED,IAAI,CAAC,CAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,cAAc,CAAA,IAAI,CAAC,CAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,aAAa,CAAA,EAAE,CAAC;wBACtD,MAAM,CAAC,IAAI,CAAC,0CAA0C,CAAC,CAAC;oBAC1D,CAAC;gBACH,CAAC;gBAED,OAAO,MAAM,CAAC;YAChB,CAAC,CAAC;YAEF,MAAM,CAAC,IAAI,CACT,wBAAwB,SAAS,2DAA2D,CAC7F,CAAC;QACJ,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC;AAGD,sBAAsB,EAAE,CAAC;AACzB,8CAAqD;AACrD,oDAAqD;AACrD,8CAA+C;AAC/C,8CAAiD;AACjD,iEAA6D;AAC7D,oDAAqD;AAErD,kDAA2D;AAY3D,MAAa,gBAAgB;IAS3B,YACU,aAAkC,EAC1C,MAAe;QADP,kBAAa,GAAb,aAAa,CAAqB;QANpC,qBAAgB,GAAG,IAAI,GAAG,EAAuC,CAAC;QAGlE,uBAAkB,GAAG,IAAI,GAAG,EAAiB,CAAC;QAgErC,sBAAiB,GAE9B;YACF,CAAC,4BAAa,CAAC,MAAM,CAAC,EAAE,CAAC,EACvB,SAAS,EACT,kBAAkB,EAClB,gBAAgB,EAChB,QAAQ,GACT,EAAE,EAAE;gBAEH,IAAI,IAAI,CAAC,2BAA2B,CAAC,SAAS,CAAC,EAAE,CAAC;oBAGhD,MAAM,gBAAgB,GAAG,CAAC,CAAC;oBAC3B,MAAM,MAAM,GAAG;wBACb,SAAS;wBACT,WAAW,EAAE,gBAAgB;wBAC7B,mBAAmB,EAAE,gBAAgB;wBACrC,SAAS,EAAE,IAAI;wBACf,YAAY,EAAE,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc;qBACrD,CAAC;oBAEF,IAAI,kBAAkB,KAAK,CAAC,EAAE,CAAC;wBAC7B,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,sCAAsC,SAAS,KAAK,kBAAkB,kDAAkD,CACzH,CAAC;oBACJ,CAAC;oBAID,MAAM,UAAU,GAAG,IAAI,mBAAU,CAAC,MAAM,CAAC,CAAC;oBAI1C,OAAO,UAAU,CAAC;gBACpB,CAAC;qBAAM,CAAC;oBAEN,MAAM,MAAM,GAAG;wBACb,SAAS;wBACT,WAAW,EAAE,kBAAkB;wBAC/B,SAAS,EAAE,gBAAgB;wBAC3B,SAAS,EAAE,IAAI;wBACf,YAAY,EAAE,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc;qBACrD,CAAC;oBAIF,MAAM,UAAU,GAAG,IAAI,mBAAU,CAAC,MAAM,CAAC,CAAC;oBAI1C,OAAO,UAAU,CAAC;gBACpB,CAAC;YACH,CAAC;YAED,CAAC,4BAAa,CAAC,SAAS,CAAC,EAAE,CAAC,EAC1B,SAAS,EACT,kBAAkB,EAClB,gBAAgB,EAChB,QAAQ,GACT,EAAE,EAAE,CACH,IAAI,yBAAa,CAAC;gBAChB,SAAS;gBACT,WAAW,EAAE,kBAAkB;gBAC/B,SAAS,EAAE,gBAAgB;gBAC3B,eAAe,EAAE,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB;aAC3D,CAAC;YAEJ,CAAC,4BAAa,CAAC,MAAM,CAAC,EAAE,CAAC,EACvB,SAAS,EACT,kBAAkB,EAClB,gBAAgB,EAChB,QAAQ,GACT,EAAE,EAAE,CACH,IAAI,mBAAU,CAAC;gBACb,KAAK,EAAE,SAAS;gBAChB,WAAW,EAAE,kBAAkB;gBAE/B,MAAM,EAAE,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc;aAC/C,CAAC;YAEJ,CAAC,4BAAa,CAAC,OAAO,CAAC,EAAE,CAAC,EACxB,SAAS,EACT,kBAAkB,EAClB,gBAAgB,EAChB,QAAQ,GACT,EAAE,EAAE,CACH,IAAI,yBAAa,CAAC;gBAChB,KAAK,EAAE,SAAS;gBAChB,WAAW,EAAE,kBAAkB;gBAC/B,SAAS,EAAE,gBAAgB;gBAC3B,MAAM,EAAE,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,eAAe;aAChD,CAAC;YAIJ,CAAC,4BAAa,CAAC,aAAa,CAAC,EAAE,CAAC,EAC9B,SAAS,EACT,kBAAkB,EAClB,gBAAgB,EAChB,QAAQ,GACT,EAAE,EAAE;gBAEH,IAAI,IAAI,CAAC,2BAA2B,CAAC,SAAS,CAAC,EAAE,CAAC;oBAGhD,MAAM,gBAAgB,GAAG,CAAC,CAAC;oBAC3B,MAAM,MAAM,GAAG;wBACb,SAAS;wBACT,WAAW,EAAE,gBAAgB;wBAC7B,mBAAmB,EAAE,gBAAgB;wBACrC,SAAS,EAAE,IAAI;wBACf,YAAY,EAAE,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc;qBACrD,CAAC;oBAEF,IAAI,kBAAkB,KAAK,CAAC,EAAE,CAAC;wBAC7B,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,6CAA6C,SAAS,KAAK,kBAAkB,kDAAkD,CAChI,CAAC;oBACJ,CAAC;oBAED,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0CAA0C,EAAE;wBAC5D,SAAS;wBACT,mBAAmB,EAAE,gBAAgB;wBACrC,WAAW,EAAE,gBAAgB;wBAC7B,mBAAmB,EAAE,kBAAkB;wBACvC,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC,YAAY;qBACjC,CAAC,CAAC;oBAEH,MAAM,UAAU,GAAG,IAAI,mBAAU,CAAC,MAAM,CAAC,CAAC;oBAG1C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,2CAA2C,EAAE;wBAC7D,SAAS,EAAE,SAAS;wBACpB,SAAS,EAAG,UAAkB,CAAC,SAAS;wBACxC,mBAAmB,EAAG,UAAkB,CAAC,mBAAmB;wBAC5D,WAAW,EAAE,UAAU,CAAC,WAAW;wBACnC,SAAS,EAAE,UAAU,CAAC,SAAS;wBAE/B,YAAY,EAAG,UAAkB,CAAC,YAAY;wBAC9C,MAAM,EAAG,UAAkB,CAAC,MAAM;qBACnC,CAAC,CAAC;oBAEH,OAAO,UAAU,CAAC;gBACpB,CAAC;qBAAM,CAAC;oBAEN,MAAM,MAAM,GAAG;wBACb,SAAS;wBACT,WAAW,EAAE,kBAAkB;wBAC/B,SAAS,EAAE,gBAAgB;wBAC3B,SAAS,EAAE,IAAI;wBACf,YAAY,EAAE,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc;qBACrD,CAAC;oBAEF,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0CAA0C,EAAE;wBAC5D,SAAS;wBACT,SAAS,EAAE,gBAAgB;wBAC3B,WAAW,EAAE,kBAAkB;wBAC/B,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC,YAAY;qBACjC,CAAC,CAAC;oBAEH,MAAM,UAAU,GAAG,IAAI,mBAAU,CAAC,MAAM,CAAC,CAAC;oBAG1C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,2CAA2C,EAAE;wBAC7D,SAAS,EAAE,SAAS;wBACpB,SAAS,EAAG,UAAkB,CAAC,SAAS;wBACxC,mBAAmB,EAAG,UAAkB,CAAC,mBAAmB;wBAC5D,WAAW,EAAE,UAAU,CAAC,WAAW;wBACnC,SAAS,EAAE,UAAU,CAAC,SAAS;wBAE/B,YAAY,EAAG,UAAkB,CAAC,YAAY;wBAC9C,MAAM,EAAG,UAAkB,CAAC,MAAM;qBACnC,CAAC,CAAC;oBAEH,OAAO,UAAU,CAAC;gBACpB,CAAC;YACH,CAAC;YAGD,CAAC,4BAAa,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE;gBAC3B,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;YACxD,CAAC;YACD,CAAC,4BAAa,CAAC,cAAc,CAAC,EAAE,GAAG,EAAE;gBACnC,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;YAChE,CAAC;YACD,CAAC,4BAAa,CAAC,gBAAgB,CAAC,EAAE,GAAG,EAAE;gBACrC,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;YAClE,CAAC;YACD,CAAC,4BAAa,CAAC,QAAQ,CAAC,EAAE,GAAG,EAAE;gBAC7B,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;YAC1D,CAAC;SACF,CAAC;QAGe,wBAAmB,GAGhC;YACF,CAAC,4BAAa,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,YAAY,EAAE,EAAE,EAAE;gBAChE,OAAO,IAAI,qBAAY,CAAC;oBACtB,MAAM,EAAE,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc;oBAC9C,KAAK,EAAE,SAAS;oBAChB,IAAI,EAAE,YAAY,IAAI,EAAE;iBACzB,CAAC,CAAC;YACL,CAAC;YAED,CAAC,4BAAa,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,YAAY,EAAE,EAAE,EAAE;gBAClE,OAAO,IAAI,gCAAc,CAAC;oBACxB,MAAM,EAAE,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB;oBAChD,KAAK,EAAE,SAAS;oBAChB,IAAI,EAAE,YAAY,IAAI,EAAE;iBACzB,CAAC,CAAC;YACL,CAAC;YAGD,CAAC,4BAAa,CAAC,MAAM,CAAC,EAAE,SAAS;YACjC,CAAC,4BAAa,CAAC,SAAS,CAAC,EAAE,SAAS;YACpC,CAAC,4BAAa,CAAC,OAAO,CAAC,EAAE,SAAS;YAClC,CAAC,4BAAa,CAAC,GAAG,CAAC,EAAE,SAAS;YAC9B,CAAC,4BAAa,CAAC,MAAM,CAAC,EAAE,SAAS;YACjC,CAAC,4BAAa,CAAC,cAAc,CAAC,EAAE,SAAS;YACzC,CAAC,4BAAa,CAAC,aAAa,CAAC,EAAE,SAAS;YACxC,CAAC,4BAAa,CAAC,gBAAgB,CAAC,EAAE,SAAS;SAC5C,CAAC;QAGe,2BAAsB,GAGnC;YACF,CAAC,4BAAa,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,EAAE,EAAE,CAClD,IAAI,yBAAgB,CAAC;gBACnB,KAAK,EAAE,SAAS;gBAChB,MAAM,EAAE,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc;aAC/C,CAAC;YAGJ,CAAC,4BAAa,CAAC,SAAS,CAAC,EAAE,SAAS;YACpC,CAAC,4BAAa,CAAC,MAAM,CAAC,EAAE,SAAS;YACjC,CAAC,4BAAa,CAAC,OAAO,CAAC,EAAE,SAAS;YAClC,CAAC,4BAAa,CAAC,GAAG,CAAC,EAAE,SAAS;YAC9B,CAAC,4BAAa,CAAC,MAAM,CAAC,EAAE,SAAS;YACjC,CAAC,4BAAa,CAAC,cAAc,CAAC,EAAE,SAAS;YACzC,CAAC,4BAAa,CAAC,aAAa,CAAC,EAAE,SAAS;YACxC,CAAC,4BAAa,CAAC,gBAAgB,CAAC,EAAE,SAAS;YAC3C,CAAC,4BAAa,CAAC,QAAQ,CAAC,EAAE,SAAS;SACpC,CAAC;QAjTA,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,IAAI,eAAM,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;IAC5D,CAAC;IAKO,qBAAqB,CAC3B,OAAe,EACf,WAAoB,EACpB,SAAkB,EAClB,SAAqB;QAErB,OAAO,GAAG,OAAO,IAAI,WAAW,IAAI,SAAS,IAAI,SAAS,IAAI,SAAS,IAAI,SAAS,IAAI,wBAAS,CAAC,IAAI,EAAE,CAAC;IAC3G,CAAC;IAgBO,2BAA2B,CAAC,SAAiB;QACnD,MAAM,WAAW,GACf,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC;YAC3B,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAC5B,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAC5B,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAC5B,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAE5B,kBAAkB,CAAC,IAAI,CAAC,SAAS,CAAC;YAClC,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAEhC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,uCAAuC,SAAS,GAAG,EAAE;YACrE,SAAS;YACT,2BAA2B,EAAE,WAAW;YACxC,MAAM,EAAE;gBACN,YAAY,EAAE,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC;gBACzC,UAAU,EAAE,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC;gBACxC,UAAU,EAAE,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC;gBACxC,UAAU,EAAE,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC;gBACxC,UAAU,EAAE,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC;gBACxC,aAAa,EAAE,kBAAkB,CAAC,IAAI,CAAC,SAAS,CAAC;gBACjD,UAAU,EAAE,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC;aAC1C;SACF,CAAC,CAAC;QAEH,OAAO,WAAW,CAAC;IACrB,CAAC;IA4PD,KAAK,CAAC,mBAAmB,CAAC,MAAuB;;QAE/C,MAAM,QAAQ,GAAG,IAAI,CAAC,qBAAqB,CACzC,MAAM,CAAC,OAAO,EACd,MAAM,CAAC,WAAW,EAClB,MAAM,CAAC,SAAS,EAChB,wBAAS,CAAC,IAAI,CACf,CAAC;QAGF,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC1D,IAAI,WAAW,EAAE,CAAC;YAChB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qCAAqC,QAAQ,EAAE,CAAC,CAAC;YACnE,OAAO,WAA4B,CAAC;QACtC,CAAC;QAED,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAEtE,IAAI,WAAW,CAAC,SAAS,KAAK,wBAAS,CAAC,IAAI,EAAE,CAAC;YAC7C,MAAM,IAAI,KAAK,CACb,SAAS,MAAM,CAAC,OAAO,+BAA+B,WAAW,CAAC,SAAS,GAAG,CAC/E,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,iBAAiB,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAC7D,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CACb,2CAA2C,WAAW,CAAC,QAAQ,EAAE,CAClE,CAAC;QACJ,CAAC;QAGD,MAAM,WAAW,mCACZ,WAAW,KACd,kBAAkB,EAAE,MAAM,CACxB,MAAA,MAAM,CAAC,WAAW,mCAAI,WAAW,CAAC,kBAAkB,CACrD,EACD,gBAAgB,EAAE,MAAM,CACtB,MAAA,MAAM,CAAC,SAAS,mCAAI,WAAW,CAAC,gBAAgB,CACjD,GACF,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qCAAqC,QAAQ,EAAE,CAAC,CAAC;QACnE,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;QAGnC,KAAK,CAAC,QAAQ,mCACT,KAAK,CAAC,QAAQ,KACjB,OAAO,EAAE,MAAM,CAAC,OAAO,GACxB,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,oCAAoC,EAAE;YACtD,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,YAAY,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,IAAI,EAAE,CAAC;YAC/C,UAAU,EAAE,CAAC,CAAC,CAAA,MAAA,KAAK,CAAC,QAAQ,0CAAE,OAAO,CAAA;SACtC,CAAC,CAAC;QAGH,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAE7C,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,qBAAqB,CACzB,MAAuB;QAGvB,MAAM,QAAQ,GAAG,IAAI,CAAC,qBAAqB,CACzC,MAAM,CAAC,OAAO,EACd,SAAS,EACT,MAAM,CAAC,SAAS,EAChB,wBAAS,CAAC,MAAM,CACjB,CAAC;QAGF,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC1D,IAAI,WAAW,EAAE,CAAC;YAChB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,uCAAuC,QAAQ,EAAE,CAAC,CAAC;YACrE,OAAO,WAAqC,CAAC;QAC/C,CAAC;QAED,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAEtE,IAAI,WAAW,CAAC,SAAS,KAAK,wBAAS,CAAC,MAAM,EAAE,CAAC;YAC/C,MAAM,IAAI,KAAK,CACb,SAAS,MAAM,CAAC,OAAO,iCAAiC,WAAW,CAAC,SAAS,GAAG,CACjF,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,mBAAmB,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAC/D,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CACb,6CAA6C,WAAW,CAAC,QAAQ,EAAE,CACpE,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,uCAAuC,QAAQ,EAAE,CAAC,CAAC;QACrE,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;QAGlC,KAAa,CAAC,QAAQ,mCACjB,KAAa,CAAC,QAAQ,KAC1B,OAAO,EAAE,MAAM,CAAC,OAAO,GACxB,CAAC;QAGF,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAE7C,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,wBAAwB,CAAC,MAAuB;QAEpD,MAAM,QAAQ,GAAG,IAAI,CAAC,qBAAqB,CACzC,MAAM,CAAC,OAAO,EACd,SAAS,EACT,SAAS,EACT,wBAAS,CAAC,SAAS,CACpB,CAAC;QAGF,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC1D,IAAI,WAAW,EAAE,CAAC;YAChB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0CAA0C,QAAQ,EAAE,CAAC,CAAC;YACxE,OAAO,WAAyB,CAAC;QACnC,CAAC;QAED,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAEtE,IAAI,WAAW,CAAC,SAAS,KAAK,wBAAS,CAAC,SAAS,EAAE,CAAC;YAClD,MAAM,IAAI,KAAK,CACb,SAAS,MAAM,CAAC,OAAO,qCAAqC,WAAW,CAAC,SAAS,GAAG,CACrF,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,sBAAsB,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAClE,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CACb,gDAAgD,WAAW,CAAC,QAAQ,EAAE,CACvE,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0CAA0C,QAAQ,EAAE,CAAC,CAAC;QACxE,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;QAGlC,KAAa,CAAC,QAAQ,mCACjB,KAAa,CAAC,QAAQ,KAC1B,OAAO,EAAE,MAAM,CAAC,OAAO,GACxB,CAAC;QAGF,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAE7C,OAAO,KAAK,CAAC;IACf,CAAC;IAID,KAAK,CAAC,mBAAmB,CAAC,OAAe;QACvC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC;QAE1D,IAAI,MAAM,CAAC,SAAS,KAAK,wBAAS,CAAC,IAAI,EAAE,CAAC;YACxC,MAAM,IAAI,KAAK,CACb,SAAS,OAAO,8BAA8B,MAAM,CAAC,SAAS,EAAE,CACjE,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,qBAAqB,CACzB,OAAe;QAEf,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC;QAE1D,IAAI,MAAM,CAAC,SAAS,KAAK,wBAAS,CAAC,MAAM,EAAE,CAAC;YAC1C,MAAM,IAAI,KAAK,CACb,SAAS,OAAO,gCAAgC,MAAM,CAAC,SAAS,EAAE,CACnE,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,wBAAwB,CAAC,OAAe;QAC5C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC;QAE1D,IAAI,MAAM,CAAC,SAAS,KAAK,wBAAS,CAAC,SAAS,EAAE,CAAC;YAC7C,MAAM,IAAI,KAAK,CACb,SAAS,OAAO,oCAAoC,MAAM,CAAC,SAAS,EAAE,CACvE,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,OAAe,EACf,YAAwB;QAExB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC;QAG1D,IAAI,YAAY,IAAI,MAAM,CAAC,SAAS,KAAK,YAAY,EAAE,CAAC;YACtD,MAAM,IAAI,KAAK,CACb,SAAS,OAAO,mBAAmB,YAAY,cAAc,MAAM,CAAC,SAAS,EAAE,CAChF,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;IAC5C,CAAC;IAID,KAAK,CAAC,qBAAqB,CAAC,MAA+B;QACzD,QAAQ,MAAM,CAAC,SAAS,EAAE,CAAC;YACzB,KAAK,wBAAS,CAAC,IAAI;gBACjB,OAAO,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;YAE1C,KAAK,wBAAS,CAAC,MAAM;gBACnB,OAAO,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;YAE5C,KAAK,wBAAS,CAAC,SAAS;gBACtB,OAAO,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,CAAC;YAE/C,KAAK,wBAAS,CAAC,KAAK,CAAC;YACrB,KAAK,wBAAS,CAAC,MAAM;gBACnB,MAAM,IAAI,KAAK,CAAC,cAAc,MAAM,CAAC,SAAS,oBAAoB,CAAC,CAAC;YAEtE;gBACE,MAAM,IAAI,KAAK,CAAC,uBAAuB,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,sBAAsB,CAC1B,OAAe;QAGf,MAAM,YAAY,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACxD,IAAI,YAAY,EAAE,CAAC;YACjB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,8BAA8B,OAAO,EAAE,CAAC,CAAC;YAC3D,OAAO,YAAY,CAAC;QACtB,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0BAA0B,OAAO,EAAE,CAAC,CAAC;QAGvD,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa;YACnC,CAAC,CAAC,MAAM,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC;YACnC,CAAC,CAAC,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAKrC,MAAM,MAAM,mCACP,UAAU,KACb,SAAS,EAAG,UAAkB,CAAC,SAAS,IAAI,wBAAS,CAAC,IAAI,EAC1D,YAAY,EAAG,UAAkB,CAAC,YAAY,EAC9C,UAAU,EAAG,UAAkB,CAAC,UAAU,EAC1C,gBAAgB,EAAG,UAAkB,CAAC,gBAAgB,GACvD,CAAC;QAKF,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAE3C,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,oBAAoB,CAAC,SAAoB;QACvC,OAAO,CAAC,wBAAS,CAAC,IAAI,EAAE,wBAAS,CAAC,MAAM,EAAE,wBAAS,CAAC,SAAS,CAAC,CAAC,QAAQ,CACrE,SAAS,CACV,CAAC;IACJ,CAAC;IAED,sBAAsB;QACpB,OAAO,CAAC,wBAAS,CAAC,IAAI,EAAE,wBAAS,CAAC,MAAM,EAAE,wBAAS,CAAC,SAAS,CAAC,CAAC;IACjE,CAAC;IAKD,UAAU;QACR,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,oCAAoC,IAAI,CAAC,gBAAgB,CAAC,IAAI,aAAa,IAAI,CAAC,kBAAkB,CAAC,IAAI,YAAY,CACpH,CAAC;QACF,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;QAC9B,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,CAAC;IAClC,CAAC;IAKD,eAAe,CAAC,OAAe;QAC7B,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAGtC,MAAM,YAAY,GAAa,EAAE,CAAC;QAClC,KAAK,MAAM,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE,EAAE,CAAC;YACtD,IAAI,GAAG,CAAC,UAAU,CAAC,GAAG,OAAO,GAAG,CAAC,EAAE,CAAC;gBAClC,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACzB,CAAC;QACH,CAAC;QAED,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE;YACzB,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,2BAA2B,OAAO,KAAK,YAAY,CAAC,MAAM,YAAY,CACvE,CAAC;IACJ,CAAC;IAKD,aAAa;QACX,OAAO;YACL,eAAe,EAAE,IAAI,CAAC,gBAAgB,CAAC,IAAI;YAC3C,iBAAiB,EAAE,IAAI,CAAC,kBAAkB,CAAC,IAAI;YAC/C,eAAe,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAI,EAAE,CAAC;YACzD,iBAAiB,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,IAAI,EAAE,CAAC;SAC9D,CAAC;IACJ,CAAC;IAGO,KAAK,CAAC,YAAY,CACxB,OAAe;QAEf,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,OAAO,IAAI,uBAAuB,CAAC;QAC9D,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;QAE7C,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC9D,CAAC;QAED,MAAM,GAAG,GAAG,GAAG,MAAM,kCAAkC,OAAO,SAAS,CAAC;QAExE,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;YAChC,MAAM,EAAE,KAAK;YACb,OAAO,EAAE;gBACP,kBAAkB,EAAE,KAAK;gBACzB,cAAc,EAAE,kBAAkB;aACnC;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CACb,iCAAiC,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,CAC1E,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QAErC,OAAO,CAAC,KAAK,CACX,oDAAoD,OAAO,GAAG,EAC9D;YACE,GAAG;YACH,UAAU,EAAE,QAAQ,CAAC,MAAM;YAC3B,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC;YAC/B,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,YAAY,EAAE,CAAC,CAAC,MAAM,CAAC,SAAS;YAChC,UAAU,EAAE,MAAM;SACnB,CACF,CAAC;QAGF,MAAM,MAAM,mCACP,MAAM,KACT,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,wBAAS,CAAC,IAAI,GAC9C,CAAC;QAEF,OAAO,CAAC,KAAK,CAAC,+CAA+C,EAAE;YAC7D,OAAO;YACP,eAAe,EAAE,MAAM,CAAC,SAAS;YACjC,YAAY,EAAE,CAAC,MAAM,CAAC,SAAS;YAC/B,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC;SAChC,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AA/rBD,4CA+rBC;AAeD,SAAgB,qBAAqB,CACnC,KAAoB,EACpB,KAAuB,EACvB,aAAkB,EAAE;IAMpB,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,OAAO;YACL,cAAc,EAAE,KAAK;YACrB,WAAW,EAAE,UAAU;YACvB,WAAW,EAAE,MAAM;SACpB,CAAC;IACJ,CAAC;IAGD,IAAI,KAAK,CAAC,SAAS,IAAI,OAAO,KAAK,CAAC,SAAS,KAAK,UAAU,EAAE,CAAC;QAC7D,IAAI,CAAC;YACH,MAAM,cAAc,GAAG,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YAC9C,OAAO;gBACL,cAAc;gBACd,WAAW,EAAE,UAAU;gBACvB,WAAW,EAAE,WAAW;aACzB,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAEf,MAAM,YAAY,GAAG,EAAE,KAAK,EAAE,CAAC;YAC/B,MAAM,WAAW,mCAAQ,UAAU,GAAK,YAAY,CAAE,CAAC;YACvD,OAAO;gBACL,cAAc,EAAE,KAAK;gBACrB,WAAW;gBACX,WAAW,EAAE,QAAQ;aACtB,CAAC;QACJ,CAAC;IACH,CAAC;IAGD,MAAM,YAAY,GAAG,EAAE,KAAK,EAAE,CAAC;IAC/B,MAAM,WAAW,mCAAQ,UAAU,GAAK,YAAY,CAAE,CAAC;IAEvD,OAAO;QACL,cAAc,EAAE,KAAK;QACrB,WAAW;QACX,WAAW,EAAE,QAAQ;KACtB,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,38 @@
|
|
|
1
|
+
import { BaseChatModel } from "@langchain/core/language_models/chat_models";
|
|
2
|
+
import { BaseDocumentCompressor } from "@langchain/core/retrievers/document_compressors";
|
|
3
|
+
import { Embeddings } from "@langchain/core/embeddings";
|
|
4
|
+
import { ModelType } from "../shared-types";
|
|
5
|
+
import { ModelByIdConfig, ModelConfigWithToken } from "./llm.types";
|
|
6
|
+
export type Model = BaseChatModel | BaseDocumentCompressor | Embeddings;
|
|
7
|
+
export interface ModelByIdWithTypeConfig extends ModelByIdConfig {
|
|
8
|
+
modelType: ModelType;
|
|
9
|
+
}
|
|
10
|
+
export interface ModelConfigWithTokenAndType extends ModelConfigWithToken {
|
|
11
|
+
modelType: ModelType;
|
|
12
|
+
maxDocuments?: number;
|
|
13
|
+
dimensions?: number;
|
|
14
|
+
supportedFormats?: string[];
|
|
15
|
+
}
|
|
16
|
+
export interface IModelInitializer {
|
|
17
|
+
createChatModelById(modelId: string): Promise<BaseChatModel>;
|
|
18
|
+
createRerankModelById(modelId: string): Promise<BaseDocumentCompressor>;
|
|
19
|
+
createEmbeddingModelById(modelId: string): Promise<Embeddings>;
|
|
20
|
+
createModelById(modelId: string, expectedType?: ModelType): Promise<Model>;
|
|
21
|
+
initializeChatModel(config: ModelByIdConfig): Promise<BaseChatModel>;
|
|
22
|
+
initializeRerankModel(config: ModelByIdConfig): Promise<BaseDocumentCompressor>;
|
|
23
|
+
initializeEmbeddingModel(config: ModelByIdConfig): Promise<Embeddings>;
|
|
24
|
+
initializeModelByType(config: ModelByIdWithTypeConfig): Promise<Model>;
|
|
25
|
+
getModelConfigWithType(modelId: string): Promise<ModelConfigWithTokenAndType>;
|
|
26
|
+
isModelTypeSupported(modelType: ModelType): boolean;
|
|
27
|
+
getSupportedModelTypes(): ModelType[];
|
|
28
|
+
}
|
|
29
|
+
export type ChatModelCreator = (config: ModelConfigWithTokenAndType) => BaseChatModel;
|
|
30
|
+
export type RerankModelCreator = (config: ModelConfigWithTokenAndType) => BaseDocumentCompressor;
|
|
31
|
+
export type EmbeddingModelCreator = (config: ModelConfigWithTokenAndType) => Embeddings;
|
|
32
|
+
export interface ModelCreators {
|
|
33
|
+
[ModelType.CHAT]: ChatModelCreator;
|
|
34
|
+
[ModelType.RERANK]: RerankModelCreator;
|
|
35
|
+
[ModelType.EMBEDDING]: EmbeddingModelCreator;
|
|
36
|
+
[ModelType.IMAGE]?: never;
|
|
37
|
+
[ModelType.SPEECH]?: never;
|
|
38
|
+
}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model.interface.js","sourceRoot":"","sources":["../../src/llm/model.interface.ts"],"names":[],"mappings":";;AAGA,kDAA4C"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
import { BaseDocumentCompressor } from "@langchain/core/retrievers/document_compressors";
|
|
2
|
+
import { Document } from "@langchain/core/documents";
|
|
3
|
+
export interface VoyageAIRerankConfig {
|
|
4
|
+
apiKey?: string;
|
|
5
|
+
model?: string;
|
|
6
|
+
topN?: number;
|
|
7
|
+
truncation?: boolean;
|
|
8
|
+
}
|
|
9
|
+
export declare class VoyageAIRerank extends BaseDocumentCompressor {
|
|
10
|
+
private apiKey;
|
|
11
|
+
private model;
|
|
12
|
+
private topN;
|
|
13
|
+
private truncation;
|
|
14
|
+
private baseUrl;
|
|
15
|
+
constructor(config: VoyageAIRerankConfig);
|
|
16
|
+
compressDocuments(documents: Document[], query: string, callbacks?: any): Promise<Document[]>;
|
|
17
|
+
}
|
|
@@ -0,0 +1,56 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.VoyageAIRerank = void 0;
|
|
4
|
+
const document_compressors_1 = require("@langchain/core/retrievers/document_compressors");
|
|
5
|
+
class VoyageAIRerank extends document_compressors_1.BaseDocumentCompressor {
|
|
6
|
+
constructor(config) {
|
|
7
|
+
var _a;
|
|
8
|
+
super();
|
|
9
|
+
this.baseUrl = "https://api.voyageai.com/v1/rerank";
|
|
10
|
+
this.apiKey = config.apiKey || process.env.VOYAGEAI_API_KEY || "";
|
|
11
|
+
this.model = config.model || "rerank-2";
|
|
12
|
+
this.topN = config.topN || 20;
|
|
13
|
+
this.truncation = (_a = config.truncation) !== null && _a !== void 0 ? _a : true;
|
|
14
|
+
if (!this.apiKey) {
|
|
15
|
+
throw new Error("VoyageAI API key is required. Set VOYAGEAI_API_KEY environment variable or pass apiKey in config.");
|
|
16
|
+
}
|
|
17
|
+
}
|
|
18
|
+
async compressDocuments(documents, query, callbacks) {
|
|
19
|
+
if (documents.length === 0) {
|
|
20
|
+
return [];
|
|
21
|
+
}
|
|
22
|
+
const texts = documents.map(doc => doc.pageContent);
|
|
23
|
+
try {
|
|
24
|
+
const response = await fetch(this.baseUrl, {
|
|
25
|
+
method: "POST",
|
|
26
|
+
headers: {
|
|
27
|
+
"Content-Type": "application/json",
|
|
28
|
+
Authorization: `Bearer ${this.apiKey}`,
|
|
29
|
+
},
|
|
30
|
+
body: JSON.stringify({
|
|
31
|
+
model: this.model,
|
|
32
|
+
query: query,
|
|
33
|
+
documents: texts,
|
|
34
|
+
top_k: Math.min(this.topN, documents.length),
|
|
35
|
+
truncation: this.truncation,
|
|
36
|
+
}),
|
|
37
|
+
});
|
|
38
|
+
if (!response.ok) {
|
|
39
|
+
const error = await response.text();
|
|
40
|
+
throw new Error(`VoyageAI API error: ${response.status} - ${error}`);
|
|
41
|
+
}
|
|
42
|
+
const data = await response.json();
|
|
43
|
+
const rerankedIndices = data.results
|
|
44
|
+
.sort((a, b) => b.relevance_score - a.relevance_score)
|
|
45
|
+
.slice(0, this.topN)
|
|
46
|
+
.map((result) => result.index);
|
|
47
|
+
return rerankedIndices.map((index) => documents[index]);
|
|
48
|
+
}
|
|
49
|
+
catch (error) {
|
|
50
|
+
console.error("Error calling VoyageAI rerank API:", error);
|
|
51
|
+
return documents.slice(0, this.topN);
|
|
52
|
+
}
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
exports.VoyageAIRerank = VoyageAIRerank;
|
|
56
|
+
//# sourceMappingURL=voyageai-rerank.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"voyageai-rerank.js","sourceRoot":"","sources":["../../../src/llm/rerankers/voyageai-rerank.ts"],"names":[],"mappings":";;;AAAA,0FAAyF;AAezF,MAAa,cAAe,SAAQ,6CAAsB;IAOxD,YAAY,MAA4B;;QACtC,KAAK,EAAE,CAAC;QAHF,YAAO,GAAG,oCAAoC,CAAC;QAIrD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,EAAE,CAAC;QAClE,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,UAAU,CAAC;QACxC,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;QAC9B,IAAI,CAAC,UAAU,GAAG,MAAA,MAAM,CAAC,UAAU,mCAAI,IAAI,CAAC;QAE5C,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CACb,mGAAmG,CACpG,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,iBAAiB,CACrB,SAAqB,EACrB,KAAa,EACb,SAAe;QAEf,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC3B,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QAEpD,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE;gBACzC,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;iBACvC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,KAAK,EAAE,KAAK;oBACZ,SAAS,EAAE,KAAK;oBAChB,KAAK,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,SAAS,CAAC,MAAM,CAAC;oBAC5C,UAAU,EAAE,IAAI,CAAC,UAAU;iBAC5B,CAAC;aACH,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,KAAK,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACpC,MAAM,IAAI,KAAK,CAAC,uBAAuB,QAAQ,CAAC,MAAM,MAAM,KAAK,EAAE,CAAC,CAAC;YACvE,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YAInC,MAAM,eAAe,GAAG,IAAI,CAAC,OAAO;iBACjC,IAAI,CAAC,CAAC,CAAM,EAAE,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,eAAe,GAAG,CAAC,CAAC,eAAe,CAAC;iBAC/D,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC;iBACnB,GAAG,CAAC,CAAC,MAAW,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAEtC,OAAO,eAAe,CAAC,GAAG,CAAC,CAAC,KAAa,EAAE,EAAE,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC;QAClE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;YAE3D,OAAO,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;CACF;AArED,wCAqEC"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import { Connection } from "mongoose";
|
|
2
|
+
export interface DatabaseConfig {
|
|
3
|
+
mainDbUri: string;
|
|
4
|
+
graphDbUri: string;
|
|
5
|
+
graphName: string;
|
|
6
|
+
}
|
|
7
|
+
export declare class DatabaseFactory {
|
|
8
|
+
private static readonly logger;
|
|
9
|
+
private static connections;
|
|
10
|
+
static getMainDbConnection(mainDbUri: string): Promise<Connection>;
|
|
11
|
+
static getGraphDbConnection(graphDbUri: string, graphName: string): Promise<Connection>;
|
|
12
|
+
static closeAllConnections(): Promise<void>;
|
|
13
|
+
static getConnectionsStatus(): Record<string, string>;
|
|
14
|
+
private static getConnectionState;
|
|
15
|
+
}
|
|
@@ -0,0 +1,83 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.DatabaseFactory = void 0;
|
|
7
|
+
const mongoose_1 = __importDefault(require("mongoose"));
|
|
8
|
+
const common_1 = require("@nestjs/common");
|
|
9
|
+
class DatabaseFactory {
|
|
10
|
+
static async getMainDbConnection(mainDbUri) {
|
|
11
|
+
const connectionKey = `main:${mainDbUri}`;
|
|
12
|
+
if (this.connections.has(connectionKey)) {
|
|
13
|
+
return this.connections.get(connectionKey);
|
|
14
|
+
}
|
|
15
|
+
this.logger.debug(`Creating connection to main database: ${mainDbUri}`);
|
|
16
|
+
const connection = await mongoose_1.default.createConnection(mainDbUri, {
|
|
17
|
+
maxPoolSize: 10,
|
|
18
|
+
serverSelectionTimeoutMS: 5000,
|
|
19
|
+
socketTimeoutMS: 45000,
|
|
20
|
+
family: 4,
|
|
21
|
+
bufferCommands: false,
|
|
22
|
+
});
|
|
23
|
+
this.connections.set(connectionKey, connection);
|
|
24
|
+
this.logger.log(`Connected to main database: ${mainDbUri}`);
|
|
25
|
+
return connection;
|
|
26
|
+
}
|
|
27
|
+
static async getGraphDbConnection(graphDbUri, graphName) {
|
|
28
|
+
const connectionKey = `graph:${graphName}:${graphDbUri}`;
|
|
29
|
+
if (this.connections.has(connectionKey)) {
|
|
30
|
+
return this.connections.get(connectionKey);
|
|
31
|
+
}
|
|
32
|
+
this.logger.debug(`Creating connection to graph database: ${graphDbUri} for graph: ${graphName}`);
|
|
33
|
+
const connection = await mongoose_1.default.createConnection(graphDbUri, {
|
|
34
|
+
maxPoolSize: 5,
|
|
35
|
+
serverSelectionTimeoutMS: 5000,
|
|
36
|
+
socketTimeoutMS: 45000,
|
|
37
|
+
family: 4,
|
|
38
|
+
bufferCommands: false,
|
|
39
|
+
});
|
|
40
|
+
this.connections.set(connectionKey, connection);
|
|
41
|
+
this.logger.log(`Connected to graph database: ${graphDbUri} for graph: ${graphName}`);
|
|
42
|
+
return connection;
|
|
43
|
+
}
|
|
44
|
+
static async closeAllConnections() {
|
|
45
|
+
this.logger.debug("Closing all database connections");
|
|
46
|
+
for (const [key, connection] of this.connections) {
|
|
47
|
+
try {
|
|
48
|
+
await connection.close();
|
|
49
|
+
this.logger.debug(`Closed connection: ${key}`);
|
|
50
|
+
}
|
|
51
|
+
catch (error) {
|
|
52
|
+
this.logger.error(`Failed to close connection ${key}: ${error instanceof Error ? error.message : String(error)}`);
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
this.connections.clear();
|
|
56
|
+
this.logger.log("All database connections closed");
|
|
57
|
+
}
|
|
58
|
+
static getConnectionsStatus() {
|
|
59
|
+
const status = {};
|
|
60
|
+
for (const [key, connection] of this.connections) {
|
|
61
|
+
status[key] = this.getConnectionState(connection.readyState);
|
|
62
|
+
}
|
|
63
|
+
return status;
|
|
64
|
+
}
|
|
65
|
+
static getConnectionState(readyState) {
|
|
66
|
+
switch (readyState) {
|
|
67
|
+
case 0:
|
|
68
|
+
return "disconnected";
|
|
69
|
+
case 1:
|
|
70
|
+
return "connected";
|
|
71
|
+
case 2:
|
|
72
|
+
return "connecting";
|
|
73
|
+
case 3:
|
|
74
|
+
return "disconnecting";
|
|
75
|
+
default:
|
|
76
|
+
return "unknown";
|
|
77
|
+
}
|
|
78
|
+
}
|
|
79
|
+
}
|
|
80
|
+
exports.DatabaseFactory = DatabaseFactory;
|
|
81
|
+
DatabaseFactory.logger = new common_1.Logger(DatabaseFactory.name);
|
|
82
|
+
DatabaseFactory.connections = new Map();
|
|
83
|
+
//# sourceMappingURL=database-factory.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"database-factory.js","sourceRoot":"","sources":["../../src/retriever/database-factory.ts"],"names":[],"mappings":";;;;;;AAAA,wDAAgD;AAChD,2CAAwC;AAwBxC,MAAa,eAAe;IAQ1B,MAAM,CAAC,KAAK,CAAC,mBAAmB,CAAC,SAAiB;QAChD,MAAM,aAAa,GAAG,QAAQ,SAAS,EAAE,CAAC;QAE1C,IAAI,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC;YACxC,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,aAAa,CAAE,CAAC;QAC9C,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,yCAAyC,SAAS,EAAE,CAAC,CAAC;QAExE,MAAM,UAAU,GAAG,MAAM,kBAAQ,CAAC,gBAAgB,CAAC,SAAS,EAAE;YAC5D,WAAW,EAAE,EAAE;YACf,wBAAwB,EAAE,IAAI;YAC9B,eAAe,EAAE,KAAK;YACtB,MAAM,EAAE,CAAC;YACT,cAAc,EAAE,KAAK;SACtB,CAAC,CAAC;QAEH,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,aAAa,EAAE,UAAU,CAAC,CAAC;QAEhD,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,+BAA+B,SAAS,EAAE,CAAC,CAAC;QAE5D,OAAO,UAAU,CAAC;IACpB,CAAC;IAMD,MAAM,CAAC,KAAK,CAAC,oBAAoB,CAC/B,UAAkB,EAClB,SAAiB;QAEjB,MAAM,aAAa,GAAG,SAAS,SAAS,IAAI,UAAU,EAAE,CAAC;QAEzD,IAAI,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC;YACxC,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,aAAa,CAAE,CAAC;QAC9C,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,0CAA0C,UAAU,eAAe,SAAS,EAAE,CAC/E,CAAC;QAEF,MAAM,UAAU,GAAG,MAAM,kBAAQ,CAAC,gBAAgB,CAAC,UAAU,EAAE;YAC7D,WAAW,EAAE,CAAC;YACd,wBAAwB,EAAE,IAAI;YAC9B,eAAe,EAAE,KAAK;YACtB,MAAM,EAAE,CAAC;YACT,cAAc,EAAE,KAAK;SACtB,CAAC,CAAC;QAEH,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,aAAa,EAAE,UAAU,CAAC,CAAC;QAEhD,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,gCAAgC,UAAU,eAAe,SAAS,EAAE,CACrE,CAAC;QAEF,OAAO,UAAU,CAAC;IACpB,CAAC;IAKD,MAAM,CAAC,KAAK,CAAC,mBAAmB;QAC9B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,kCAAkC,CAAC,CAAC;QAEtD,KAAK,MAAM,CAAC,GAAG,EAAE,UAAU,CAAC,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACjD,IAAI,CAAC;gBACH,MAAM,UAAU,CAAC,KAAK,EAAE,CAAC;gBACzB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,sBAAsB,GAAG,EAAE,CAAC,CAAC;YACjD,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,8BAA8B,GAAG,KAAK,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAC/F,CAAC;YACJ,CAAC;QACH,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;IACrD,CAAC;IAKD,MAAM,CAAC,oBAAoB;QACzB,MAAM,MAAM,GAA2B,EAAE,CAAC;QAE1C,KAAK,MAAM,CAAC,GAAG,EAAE,UAAU,CAAC,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACjD,MAAM,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,kBAAkB,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;QAC/D,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,MAAM,CAAC,kBAAkB,CAAC,UAAkB;QAClD,QAAQ,UAAU,EAAE,CAAC;YACnB,KAAK,CAAC;gBACJ,OAAO,cAAc,CAAC;YACxB,KAAK,CAAC;gBACJ,OAAO,WAAW,CAAC;YACrB,KAAK,CAAC;gBACJ,OAAO,YAAY,CAAC;YACtB,KAAK,CAAC;gBACJ,OAAO,eAAe,CAAC;YACzB;gBACE,OAAO,SAAS,CAAC;QACrB,CAAC;IACH,CAAC;;AAlHH,0CAmHC;AAlHyB,sBAAM,GAAG,IAAI,eAAM,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;AACnD,2BAAW,GAA4B,IAAI,GAAG,EAAE,CAAC"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
3
|
+
if (k2 === undefined) k2 = k;
|
|
4
|
+
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
5
|
+
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
6
|
+
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
7
|
+
}
|
|
8
|
+
Object.defineProperty(o, k2, desc);
|
|
9
|
+
}) : (function(o, m, k, k2) {
|
|
10
|
+
if (k2 === undefined) k2 = k;
|
|
11
|
+
o[k2] = m[k];
|
|
12
|
+
}));
|
|
13
|
+
var __exportStar = (this && this.__exportStar) || function(m, exports) {
|
|
14
|
+
for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
|
|
15
|
+
};
|
|
16
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
17
|
+
__exportStar(require("./retriever.service"), exports);
|
|
18
|
+
__exportStar(require("./types"), exports);
|
|
19
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/retriever/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,sDAAoC;AAGpC,0CAAwB"}
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
import { HttpService } from "@nestjs/axios";
|
|
2
|
+
import { RetrieverSearchType } from "../shared-types";
|
|
3
|
+
import { CustomDocument } from "./types";
|
|
4
|
+
export interface RetrieverClientConfig {
|
|
5
|
+
baseUrl: string;
|
|
6
|
+
timeout?: number;
|
|
7
|
+
retries?: number;
|
|
8
|
+
}
|
|
9
|
+
export declare class RetrieverClient {
|
|
10
|
+
private readonly httpService;
|
|
11
|
+
private readonly logger;
|
|
12
|
+
private readonly baseUrl;
|
|
13
|
+
private readonly timeout;
|
|
14
|
+
private readonly retries;
|
|
15
|
+
constructor(httpService: HttpService, config: RetrieverClientConfig);
|
|
16
|
+
search(query: string, searchType: RetrieverSearchType, knowledgeBaseIds: string[], options?: {
|
|
17
|
+
k?: number;
|
|
18
|
+
threshold?: number;
|
|
19
|
+
metadata?: Record<string, any>;
|
|
20
|
+
}): Promise<CustomDocument[]>;
|
|
21
|
+
healthCheck(): Promise<boolean>;
|
|
22
|
+
getKnowledgeBaseInfo(knowledgeBaseId: string): Promise<{
|
|
23
|
+
id: string;
|
|
24
|
+
name: string;
|
|
25
|
+
documentsCount: number;
|
|
26
|
+
lastUpdated: string;
|
|
27
|
+
} | null>;
|
|
28
|
+
}
|
|
@@ -0,0 +1,97 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
|
3
|
+
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
4
|
+
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
|
5
|
+
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
|
|
6
|
+
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
|
7
|
+
};
|
|
8
|
+
var __metadata = (this && this.__metadata) || function (k, v) {
|
|
9
|
+
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
|
|
10
|
+
};
|
|
11
|
+
var RetrieverClient_1;
|
|
12
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
13
|
+
exports.RetrieverClient = void 0;
|
|
14
|
+
const common_1 = require("@nestjs/common");
|
|
15
|
+
const axios_1 = require("@nestjs/axios");
|
|
16
|
+
const rxjs_1 = require("rxjs");
|
|
17
|
+
let RetrieverClient = RetrieverClient_1 = class RetrieverClient {
|
|
18
|
+
constructor(httpService, config) {
|
|
19
|
+
this.httpService = httpService;
|
|
20
|
+
this.logger = new common_1.Logger(RetrieverClient_1.name);
|
|
21
|
+
this.baseUrl = config.baseUrl;
|
|
22
|
+
this.timeout = config.timeout || 180000;
|
|
23
|
+
this.retries = config.retries || 3;
|
|
24
|
+
}
|
|
25
|
+
async search(query, searchType, knowledgeBaseIds, options) {
|
|
26
|
+
var _a;
|
|
27
|
+
const url = `${this.baseUrl}/api/internal/retriever/search`;
|
|
28
|
+
const payload = {
|
|
29
|
+
query,
|
|
30
|
+
searchType,
|
|
31
|
+
knowledgeBaseIds,
|
|
32
|
+
options: {
|
|
33
|
+
k: (options === null || options === void 0 ? void 0 : options.k) || 10,
|
|
34
|
+
threshold: (options === null || options === void 0 ? void 0 : options.threshold) || 0.7,
|
|
35
|
+
metadata: (options === null || options === void 0 ? void 0 : options.metadata) || {},
|
|
36
|
+
},
|
|
37
|
+
};
|
|
38
|
+
this.logger.debug(`Searching documents: ${query}`, {
|
|
39
|
+
searchType,
|
|
40
|
+
knowledgeBaseIds,
|
|
41
|
+
options: payload.options,
|
|
42
|
+
});
|
|
43
|
+
try {
|
|
44
|
+
const response = await (0, rxjs_1.firstValueFrom)(this.httpService.post(url, payload, {
|
|
45
|
+
timeout: this.timeout,
|
|
46
|
+
}));
|
|
47
|
+
const documents = response.data.documents || [];
|
|
48
|
+
this.logger.debug(`Found ${documents.length} documents`);
|
|
49
|
+
return documents;
|
|
50
|
+
}
|
|
51
|
+
catch (error) {
|
|
52
|
+
this.logger.error(`Failed to search documents: ${error instanceof Error ? error.message : String(error)}`, {
|
|
53
|
+
query,
|
|
54
|
+
searchType,
|
|
55
|
+
knowledgeBaseIds,
|
|
56
|
+
error: error instanceof Error && "response" in error
|
|
57
|
+
? ((_a = error.response) === null || _a === void 0 ? void 0 : _a.data) || error.message
|
|
58
|
+
: String(error),
|
|
59
|
+
});
|
|
60
|
+
return [];
|
|
61
|
+
}
|
|
62
|
+
}
|
|
63
|
+
async healthCheck() {
|
|
64
|
+
const url = `${this.baseUrl}/api/internal/retriever/health`;
|
|
65
|
+
try {
|
|
66
|
+
const response = await (0, rxjs_1.firstValueFrom)(this.httpService.get(url, {
|
|
67
|
+
timeout: 5000,
|
|
68
|
+
}));
|
|
69
|
+
return response.status === 200;
|
|
70
|
+
}
|
|
71
|
+
catch (error) {
|
|
72
|
+
this.logger.warn(`Retriever service is not available: ${error instanceof Error ? error.message : String(error)}`);
|
|
73
|
+
return false;
|
|
74
|
+
}
|
|
75
|
+
}
|
|
76
|
+
async getKnowledgeBaseInfo(knowledgeBaseId) {
|
|
77
|
+
const url = `${this.baseUrl}/api/internal/retriever/kb/${knowledgeBaseId}`;
|
|
78
|
+
try {
|
|
79
|
+
const response = await (0, rxjs_1.firstValueFrom)(this.httpService.get(url, {
|
|
80
|
+
timeout: this.timeout,
|
|
81
|
+
}));
|
|
82
|
+
return response.data;
|
|
83
|
+
}
|
|
84
|
+
catch (error) {
|
|
85
|
+
this.logger.error(`Failed to get KB info: ${error instanceof Error ? error.message : String(error)}`, {
|
|
86
|
+
knowledgeBaseId,
|
|
87
|
+
});
|
|
88
|
+
return null;
|
|
89
|
+
}
|
|
90
|
+
}
|
|
91
|
+
};
|
|
92
|
+
exports.RetrieverClient = RetrieverClient;
|
|
93
|
+
exports.RetrieverClient = RetrieverClient = RetrieverClient_1 = __decorate([
|
|
94
|
+
(0, common_1.Injectable)(),
|
|
95
|
+
__metadata("design:paramtypes", [axios_1.HttpService, Object])
|
|
96
|
+
], RetrieverClient);
|
|
97
|
+
//# sourceMappingURL=retriever.client.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"retriever.client.js","sourceRoot":"","sources":["../../src/retriever/retriever.client.ts"],"names":[],"mappings":";;;;;;;;;;;;;AAAA,2CAAoD;AACpD,yCAA4C;AAC5C,+BAAsC;AAe/B,IAAM,eAAe,uBAArB,MAAM,eAAe;IAM1B,YACmB,WAAwB,EACzC,MAA6B;QADZ,gBAAW,GAAX,WAAW,CAAa;QAN1B,WAAM,GAAG,IAAI,eAAM,CAAC,iBAAe,CAAC,IAAI,CAAC,CAAC;QASzD,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QAC9B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC;QACxC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,CAAC,CAAC;IACrC,CAAC;IAKD,KAAK,CAAC,MAAM,CACV,KAAa,EACb,UAA+B,EAC/B,gBAA0B,EAC1B,OAIC;;QAED,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,gCAAgC,CAAC;QAE5D,MAAM,OAAO,GAAG;YACd,KAAK;YACL,UAAU;YACV,gBAAgB;YAChB,OAAO,EAAE;gBACP,CAAC,EAAE,CAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,CAAC,KAAI,EAAE;gBACnB,SAAS,EAAE,CAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,SAAS,KAAI,GAAG;gBACpC,QAAQ,EAAE,CAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,QAAQ,KAAI,EAAE;aAClC;SACF,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,wBAAwB,KAAK,EAAE,EAAE;YACjD,UAAU;YACV,gBAAgB;YAChB,OAAO,EAAE,OAAO,CAAC,OAAO;SACzB,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAA,qBAAc,EACnC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAkC,GAAG,EAAE,OAAO,EAAE;gBACnE,OAAO,EAAE,IAAI,CAAC,OAAO;aACtB,CAAC,CACH,CAAC;YAEF,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,CAAC,SAAS,IAAI,EAAE,CAAC;YAEhD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,SAAS,CAAC,MAAM,YAAY,CAAC,CAAC;YAEzD,OAAO,SAAS,CAAC;QACnB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,+BAA+B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EACvF;gBACE,KAAK;gBACL,UAAU;gBACV,gBAAgB;gBAChB,KAAK,EACH,KAAK,YAAY,KAAK,IAAI,UAAU,IAAI,KAAK;oBAC3C,CAAC,CAAC,CAAA,MAAC,KAAa,CAAC,QAAQ,0CAAE,IAAI,KAAI,KAAK,CAAC,OAAO;oBAChD,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aACpB,CACF,CAAC;YAIF,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAKD,KAAK,CAAC,WAAW;QACf,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,gCAAgC,CAAC;QAE5D,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAA,qBAAc,EACnC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,EAAE;gBACxB,OAAO,EAAE,IAAI;aACd,CAAC,CACH,CAAC;YAEF,OAAO,QAAQ,CAAC,MAAM,KAAK,GAAG,CAAC;QACjC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,uCAAuC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAChG,CAAC;YACF,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAKD,KAAK,CAAC,oBAAoB,CAAC,eAAuB;QAMhD,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,8BAA8B,eAAe,EAAE,CAAC;QAE3E,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAA,qBAAc,EACnC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,EAAE;gBACxB,OAAO,EAAE,IAAI,CAAC,OAAO;aACtB,CAAC,CACH,CAAC;YAEF,OAAO,QAAQ,CAAC,IAAI,CAAC;QACvB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,0BAA0B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EAClF;gBACE,eAAe;aAChB,CACF,CAAC;YAEF,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;CACF,CAAA;AAnIY,0CAAe;0BAAf,eAAe;IAD3B,IAAA,mBAAU,GAAE;qCAQqB,mBAAW;GAPhC,eAAe,CAmI3B"}
|
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
import { RetrieverSearchType } from "../shared-types";
|
|
2
|
+
import { CustomDocument, RetrieveQueryOptions, KnowledgeBaseInfo } from "./types";
|
|
3
|
+
export interface RetrieverConfig {
|
|
4
|
+
apiUrl?: string;
|
|
5
|
+
internalToken?: string;
|
|
6
|
+
timeout?: number;
|
|
7
|
+
retries?: number;
|
|
8
|
+
}
|
|
9
|
+
export declare class RetrieverService {
|
|
10
|
+
private readonly logger;
|
|
11
|
+
private readonly apiUrl;
|
|
12
|
+
private readonly internalToken;
|
|
13
|
+
private readonly timeout;
|
|
14
|
+
private readonly retries;
|
|
15
|
+
constructor(config?: RetrieverConfig);
|
|
16
|
+
private makeRequest;
|
|
17
|
+
healthCheck(): Promise<boolean>;
|
|
18
|
+
search(query: string, searchType: RetrieverSearchType, knowledgeBaseIds: string[], options?: RetrieveQueryOptions): Promise<CustomDocument[]>;
|
|
19
|
+
getDocumentsByIds(ids: string[]): Promise<CustomDocument[]>;
|
|
20
|
+
knowledgeBaseExists(knowledgeBaseId: string): Promise<boolean>;
|
|
21
|
+
getKnowledgeBaseInfo(knowledgeBaseId: string): Promise<KnowledgeBaseInfo | null>;
|
|
22
|
+
textSearch(query: string, knowledgeBaseIds: string[], limit?: number): Promise<CustomDocument[]>;
|
|
23
|
+
getAvailableKnowledgeBases(): Promise<KnowledgeBaseInfo[]>;
|
|
24
|
+
check(): string;
|
|
25
|
+
}
|