@compresr/sdk 1.3.1 → 1.5.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/LICENSE +201 -0
- package/README.md +237 -173
- package/dist/cjs/agents/engine.d.ts +121 -0
- package/dist/cjs/agents/engine.d.ts.map +1 -0
- package/dist/cjs/agents/engine.js +517 -0
- package/dist/cjs/agents/engine.js.map +1 -0
- package/dist/cjs/agents/facades/anthropic.d.ts +26 -0
- package/dist/cjs/agents/facades/anthropic.d.ts.map +1 -0
- package/dist/cjs/agents/facades/anthropic.js +46 -0
- package/dist/cjs/agents/facades/anthropic.js.map +1 -0
- package/dist/cjs/agents/facades/index.d.ts +4 -0
- package/dist/cjs/agents/facades/index.d.ts.map +1 -0
- package/dist/cjs/agents/facades/index.js +10 -0
- package/dist/cjs/agents/facades/index.js.map +1 -0
- package/dist/cjs/agents/facades/native.d.ts +29 -0
- package/dist/cjs/agents/facades/native.d.ts.map +1 -0
- package/dist/cjs/agents/facades/native.js +27 -0
- package/dist/cjs/agents/facades/native.js.map +1 -0
- package/dist/cjs/agents/facades/openai.d.ts +32 -0
- package/dist/cjs/agents/facades/openai.d.ts.map +1 -0
- package/dist/cjs/agents/facades/openai.js +55 -0
- package/dist/cjs/agents/facades/openai.js.map +1 -0
- package/dist/cjs/agents/index.d.ts +23 -0
- package/dist/cjs/agents/index.d.ts.map +1 -0
- package/dist/cjs/agents/index.js +37 -0
- package/dist/cjs/agents/index.js.map +1 -0
- package/dist/cjs/agents/normalized.d.ts +59 -0
- package/dist/cjs/agents/normalized.d.ts.map +1 -0
- package/dist/cjs/agents/normalized.js +35 -0
- package/dist/cjs/agents/normalized.js.map +1 -0
- package/dist/cjs/agents/schemas/anthropic.d.ts +46 -0
- package/dist/cjs/agents/schemas/anthropic.d.ts.map +1 -0
- package/dist/cjs/agents/schemas/anthropic.js +55 -0
- package/dist/cjs/agents/schemas/anthropic.js.map +1 -0
- package/dist/cjs/agents/schemas/index.d.ts +3 -0
- package/dist/cjs/agents/schemas/index.d.ts.map +1 -0
- package/dist/cjs/agents/schemas/index.js +8 -0
- package/dist/cjs/agents/schemas/index.js.map +1 -0
- package/dist/cjs/agents/schemas/openai.d.ts +51 -0
- package/dist/cjs/agents/schemas/openai.d.ts.map +1 -0
- package/dist/cjs/agents/schemas/openai.js +60 -0
- package/dist/cjs/agents/schemas/openai.js.map +1 -0
- package/dist/cjs/agents/tools/index.d.ts +2 -0
- package/dist/cjs/agents/tools/index.d.ts.map +1 -0
- package/dist/cjs/agents/tools/index.js +7 -0
- package/dist/cjs/agents/tools/index.js.map +1 -0
- package/dist/cjs/agents/tools/web-search.d.ts +56 -0
- package/dist/cjs/agents/tools/web-search.d.ts.map +1 -0
- package/dist/cjs/agents/tools/web-search.js +128 -0
- package/dist/cjs/agents/tools/web-search.js.map +1 -0
- package/dist/cjs/clients/compression.d.ts +39 -147
- package/dist/cjs/clients/compression.d.ts.map +1 -1
- package/dist/cjs/clients/compression.js +114 -194
- package/dist/cjs/clients/compression.js.map +1 -1
- package/dist/cjs/clients/index.d.ts +1 -1
- package/dist/cjs/clients/index.d.ts.map +1 -1
- package/dist/cjs/clients/index.js.map +1 -1
- package/dist/cjs/config/constants.d.ts +3 -28
- package/dist/cjs/config/constants.d.ts.map +1 -1
- package/dist/cjs/config/constants.js +4 -43
- package/dist/cjs/config/constants.js.map +1 -1
- package/dist/cjs/config/endpoints.d.ts +3 -15
- package/dist/cjs/config/endpoints.d.ts.map +1 -1
- package/dist/cjs/config/endpoints.js +3 -20
- package/dist/cjs/config/endpoints.js.map +1 -1
- package/dist/cjs/config/index.d.ts +1 -1
- package/dist/cjs/config/index.d.ts.map +1 -1
- package/dist/cjs/config/index.js +1 -7
- package/dist/cjs/config/index.js.map +1 -1
- package/dist/cjs/errors/exceptions.d.ts.map +1 -1
- package/dist/cjs/errors/exceptions.js +0 -6
- package/dist/cjs/errors/exceptions.js.map +1 -1
- package/dist/cjs/http/client.d.ts +5 -17
- package/dist/cjs/http/client.d.ts.map +1 -1
- package/dist/cjs/http/client.js +110 -115
- package/dist/cjs/http/client.js.map +1 -1
- package/dist/cjs/index.d.ts +7 -15
- package/dist/cjs/index.d.ts.map +1 -1
- package/dist/cjs/index.js +22 -25
- package/dist/cjs/index.js.map +1 -1
- package/dist/cjs/integrations/_shared/client.d.ts +13 -0
- package/dist/cjs/integrations/_shared/client.d.ts.map +1 -0
- package/dist/cjs/integrations/_shared/client.js +21 -0
- package/dist/cjs/integrations/_shared/client.js.map +1 -0
- package/dist/cjs/integrations/_shared/compress.d.ts +28 -0
- package/dist/cjs/integrations/_shared/compress.d.ts.map +1 -0
- package/dist/cjs/integrations/_shared/compress.js +38 -0
- package/dist/cjs/integrations/_shared/compress.js.map +1 -0
- package/dist/cjs/integrations/_shared/errors.d.ts +20 -0
- package/dist/cjs/integrations/_shared/errors.d.ts.map +1 -0
- package/dist/cjs/integrations/_shared/errors.js +49 -0
- package/dist/cjs/integrations/_shared/errors.js.map +1 -0
- package/dist/cjs/integrations/_shared/filters.d.ts +20 -0
- package/dist/cjs/integrations/_shared/filters.d.ts.map +1 -0
- package/dist/cjs/integrations/_shared/filters.js +33 -0
- package/dist/cjs/integrations/_shared/filters.js.map +1 -0
- package/dist/cjs/integrations/_shared/index.d.ts +7 -0
- package/dist/cjs/integrations/_shared/index.d.ts.map +1 -0
- package/dist/cjs/integrations/_shared/index.js +26 -0
- package/dist/cjs/integrations/_shared/index.js.map +1 -0
- package/dist/cjs/integrations/_shared/query.d.ts +48 -0
- package/dist/cjs/integrations/_shared/query.d.ts.map +1 -0
- package/dist/cjs/integrations/_shared/query.js +141 -0
- package/dist/cjs/integrations/_shared/query.js.map +1 -0
- package/dist/cjs/integrations/_shared/tokens.d.ts +20 -0
- package/dist/cjs/integrations/_shared/tokens.d.ts.map +1 -0
- package/dist/cjs/integrations/_shared/tokens.js +26 -0
- package/dist/cjs/integrations/_shared/tokens.js.map +1 -0
- package/dist/cjs/integrations/langchain/index.d.ts +10 -0
- package/dist/cjs/integrations/langchain/index.d.ts.map +1 -0
- package/dist/cjs/integrations/langchain/index.js +18 -0
- package/dist/cjs/integrations/langchain/index.js.map +1 -0
- package/dist/cjs/integrations/langchain/middleware.d.ts +94 -0
- package/dist/cjs/integrations/langchain/middleware.d.ts.map +1 -0
- package/dist/cjs/integrations/langchain/middleware.js +273 -0
- package/dist/cjs/integrations/langchain/middleware.js.map +1 -0
- package/dist/cjs/integrations/langchain/retriever.d.ts +54 -0
- package/dist/cjs/integrations/langchain/retriever.d.ts.map +1 -0
- package/dist/cjs/integrations/langchain/retriever.js +118 -0
- package/dist/cjs/integrations/langchain/retriever.js.map +1 -0
- package/dist/cjs/integrations/langchain/wrappers.d.ts +39 -0
- package/dist/cjs/integrations/langchain/wrappers.d.ts.map +1 -0
- package/dist/cjs/integrations/langchain/wrappers.js +88 -0
- package/dist/cjs/integrations/langchain/wrappers.js.map +1 -0
- package/dist/cjs/integrations/langgraph/checkpoint.d.ts +49 -0
- package/dist/cjs/integrations/langgraph/checkpoint.d.ts.map +1 -0
- package/dist/cjs/integrations/langgraph/checkpoint.js +79 -0
- package/dist/cjs/integrations/langgraph/checkpoint.js.map +1 -0
- package/dist/cjs/integrations/langgraph/handoff.d.ts +44 -0
- package/dist/cjs/integrations/langgraph/handoff.d.ts.map +1 -0
- package/dist/cjs/integrations/langgraph/handoff.js +86 -0
- package/dist/cjs/integrations/langgraph/handoff.js.map +1 -0
- package/dist/cjs/integrations/langgraph/index.d.ts +18 -0
- package/dist/cjs/integrations/langgraph/index.d.ts.map +1 -0
- package/dist/cjs/integrations/langgraph/index.js +28 -0
- package/dist/cjs/integrations/langgraph/index.js.map +1 -0
- package/dist/cjs/integrations/langgraph/nodes.d.ts +34 -0
- package/dist/cjs/integrations/langgraph/nodes.d.ts.map +1 -0
- package/dist/cjs/integrations/langgraph/nodes.js +54 -0
- package/dist/cjs/integrations/langgraph/nodes.js.map +1 -0
- package/dist/cjs/integrations/langgraph/store.d.ts +76 -0
- package/dist/cjs/integrations/langgraph/store.d.ts.map +1 -0
- package/dist/cjs/integrations/langgraph/store.js +121 -0
- package/dist/cjs/integrations/langgraph/store.js.map +1 -0
- package/dist/cjs/integrations/llamaindex/index.d.ts +10 -0
- package/dist/cjs/integrations/llamaindex/index.d.ts.map +1 -0
- package/dist/cjs/integrations/llamaindex/index.js +15 -0
- package/dist/cjs/integrations/llamaindex/index.js.map +1 -0
- package/dist/cjs/integrations/llamaindex/memory.d.ts +74 -0
- package/dist/cjs/integrations/llamaindex/memory.d.ts.map +1 -0
- package/dist/cjs/integrations/llamaindex/memory.js +95 -0
- package/dist/cjs/integrations/llamaindex/memory.js.map +1 -0
- package/dist/cjs/integrations/llamaindex/postprocessor.d.ts +66 -0
- package/dist/cjs/integrations/llamaindex/postprocessor.d.ts.map +1 -0
- package/dist/cjs/integrations/llamaindex/postprocessor.js +172 -0
- package/dist/cjs/integrations/llamaindex/postprocessor.js.map +1 -0
- package/dist/cjs/integrations/llamaindex/wrappers.d.ts +38 -0
- package/dist/cjs/integrations/llamaindex/wrappers.d.ts.map +1 -0
- package/dist/cjs/integrations/llamaindex/wrappers.js +46 -0
- package/dist/cjs/integrations/llamaindex/wrappers.js.map +1 -0
- package/dist/cjs/logger.d.ts +18 -0
- package/dist/cjs/logger.d.ts.map +1 -0
- package/dist/cjs/logger.js +41 -0
- package/dist/cjs/logger.js.map +1 -0
- package/dist/cjs/schemas/compression.d.ts +33 -103
- package/dist/cjs/schemas/compression.d.ts.map +1 -1
- package/dist/cjs/schemas/compression.js +6 -59
- package/dist/cjs/schemas/compression.js.map +1 -1
- package/dist/cjs/schemas/index.d.ts +1 -4
- package/dist/cjs/schemas/index.d.ts.map +1 -1
- package/dist/cjs/schemas/index.js +1 -8
- package/dist/cjs/schemas/index.js.map +1 -1
- package/dist/cjs/version.d.ts +1 -1
- package/dist/cjs/version.js +1 -1
- package/dist/esm/agents/engine.d.ts +121 -0
- package/dist/esm/agents/engine.d.ts.map +1 -0
- package/dist/esm/agents/engine.js +478 -0
- package/dist/esm/agents/engine.js.map +1 -0
- package/dist/esm/agents/facades/anthropic.d.ts +26 -0
- package/dist/esm/agents/facades/anthropic.d.ts.map +1 -0
- package/dist/esm/agents/facades/anthropic.js +43 -0
- package/dist/esm/agents/facades/anthropic.js.map +1 -0
- package/dist/esm/agents/facades/index.d.ts +4 -0
- package/dist/esm/agents/facades/index.d.ts.map +1 -0
- package/dist/esm/agents/facades/index.js +4 -0
- package/dist/esm/agents/facades/index.js.map +1 -0
- package/dist/esm/agents/facades/native.d.ts +29 -0
- package/dist/esm/agents/facades/native.d.ts.map +1 -0
- package/dist/esm/agents/facades/native.js +24 -0
- package/dist/esm/agents/facades/native.js.map +1 -0
- package/dist/esm/agents/facades/openai.d.ts +32 -0
- package/dist/esm/agents/facades/openai.d.ts.map +1 -0
- package/dist/esm/agents/facades/openai.js +52 -0
- package/dist/esm/agents/facades/openai.js.map +1 -0
- package/dist/esm/agents/index.d.ts +23 -0
- package/dist/esm/agents/index.d.ts.map +1 -0
- package/dist/esm/agents/index.js +23 -0
- package/dist/esm/agents/index.js.map +1 -0
- package/dist/esm/agents/normalized.d.ts +59 -0
- package/dist/esm/agents/normalized.d.ts.map +1 -0
- package/dist/esm/agents/normalized.js +31 -0
- package/dist/esm/agents/normalized.js.map +1 -0
- package/dist/esm/agents/schemas/anthropic.d.ts +46 -0
- package/dist/esm/agents/schemas/anthropic.d.ts.map +1 -0
- package/dist/esm/agents/schemas/anthropic.js +52 -0
- package/dist/esm/agents/schemas/anthropic.js.map +1 -0
- package/dist/esm/agents/schemas/index.d.ts +3 -0
- package/dist/esm/agents/schemas/index.d.ts.map +1 -0
- package/dist/esm/agents/schemas/index.js +3 -0
- package/dist/esm/agents/schemas/index.js.map +1 -0
- package/dist/esm/agents/schemas/openai.d.ts +51 -0
- package/dist/esm/agents/schemas/openai.d.ts.map +1 -0
- package/dist/esm/agents/schemas/openai.js +57 -0
- package/dist/esm/agents/schemas/openai.js.map +1 -0
- package/dist/esm/agents/tools/index.d.ts +2 -0
- package/dist/esm/agents/tools/index.d.ts.map +1 -0
- package/dist/esm/agents/tools/index.js +2 -0
- package/dist/esm/agents/tools/index.js.map +1 -0
- package/dist/esm/agents/tools/web-search.d.ts +56 -0
- package/dist/esm/agents/tools/web-search.d.ts.map +1 -0
- package/dist/esm/agents/tools/web-search.js +91 -0
- package/dist/esm/agents/tools/web-search.js.map +1 -0
- package/dist/esm/clients/compression.d.ts +39 -147
- package/dist/esm/clients/compression.d.ts.map +1 -1
- package/dist/esm/clients/compression.js +117 -197
- package/dist/esm/clients/compression.js.map +1 -1
- package/dist/esm/clients/index.d.ts +1 -1
- package/dist/esm/clients/index.d.ts.map +1 -1
- package/dist/esm/clients/index.js.map +1 -1
- package/dist/esm/config/constants.d.ts +3 -28
- package/dist/esm/config/constants.d.ts.map +1 -1
- package/dist/esm/config/constants.js +3 -42
- package/dist/esm/config/constants.js.map +1 -1
- package/dist/esm/config/endpoints.d.ts +3 -15
- package/dist/esm/config/endpoints.d.ts.map +1 -1
- package/dist/esm/config/endpoints.js +3 -20
- package/dist/esm/config/endpoints.js.map +1 -1
- package/dist/esm/config/index.d.ts +1 -1
- package/dist/esm/config/index.d.ts.map +1 -1
- package/dist/esm/config/index.js +1 -1
- package/dist/esm/config/index.js.map +1 -1
- package/dist/esm/errors/exceptions.d.ts.map +1 -1
- package/dist/esm/errors/exceptions.js +0 -6
- package/dist/esm/errors/exceptions.js.map +1 -1
- package/dist/esm/http/client.d.ts +5 -17
- package/dist/esm/http/client.d.ts.map +1 -1
- package/dist/esm/http/client.js +110 -115
- package/dist/esm/http/client.js.map +1 -1
- package/dist/esm/index.d.ts +7 -15
- package/dist/esm/index.d.ts.map +1 -1
- package/dist/esm/index.js +7 -13
- package/dist/esm/index.js.map +1 -1
- package/dist/esm/integrations/_shared/client.d.ts +13 -0
- package/dist/esm/integrations/_shared/client.d.ts.map +1 -0
- package/dist/esm/integrations/_shared/client.js +17 -0
- package/dist/esm/integrations/_shared/client.js.map +1 -0
- package/dist/esm/integrations/_shared/compress.d.ts +28 -0
- package/dist/esm/integrations/_shared/compress.d.ts.map +1 -0
- package/dist/esm/integrations/_shared/compress.js +35 -0
- package/dist/esm/integrations/_shared/compress.js.map +1 -0
- package/dist/esm/integrations/_shared/errors.d.ts +20 -0
- package/dist/esm/integrations/_shared/errors.d.ts.map +1 -0
- package/dist/esm/integrations/_shared/errors.js +44 -0
- package/dist/esm/integrations/_shared/errors.js.map +1 -0
- package/dist/esm/integrations/_shared/filters.d.ts +20 -0
- package/dist/esm/integrations/_shared/filters.d.ts.map +1 -0
- package/dist/esm/integrations/_shared/filters.js +30 -0
- package/dist/esm/integrations/_shared/filters.js.map +1 -0
- package/dist/esm/integrations/_shared/index.d.ts +7 -0
- package/dist/esm/integrations/_shared/index.d.ts.map +1 -0
- package/dist/esm/integrations/_shared/index.js +7 -0
- package/dist/esm/integrations/_shared/index.js.map +1 -0
- package/dist/esm/integrations/_shared/query.d.ts +48 -0
- package/dist/esm/integrations/_shared/query.d.ts.map +1 -0
- package/dist/esm/integrations/_shared/query.js +135 -0
- package/dist/esm/integrations/_shared/query.js.map +1 -0
- package/dist/esm/integrations/_shared/tokens.d.ts +20 -0
- package/dist/esm/integrations/_shared/tokens.d.ts.map +1 -0
- package/dist/esm/integrations/_shared/tokens.js +23 -0
- package/dist/esm/integrations/_shared/tokens.js.map +1 -0
- package/dist/esm/integrations/langchain/index.d.ts +10 -0
- package/dist/esm/integrations/langchain/index.d.ts.map +1 -0
- package/dist/esm/integrations/langchain/index.js +9 -0
- package/dist/esm/integrations/langchain/index.js.map +1 -0
- package/dist/esm/integrations/langchain/middleware.d.ts +94 -0
- package/dist/esm/integrations/langchain/middleware.d.ts.map +1 -0
- package/dist/esm/integrations/langchain/middleware.js +268 -0
- package/dist/esm/integrations/langchain/middleware.js.map +1 -0
- package/dist/esm/integrations/langchain/retriever.d.ts +54 -0
- package/dist/esm/integrations/langchain/retriever.d.ts.map +1 -0
- package/dist/esm/integrations/langchain/retriever.js +114 -0
- package/dist/esm/integrations/langchain/retriever.js.map +1 -0
- package/dist/esm/integrations/langchain/wrappers.d.ts +39 -0
- package/dist/esm/integrations/langchain/wrappers.d.ts.map +1 -0
- package/dist/esm/integrations/langchain/wrappers.js +84 -0
- package/dist/esm/integrations/langchain/wrappers.js.map +1 -0
- package/dist/esm/integrations/langgraph/checkpoint.d.ts +49 -0
- package/dist/esm/integrations/langgraph/checkpoint.d.ts.map +1 -0
- package/dist/esm/integrations/langgraph/checkpoint.js +75 -0
- package/dist/esm/integrations/langgraph/checkpoint.js.map +1 -0
- package/dist/esm/integrations/langgraph/handoff.d.ts +44 -0
- package/dist/esm/integrations/langgraph/handoff.d.ts.map +1 -0
- package/dist/esm/integrations/langgraph/handoff.js +83 -0
- package/dist/esm/integrations/langgraph/handoff.js.map +1 -0
- package/dist/esm/integrations/langgraph/index.d.ts +18 -0
- package/dist/esm/integrations/langgraph/index.d.ts.map +1 -0
- package/dist/esm/integrations/langgraph/index.js +17 -0
- package/dist/esm/integrations/langgraph/index.js.map +1 -0
- package/dist/esm/integrations/langgraph/nodes.d.ts +34 -0
- package/dist/esm/integrations/langgraph/nodes.d.ts.map +1 -0
- package/dist/esm/integrations/langgraph/nodes.js +51 -0
- package/dist/esm/integrations/langgraph/nodes.js.map +1 -0
- package/dist/esm/integrations/langgraph/store.d.ts +76 -0
- package/dist/esm/integrations/langgraph/store.d.ts.map +1 -0
- package/dist/esm/integrations/langgraph/store.js +117 -0
- package/dist/esm/integrations/langgraph/store.js.map +1 -0
- package/dist/esm/integrations/llamaindex/index.d.ts +10 -0
- package/dist/esm/integrations/llamaindex/index.d.ts.map +1 -0
- package/dist/esm/integrations/llamaindex/index.js +9 -0
- package/dist/esm/integrations/llamaindex/index.js.map +1 -0
- package/dist/esm/integrations/llamaindex/memory.d.ts +74 -0
- package/dist/esm/integrations/llamaindex/memory.d.ts.map +1 -0
- package/dist/esm/integrations/llamaindex/memory.js +91 -0
- package/dist/esm/integrations/llamaindex/memory.js.map +1 -0
- package/dist/esm/integrations/llamaindex/postprocessor.d.ts +66 -0
- package/dist/esm/integrations/llamaindex/postprocessor.d.ts.map +1 -0
- package/dist/esm/integrations/llamaindex/postprocessor.js +168 -0
- package/dist/esm/integrations/llamaindex/postprocessor.js.map +1 -0
- package/dist/esm/integrations/llamaindex/wrappers.d.ts +38 -0
- package/dist/esm/integrations/llamaindex/wrappers.d.ts.map +1 -0
- package/dist/esm/integrations/llamaindex/wrappers.js +43 -0
- package/dist/esm/integrations/llamaindex/wrappers.js.map +1 -0
- package/dist/esm/logger.d.ts +18 -0
- package/dist/esm/logger.d.ts.map +1 -0
- package/dist/esm/logger.js +37 -0
- package/dist/esm/logger.js.map +1 -0
- package/dist/esm/schemas/compression.d.ts +33 -103
- package/dist/esm/schemas/compression.d.ts.map +1 -1
- package/dist/esm/schemas/compression.js +5 -58
- package/dist/esm/schemas/compression.js.map +1 -1
- package/dist/esm/schemas/index.d.ts +1 -4
- package/dist/esm/schemas/index.d.ts.map +1 -1
- package/dist/esm/schemas/index.js +1 -6
- package/dist/esm/schemas/index.js.map +1 -1
- package/dist/esm/version.d.ts +1 -1
- package/dist/esm/version.js +1 -1
- package/package.json +91 -6
|
@@ -0,0 +1,75 @@
|
|
|
1
|
+
import { buildClient, compressSafe, DEFAULT_MODEL, DEFAULT_POLICY, DEFAULT_RATIO, estimateTokens, } from '../_shared/index.js';
|
|
2
|
+
const SENTINEL_KEY = '__compresr__';
|
|
3
|
+
/**
|
|
4
|
+
* Drop-in for LangGraph.js checkpoint serializers. Calls
|
|
5
|
+
* `dumpsTyped(obj) → [tag, bytes]`; recursively rewrites long strings
|
|
6
|
+
* into a `{ __compresr__: true, v: <compressed> }` sentinel before
|
|
7
|
+
* JSON-encoding. `loadsTyped` returns the payload as-is (sentinels and all).
|
|
8
|
+
*/
|
|
9
|
+
export class CompresrCheckpointSerializer {
|
|
10
|
+
compresr;
|
|
11
|
+
compressionModel;
|
|
12
|
+
query;
|
|
13
|
+
targetCompressionRatio;
|
|
14
|
+
minTokens;
|
|
15
|
+
coarse;
|
|
16
|
+
fields;
|
|
17
|
+
onError;
|
|
18
|
+
constructor(options) {
|
|
19
|
+
const { apiKey, client, baseUrl, compressionModel = DEFAULT_MODEL, query, targetCompressionRatio = DEFAULT_RATIO, minTokens = 500, coarse, fields, onError = DEFAULT_POLICY, } = options;
|
|
20
|
+
this.compresr =
|
|
21
|
+
client ?? buildClient({ apiKey, baseUrl, caller: 'CompresrCheckpointSerializer' });
|
|
22
|
+
this.compressionModel = compressionModel;
|
|
23
|
+
this.query = query;
|
|
24
|
+
this.targetCompressionRatio = targetCompressionRatio;
|
|
25
|
+
this.minTokens = Math.max(1, minTokens);
|
|
26
|
+
this.coarse = coarse;
|
|
27
|
+
this.fields = fields;
|
|
28
|
+
this.onError = onError;
|
|
29
|
+
}
|
|
30
|
+
async dumpsTyped(obj) {
|
|
31
|
+
const transformed = await this.walkCompress(obj, undefined);
|
|
32
|
+
const encoded = new TextEncoder().encode(JSON.stringify(transformed));
|
|
33
|
+
return ['json', encoded];
|
|
34
|
+
}
|
|
35
|
+
loadsTyped([_tag, bytes]) {
|
|
36
|
+
return JSON.parse(new TextDecoder().decode(bytes));
|
|
37
|
+
}
|
|
38
|
+
async walkCompress(value, parentKey) {
|
|
39
|
+
if (typeof value === 'string') {
|
|
40
|
+
if (!this.shouldCompress(value, parentKey))
|
|
41
|
+
return value;
|
|
42
|
+
return await this.wrapCompressed(value, parentKey);
|
|
43
|
+
}
|
|
44
|
+
if (Array.isArray(value)) {
|
|
45
|
+
return Promise.all(value.map((v) => this.walkCompress(v, parentKey)));
|
|
46
|
+
}
|
|
47
|
+
if (value !== null && typeof value === 'object') {
|
|
48
|
+
const entries = await Promise.all(Object.entries(value).map(async ([k, v]) => [k, await this.walkCompress(v, k)]));
|
|
49
|
+
return Object.fromEntries(entries);
|
|
50
|
+
}
|
|
51
|
+
return value;
|
|
52
|
+
}
|
|
53
|
+
shouldCompress(text, parentKey) {
|
|
54
|
+
if (this.fields !== undefined && (parentKey === undefined || !this.fields.has(parentKey))) {
|
|
55
|
+
return false;
|
|
56
|
+
}
|
|
57
|
+
return estimateTokens(text) >= this.minTokens;
|
|
58
|
+
}
|
|
59
|
+
async wrapCompressed(text, parentKey) {
|
|
60
|
+
const compressed = await compressSafe(this.compresr, {
|
|
61
|
+
context: text,
|
|
62
|
+
query: this.query ?? parentKey ?? 'stored content',
|
|
63
|
+
compressionModel: this.compressionModel,
|
|
64
|
+
targetCompressionRatio: this.targetCompressionRatio,
|
|
65
|
+
...(this.coarse !== undefined ? { coarse: this.coarse } : {}),
|
|
66
|
+
minTokens: 1,
|
|
67
|
+
onError: this.onError,
|
|
68
|
+
contextLabel: `checkpoint:${parentKey ?? '?'}`,
|
|
69
|
+
});
|
|
70
|
+
if (compressed === text)
|
|
71
|
+
return text;
|
|
72
|
+
return { [SENTINEL_KEY]: true, v: compressed };
|
|
73
|
+
}
|
|
74
|
+
}
|
|
75
|
+
//# sourceMappingURL=checkpoint.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"checkpoint.js","sourceRoot":"","sources":["../../../../src/integrations/langgraph/checkpoint.ts"],"names":[],"mappings":"AASA,OAAO,EACL,WAAW,EACX,YAAY,EACZ,aAAa,EACb,cAAc,EACd,aAAa,EAEb,cAAc,GACf,MAAM,qBAAqB,CAAC;AAE7B,MAAM,YAAY,GAAG,cAAc,CAAC;AAmBpC;;;;;GAKG;AACH,MAAM,OAAO,4BAA4B;IACtB,QAAQ,CAAoB;IAC5B,gBAAgB,CAAS;IACzB,KAAK,CAAqB;IAC1B,sBAAsB,CAAS;IAC/B,SAAS,CAAS;IAClB,MAAM,CAAsB;IAC5B,MAAM,CAAkC;IACxC,OAAO,CAAc;IAEtC,YAAY,OAA4C;QACtD,MAAM,EACJ,MAAM,EACN,MAAM,EACN,OAAO,EACP,gBAAgB,GAAG,aAAa,EAChC,KAAK,EACL,sBAAsB,GAAG,aAAa,EACtC,SAAS,GAAG,GAAG,EACf,MAAM,EACN,MAAM,EACN,OAAO,GAAG,cAAc,GACzB,GAAG,OAAO,CAAC;QAEZ,IAAI,CAAC,QAAQ;YACX,MAAM,IAAI,WAAW,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,8BAA8B,EAAE,CAAC,CAAC;QACrF,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,sBAAsB,GAAG,sBAAsB,CAAC;QACrD,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;QACxC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,GAAY;QAC3B,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QAC5D,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,CAAC;QACtE,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC3B,CAAC;IAED,UAAU,CAAC,CAAC,IAAI,EAAE,KAAK,CAAuB;QAC5C,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IACrD,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,KAAc,EAAE,SAA6B;QACtE,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,SAAS,CAAC;gBAAE,OAAO,KAAK,CAAC;YACzD,OAAO,MAAM,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QACrD,CAAC;QACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACzB,OAAO,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC;QACxE,CAAC;QACD,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAChD,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,MAAM,CAAC,OAAO,CAAC,KAAgC,CAAC,CAAC,GAAG,CAClD,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC,CAAC,CAAU,CAC9D,CACF,CAAC;YACF,OAAO,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QACrC,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,cAAc,CAAC,IAAY,EAAE,SAA6B;QAChE,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS,IAAI,CAAC,SAAS,KAAK,SAAS,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC;YAC1F,OAAO,KAAK,CAAC;QACf,CAAC;QACD,OAAO,cAAc,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,IAAY,EAAE,SAA6B;QACtE,MAAM,UAAU,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,QAAQ,EAAE;YACnD,OAAO,EAAE,IAAI;YACb,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,SAAS,IAAI,gBAAgB;YAClD,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,sBAAsB,EAAE,IAAI,CAAC,sBAAsB;YACnD,GAAG,CAAC,IAAI,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7D,SAAS,EAAE,CAAC;YACZ,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,YAAY,EAAE,cAAc,SAAS,IAAI,GAAG,EAAE;SAC/C,CAAC,CAAC;QACH,IAAI,UAAU,KAAK,IAAI;YAAE,OAAO,IAAI,CAAC;QACrC,OAAO,EAAE,CAAC,YAAY,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC;IACjD,CAAC;CACF"}
|
|
@@ -0,0 +1,44 @@
|
|
|
1
|
+
import { Command } from '@langchain/langgraph';
|
|
2
|
+
import { z } from 'zod';
|
|
3
|
+
import type { CompressionClient } from '../../clients/compression.js';
|
|
4
|
+
import { type ErrorPolicy } from '../_shared/index.js';
|
|
5
|
+
export interface CompresrHandoffToolOptions {
|
|
6
|
+
description?: string;
|
|
7
|
+
apiKey?: string;
|
|
8
|
+
client?: CompressionClient;
|
|
9
|
+
baseUrl?: string;
|
|
10
|
+
compressionModel?: string;
|
|
11
|
+
targetCompressionRatio?: number;
|
|
12
|
+
minTokens?: number;
|
|
13
|
+
coarse?: boolean;
|
|
14
|
+
onError?: ErrorPolicy;
|
|
15
|
+
}
|
|
16
|
+
/**
|
|
17
|
+
* Build a handoff tool for `agentName`.
|
|
18
|
+
*
|
|
19
|
+
* @example
|
|
20
|
+
* const supervisor = createAgent({
|
|
21
|
+
* model,
|
|
22
|
+
* tools: [
|
|
23
|
+
* compresrHandoffTool('researcher', { apiKey: process.env.COMPRESR_API_KEY }),
|
|
24
|
+
* compresrHandoffTool('writer', { apiKey: process.env.COMPRESR_API_KEY }),
|
|
25
|
+
* ],
|
|
26
|
+
* });
|
|
27
|
+
*/
|
|
28
|
+
export declare function compresrHandoffTool(agentName: string, options?: CompresrHandoffToolOptions): import("@langchain/core/tools").DynamicStructuredTool<z.ZodObject<{
|
|
29
|
+
task_description: z.ZodString;
|
|
30
|
+
context: z.ZodDefault<z.ZodOptional<z.ZodString>>;
|
|
31
|
+
}, "strip", z.ZodTypeAny, {
|
|
32
|
+
context: string;
|
|
33
|
+
task_description: string;
|
|
34
|
+
}, {
|
|
35
|
+
task_description: string;
|
|
36
|
+
context?: string | undefined;
|
|
37
|
+
}>, {
|
|
38
|
+
task_description: string;
|
|
39
|
+
context?: string;
|
|
40
|
+
}, {
|
|
41
|
+
task_description: string;
|
|
42
|
+
context?: string | undefined;
|
|
43
|
+
}, Command<unknown, Record<string, unknown>, string>, unknown, string>;
|
|
44
|
+
//# sourceMappingURL=handoff.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"handoff.d.ts","sourceRoot":"","sources":["../../../../src/integrations/langgraph/handoff.ts"],"names":[],"mappings":"AAUA,OAAO,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAC;AAC/C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AACtE,OAAO,EAOL,KAAK,WAAW,EACjB,MAAM,qBAAqB,CAAC;AAE7B,MAAM,WAAW,0BAA0B;IACzC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,iBAAiB,CAAC;IAC3B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,sBAAsB,CAAC,EAAE,MAAM,CAAC;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,OAAO,CAAC,EAAE,WAAW,CAAC;CACvB;AAED;;;;;;;;;;;GAWG;AACH,wBAAgB,mBAAmB,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,GAAE,0BAA+B;;;;;;;;;;sBA6C9D,MAAM;cAAY,MAAM;;;;uEAyCxD"}
|
|
@@ -0,0 +1,83 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* `compresrHandoffTool` — supervisor → subagent handoff tool with
|
|
3
|
+
* compression baked in for LangGraph.js multi-agent systems.
|
|
4
|
+
*
|
|
5
|
+
* Returns a LangChain `DynamicStructuredTool` that emits a `Command`
|
|
6
|
+
* routing control to `agentName` with a compressed `task_description`
|
|
7
|
+
* and (optionally) compressed `context` field.
|
|
8
|
+
*/
|
|
9
|
+
import { ToolMessage } from '@langchain/core/messages';
|
|
10
|
+
import { tool } from '@langchain/core/tools';
|
|
11
|
+
import { Command } from '@langchain/langgraph';
|
|
12
|
+
import { z } from 'zod';
|
|
13
|
+
import { buildClient, compressSafe, DEFAULT_MIN_TOKENS, DEFAULT_MODEL, DEFAULT_POLICY, DEFAULT_RATIO, } from '../_shared/index.js';
|
|
14
|
+
/**
|
|
15
|
+
* Build a handoff tool for `agentName`.
|
|
16
|
+
*
|
|
17
|
+
* @example
|
|
18
|
+
* const supervisor = createAgent({
|
|
19
|
+
* model,
|
|
20
|
+
* tools: [
|
|
21
|
+
* compresrHandoffTool('researcher', { apiKey: process.env.COMPRESR_API_KEY }),
|
|
22
|
+
* compresrHandoffTool('writer', { apiKey: process.env.COMPRESR_API_KEY }),
|
|
23
|
+
* ],
|
|
24
|
+
* });
|
|
25
|
+
*/
|
|
26
|
+
export function compresrHandoffTool(agentName, options = {}) {
|
|
27
|
+
const { description, apiKey, client, baseUrl, compressionModel = DEFAULT_MODEL, targetCompressionRatio = DEFAULT_RATIO, minTokens = DEFAULT_MIN_TOKENS, coarse, onError = DEFAULT_POLICY, } = options;
|
|
28
|
+
const compresr = client ?? buildClient({ apiKey, baseUrl, caller: 'compresrHandoffTool' });
|
|
29
|
+
const toolName = `transfer_to_${agentName}`;
|
|
30
|
+
const toolDescription = description ??
|
|
31
|
+
`Hand off the task to the \`${agentName}\` agent. ` +
|
|
32
|
+
'Provide a clear task_description and any context the agent needs.';
|
|
33
|
+
const schema = z.object({
|
|
34
|
+
task_description: z.string().describe(`Task description for the ${agentName} agent.`),
|
|
35
|
+
context: z
|
|
36
|
+
.string()
|
|
37
|
+
.optional()
|
|
38
|
+
.default('')
|
|
39
|
+
.describe('Optional context excerpts to share with the agent.'),
|
|
40
|
+
});
|
|
41
|
+
async function maybeCompress(text, query, label) {
|
|
42
|
+
if (!text)
|
|
43
|
+
return text;
|
|
44
|
+
return await compressSafe(compresr, {
|
|
45
|
+
context: text,
|
|
46
|
+
query,
|
|
47
|
+
compressionModel,
|
|
48
|
+
targetCompressionRatio,
|
|
49
|
+
...(coarse !== undefined ? { coarse } : {}),
|
|
50
|
+
minTokens,
|
|
51
|
+
onError,
|
|
52
|
+
contextLabel: label,
|
|
53
|
+
});
|
|
54
|
+
}
|
|
55
|
+
return tool(async (input, runtime) => {
|
|
56
|
+
// latte_v1 requires a query — the task description itself is the natural
|
|
57
|
+
// one for the context excerpt (what the subagent will do with it).
|
|
58
|
+
const compressedTask = await maybeCompress(input.task_description, `task for ${agentName}`, 'handoff:task');
|
|
59
|
+
const ctx = input.context ?? '';
|
|
60
|
+
const compressedCtx = ctx
|
|
61
|
+
? await maybeCompress(ctx, input.task_description || `context for ${agentName}`, 'handoff:context')
|
|
62
|
+
: '';
|
|
63
|
+
const ack = new ToolMessage({
|
|
64
|
+
content: `Successfully transferred to ${agentName}.`,
|
|
65
|
+
tool_call_id: runtime.toolCallId ?? '',
|
|
66
|
+
name: toolName,
|
|
67
|
+
});
|
|
68
|
+
return new Command({
|
|
69
|
+
goto: agentName,
|
|
70
|
+
update: {
|
|
71
|
+
messages: [ack],
|
|
72
|
+
task_description: compressedTask,
|
|
73
|
+
context: compressedCtx,
|
|
74
|
+
},
|
|
75
|
+
graph: Command.PARENT,
|
|
76
|
+
});
|
|
77
|
+
}, {
|
|
78
|
+
name: toolName,
|
|
79
|
+
description: toolDescription,
|
|
80
|
+
schema,
|
|
81
|
+
});
|
|
82
|
+
}
|
|
83
|
+
//# sourceMappingURL=handoff.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"handoff.js","sourceRoot":"","sources":["../../../../src/integrations/langgraph/handoff.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AACH,OAAO,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AACvD,OAAO,EAAE,IAAI,EAAoB,MAAM,uBAAuB,CAAC;AAC/D,OAAO,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAC;AAC/C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,OAAO,EACL,WAAW,EACX,YAAY,EACZ,kBAAkB,EAClB,aAAa,EACb,cAAc,EACd,aAAa,GAEd,MAAM,qBAAqB,CAAC;AAc7B;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,mBAAmB,CAAC,SAAiB,EAAE,UAAsC,EAAE;IAC7F,MAAM,EACJ,WAAW,EACX,MAAM,EACN,MAAM,EACN,OAAO,EACP,gBAAgB,GAAG,aAAa,EAChC,sBAAsB,GAAG,aAAa,EACtC,SAAS,GAAG,kBAAkB,EAC9B,MAAM,EACN,OAAO,GAAG,cAAc,GACzB,GAAG,OAAO,CAAC;IAEZ,MAAM,QAAQ,GAAG,MAAM,IAAI,WAAW,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,qBAAqB,EAAE,CAAC,CAAC;IAC3F,MAAM,QAAQ,GAAG,eAAe,SAAS,EAAE,CAAC;IAC5C,MAAM,eAAe,GACnB,WAAW;QACX,8BAA8B,SAAS,YAAY;YACjD,mEAAmE,CAAC;IAExE,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC;QACtB,gBAAgB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,4BAA4B,SAAS,SAAS,CAAC;QACrF,OAAO,EAAE,CAAC;aACP,MAAM,EAAE;aACR,QAAQ,EAAE;aACV,OAAO,CAAC,EAAE,CAAC;aACX,QAAQ,CAAC,oDAAoD,CAAC;KAClE,CAAC,CAAC;IAEH,KAAK,UAAU,aAAa,CAAC,IAAY,EAAE,KAAa,EAAE,KAAa;QACrE,IAAI,CAAC,IAAI;YAAE,OAAO,IAAI,CAAC;QACvB,OAAO,MAAM,YAAY,CAAC,QAAQ,EAAE;YAClC,OAAO,EAAE,IAAI;YACb,KAAK;YACL,gBAAgB;YAChB,sBAAsB;YACtB,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3C,SAAS;YACT,OAAO;YACP,YAAY,EAAE,KAAK;SACpB,CAAC,CAAC;IACL,CAAC;IAED,OAAO,IAAI,CACT,KAAK,EACH,KAAqD,EACrD,OAAoB,EACF,EAAE;QACpB,yEAAyE;QACzE,mEAAmE;QACnE,MAAM,cAAc,GAAG,MAAM,aAAa,CACxC,KAAK,CAAC,gBAAgB,EACtB,YAAY,SAAS,EAAE,EACvB,cAAc,CACf,CAAC;QACF,MAAM,GAAG,GAAG,KAAK,CAAC,OAAO,IAAI,EAAE,CAAC;QAChC,MAAM,aAAa,GAAG,GAAG;YACvB,CAAC,CAAC,MAAM,aAAa,CACjB,GAAG,EACH,KAAK,CAAC,gBAAgB,IAAI,eAAe,SAAS,EAAE,EACpD,iBAAiB,CAClB;YACH,CAAC,CAAC,EAAE,CAAC;QAEP,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC;YAC1B,OAAO,EAAE,+BAA+B,SAAS,GAAG;YACpD,YAAY,EAAE,OAAO,CAAC,UAAU,IAAI,EAAE;YACtC,IAAI,EAAE,QAAQ;SACf,CAAC,CAAC;QAEH,OAAO,IAAI,OAAO,CAAC;YACjB,IAAI,EAAE,SAAS;YACf,MAAM,EAAE;gBACN,QAAQ,EAAE,CAAC,GAAG,CAAC;gBACf,gBAAgB,EAAE,cAAc;gBAChC,OAAO,EAAE,aAAa;aACvB;YACD,KAAK,EAAE,OAAO,CAAC,MAAM;SACtB,CAAC,CAAC;IACL,CAAC,EACD;QACE,IAAI,EAAE,QAAQ;QACd,WAAW,EAAE,eAAe;QAC5B,MAAM;KACP,CACF,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* LangGraph.js integration for Compresr.
|
|
3
|
+
*
|
|
4
|
+
* npm install @langchain/langgraph @langchain/core # (peers)
|
|
5
|
+
*
|
|
6
|
+
* LangGraph 1.0+ uses the same middleware mechanism as LangChain's
|
|
7
|
+
* `createAgent`. Both middlewares below are re-exported from
|
|
8
|
+
* `@compresr/sdk/integrations/langchain` so they're discoverable here
|
|
9
|
+
* too — import either path you prefer.
|
|
10
|
+
*/
|
|
11
|
+
export { compresrToolMiddleware, compresrSummarizationMiddleware, compresrPromptMiddleware, type CompresrMiddleware, type CompresrToolMiddlewareOptions, type CompresrSummarizationMiddlewareOptions, type CompresrPromptMiddlewareOptions, } from '../langchain/middleware.js';
|
|
12
|
+
export { makeCompresrNode, type MakeCompresrNodeOptions } from './nodes.js';
|
|
13
|
+
export { makeCompresrNode as compresrNode } from './nodes.js';
|
|
14
|
+
export { CompresrCheckpointSerializer, type CompresrCheckpointSerializerOptions, } from './checkpoint.js';
|
|
15
|
+
export { compresrHandoffTool, type CompresrHandoffToolOptions } from './handoff.js';
|
|
16
|
+
export { CompresrStore, type CompresrStoreOptions } from './store.js';
|
|
17
|
+
export type { ErrorPolicy } from '../_shared/index.js';
|
|
18
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/integrations/langgraph/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AACH,OAAO,EACL,sBAAsB,EACtB,+BAA+B,EAC/B,wBAAwB,EACxB,KAAK,kBAAkB,EACvB,KAAK,6BAA6B,EAClC,KAAK,sCAAsC,EAC3C,KAAK,+BAA+B,GACrC,MAAM,4BAA4B,CAAC;AAEpC,OAAO,EAAE,gBAAgB,EAAE,KAAK,uBAAuB,EAAE,MAAM,YAAY,CAAC;AAC5E,OAAO,EAAE,gBAAgB,IAAI,YAAY,EAAE,MAAM,YAAY,CAAC;AAC9D,OAAO,EACL,4BAA4B,EAC5B,KAAK,mCAAmC,GACzC,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,mBAAmB,EAAE,KAAK,0BAA0B,EAAE,MAAM,cAAc,CAAC;AACpF,OAAO,EAAE,aAAa,EAAE,KAAK,oBAAoB,EAAE,MAAM,YAAY,CAAC;AAEtE,YAAY,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* LangGraph.js integration for Compresr.
|
|
3
|
+
*
|
|
4
|
+
* npm install @langchain/langgraph @langchain/core # (peers)
|
|
5
|
+
*
|
|
6
|
+
* LangGraph 1.0+ uses the same middleware mechanism as LangChain's
|
|
7
|
+
* `createAgent`. Both middlewares below are re-exported from
|
|
8
|
+
* `@compresr/sdk/integrations/langchain` so they're discoverable here
|
|
9
|
+
* too — import either path you prefer.
|
|
10
|
+
*/
|
|
11
|
+
export { compresrToolMiddleware, compresrSummarizationMiddleware, compresrPromptMiddleware, } from '../langchain/middleware.js';
|
|
12
|
+
export { makeCompresrNode } from './nodes.js';
|
|
13
|
+
export { makeCompresrNode as compresrNode } from './nodes.js';
|
|
14
|
+
export { CompresrCheckpointSerializer, } from './checkpoint.js';
|
|
15
|
+
export { compresrHandoffTool } from './handoff.js';
|
|
16
|
+
export { CompresrStore } from './store.js';
|
|
17
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/integrations/langgraph/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AACH,OAAO,EACL,sBAAsB,EACtB,+BAA+B,EAC/B,wBAAwB,GAKzB,MAAM,4BAA4B,CAAC;AAEpC,OAAO,EAAE,gBAAgB,EAAgC,MAAM,YAAY,CAAC;AAC5E,OAAO,EAAE,gBAAgB,IAAI,YAAY,EAAE,MAAM,YAAY,CAAC;AAC9D,OAAO,EACL,4BAA4B,GAE7B,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,mBAAmB,EAAmC,MAAM,cAAc,CAAC;AACpF,OAAO,EAAE,aAAa,EAA6B,MAAM,YAAY,CAAC"}
|
|
@@ -0,0 +1,34 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Drop-in compression node for a custom `StateGraph`.
|
|
3
|
+
*
|
|
4
|
+
* Mirrors Python `make_compresr_node`. Reads a string field from state,
|
|
5
|
+
* compresses it, writes back. Same unified query API.
|
|
6
|
+
*/
|
|
7
|
+
import type { CompressionClient } from '../../clients/compression.js';
|
|
8
|
+
import { type ErrorPolicy } from '../_shared/index.js';
|
|
9
|
+
export interface MakeCompresrNodeOptions<S extends Record<string, unknown>> {
|
|
10
|
+
apiKey?: string;
|
|
11
|
+
client?: CompressionClient;
|
|
12
|
+
baseUrl?: string;
|
|
13
|
+
/** State key holding the text to compress. Required. */
|
|
14
|
+
contextKey: keyof S & string;
|
|
15
|
+
compressionModel?: string;
|
|
16
|
+
targetCompressionRatio?: number;
|
|
17
|
+
minTokens?: number;
|
|
18
|
+
coarse?: boolean;
|
|
19
|
+
/** Static query (latte_v1 only). */
|
|
20
|
+
query?: string;
|
|
21
|
+
/** State key holding the query string. */
|
|
22
|
+
queryKey?: keyof S & string;
|
|
23
|
+
/** Custom extractor over state. Highest priority after `query`. */
|
|
24
|
+
queryExtractor?: (state: S) => string | undefined;
|
|
25
|
+
onError?: ErrorPolicy;
|
|
26
|
+
}
|
|
27
|
+
/**
|
|
28
|
+
* Build a state-graph node that compresses `state[contextKey]`.
|
|
29
|
+
*
|
|
30
|
+
* Returns an async function `(state) => Partial<State>` compatible with
|
|
31
|
+
* `graph.addNode("compress", node)` in LangGraph.js.
|
|
32
|
+
*/
|
|
33
|
+
export declare function makeCompresrNode<S extends Record<string, unknown>>(options: MakeCompresrNodeOptions<S>): (state: S) => Promise<Partial<S>>;
|
|
34
|
+
//# sourceMappingURL=nodes.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"nodes.d.ts","sourceRoot":"","sources":["../../../../src/integrations/langgraph/nodes.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AACH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AACtE,OAAO,EAOL,KAAK,WAAW,EAEjB,MAAM,qBAAqB,CAAC;AAE7B,MAAM,WAAW,uBAAuB,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IACxE,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,iBAAiB,CAAC;IAC3B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,wDAAwD;IACxD,UAAU,EAAE,MAAM,CAAC,GAAG,MAAM,CAAC;IAC7B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,sBAAsB,CAAC,EAAE,MAAM,CAAC;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,oCAAoC;IACpC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,0CAA0C;IAC1C,QAAQ,CAAC,EAAE,MAAM,CAAC,GAAG,MAAM,CAAC;IAC5B,mEAAmE;IACnE,cAAc,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,KAAK,MAAM,GAAG,SAAS,CAAC;IAClD,OAAO,CAAC,EAAE,WAAW,CAAC;CACvB;AAED;;;;;GAKG;AACH,wBAAgB,gBAAgB,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAChE,OAAO,EAAE,uBAAuB,CAAC,CAAC,CAAC,GAClC,CAAC,KAAK,EAAE,CAAC,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CA2DnC"}
|
|
@@ -0,0 +1,51 @@
|
|
|
1
|
+
import { buildClient, compressSafe, DEFAULT_MIN_TOKENS, DEFAULT_MODEL, DEFAULT_POLICY, DEFAULT_RATIO, resolveQuery, } from '../_shared/index.js';
|
|
2
|
+
/**
|
|
3
|
+
* Build a state-graph node that compresses `state[contextKey]`.
|
|
4
|
+
*
|
|
5
|
+
* Returns an async function `(state) => Partial<State>` compatible with
|
|
6
|
+
* `graph.addNode("compress", node)` in LangGraph.js.
|
|
7
|
+
*/
|
|
8
|
+
export function makeCompresrNode(options) {
|
|
9
|
+
const { apiKey, client, baseUrl, contextKey, compressionModel = DEFAULT_MODEL, targetCompressionRatio = DEFAULT_RATIO, minTokens = DEFAULT_MIN_TOKENS, coarse, query, queryKey, queryExtractor, onError = DEFAULT_POLICY, } = options;
|
|
10
|
+
const compresr = client ?? buildClient({ apiKey, baseUrl, caller: 'makeCompresrNode' });
|
|
11
|
+
function resolve(state) {
|
|
12
|
+
let argsDict = null;
|
|
13
|
+
if (queryKey !== undefined) {
|
|
14
|
+
const v = state[queryKey];
|
|
15
|
+
if (typeof v === 'string' && v.trim()) {
|
|
16
|
+
argsDict = { [queryKey]: v };
|
|
17
|
+
}
|
|
18
|
+
}
|
|
19
|
+
return resolveQuery({
|
|
20
|
+
staticQuery: query,
|
|
21
|
+
extractor: queryExtractor
|
|
22
|
+
? (arg) => queryExtractor(arg)
|
|
23
|
+
: undefined,
|
|
24
|
+
extractorArg: queryExtractor ? state : undefined,
|
|
25
|
+
args: argsDict,
|
|
26
|
+
argsKey: queryKey,
|
|
27
|
+
});
|
|
28
|
+
}
|
|
29
|
+
return async (state) => {
|
|
30
|
+
const ctx = state[contextKey];
|
|
31
|
+
if (typeof ctx !== 'string' || !ctx.trim()) {
|
|
32
|
+
return {};
|
|
33
|
+
}
|
|
34
|
+
const q = resolve(state);
|
|
35
|
+
const compressed = await compressSafe(compresr, {
|
|
36
|
+
context: ctx,
|
|
37
|
+
...(q !== undefined ? { query: q } : {}),
|
|
38
|
+
compressionModel,
|
|
39
|
+
targetCompressionRatio,
|
|
40
|
+
...(coarse !== undefined ? { coarse } : {}),
|
|
41
|
+
minTokens,
|
|
42
|
+
onError,
|
|
43
|
+
contextLabel: `node:${String(contextKey)}`,
|
|
44
|
+
});
|
|
45
|
+
if (compressed === ctx) {
|
|
46
|
+
return {};
|
|
47
|
+
}
|
|
48
|
+
return { [contextKey]: compressed };
|
|
49
|
+
};
|
|
50
|
+
}
|
|
51
|
+
//# sourceMappingURL=nodes.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"nodes.js","sourceRoot":"","sources":["../../../../src/integrations/langgraph/nodes.ts"],"names":[],"mappings":"AAOA,OAAO,EACL,WAAW,EACX,YAAY,EACZ,kBAAkB,EAClB,aAAa,EACb,cAAc,EACd,aAAa,EAEb,YAAY,GACb,MAAM,qBAAqB,CAAC;AAqB7B;;;;;GAKG;AACH,MAAM,UAAU,gBAAgB,CAC9B,OAAmC;IAEnC,MAAM,EACJ,MAAM,EACN,MAAM,EACN,OAAO,EACP,UAAU,EACV,gBAAgB,GAAG,aAAa,EAChC,sBAAsB,GAAG,aAAa,EACtC,SAAS,GAAG,kBAAkB,EAC9B,MAAM,EACN,KAAK,EACL,QAAQ,EACR,cAAc,EACd,OAAO,GAAG,cAAc,GACzB,GAAG,OAAO,CAAC;IAEZ,MAAM,QAAQ,GACZ,MAAM,IAAI,WAAW,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,kBAAkB,EAAE,CAAC,CAAC;IAEzE,SAAS,OAAO,CAAC,KAAQ;QACvB,IAAI,QAAQ,GAAmC,IAAI,CAAC;QACpD,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC3B,MAAM,CAAC,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAC;YAC1B,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;gBACtC,QAAQ,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,CAAC;YAC/B,CAAC;QACH,CAAC;QACD,OAAO,YAAY,CAAC;YAClB,WAAW,EAAE,KAAK;YAClB,SAAS,EAAE,cAAc;gBACvB,CAAC,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,cAAc,CAAC,GAAQ,CAAC;gBACnC,CAAC,CAAC,SAAS;YACb,YAAY,EAAE,cAAc,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;YAChD,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,QAAQ;SAClB,CAAC,CAAC;IACL,CAAC;IAED,OAAO,KAAK,EAAE,KAAQ,EAAE,EAAE;QACxB,MAAM,GAAG,GAAG,KAAK,CAAC,UAAU,CAAC,CAAC;QAC9B,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC;YAC3C,OAAO,EAAgB,CAAC;QAC1B,CAAC;QACD,MAAM,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC;QACzB,MAAM,UAAU,GAAG,MAAM,YAAY,CAAC,QAAQ,EAAE;YAC9C,OAAO,EAAE,GAAG;YACZ,GAAG,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxC,gBAAgB;YAChB,sBAAsB;YACtB,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3C,SAAS;YACT,OAAO;YACP,YAAY,EAAE,QAAQ,MAAM,CAAC,UAAU,CAAC,EAAE;SAC3C,CAAC,CAAC;QACH,IAAI,UAAU,KAAK,GAAG,EAAE,CAAC;YACvB,OAAO,EAAgB,CAAC;QAC1B,CAAC;QACD,OAAO,EAAE,CAAC,UAAU,CAAC,EAAE,UAAU,EAAgB,CAAC;IACpD,CAAC,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,76 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* `CompresrStore` — wrap a LangGraph.js `BaseStore` so long string fields
|
|
3
|
+
* are compressed before being persisted to the underlying store.
|
|
4
|
+
*
|
|
5
|
+
* Composition over inheritance: holds an inner store and proxies all
|
|
6
|
+
* methods, intercepting `put` and `batch` to walk the value object and
|
|
7
|
+
* rewrite strings above `minTokens` in place. Lossy — there is no
|
|
8
|
+
* decompression on read.
|
|
9
|
+
*/
|
|
10
|
+
import { BaseStore, type Item, type Operation, type OperationResults, type SearchItem } from '@langchain/langgraph-checkpoint';
|
|
11
|
+
import type { CompressionClient } from '../../clients/compression.js';
|
|
12
|
+
import { type ErrorPolicy } from '../_shared/index.js';
|
|
13
|
+
export interface CompresrStoreOptions {
|
|
14
|
+
apiKey?: string;
|
|
15
|
+
client?: CompressionClient;
|
|
16
|
+
baseUrl?: string;
|
|
17
|
+
compressionModel?: string;
|
|
18
|
+
/**
|
|
19
|
+
* Query passed to `latte_v1`. If unset, the field name (e.g. `retrievedText`)
|
|
20
|
+
* is used as the query.
|
|
21
|
+
*/
|
|
22
|
+
query?: string;
|
|
23
|
+
targetCompressionRatio?: number;
|
|
24
|
+
minTokens?: number;
|
|
25
|
+
coarse?: boolean;
|
|
26
|
+
fields?: ReadonlySet<string>;
|
|
27
|
+
onError?: ErrorPolicy;
|
|
28
|
+
}
|
|
29
|
+
/**
|
|
30
|
+
* Drop-in `BaseStore` wrapper that compresses string fields on write.
|
|
31
|
+
*
|
|
32
|
+
* @example
|
|
33
|
+
* import { InMemoryStore } from '@langchain/langgraph';
|
|
34
|
+
* import { CompresrStore } from '@compresr/sdk/integrations/langgraph';
|
|
35
|
+
*
|
|
36
|
+
* const store = new CompresrStore(new InMemoryStore(), {
|
|
37
|
+
* apiKey: process.env.COMPRESR_API_KEY!,
|
|
38
|
+
* fields: new Set(['retrievedText']),
|
|
39
|
+
* minTokens: 500,
|
|
40
|
+
* });
|
|
41
|
+
*/
|
|
42
|
+
export declare class CompresrStore extends BaseStore {
|
|
43
|
+
private readonly inner;
|
|
44
|
+
private readonly compresr;
|
|
45
|
+
private readonly compressionModel;
|
|
46
|
+
private readonly query;
|
|
47
|
+
private readonly targetCompressionRatio;
|
|
48
|
+
private readonly minTokens;
|
|
49
|
+
private readonly coarse;
|
|
50
|
+
private readonly fields;
|
|
51
|
+
private readonly onError;
|
|
52
|
+
constructor(inner: BaseStore, options?: CompresrStoreOptions);
|
|
53
|
+
private shouldCompress;
|
|
54
|
+
private compressValue;
|
|
55
|
+
private walk;
|
|
56
|
+
put(namespace: string[], key: string, value: Record<string, unknown>, index?: false | string[]): Promise<void>;
|
|
57
|
+
batch<Op extends Operation[]>(operations: Op): Promise<OperationResults<Op>>;
|
|
58
|
+
get(namespace: string[], key: string): Promise<Item | null>;
|
|
59
|
+
search(namespacePrefix: string[], options?: {
|
|
60
|
+
filter?: Record<string, unknown>;
|
|
61
|
+
limit?: number;
|
|
62
|
+
offset?: number;
|
|
63
|
+
query?: string;
|
|
64
|
+
}): Promise<SearchItem[]>;
|
|
65
|
+
delete(namespace: string[], key: string): Promise<void>;
|
|
66
|
+
listNamespaces(options?: {
|
|
67
|
+
prefix?: string[];
|
|
68
|
+
suffix?: string[];
|
|
69
|
+
maxDepth?: number;
|
|
70
|
+
limit?: number;
|
|
71
|
+
offset?: number;
|
|
72
|
+
}): Promise<string[][]>;
|
|
73
|
+
start(): Promise<void>;
|
|
74
|
+
stop(): Promise<void>;
|
|
75
|
+
}
|
|
76
|
+
//# sourceMappingURL=store.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"store.d.ts","sourceRoot":"","sources":["../../../../src/integrations/langgraph/store.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AACH,OAAO,EACL,SAAS,EACT,KAAK,IAAI,EACT,KAAK,SAAS,EACd,KAAK,gBAAgB,EACrB,KAAK,UAAU,EAChB,MAAM,iCAAiC,CAAC;AAEzC,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AACtE,OAAO,EAML,KAAK,WAAW,EAEjB,MAAM,qBAAqB,CAAC;AAE7B,MAAM,WAAW,oBAAoB;IACnC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,iBAAiB,CAAC;IAC3B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B;;;OAGG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,sBAAsB,CAAC,EAAE,MAAM,CAAC;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,MAAM,CAAC,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC;IAC7B,OAAO,CAAC,EAAE,WAAW,CAAC;CACvB;AAcD;;;;;;;;;;;;GAYG;AACH,qBAAa,aAAc,SAAQ,SAAS;IAC1C,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAY;IAClC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAoB;IAC7C,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAS;IAC1C,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAqB;IAC3C,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAAS;IAChD,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAsB;IAC7C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAkC;IACzD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAc;gBAE1B,KAAK,EAAE,SAAS,EAAE,OAAO,GAAE,oBAAyB;IA0BhE,OAAO,CAAC,cAAc;YAOR,aAAa;YAab,IAAI;IAmBZ,GAAG,CACP,SAAS,EAAE,MAAM,EAAE,EACnB,GAAG,EAAE,MAAM,EACX,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC9B,KAAK,CAAC,EAAE,KAAK,GAAG,MAAM,EAAE,GACvB,OAAO,CAAC,IAAI,CAAC;IAKV,KAAK,CAAC,EAAE,SAAS,SAAS,EAAE,EAAE,UAAU,EAAE,EAAE,GAAG,OAAO,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;IAY5E,GAAG,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;IAI3D,MAAM,CACV,eAAe,EAAE,MAAM,EAAE,EACzB,OAAO,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GAC9F,OAAO,CAAC,UAAU,EAAE,CAAC;IAIlB,MAAM,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIvD,cAAc,CAAC,OAAO,CAAC,EAAE;QAC7B,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;QAClB,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;QAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;IAIjB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAItB,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;CAG5B"}
|
|
@@ -0,0 +1,117 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* `CompresrStore` — wrap a LangGraph.js `BaseStore` so long string fields
|
|
3
|
+
* are compressed before being persisted to the underlying store.
|
|
4
|
+
*
|
|
5
|
+
* Composition over inheritance: holds an inner store and proxies all
|
|
6
|
+
* methods, intercepting `put` and `batch` to walk the value object and
|
|
7
|
+
* rewrite strings above `minTokens` in place. Lossy — there is no
|
|
8
|
+
* decompression on read.
|
|
9
|
+
*/
|
|
10
|
+
import { BaseStore, } from '@langchain/langgraph-checkpoint';
|
|
11
|
+
import { buildClient, compressSafe, DEFAULT_MODEL, DEFAULT_POLICY, DEFAULT_RATIO, estimateTokens, } from '../_shared/index.js';
|
|
12
|
+
function isPutOp(op) {
|
|
13
|
+
const o = op;
|
|
14
|
+
return Object.prototype.hasOwnProperty.call(op, 'value') && typeof o.value !== 'undefined';
|
|
15
|
+
}
|
|
16
|
+
/**
|
|
17
|
+
* Drop-in `BaseStore` wrapper that compresses string fields on write.
|
|
18
|
+
*
|
|
19
|
+
* @example
|
|
20
|
+
* import { InMemoryStore } from '@langchain/langgraph';
|
|
21
|
+
* import { CompresrStore } from '@compresr/sdk/integrations/langgraph';
|
|
22
|
+
*
|
|
23
|
+
* const store = new CompresrStore(new InMemoryStore(), {
|
|
24
|
+
* apiKey: process.env.COMPRESR_API_KEY!,
|
|
25
|
+
* fields: new Set(['retrievedText']),
|
|
26
|
+
* minTokens: 500,
|
|
27
|
+
* });
|
|
28
|
+
*/
|
|
29
|
+
export class CompresrStore extends BaseStore {
|
|
30
|
+
inner;
|
|
31
|
+
compresr;
|
|
32
|
+
compressionModel;
|
|
33
|
+
query;
|
|
34
|
+
targetCompressionRatio;
|
|
35
|
+
minTokens;
|
|
36
|
+
coarse;
|
|
37
|
+
fields;
|
|
38
|
+
onError;
|
|
39
|
+
constructor(inner, options = {}) {
|
|
40
|
+
super();
|
|
41
|
+
const { apiKey, client, baseUrl, compressionModel = DEFAULT_MODEL, query, targetCompressionRatio = DEFAULT_RATIO, minTokens = 500, coarse, fields, onError = DEFAULT_POLICY, } = options;
|
|
42
|
+
this.inner = inner;
|
|
43
|
+
this.compresr = client ?? buildClient({ apiKey, baseUrl, caller: 'CompresrStore' });
|
|
44
|
+
this.compressionModel = compressionModel;
|
|
45
|
+
this.query = query;
|
|
46
|
+
this.targetCompressionRatio = targetCompressionRatio;
|
|
47
|
+
this.minTokens = Math.max(1, minTokens);
|
|
48
|
+
this.coarse = coarse;
|
|
49
|
+
this.fields = fields;
|
|
50
|
+
this.onError = onError;
|
|
51
|
+
}
|
|
52
|
+
shouldCompress(text, parentKey) {
|
|
53
|
+
if (this.fields !== undefined && (parentKey === undefined || !this.fields.has(parentKey))) {
|
|
54
|
+
return false;
|
|
55
|
+
}
|
|
56
|
+
return estimateTokens(text) >= this.minTokens;
|
|
57
|
+
}
|
|
58
|
+
async compressValue(text, parentKey) {
|
|
59
|
+
return await compressSafe(this.compresr, {
|
|
60
|
+
context: text,
|
|
61
|
+
query: this.query ?? parentKey ?? 'stored content',
|
|
62
|
+
compressionModel: this.compressionModel,
|
|
63
|
+
targetCompressionRatio: this.targetCompressionRatio,
|
|
64
|
+
...(this.coarse !== undefined ? { coarse: this.coarse } : {}),
|
|
65
|
+
minTokens: 1,
|
|
66
|
+
onError: this.onError,
|
|
67
|
+
contextLabel: `store:${parentKey ?? '?'}`,
|
|
68
|
+
});
|
|
69
|
+
}
|
|
70
|
+
async walk(value, parentKey) {
|
|
71
|
+
if (typeof value === 'string') {
|
|
72
|
+
if (!this.shouldCompress(value, parentKey))
|
|
73
|
+
return value;
|
|
74
|
+
return await this.compressValue(value, parentKey);
|
|
75
|
+
}
|
|
76
|
+
if (Array.isArray(value)) {
|
|
77
|
+
return Promise.all(value.map((v) => this.walk(v, parentKey)));
|
|
78
|
+
}
|
|
79
|
+
if (value !== null && typeof value === 'object') {
|
|
80
|
+
const entries = await Promise.all(Object.entries(value).map(async ([k, v]) => [k, await this.walk(v, k)]));
|
|
81
|
+
return Object.fromEntries(entries);
|
|
82
|
+
}
|
|
83
|
+
return value;
|
|
84
|
+
}
|
|
85
|
+
async put(namespace, key, value, index) {
|
|
86
|
+
const compressed = (await this.walk(value, undefined));
|
|
87
|
+
return this.inner.put(namespace, key, compressed, index);
|
|
88
|
+
}
|
|
89
|
+
async batch(operations) {
|
|
90
|
+
const rewritten = await Promise.all(operations.map(async (op) => {
|
|
91
|
+
if (isPutOp(op) && op.value !== null) {
|
|
92
|
+
return { ...op, value: (await this.walk(op.value, undefined)) };
|
|
93
|
+
}
|
|
94
|
+
return op;
|
|
95
|
+
}));
|
|
96
|
+
return this.inner.batch(rewritten);
|
|
97
|
+
}
|
|
98
|
+
async get(namespace, key) {
|
|
99
|
+
return this.inner.get(namespace, key);
|
|
100
|
+
}
|
|
101
|
+
async search(namespacePrefix, options) {
|
|
102
|
+
return this.inner.search(namespacePrefix, options);
|
|
103
|
+
}
|
|
104
|
+
async delete(namespace, key) {
|
|
105
|
+
return this.inner.delete(namespace, key);
|
|
106
|
+
}
|
|
107
|
+
async listNamespaces(options) {
|
|
108
|
+
return this.inner.listNamespaces(options);
|
|
109
|
+
}
|
|
110
|
+
async start() {
|
|
111
|
+
await this.inner.start();
|
|
112
|
+
}
|
|
113
|
+
async stop() {
|
|
114
|
+
await this.inner.stop();
|
|
115
|
+
}
|
|
116
|
+
}
|
|
117
|
+
//# sourceMappingURL=store.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"store.js","sourceRoot":"","sources":["../../../../src/integrations/langgraph/store.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AACH,OAAO,EACL,SAAS,GAKV,MAAM,iCAAiC,CAAC;AAGzC,OAAO,EACL,WAAW,EACX,YAAY,EACZ,aAAa,EACb,cAAc,EACd,aAAa,EAEb,cAAc,GACf,MAAM,qBAAqB,CAAC;AA0B7B,SAAS,OAAO,CAAC,EAAa;IAC5B,MAAM,CAAC,GAAG,EAAyB,CAAC;IACpC,OAAO,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,OAAO,CAAC,CAAC,KAAK,KAAK,WAAW,CAAC;AAC7F,CAAC;AAED;;;;;;;;;;;;GAYG;AACH,MAAM,OAAO,aAAc,SAAQ,SAAS;IACzB,KAAK,CAAY;IACjB,QAAQ,CAAoB;IAC5B,gBAAgB,CAAS;IACzB,KAAK,CAAqB;IAC1B,sBAAsB,CAAS;IAC/B,SAAS,CAAS;IAClB,MAAM,CAAsB;IAC5B,MAAM,CAAkC;IACxC,OAAO,CAAc;IAEtC,YAAY,KAAgB,EAAE,UAAgC,EAAE;QAC9D,KAAK,EAAE,CAAC;QACR,MAAM,EACJ,MAAM,EACN,MAAM,EACN,OAAO,EACP,gBAAgB,GAAG,aAAa,EAChC,KAAK,EACL,sBAAsB,GAAG,aAAa,EACtC,SAAS,GAAG,GAAG,EACf,MAAM,EACN,MAAM,EACN,OAAO,GAAG,cAAc,GACzB,GAAG,OAAO,CAAC;QAEZ,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,QAAQ,GAAG,MAAM,IAAI,WAAW,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,eAAe,EAAE,CAAC,CAAC;QACpF,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,sBAAsB,GAAG,sBAAsB,CAAC;QACrD,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;QACxC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAEO,cAAc,CAAC,IAAY,EAAE,SAA6B;QAChE,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS,IAAI,CAAC,SAAS,KAAK,SAAS,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC;YAC1F,OAAO,KAAK,CAAC;QACf,CAAC;QACD,OAAO,cAAc,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,IAAY,EAAE,SAA6B;QACrE,OAAO,MAAM,YAAY,CAAC,IAAI,CAAC,QAAQ,EAAE;YACvC,OAAO,EAAE,IAAI;YACb,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,SAAS,IAAI,gBAAgB;YAClD,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,sBAAsB,EAAE,IAAI,CAAC,sBAAsB;YACnD,GAAG,CAAC,IAAI,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7D,SAAS,EAAE,CAAC;YACZ,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,YAAY,EAAE,SAAS,SAAS,IAAI,GAAG,EAAE;SAC1C,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,IAAI,CAAC,KAAc,EAAE,SAA6B;QAC9D,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,SAAS,CAAC;gBAAE,OAAO,KAAK,CAAC;YACzD,OAAO,MAAM,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QACpD,CAAC;QACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACzB,OAAO,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC;QAChE,CAAC;QACD,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAChD,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,MAAM,CAAC,OAAO,CAAC,KAAgC,CAAC,CAAC,GAAG,CAClD,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAU,CACtD,CACF,CAAC;YACF,OAAO,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QACrC,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,GAAG,CACP,SAAmB,EACnB,GAAW,EACX,KAA8B,EAC9B,KAAwB;QAExB,MAAM,UAAU,GAAG,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,SAAS,CAAC,CAA4B,CAAC;QAClF,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,EAAE,UAAU,EAAE,KAAK,CAAC,CAAC;IAC3D,CAAC;IAED,KAAK,CAAC,KAAK,CAAyB,UAAc;QAChD,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,GAAG,CACjC,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE;YAC1B,IAAI,OAAO,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,KAAK,IAAI,EAAE,CAAC;gBACrC,OAAO,EAAE,GAAG,EAAE,EAAE,KAAK,EAAE,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,CAAC,CAA4B,EAAE,CAAC;YAC7F,CAAC;YACD,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC,CACH,CAAC;QACF,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,SAAe,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,SAAmB,EAAE,GAAW;QACxC,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,MAAM,CACV,eAAyB,EACzB,OAA+F;QAE/F,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,SAAmB,EAAE,GAAW;QAC3C,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,OAMpB;QACC,OAAO,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,IAAI;QACR,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;IAC1B,CAAC;CACF"}
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* LlamaIndex.TS integration for Compresr.
|
|
3
|
+
*
|
|
4
|
+
* npm install llamaindex # (peer)
|
|
5
|
+
*/
|
|
6
|
+
export { CompresrNodePostprocessor, type CompresrNodePostprocessorOptions, } from './postprocessor.js';
|
|
7
|
+
export { wrapToolWithCompresr, type WrapToolOptions, type CompresrWrappableTool, } from './wrappers.js';
|
|
8
|
+
export { CompresrMemoryBlock, type CompresrMemoryBlockOptions } from './memory.js';
|
|
9
|
+
export type { ErrorPolicy } from '../_shared/index.js';
|
|
10
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/integrations/llamaindex/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EACL,yBAAyB,EACzB,KAAK,gCAAgC,GACtC,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,oBAAoB,EACpB,KAAK,eAAe,EACpB,KAAK,qBAAqB,GAC3B,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,mBAAmB,EAAE,KAAK,0BAA0B,EAAE,MAAM,aAAa,CAAC;AACnF,YAAY,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* LlamaIndex.TS integration for Compresr.
|
|
3
|
+
*
|
|
4
|
+
* npm install llamaindex # (peer)
|
|
5
|
+
*/
|
|
6
|
+
export { CompresrNodePostprocessor, } from './postprocessor.js';
|
|
7
|
+
export { wrapToolWithCompresr, } from './wrappers.js';
|
|
8
|
+
export { CompresrMemoryBlock } from './memory.js';
|
|
9
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/integrations/llamaindex/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EACL,yBAAyB,GAE1B,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,oBAAoB,GAGrB,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,mBAAmB,EAAmC,MAAM,aAAa,CAAC"}
|