@ax-llm/ax 9.0.21 → 9.0.23
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/ai/anthropic/api.d.ts +19 -0
- package/ai/anthropic/api.js +326 -0
- package/ai/anthropic/api.js.map +1 -0
- package/ai/anthropic/index.d.ts +2 -0
- package/ai/anthropic/index.js +3 -0
- package/ai/anthropic/index.js.map +1 -0
- package/ai/anthropic/info.d.ts +2 -0
- package/ai/anthropic/info.js +40 -0
- package/ai/anthropic/info.js.map +1 -0
- package/ai/anthropic/types.d.ts +171 -0
- package/ai/anthropic/types.js +10 -0
- package/ai/anthropic/types.js.map +1 -0
- package/ai/azure-openai/api.d.ts +19 -0
- package/ai/azure-openai/api.js +30 -0
- package/ai/azure-openai/api.js.map +1 -0
- package/ai/azure-openai/index.d.ts +1 -0
- package/ai/azure-openai/index.js +2 -0
- package/ai/azure-openai/index.js.map +1 -0
- package/ai/balance.d.ts +23 -0
- package/ai/balance.js +87 -0
- package/ai/balance.js.map +1 -0
- package/ai/base.d.ts +55 -0
- package/ai/base.js +343 -0
- package/ai/base.js.map +1 -0
- package/ai/cohere/api.d.ts +22 -0
- package/ai/cohere/api.js +264 -0
- package/ai/cohere/api.js.map +1 -0
- package/ai/cohere/index.d.ts +2 -0
- package/ai/cohere/index.js +3 -0
- package/ai/cohere/index.js.map +1 -0
- package/ai/cohere/info.d.ts +2 -0
- package/ai/cohere/info.js +52 -0
- package/ai/cohere/info.js.map +1 -0
- package/ai/cohere/types.d.ts +102 -0
- package/ai/cohere/types.js +23 -0
- package/ai/cohere/types.js.map +1 -0
- package/ai/deepseek/api.d.ts +16 -0
- package/ai/deepseek/api.js +32 -0
- package/ai/deepseek/api.js.map +1 -0
- package/ai/deepseek/index.d.ts +2 -0
- package/ai/deepseek/index.js +3 -0
- package/ai/deepseek/index.js.map +1 -0
- package/ai/deepseek/info.d.ts +2 -0
- package/ai/deepseek/info.js +16 -0
- package/ai/deepseek/info.js.map +1 -0
- package/ai/deepseek/types.d.ts +8 -0
- package/ai/deepseek/types.js +10 -0
- package/ai/deepseek/types.js.map +1 -0
- package/ai/google-gemini/api.d.ts +36 -0
- package/ai/google-gemini/api.js +299 -0
- package/ai/google-gemini/api.js.map +1 -0
- package/ai/google-gemini/index.d.ts +2 -0
- package/ai/google-gemini/index.js +3 -0
- package/ai/google-gemini/index.js.map +1 -0
- package/ai/google-gemini/info.d.ts +6 -0
- package/ai/google-gemini/info.js +29 -0
- package/ai/google-gemini/info.js.map +1 -0
- package/ai/google-gemini/types.d.ts +146 -0
- package/ai/google-gemini/types.js +27 -0
- package/ai/google-gemini/types.js.map +1 -0
- package/ai/groq/api.d.ts +18 -0
- package/ai/groq/api.js +52 -0
- package/ai/groq/api.js.map +1 -0
- package/ai/groq/index.d.ts +1 -0
- package/ai/groq/index.js +2 -0
- package/ai/groq/index.js.map +1 -0
- package/ai/groq/types.d.ts +6 -0
- package/ai/groq/types.js +8 -0
- package/ai/groq/types.js.map +1 -0
- package/ai/huggingface/api.d.ts +19 -0
- package/ai/huggingface/api.js +115 -0
- package/ai/huggingface/api.js.map +1 -0
- package/ai/huggingface/index.d.ts +2 -0
- package/ai/huggingface/index.js +3 -0
- package/ai/huggingface/index.js.map +1 -0
- package/ai/huggingface/info.d.ts +6 -0
- package/ai/huggingface/info.js +6 -0
- package/ai/huggingface/info.js.map +1 -0
- package/ai/huggingface/types.d.ts +34 -0
- package/ai/huggingface/types.js +5 -0
- package/ai/huggingface/types.js.map +1 -0
- package/ai/index.d.ts +16 -0
- package/ai/index.js +16 -0
- package/ai/index.js.map +1 -0
- package/ai/mistral/api.d.ts +16 -0
- package/ai/mistral/api.js +32 -0
- package/ai/mistral/api.js.map +1 -0
- package/ai/mistral/index.d.ts +2 -0
- package/ai/mistral/index.js +3 -0
- package/ai/mistral/index.js.map +1 -0
- package/ai/mistral/info.d.ts +2 -0
- package/ai/mistral/info.js +34 -0
- package/ai/mistral/info.js.map +1 -0
- package/ai/mistral/types.d.ts +10 -0
- package/ai/mistral/types.js +13 -0
- package/ai/mistral/types.js.map +1 -0
- package/ai/ollama/api.d.ts +22 -0
- package/ai/ollama/api.js +32 -0
- package/ai/ollama/api.js.map +1 -0
- package/ai/ollama/index.d.ts +1 -0
- package/ai/ollama/index.js +2 -0
- package/ai/ollama/index.js.map +1 -0
- package/ai/openai/api.d.ts +29 -0
- package/ai/openai/api.js +283 -0
- package/ai/openai/api.js.map +1 -0
- package/ai/openai/index.d.ts +2 -0
- package/ai/openai/index.js +3 -0
- package/ai/openai/index.js.map +1 -0
- package/ai/openai/info.d.ts +6 -0
- package/ai/openai/info.js +50 -0
- package/ai/openai/info.js.map +1 -0
- package/ai/openai/types.d.ts +165 -0
- package/ai/openai/types.js +18 -0
- package/ai/openai/types.js.map +1 -0
- package/ai/together/api.d.ts +15 -0
- package/ai/together/api.js +28 -0
- package/ai/together/api.js.map +1 -0
- package/ai/together/index.d.ts +1 -0
- package/ai/together/index.js +2 -0
- package/ai/together/index.js.map +1 -0
- package/ai/together/info.d.ts +2 -0
- package/ai/together/info.js +2 -0
- package/ai/together/info.js.map +1 -0
- package/ai/types.d.ts +159 -0
- package/ai/types.js +2 -0
- package/ai/types.js.map +1 -0
- package/ai/util.d.ts +12 -0
- package/ai/util.js +63 -0
- package/ai/util.js.map +1 -0
- package/ai/wrap.d.ts +47 -0
- package/ai/wrap.js +95 -0
- package/ai/wrap.js.map +1 -0
- package/cjs/ai/anthropic/api.d.ts +19 -0
- package/cjs/ai/anthropic/api.js +331 -0
- package/cjs/ai/anthropic/api.js.map +1 -0
- package/cjs/ai/anthropic/index.d.ts +2 -0
- package/cjs/ai/anthropic/index.js +19 -0
- package/cjs/ai/anthropic/index.js.map +1 -0
- package/cjs/ai/anthropic/info.d.ts +2 -0
- package/cjs/ai/anthropic/info.js +43 -0
- package/cjs/ai/anthropic/info.js.map +1 -0
- package/cjs/ai/anthropic/types.d.ts +171 -0
- package/cjs/ai/anthropic/types.js +13 -0
- package/cjs/ai/anthropic/types.js.map +1 -0
- package/cjs/ai/azure-openai/api.d.ts +19 -0
- package/cjs/ai/azure-openai/api.js +34 -0
- package/cjs/ai/azure-openai/api.js.map +1 -0
- package/cjs/ai/azure-openai/index.d.ts +1 -0
- package/cjs/ai/azure-openai/index.js +18 -0
- package/cjs/ai/azure-openai/index.js.map +1 -0
- package/cjs/ai/balance.d.ts +23 -0
- package/cjs/ai/balance.js +91 -0
- package/cjs/ai/balance.js.map +1 -0
- package/cjs/ai/base.d.ts +55 -0
- package/cjs/ai/base.js +349 -0
- package/cjs/ai/base.js.map +1 -0
- package/cjs/ai/cohere/api.d.ts +22 -0
- package/cjs/ai/cohere/api.js +270 -0
- package/cjs/ai/cohere/api.js.map +1 -0
- package/cjs/ai/cohere/index.d.ts +2 -0
- package/cjs/ai/cohere/index.js +19 -0
- package/cjs/ai/cohere/index.js.map +1 -0
- package/cjs/ai/cohere/info.d.ts +2 -0
- package/cjs/ai/cohere/info.js +55 -0
- package/cjs/ai/cohere/info.js.map +1 -0
- package/cjs/ai/cohere/types.d.ts +102 -0
- package/cjs/ai/cohere/types.js +26 -0
- package/cjs/ai/cohere/types.js.map +1 -0
- package/cjs/ai/deepseek/api.d.ts +16 -0
- package/cjs/ai/deepseek/api.js +38 -0
- package/cjs/ai/deepseek/api.js.map +1 -0
- package/cjs/ai/deepseek/index.d.ts +2 -0
- package/cjs/ai/deepseek/index.js +19 -0
- package/cjs/ai/deepseek/index.js.map +1 -0
- package/cjs/ai/deepseek/info.d.ts +2 -0
- package/cjs/ai/deepseek/info.js +19 -0
- package/cjs/ai/deepseek/info.js.map +1 -0
- package/cjs/ai/deepseek/types.d.ts +8 -0
- package/cjs/ai/deepseek/types.js +13 -0
- package/cjs/ai/deepseek/types.js.map +1 -0
- package/cjs/ai/google-gemini/api.d.ts +36 -0
- package/cjs/ai/google-gemini/api.js +305 -0
- package/cjs/ai/google-gemini/api.js.map +1 -0
- package/cjs/ai/google-gemini/index.d.ts +2 -0
- package/cjs/ai/google-gemini/index.js +19 -0
- package/cjs/ai/google-gemini/index.js.map +1 -0
- package/cjs/ai/google-gemini/info.d.ts +6 -0
- package/cjs/ai/google-gemini/info.js +32 -0
- package/cjs/ai/google-gemini/info.js.map +1 -0
- package/cjs/ai/google-gemini/types.d.ts +146 -0
- package/cjs/ai/google-gemini/types.js +30 -0
- package/cjs/ai/google-gemini/types.js.map +1 -0
- package/cjs/ai/groq/api.d.ts +18 -0
- package/cjs/ai/groq/api.js +56 -0
- package/cjs/ai/groq/api.js.map +1 -0
- package/cjs/ai/groq/index.d.ts +1 -0
- package/cjs/ai/groq/index.js +18 -0
- package/cjs/ai/groq/index.js.map +1 -0
- package/cjs/ai/groq/types.d.ts +6 -0
- package/cjs/ai/groq/types.js +11 -0
- package/cjs/ai/groq/types.js.map +1 -0
- package/cjs/ai/huggingface/api.d.ts +19 -0
- package/cjs/ai/huggingface/api.js +121 -0
- package/cjs/ai/huggingface/api.js.map +1 -0
- package/cjs/ai/huggingface/index.d.ts +2 -0
- package/cjs/ai/huggingface/index.js +19 -0
- package/cjs/ai/huggingface/index.js.map +1 -0
- package/cjs/ai/huggingface/info.d.ts +6 -0
- package/cjs/ai/huggingface/info.js +9 -0
- package/cjs/ai/huggingface/info.js.map +1 -0
- package/cjs/ai/huggingface/types.d.ts +34 -0
- package/cjs/ai/huggingface/types.js +8 -0
- package/cjs/ai/huggingface/types.js.map +1 -0
- package/cjs/ai/index.d.ts +16 -0
- package/cjs/ai/index.js +32 -0
- package/cjs/ai/index.js.map +1 -0
- package/cjs/ai/mistral/api.d.ts +16 -0
- package/cjs/ai/mistral/api.js +38 -0
- package/cjs/ai/mistral/api.js.map +1 -0
- package/cjs/ai/mistral/index.d.ts +2 -0
- package/cjs/ai/mistral/index.js +19 -0
- package/cjs/ai/mistral/index.js.map +1 -0
- package/cjs/ai/mistral/info.d.ts +2 -0
- package/cjs/ai/mistral/info.js +37 -0
- package/cjs/ai/mistral/info.js.map +1 -0
- package/cjs/ai/mistral/types.d.ts +10 -0
- package/cjs/ai/mistral/types.js +16 -0
- package/cjs/ai/mistral/types.js.map +1 -0
- package/cjs/ai/ollama/api.d.ts +22 -0
- package/cjs/ai/ollama/api.js +38 -0
- package/cjs/ai/ollama/api.js.map +1 -0
- package/cjs/ai/ollama/index.d.ts +1 -0
- package/cjs/ai/ollama/index.js +18 -0
- package/cjs/ai/ollama/index.js.map +1 -0
- package/cjs/ai/openai/api.d.ts +29 -0
- package/cjs/ai/openai/api.js +291 -0
- package/cjs/ai/openai/api.js.map +1 -0
- package/cjs/ai/openai/index.d.ts +2 -0
- package/cjs/ai/openai/index.js +19 -0
- package/cjs/ai/openai/index.js.map +1 -0
- package/cjs/ai/openai/info.d.ts +6 -0
- package/cjs/ai/openai/info.js +53 -0
- package/cjs/ai/openai/info.js.map +1 -0
- package/cjs/ai/openai/types.d.ts +165 -0
- package/cjs/ai/openai/types.js +21 -0
- package/cjs/ai/openai/types.js.map +1 -0
- package/cjs/ai/together/api.d.ts +15 -0
- package/cjs/ai/together/api.js +33 -0
- package/cjs/ai/together/api.js.map +1 -0
- package/cjs/ai/together/index.d.ts +1 -0
- package/cjs/ai/together/index.js +18 -0
- package/cjs/ai/together/index.js.map +1 -0
- package/cjs/ai/together/info.d.ts +2 -0
- package/cjs/ai/together/info.js +5 -0
- package/cjs/ai/together/info.js.map +1 -0
- package/cjs/ai/types.d.ts +159 -0
- package/cjs/ai/types.js +3 -0
- package/cjs/ai/types.js.map +1 -0
- package/cjs/ai/util.d.ts +12 -0
- package/cjs/ai/util.js +71 -0
- package/cjs/ai/util.js.map +1 -0
- package/cjs/ai/wrap.d.ts +47 -0
- package/cjs/ai/wrap.js +99 -0
- package/cjs/ai/wrap.js.map +1 -0
- package/cjs/db/base.d.ts +23 -0
- package/cjs/db/base.js +91 -0
- package/cjs/db/base.js.map +1 -0
- package/cjs/db/cloudflare.d.ts +21 -0
- package/cjs/db/cloudflare.js +102 -0
- package/cjs/db/cloudflare.js.map +1 -0
- package/cjs/db/index.d.ts +6 -0
- package/cjs/db/index.js +23 -0
- package/cjs/db/index.js.map +1 -0
- package/cjs/db/memory.d.ts +21 -0
- package/cjs/db/memory.js +103 -0
- package/cjs/db/memory.js.map +1 -0
- package/cjs/db/pinecone.d.ts +21 -0
- package/cjs/db/pinecone.js +75 -0
- package/cjs/db/pinecone.js.map +1 -0
- package/cjs/db/types.d.ts +35 -0
- package/cjs/db/types.js +4 -0
- package/cjs/db/types.js.map +1 -0
- package/cjs/db/weaviate.d.ts +21 -0
- package/cjs/db/weaviate.js +131 -0
- package/cjs/db/weaviate.js.map +1 -0
- package/cjs/db/wrap.d.ts +13 -0
- package/cjs/db/wrap.js +39 -0
- package/cjs/db/wrap.js.map +1 -0
- package/cjs/docs/index.d.ts +4 -0
- package/cjs/docs/index.js +21 -0
- package/cjs/docs/index.js.map +1 -0
- package/cjs/docs/manager.d.ts +45 -0
- package/cjs/docs/manager.js +162 -0
- package/cjs/docs/manager.js.map +1 -0
- package/cjs/docs/reranker.d.ts +7 -0
- package/cjs/docs/reranker.js +30 -0
- package/cjs/docs/reranker.js.map +1 -0
- package/cjs/docs/rewriter.d.ts +4 -0
- package/cjs/docs/rewriter.js +13 -0
- package/cjs/docs/rewriter.js.map +1 -0
- package/cjs/docs/tika.d.ts +18 -0
- package/cjs/docs/tika.js +50 -0
- package/cjs/docs/tika.js.map +1 -0
- package/cjs/dsp/asserts.d.ts +32 -0
- package/cjs/dsp/asserts.js +98 -0
- package/cjs/dsp/asserts.js.map +1 -0
- package/cjs/dsp/eval.d.ts +43 -0
- package/cjs/dsp/eval.js +128 -0
- package/cjs/dsp/eval.js.map +1 -0
- package/cjs/dsp/evaluate.d.ts +12 -0
- package/cjs/dsp/evaluate.js +36 -0
- package/cjs/dsp/evaluate.js.map +1 -0
- package/cjs/dsp/extract.d.ts +25 -0
- package/cjs/dsp/extract.js +164 -0
- package/cjs/dsp/extract.js.map +1 -0
- package/cjs/dsp/generate.d.ts +55 -0
- package/cjs/dsp/generate.js +314 -0
- package/cjs/dsp/generate.js.map +1 -0
- package/cjs/dsp/index.d.ts +15 -0
- package/cjs/dsp/index.js +27 -0
- package/cjs/dsp/index.js.map +1 -0
- package/cjs/dsp/loader.d.ts +30 -0
- package/cjs/dsp/loader.js +85 -0
- package/cjs/dsp/loader.js.map +1 -0
- package/cjs/dsp/optimize.d.ts +27 -0
- package/cjs/dsp/optimize.js +99 -0
- package/cjs/dsp/optimize.js.map +1 -0
- package/cjs/dsp/parser.d.ts +87 -0
- package/cjs/dsp/parser.js +1545 -0
- package/cjs/dsp/parser.js.map +1 -0
- package/cjs/dsp/program.d.ts +73 -0
- package/cjs/dsp/program.js +120 -0
- package/cjs/dsp/program.js.map +1 -0
- package/cjs/dsp/prompt.d.ts +48 -0
- package/cjs/dsp/prompt.js +263 -0
- package/cjs/dsp/prompt.js.map +1 -0
- package/cjs/dsp/registry.d.ts +6 -0
- package/cjs/dsp/registry.js +19 -0
- package/cjs/dsp/registry.js.map +1 -0
- package/cjs/dsp/router.d.ts +26 -0
- package/cjs/dsp/router.js +76 -0
- package/cjs/dsp/router.js.map +1 -0
- package/cjs/dsp/sig.d.ts +37 -0
- package/cjs/dsp/sig.js +197 -0
- package/cjs/dsp/sig.js.map +1 -0
- package/cjs/dsp/stopwords.d.ts +1 -0
- package/cjs/dsp/stopwords.js +1167 -0
- package/cjs/dsp/stopwords.js.map +1 -0
- package/cjs/dsp/strutil.d.ts +11 -0
- package/cjs/dsp/strutil.js +63 -0
- package/cjs/dsp/strutil.js.map +1 -0
- package/cjs/dsp/util.d.ts +5 -0
- package/cjs/dsp/util.js +82 -0
- package/cjs/dsp/util.js.map +1 -0
- package/cjs/funcs/code.d.ts +9 -0
- package/cjs/funcs/code.js +90 -0
- package/cjs/funcs/code.js.map +1 -0
- package/cjs/funcs/embed.d.ts +6 -0
- package/cjs/funcs/embed.js +33 -0
- package/cjs/funcs/embed.js.map +1 -0
- package/cjs/funcs/functions.d.ts +16 -0
- package/cjs/funcs/functions.js +62 -0
- package/cjs/funcs/functions.js.map +1 -0
- package/cjs/funcs/index.d.ts +3 -0
- package/cjs/funcs/index.js +20 -0
- package/cjs/funcs/index.js.map +1 -0
- package/cjs/funcs/jsonschema.d.ts +2 -0
- package/cjs/funcs/jsonschema.js +101 -0
- package/cjs/funcs/jsonschema.js.map +1 -0
- package/cjs/index.d.ts +9 -0
- package/cjs/index.js +34 -0
- package/cjs/index.js.map +1 -0
- package/cjs/mem/index.d.ts +2 -0
- package/cjs/mem/index.js +19 -0
- package/cjs/mem/index.js.map +1 -0
- package/cjs/mem/memory.d.ts +16 -0
- package/cjs/mem/memory.js +81 -0
- package/cjs/mem/memory.js.map +1 -0
- package/cjs/mem/types.d.ts +10 -0
- package/cjs/mem/types.js +3 -0
- package/cjs/mem/types.js.map +1 -0
- package/cjs/prompts/agent.d.ts +34 -0
- package/cjs/prompts/agent.js +110 -0
- package/cjs/prompts/agent.js.map +1 -0
- package/cjs/prompts/cot.d.ts +9 -0
- package/cjs/prompts/cot.js +21 -0
- package/cjs/prompts/cot.js.map +1 -0
- package/cjs/prompts/index.d.ts +4 -0
- package/cjs/prompts/index.js +21 -0
- package/cjs/prompts/index.js.map +1 -0
- package/cjs/prompts/rag.d.ts +23 -0
- package/cjs/prompts/rag.js +33 -0
- package/cjs/prompts/rag.js.map +1 -0
- package/cjs/prompts/react.d.ts +8 -0
- package/cjs/prompts/react.js +26 -0
- package/cjs/prompts/react.js.map +1 -0
- package/cjs/trace/index.d.ts +171 -0
- package/cjs/trace/index.js +70 -0
- package/cjs/trace/index.js.map +1 -0
- package/cjs/util/apicall.d.ts +18 -0
- package/cjs/util/apicall.js +57 -0
- package/cjs/util/apicall.js.map +1 -0
- package/cjs/util/log.d.ts +14 -0
- package/cjs/util/log.js +35 -0
- package/cjs/util/log.js.map +1 -0
- package/cjs/util/other.d.ts +1 -0
- package/cjs/util/other.js +8 -0
- package/cjs/util/other.js.map +1 -0
- package/cjs/util/rate-limit.d.ts +24 -0
- package/cjs/util/rate-limit.js +53 -0
- package/cjs/util/rate-limit.js.map +1 -0
- package/cjs/util/stream.d.ts +9 -0
- package/cjs/util/stream.js +33 -0
- package/cjs/util/stream.js.map +1 -0
- package/cjs/util/transform.d.ts +7 -0
- package/cjs/util/transform.js +64 -0
- package/cjs/util/transform.js.map +1 -0
- package/db/base.d.ts +23 -0
- package/db/base.js +87 -0
- package/db/base.js.map +1 -0
- package/db/cloudflare.d.ts +21 -0
- package/db/cloudflare.js +98 -0
- package/db/cloudflare.js.map +1 -0
- package/db/index.d.ts +6 -0
- package/db/index.js +7 -0
- package/db/index.js.map +1 -0
- package/db/memory.d.ts +21 -0
- package/db/memory.js +99 -0
- package/db/memory.js.map +1 -0
- package/db/pinecone.d.ts +21 -0
- package/db/pinecone.js +71 -0
- package/db/pinecone.js.map +1 -0
- package/db/types.d.ts +35 -0
- package/db/types.js +3 -0
- package/db/types.js.map +1 -0
- package/db/weaviate.d.ts +21 -0
- package/db/weaviate.js +127 -0
- package/db/weaviate.js.map +1 -0
- package/db/wrap.d.ts +13 -0
- package/db/wrap.js +35 -0
- package/db/wrap.js.map +1 -0
- package/docs/index.d.ts +4 -0
- package/docs/index.js +5 -0
- package/docs/index.js.map +1 -0
- package/docs/manager.d.ts +45 -0
- package/docs/manager.js +159 -0
- package/docs/manager.js.map +1 -0
- package/docs/reranker.d.ts +7 -0
- package/docs/reranker.js +26 -0
- package/docs/reranker.js.map +1 -0
- package/docs/rewriter.d.ts +4 -0
- package/docs/rewriter.js +9 -0
- package/docs/rewriter.js.map +1 -0
- package/docs/tika.d.ts +18 -0
- package/docs/tika.js +46 -0
- package/docs/tika.js.map +1 -0
- package/dsp/asserts.d.ts +32 -0
- package/dsp/asserts.js +91 -0
- package/dsp/asserts.js.map +1 -0
- package/dsp/eval.d.ts +43 -0
- package/dsp/eval.js +125 -0
- package/dsp/eval.js.map +1 -0
- package/dsp/evaluate.d.ts +12 -0
- package/dsp/evaluate.js +32 -0
- package/dsp/evaluate.js.map +1 -0
- package/dsp/extract.d.ts +25 -0
- package/dsp/extract.js +153 -0
- package/dsp/extract.js.map +1 -0
- package/dsp/generate.d.ts +55 -0
- package/dsp/generate.js +311 -0
- package/dsp/generate.js.map +1 -0
- package/dsp/index.d.ts +15 -0
- package/dsp/index.js +11 -0
- package/dsp/index.js.map +1 -0
- package/dsp/loader.d.ts +30 -0
- package/dsp/loader.js +81 -0
- package/dsp/loader.js.map +1 -0
- package/dsp/optimize.d.ts +27 -0
- package/dsp/optimize.js +95 -0
- package/dsp/optimize.js.map +1 -0
- package/dsp/parser.d.ts +87 -0
- package/dsp/parser.js +1542 -0
- package/dsp/parser.js.map +1 -0
- package/dsp/program.d.ts +73 -0
- package/dsp/program.js +116 -0
- package/dsp/program.js.map +1 -0
- package/dsp/prompt.d.ts +48 -0
- package/dsp/prompt.js +260 -0
- package/dsp/prompt.js.map +1 -0
- package/dsp/registry.d.ts +6 -0
- package/dsp/registry.js +15 -0
- package/dsp/registry.js.map +1 -0
- package/dsp/router.d.ts +26 -0
- package/dsp/router.js +71 -0
- package/dsp/router.js.map +1 -0
- package/dsp/sig.d.ts +37 -0
- package/dsp/sig.js +193 -0
- package/dsp/sig.js.map +1 -0
- package/dsp/stopwords.d.ts +1 -0
- package/dsp/stopwords.js +1164 -0
- package/dsp/stopwords.js.map +1 -0
- package/dsp/strutil.d.ts +11 -0
- package/dsp/strutil.js +60 -0
- package/dsp/strutil.js.map +1 -0
- package/dsp/util.d.ts +5 -0
- package/dsp/util.js +76 -0
- package/dsp/util.js.map +1 -0
- package/funcs/code.d.ts +9 -0
- package/funcs/code.js +63 -0
- package/funcs/code.js.map +1 -0
- package/funcs/embed.d.ts +6 -0
- package/funcs/embed.js +29 -0
- package/funcs/embed.js.map +1 -0
- package/funcs/functions.d.ts +16 -0
- package/funcs/functions.js +55 -0
- package/funcs/functions.js.map +1 -0
- package/funcs/index.d.ts +3 -0
- package/funcs/index.js +4 -0
- package/funcs/index.js.map +1 -0
- package/funcs/jsonschema.d.ts +2 -0
- package/funcs/jsonschema.js +97 -0
- package/funcs/jsonschema.js.map +1 -0
- package/index.d.ts +9 -0
- package/index.js.map +1 -0
- package/mem/index.d.ts +2 -0
- package/mem/index.js +3 -0
- package/mem/index.js.map +1 -0
- package/mem/memory.d.ts +16 -0
- package/mem/memory.js +77 -0
- package/mem/memory.js.map +1 -0
- package/mem/types.d.ts +10 -0
- package/mem/types.js +2 -0
- package/mem/types.js.map +1 -0
- package/package.json +7 -9
- package/prompts/agent.d.ts +34 -0
- package/prompts/agent.js +108 -0
- package/prompts/agent.js.map +1 -0
- package/prompts/cot.d.ts +9 -0
- package/prompts/cot.js +17 -0
- package/prompts/cot.js.map +1 -0
- package/prompts/index.d.ts +4 -0
- package/prompts/index.js +5 -0
- package/prompts/index.js.map +1 -0
- package/prompts/rag.d.ts +23 -0
- package/prompts/rag.js +30 -0
- package/prompts/rag.js.map +1 -0
- package/prompts/react.d.ts +8 -0
- package/prompts/react.js +22 -0
- package/prompts/react.js.map +1 -0
- package/trace/index.d.ts +171 -0
- package/trace/index.js +67 -0
- package/trace/index.js.map +1 -0
- package/util/apicall.d.ts +18 -0
- package/util/apicall.js +50 -0
- package/util/apicall.js.map +1 -0
- package/util/log.d.ts +14 -0
- package/util/log.js +31 -0
- package/util/log.js.map +1 -0
- package/util/other.d.ts +1 -0
- package/util/other.js +5 -0
- package/util/other.js.map +1 -0
- package/util/rate-limit.d.ts +24 -0
- package/util/rate-limit.js +49 -0
- package/util/rate-limit.js.map +1 -0
- package/util/stream.d.ts +9 -0
- package/util/stream.js +29 -0
- package/util/stream.js.map +1 -0
- package/util/transform.d.ts +7 -0
- package/util/transform.js +59 -0
- package/util/transform.js.map +1 -0
|
@@ -0,0 +1,70 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.AxSpanKind = exports.AxSpanStatusCode = exports.AxSpanKindValues = exports.AxLLMRequestTypeValues = exports.axSpanEvents = exports.axSpanAttributes = void 0;
|
|
4
|
+
exports.axSpanAttributes = {
|
|
5
|
+
// LLM
|
|
6
|
+
LLM_SYSTEM: 'gen_ai.system',
|
|
7
|
+
LLM_REQUEST_MODEL: 'gen_ai.request.model',
|
|
8
|
+
LLM_REQUEST_MAX_TOKENS: 'gen_ai.request.max_tokens',
|
|
9
|
+
LLM_REQUEST_TEMPERATURE: 'gen_ai.request.temperature',
|
|
10
|
+
LLM_REQUEST_TOP_K: 'gen_ai.request.top_k',
|
|
11
|
+
LLM_REQUEST_FREQUENCY_PENALTY: 'gen_ai.request.frequency_penalty',
|
|
12
|
+
LLM_REQUEST_PRESENCE_PENALTY: 'gen_ai.request.presence_penalty',
|
|
13
|
+
LLM_REQUEST_STOP_SEQUENCES: 'gen_ai.request.stop_sequences',
|
|
14
|
+
LLM_REQUEST_LLM_IS_STREAMING: 'gen_ai.request.llm_is_streaming',
|
|
15
|
+
LLM_REQUEST_TOP_P: 'gen_ai.request.top_p',
|
|
16
|
+
LLM_USAGE_PROMPT_TOKENS: 'gen_ai.usage.prompt_tokens',
|
|
17
|
+
LLM_USAGE_COMPLETION_TOKENS: 'gen_ai.usage.completion_tokens',
|
|
18
|
+
// Vector DB
|
|
19
|
+
DB_SYSTEM: 'db.system',
|
|
20
|
+
DB_TABLE: 'db.table',
|
|
21
|
+
DB_NAMESPACE: 'db.namespace',
|
|
22
|
+
DB_ID: 'db.id',
|
|
23
|
+
DB_QUERY_TEXT: 'db.query.text',
|
|
24
|
+
DB_VECTOR: 'db.vector',
|
|
25
|
+
DB_OPERATION_NAME: 'db.operation.name',
|
|
26
|
+
DB_VECTOR_QUERY_TOP_K: 'db.vector.query.top_k',
|
|
27
|
+
DB_QUERY_EMBEDDINGS: 'db.query.embeddings',
|
|
28
|
+
DB_QUERY_RESULT: 'db.query.result',
|
|
29
|
+
// Query Embeddings
|
|
30
|
+
DB_QUERY_EMBEDDINGS_VECTOR: 'db.query.embeddings.vector',
|
|
31
|
+
// Query Result (canonical format)
|
|
32
|
+
DB_QUERY_RESULT_ID: 'db.query.result.id',
|
|
33
|
+
DB_QUERY_RESULT_SCORE: 'db.query.result.score',
|
|
34
|
+
DB_QUERY_RESULT_DISTANCE: 'db.query.result.distance',
|
|
35
|
+
DB_QUERY_RESULT_METADATA: 'db.query.result.metadata',
|
|
36
|
+
DB_QUERY_RESULT_VECTOR: 'db.query.result.vector',
|
|
37
|
+
DB_QUERY_RESULT_DOCUMENT: 'db.query.result.document'
|
|
38
|
+
};
|
|
39
|
+
exports.axSpanEvents = {
|
|
40
|
+
LLM_PROMPT: 'gen_ai.prompt'
|
|
41
|
+
};
|
|
42
|
+
var AxLLMRequestTypeValues;
|
|
43
|
+
(function (AxLLMRequestTypeValues) {
|
|
44
|
+
AxLLMRequestTypeValues["COMPLETION"] = "completion";
|
|
45
|
+
AxLLMRequestTypeValues["CHAT"] = "chat";
|
|
46
|
+
AxLLMRequestTypeValues["RERANK"] = "rerank";
|
|
47
|
+
AxLLMRequestTypeValues["UNKNOWN"] = "unknown";
|
|
48
|
+
})(AxLLMRequestTypeValues || (exports.AxLLMRequestTypeValues = AxLLMRequestTypeValues = {}));
|
|
49
|
+
var AxSpanKindValues;
|
|
50
|
+
(function (AxSpanKindValues) {
|
|
51
|
+
AxSpanKindValues["WORKFLOW"] = "workflow";
|
|
52
|
+
AxSpanKindValues["TASK"] = "task";
|
|
53
|
+
AxSpanKindValues["AGENT"] = "agent";
|
|
54
|
+
AxSpanKindValues["TOOL"] = "tool";
|
|
55
|
+
AxSpanKindValues["UNKNOWN"] = "unknown";
|
|
56
|
+
})(AxSpanKindValues || (exports.AxSpanKindValues = AxSpanKindValues = {}));
|
|
57
|
+
var AxSpanStatusCode;
|
|
58
|
+
(function (AxSpanStatusCode) {
|
|
59
|
+
AxSpanStatusCode["OK"] = "OK";
|
|
60
|
+
AxSpanStatusCode["ERROR"] = "ERROR";
|
|
61
|
+
})(AxSpanStatusCode || (exports.AxSpanStatusCode = AxSpanStatusCode = {}));
|
|
62
|
+
var AxSpanKind;
|
|
63
|
+
(function (AxSpanKind) {
|
|
64
|
+
AxSpanKind["INTERNAL"] = "INTERNAL";
|
|
65
|
+
AxSpanKind["SERVER"] = "SERVER";
|
|
66
|
+
AxSpanKind["CLIENT"] = "CLIENT";
|
|
67
|
+
AxSpanKind["PRODUCER"] = "PRODUCER";
|
|
68
|
+
AxSpanKind["CONSUMER"] = "CONSUMER";
|
|
69
|
+
})(AxSpanKind || (exports.AxSpanKind = AxSpanKind = {}));
|
|
70
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../trace/index.ts"],"names":[],"mappings":";;;AAAa,QAAA,gBAAgB,GAAG;IAC9B,MAAM;IACN,UAAU,EAAE,eAAe;IAC3B,iBAAiB,EAAE,sBAAsB;IACzC,sBAAsB,EAAE,2BAA2B;IACnD,uBAAuB,EAAE,4BAA4B;IACrD,iBAAiB,EAAE,sBAAsB;IACzC,6BAA6B,EAAE,kCAAkC;IACjE,4BAA4B,EAAE,iCAAiC;IAC/D,0BAA0B,EAAE,+BAA+B;IAC3D,4BAA4B,EAAE,iCAAiC;IAC/D,iBAAiB,EAAE,sBAAsB;IAEzC,uBAAuB,EAAE,4BAA4B;IACrD,2BAA2B,EAAE,gCAAgC;IAE7D,YAAY;IACZ,SAAS,EAAE,WAAW;IACtB,QAAQ,EAAE,UAAU;IACpB,YAAY,EAAE,cAAc;IAC5B,KAAK,EAAE,OAAO;IACd,aAAa,EAAE,eAAe;IAC9B,SAAS,EAAE,WAAW;IACtB,iBAAiB,EAAE,mBAAmB;IACtC,qBAAqB,EAAE,uBAAuB;IAE9C,mBAAmB,EAAE,qBAAqB;IAC1C,eAAe,EAAE,iBAAiB;IAElC,mBAAmB;IACnB,0BAA0B,EAAE,4BAA4B;IAExD,kCAAkC;IAClC,kBAAkB,EAAE,oBAAoB;IACxC,qBAAqB,EAAE,uBAAuB;IAC9C,wBAAwB,EAAE,0BAA0B;IACpD,wBAAwB,EAAE,0BAA0B;IACpD,sBAAsB,EAAE,wBAAwB;IAChD,wBAAwB,EAAE,0BAA0B;CACrD,CAAC;AAEW,QAAA,YAAY,GAAG;IAC1B,UAAU,EAAE,eAAe;CAC5B,CAAC;AAEF,IAAY,sBAKX;AALD,WAAY,sBAAsB;IAChC,mDAAyB,CAAA;IACzB,uCAAa,CAAA;IACb,2CAAiB,CAAA;IACjB,6CAAmB,CAAA;AACrB,CAAC,EALW,sBAAsB,sCAAtB,sBAAsB,QAKjC;AAED,IAAY,gBAMX;AAND,WAAY,gBAAgB;IAC1B,yCAAqB,CAAA;IACrB,iCAAa,CAAA;IACb,mCAAe,CAAA;IACf,iCAAa,CAAA;IACb,uCAAmB,CAAA;AACrB,CAAC,EANW,gBAAgB,gCAAhB,gBAAgB,QAM3B;AAuBD,IAAY,gBAGX;AAHD,WAAY,gBAAgB;IAC1B,6BAAS,CAAA;IACT,mCAAe,CAAA;AACjB,CAAC,EAHW,gBAAgB,gCAAhB,gBAAgB,QAG3B;AA2DD,IAAY,UAMX;AAND,WAAY,UAAU;IACpB,mCAAqB,CAAA;IACrB,+BAAiB,CAAA;IACjB,+BAAiB,CAAA;IACjB,mCAAqB,CAAA;IACrB,mCAAqB,CAAA;AACvB,CAAC,EANW,UAAU,0BAAV,UAAU,QAMrB"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
import { type ReadableStream } from 'stream/web';
|
|
2
|
+
import type { AxSpan } from '../trace/index.js';
|
|
3
|
+
/**
|
|
4
|
+
* Util: API details
|
|
5
|
+
* @export
|
|
6
|
+
*/
|
|
7
|
+
export type API = {
|
|
8
|
+
name?: string;
|
|
9
|
+
headers?: Record<string, string>;
|
|
10
|
+
put?: boolean;
|
|
11
|
+
};
|
|
12
|
+
export declare const apiCall: <TRequest = unknown, TResponse = unknown>(api: Readonly<API & {
|
|
13
|
+
url: string | URL;
|
|
14
|
+
stream?: boolean;
|
|
15
|
+
debug?: boolean;
|
|
16
|
+
fetch?: typeof fetch;
|
|
17
|
+
span?: AxSpan;
|
|
18
|
+
}>, json: TRequest) => Promise<TResponse | ReadableStream<TResponse>>;
|
|
@@ -0,0 +1,57 @@
|
|
|
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.apiCall = void 0;
|
|
7
|
+
const path_1 = __importDefault(require("path"));
|
|
8
|
+
const web_1 = require("stream/web");
|
|
9
|
+
const stream_js_1 = require("./stream.js");
|
|
10
|
+
const transform_js_1 = require("./transform.js");
|
|
11
|
+
const textDecoderStream = web_1.TextDecoderStream ?? stream_js_1.TextDecoderStreamPolyfill;
|
|
12
|
+
// eslint-disable-next-line @typescript-eslint/naming-convention
|
|
13
|
+
const apiCall = async (api, json) => {
|
|
14
|
+
const baseUrl = new URL(process.env.PROXY ?? api.url);
|
|
15
|
+
const apiPath = path_1.default.join(baseUrl.pathname, api.name ?? '/', baseUrl.search);
|
|
16
|
+
const apiUrl = new URL(apiPath, baseUrl);
|
|
17
|
+
if (api.span?.isRecording()) {
|
|
18
|
+
api.span.setAttributes({
|
|
19
|
+
'http.request.method': api.put ? 'PUT' : 'POST',
|
|
20
|
+
'url.full': apiUrl.href
|
|
21
|
+
});
|
|
22
|
+
}
|
|
23
|
+
let res;
|
|
24
|
+
try {
|
|
25
|
+
res = await (api.fetch ?? fetch)(apiUrl, {
|
|
26
|
+
method: api.put ? 'PUT' : 'POST',
|
|
27
|
+
headers: {
|
|
28
|
+
'Content-Type': 'application/json',
|
|
29
|
+
...api.headers
|
|
30
|
+
},
|
|
31
|
+
body: JSON.stringify(json)
|
|
32
|
+
});
|
|
33
|
+
if (res.status >= 400) {
|
|
34
|
+
const respBody = JSON.stringify(await res.json(), null, 2);
|
|
35
|
+
throw new Error(`${res.status}, ${res.statusText}\n:Response Body: ${respBody}`);
|
|
36
|
+
}
|
|
37
|
+
if (!api.stream) {
|
|
38
|
+
return (await res.json());
|
|
39
|
+
}
|
|
40
|
+
if (!res.body) {
|
|
41
|
+
throw new Error('Response body is null');
|
|
42
|
+
}
|
|
43
|
+
const st = res.body
|
|
44
|
+
.pipeThrough(new textDecoderStream())
|
|
45
|
+
.pipeThrough(new transform_js_1.JSONStringifyStream());
|
|
46
|
+
return st;
|
|
47
|
+
}
|
|
48
|
+
catch (e) {
|
|
49
|
+
if (api.span?.isRecording()) {
|
|
50
|
+
api.span.recordAxSpanException(e);
|
|
51
|
+
}
|
|
52
|
+
const reqBody = JSON.stringify(json, null, 2);
|
|
53
|
+
throw new Error(`API Error: ${apiUrl.href}, ${e}\nRequest Body: ${reqBody}`);
|
|
54
|
+
}
|
|
55
|
+
};
|
|
56
|
+
exports.apiCall = apiCall;
|
|
57
|
+
//# sourceMappingURL=apicall.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"apicall.js","sourceRoot":"","sources":["../../../util/apicall.ts"],"names":[],"mappings":";;;;;;AAAA,gDAAwB;AACxB,oCAGoB;AAIpB,2CAAwD;AACxD,iDAAqD;AAarD,MAAM,iBAAiB,GAAG,uBAAuB,IAAI,qCAAyB,CAAC;AAE/E,gEAAgE;AACzD,MAAM,OAAO,GAAG,KAAK,EAC1B,GAQC,EACD,IAAc,EACkC,EAAE;IAClD,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;IACtD,MAAM,OAAO,GAAG,cAAI,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,IAAI,IAAI,GAAG,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;IAC7E,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAEzC,IAAI,GAAG,CAAC,IAAI,EAAE,WAAW,EAAE,EAAE,CAAC;QAC5B,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC;YACrB,qBAAqB,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM;YAC/C,UAAU,EAAE,MAAM,CAAC,IAAI;SACxB,CAAC,CAAC;IACL,CAAC;IAED,IAAI,GAAqD,CAAC;IAE1D,IAAI,CAAC;QACH,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC,KAAK,IAAI,KAAK,CAAC,CAAC,MAAM,EAAE;YACvC,MAAM,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM;YAChC,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,GAAG,GAAG,CAAC,OAAO;aACf;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CAAC,CAAC;QAEH,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,EAAE,CAAC;YACtB,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YAC3D,MAAM,IAAI,KAAK,CACb,GAAG,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,UAAU,qBAAqB,QAAQ,EAAE,CAChE,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAc,CAAC;QACzC,CAAC;QAED,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;QAED,MAAM,EAAE,GAAG,GAAG,CAAC,IAAI;aAChB,WAAW,CAAC,IAAI,iBAAiB,EAAE,CAAC;aACpC,WAAW,CAAC,IAAI,kCAAmB,EAAa,CAAC,CAAC;QAErD,OAAO,EAAE,CAAC;IACZ,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,IAAI,GAAG,CAAC,IAAI,EAAE,WAAW,EAAE,EAAE,CAAC;YAC5B,GAAG,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAU,CAAC,CAAC;QAC7C,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAC9C,MAAM,IAAI,KAAK,CACb,cAAc,MAAM,CAAC,IAAI,KAAK,CAAC,mBAAmB,OAAO,EAAE,CAC5D,CAAC;IACJ,CAAC;AACH,CAAC,CAAC;AAhEW,QAAA,OAAO,WAgElB"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
export declare class ColorLog {
|
|
2
|
+
private readonly ANSI_WHITE_BRIGHT;
|
|
3
|
+
private readonly ANSI_GREEN_BRIGHT;
|
|
4
|
+
private readonly ANSI_BLUE_BRIGHT;
|
|
5
|
+
private readonly ANSI_YELLOW;
|
|
6
|
+
private readonly ANSI_RED;
|
|
7
|
+
private readonly ANSI_RESET;
|
|
8
|
+
private colorize;
|
|
9
|
+
whiteBright(text: string): string;
|
|
10
|
+
greenBright(text: string): string;
|
|
11
|
+
blueBright(text: string): string;
|
|
12
|
+
yellow(text: string): string;
|
|
13
|
+
red(text: string): string;
|
|
14
|
+
}
|
package/cjs/util/log.js
ADDED
|
@@ -0,0 +1,35 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.ColorLog = void 0;
|
|
4
|
+
// eslint-disable-next-line @typescript-eslint/naming-convention
|
|
5
|
+
class ColorLog {
|
|
6
|
+
// ANSI escape codes for different colors
|
|
7
|
+
ANSI_WHITE_BRIGHT = '\x1b[97m';
|
|
8
|
+
ANSI_GREEN_BRIGHT = '\x1b[92m';
|
|
9
|
+
ANSI_BLUE_BRIGHT = '\x1b[94m';
|
|
10
|
+
ANSI_YELLOW = '\x1b[93m';
|
|
11
|
+
ANSI_RED = '\x1b[91m';
|
|
12
|
+
ANSI_RESET = '\x1b[0m';
|
|
13
|
+
// Method to wrap text with the specified ANSI color code
|
|
14
|
+
colorize(text, colorCode) {
|
|
15
|
+
return `${colorCode}${text}${this.ANSI_RESET}`;
|
|
16
|
+
}
|
|
17
|
+
// Public methods to colorize text in various colors
|
|
18
|
+
whiteBright(text) {
|
|
19
|
+
return this.colorize(text, this.ANSI_WHITE_BRIGHT);
|
|
20
|
+
}
|
|
21
|
+
greenBright(text) {
|
|
22
|
+
return this.colorize(text, this.ANSI_GREEN_BRIGHT);
|
|
23
|
+
}
|
|
24
|
+
blueBright(text) {
|
|
25
|
+
return this.colorize(text, this.ANSI_BLUE_BRIGHT);
|
|
26
|
+
}
|
|
27
|
+
yellow(text) {
|
|
28
|
+
return this.colorize(text, this.ANSI_YELLOW);
|
|
29
|
+
}
|
|
30
|
+
red(text) {
|
|
31
|
+
return this.colorize(text, this.ANSI_RED);
|
|
32
|
+
}
|
|
33
|
+
}
|
|
34
|
+
exports.ColorLog = ColorLog;
|
|
35
|
+
//# sourceMappingURL=log.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"log.js","sourceRoot":"","sources":["../../../util/log.ts"],"names":[],"mappings":";;;AAAA,gEAAgE;AAChE,MAAa,QAAQ;IACnB,yCAAyC;IACxB,iBAAiB,GAAG,UAAU,CAAC;IAC/B,iBAAiB,GAAG,UAAU,CAAC;IAC/B,gBAAgB,GAAG,UAAU,CAAC;IAC9B,WAAW,GAAG,UAAU,CAAC;IACzB,QAAQ,GAAG,UAAU,CAAC;IACtB,UAAU,GAAG,SAAS,CAAC;IAExC,yDAAyD;IACjD,QAAQ,CAAC,IAAY,EAAE,SAAiB;QAC9C,OAAO,GAAG,SAAS,GAAG,IAAI,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;IACjD,CAAC;IAED,oDAAoD;IAC7C,WAAW,CAAC,IAAY;QAC7B,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACrD,CAAC;IAEM,WAAW,CAAC,IAAY;QAC7B,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACrD,CAAC;IAEM,UAAU,CAAC,IAAY;QAC5B,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;IACpD,CAAC;IAEM,MAAM,CAAC,IAAY;QACxB,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;IAC/C,CAAC;IAEM,GAAG,CAAC,IAAY;QACrB,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC5C,CAAC;CACF;AAlCD,4BAkCC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export declare function sleep(ms: number): Promise<unknown>;
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.sleep = sleep;
|
|
4
|
+
// eslint-disable-next-line @typescript-eslint/naming-convention
|
|
5
|
+
async function sleep(ms) {
|
|
6
|
+
return new Promise((resolve) => setTimeout(resolve, ms));
|
|
7
|
+
}
|
|
8
|
+
//# sourceMappingURL=other.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"other.js","sourceRoot":"","sources":["../../../util/other.ts"],"names":[],"mappings":";;AACA,sBAEC;AAHD,gEAAgE;AACzD,KAAK,UAAU,KAAK,CAAC,EAAU;IACpC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3D,CAAC"}
|
|
@@ -0,0 +1,24 @@
|
|
|
1
|
+
export interface AxRateLimiterTokenUsageOptions {
|
|
2
|
+
debug?: boolean;
|
|
3
|
+
}
|
|
4
|
+
export declare class AxRateLimiterTokenUsage {
|
|
5
|
+
private options?;
|
|
6
|
+
private maxTokens;
|
|
7
|
+
private refillRate;
|
|
8
|
+
private currentTokens;
|
|
9
|
+
private lastRefillTime;
|
|
10
|
+
constructor(maxTokens: number, refillRate: number, options?: Readonly<AxRateLimiterTokenUsageOptions>);
|
|
11
|
+
private refillTokens;
|
|
12
|
+
private waitUntilTokensAvailable;
|
|
13
|
+
acquire(tokens: number): Promise<void>;
|
|
14
|
+
}
|
|
15
|
+
/**
|
|
16
|
+
* Example usage of the rate limiter. Limits to 5800 tokens per minute.
|
|
17
|
+
const rateLimiter = new AxRateLimiterTokenUsage(5800, 5800 / 60);
|
|
18
|
+
|
|
19
|
+
const axRateLimiterFunction = async (func, info) => {
|
|
20
|
+
const totalTokens = info.modelUsage?.totalTokens || 0;
|
|
21
|
+
await rateLimiter.acquire(totalTokens);
|
|
22
|
+
return func();
|
|
23
|
+
};
|
|
24
|
+
**/
|
|
@@ -0,0 +1,53 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.AxRateLimiterTokenUsage = void 0;
|
|
4
|
+
const log_js_1 = require("./log.js");
|
|
5
|
+
const colorLog = new log_js_1.ColorLog();
|
|
6
|
+
class AxRateLimiterTokenUsage {
|
|
7
|
+
options;
|
|
8
|
+
maxTokens;
|
|
9
|
+
refillRate;
|
|
10
|
+
currentTokens;
|
|
11
|
+
lastRefillTime;
|
|
12
|
+
constructor(maxTokens, refillRate, options) {
|
|
13
|
+
this.maxTokens = maxTokens;
|
|
14
|
+
this.refillRate = refillRate;
|
|
15
|
+
this.currentTokens = maxTokens;
|
|
16
|
+
this.lastRefillTime = Date.now();
|
|
17
|
+
this.options = options;
|
|
18
|
+
}
|
|
19
|
+
refillTokens() {
|
|
20
|
+
const now = Date.now();
|
|
21
|
+
const timeElapsed = (now - this.lastRefillTime) / 1000; // Convert ms to seconds
|
|
22
|
+
const tokensToAdd = timeElapsed * this.refillRate;
|
|
23
|
+
this.currentTokens = Math.min(this.maxTokens, this.currentTokens + tokensToAdd);
|
|
24
|
+
this.lastRefillTime = now;
|
|
25
|
+
}
|
|
26
|
+
async waitUntilTokensAvailable(tokens) {
|
|
27
|
+
this.refillTokens();
|
|
28
|
+
if (this.currentTokens >= tokens) {
|
|
29
|
+
this.currentTokens -= tokens;
|
|
30
|
+
return;
|
|
31
|
+
}
|
|
32
|
+
if (this.options?.debug) {
|
|
33
|
+
console.log(colorLog.red(`Rate limiter: Waiting for ${tokens - this.currentTokens} tokens`));
|
|
34
|
+
}
|
|
35
|
+
await new Promise((resolve) => setTimeout(resolve, 100)); // Wait for 100ms before checking again
|
|
36
|
+
return this.waitUntilTokensAvailable(tokens); // Recursive call
|
|
37
|
+
}
|
|
38
|
+
async acquire(tokens) {
|
|
39
|
+
await this.waitUntilTokensAvailable(tokens);
|
|
40
|
+
}
|
|
41
|
+
}
|
|
42
|
+
exports.AxRateLimiterTokenUsage = AxRateLimiterTokenUsage;
|
|
43
|
+
/**
|
|
44
|
+
* Example usage of the rate limiter. Limits to 5800 tokens per minute.
|
|
45
|
+
const rateLimiter = new AxRateLimiterTokenUsage(5800, 5800 / 60);
|
|
46
|
+
|
|
47
|
+
const axRateLimiterFunction = async (func, info) => {
|
|
48
|
+
const totalTokens = info.modelUsage?.totalTokens || 0;
|
|
49
|
+
await rateLimiter.acquire(totalTokens);
|
|
50
|
+
return func();
|
|
51
|
+
};
|
|
52
|
+
**/
|
|
53
|
+
//# sourceMappingURL=rate-limit.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"rate-limit.js","sourceRoot":"","sources":["../../../util/rate-limit.ts"],"names":[],"mappings":";;;AAAA,qCAAoC;AAEpC,MAAM,QAAQ,GAAG,IAAI,iBAAQ,EAAE,CAAC;AAMhC,MAAa,uBAAuB;IAC1B,OAAO,CAA4C;IACnD,SAAS,CAAS;IAClB,UAAU,CAAS;IACnB,aAAa,CAAS;IACtB,cAAc,CAAS;IAE/B,YACE,SAAiB,EACjB,UAAkB,EAClB,OAAkD;QAElD,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;QAC/B,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACjC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAEO,YAAY;QAClB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,WAAW,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,IAAI,CAAC,CAAC,wBAAwB;QAChF,MAAM,WAAW,GAAG,WAAW,GAAG,IAAI,CAAC,UAAU,CAAC;QAClD,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,GAAG,CAC3B,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,aAAa,GAAG,WAAW,CACjC,CAAC;QACF,IAAI,CAAC,cAAc,GAAG,GAAG,CAAC;IAC5B,CAAC;IAEO,KAAK,CAAC,wBAAwB,CAAC,MAAc;QACnD,IAAI,CAAC,YAAY,EAAE,CAAC;QACpB,IAAI,IAAI,CAAC,aAAa,IAAI,MAAM,EAAE,CAAC;YACjC,IAAI,CAAC,aAAa,IAAI,MAAM,CAAC;YAC7B,OAAO;QACT,CAAC;QACD,IAAI,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC;YACxB,OAAO,CAAC,GAAG,CACT,QAAQ,CAAC,GAAG,CACV,6BAA6B,MAAM,GAAG,IAAI,CAAC,aAAa,SAAS,CAClE,CACF,CAAC;QACJ,CAAC;QACD,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,uCAAuC;QACjG,OAAO,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,CAAC,CAAC,iBAAiB;IACjE,CAAC;IAEM,KAAK,CAAC,OAAO,CAAC,MAAc;QACjC,MAAM,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,CAAC;IAC9C,CAAC;CACF;AAlDD,0DAkDC;AAED;;;;;;;;;GASG"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import { TransformStream } from 'stream/web';
|
|
2
|
+
export interface TextDecoderCommon {
|
|
3
|
+
readonly encoding: string;
|
|
4
|
+
readonly fatal: boolean;
|
|
5
|
+
readonly ignoreBOM: boolean;
|
|
6
|
+
}
|
|
7
|
+
export declare class TextDecoderStreamPolyfill extends TransformStream<ArrayBuffer | Uint8Array, string> {
|
|
8
|
+
constructor();
|
|
9
|
+
}
|
|
@@ -0,0 +1,33 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.TextDecoderStreamPolyfill = void 0;
|
|
4
|
+
const web_1 = require("stream/web");
|
|
5
|
+
class TextDecodeTransformer {
|
|
6
|
+
decoder;
|
|
7
|
+
constructor() {
|
|
8
|
+
this.decoder = new TextDecoder();
|
|
9
|
+
}
|
|
10
|
+
transform(chunk, controller) {
|
|
11
|
+
if (!(chunk instanceof ArrayBuffer || ArrayBuffer.isView(chunk))) {
|
|
12
|
+
throw new TypeError('Input data must be a BufferSource');
|
|
13
|
+
}
|
|
14
|
+
const text = this.decoder.decode(chunk, { stream: true });
|
|
15
|
+
if (text.length !== 0) {
|
|
16
|
+
controller.enqueue(text);
|
|
17
|
+
}
|
|
18
|
+
}
|
|
19
|
+
flush(controller) {
|
|
20
|
+
const text = this.decoder.decode();
|
|
21
|
+
if (text.length !== 0) {
|
|
22
|
+
controller.enqueue(text);
|
|
23
|
+
}
|
|
24
|
+
}
|
|
25
|
+
}
|
|
26
|
+
// eslint-disable-next-line @typescript-eslint/naming-convention
|
|
27
|
+
class TextDecoderStreamPolyfill extends web_1.TransformStream {
|
|
28
|
+
constructor() {
|
|
29
|
+
super(new TextDecodeTransformer());
|
|
30
|
+
}
|
|
31
|
+
}
|
|
32
|
+
exports.TextDecoderStreamPolyfill = TextDecoderStreamPolyfill;
|
|
33
|
+
//# sourceMappingURL=stream.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"stream.js","sourceRoot":"","sources":["../../../util/stream.ts"],"names":[],"mappings":";;;AAAA,oCAIoB;AASpB,MAAM,qBAAqB;IAGjB,OAAO,CAAC;IAEhB;QACE,IAAI,CAAC,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;IACnC,CAAC;IAED,SAAS,CACP,KAA+B,EAC/B,UAAoD;QAEpD,IAAI,CAAC,CAAC,KAAK,YAAY,WAAW,IAAI,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;YACjE,MAAM,IAAI,SAAS,CAAC,mCAAmC,CAAC,CAAC;QAC3D,CAAC;QACD,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1D,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtB,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAoD;QACxD,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;QACnC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtB,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;CACF;AAED,gEAAgE;AAChE,MAAa,yBAA0B,SAAQ,qBAG9C;IACC;QACE,KAAK,CAAC,IAAI,qBAAqB,EAAE,CAAC,CAAC;IACrC,CAAC;CACF;AAPD,8DAOC"}
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
import { TransformStream } from 'stream/web';
|
|
2
|
+
export declare class JSONStringifyStream<O> extends TransformStream<string, O> {
|
|
3
|
+
constructor();
|
|
4
|
+
}
|
|
5
|
+
export declare class RespTransformStream<I, O> extends TransformStream<I, O> {
|
|
6
|
+
constructor(transformFn: (arg0: I) => O, doneCallback?: (args0: readonly O[]) => Promise<void>);
|
|
7
|
+
}
|
|
@@ -0,0 +1,64 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.RespTransformStream = exports.JSONStringifyStream = void 0;
|
|
4
|
+
const web_1 = require("stream/web");
|
|
5
|
+
class JSONTransformer {
|
|
6
|
+
async transform(obj, controller) {
|
|
7
|
+
(obj.split('\n') ?? [])
|
|
8
|
+
.map(extractJson)
|
|
9
|
+
.map((v) => {
|
|
10
|
+
try {
|
|
11
|
+
return v && v.length > 0 ? JSON.parse(v) : null;
|
|
12
|
+
}
|
|
13
|
+
catch (e) {
|
|
14
|
+
return null;
|
|
15
|
+
}
|
|
16
|
+
})
|
|
17
|
+
.filter((v) => v)
|
|
18
|
+
.forEach((v) => v && controller.enqueue(v));
|
|
19
|
+
}
|
|
20
|
+
}
|
|
21
|
+
// eslint-disable-next-line @typescript-eslint/naming-convention
|
|
22
|
+
class JSONStringifyStream extends web_1.TransformStream {
|
|
23
|
+
constructor() {
|
|
24
|
+
super(new JSONTransformer());
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
exports.JSONStringifyStream = JSONStringifyStream;
|
|
28
|
+
class TypeTransformer {
|
|
29
|
+
buffer;
|
|
30
|
+
doneCallback;
|
|
31
|
+
transformFn;
|
|
32
|
+
constructor(transformFn, doneCallback) {
|
|
33
|
+
this.transformFn = transformFn;
|
|
34
|
+
this.doneCallback = doneCallback;
|
|
35
|
+
this.buffer = doneCallback ? [] : undefined;
|
|
36
|
+
}
|
|
37
|
+
async transform(obj, controller) {
|
|
38
|
+
const val = this.transformFn(obj);
|
|
39
|
+
if (val) {
|
|
40
|
+
controller.enqueue(val);
|
|
41
|
+
this.buffer?.push(val);
|
|
42
|
+
}
|
|
43
|
+
}
|
|
44
|
+
async flush(controller) {
|
|
45
|
+
await this.doneCallback?.(this.buffer ?? []);
|
|
46
|
+
controller.terminate();
|
|
47
|
+
}
|
|
48
|
+
}
|
|
49
|
+
// eslint-disable-next-line @typescript-eslint/naming-convention
|
|
50
|
+
class RespTransformStream extends web_1.TransformStream {
|
|
51
|
+
constructor(transformFn, doneCallback) {
|
|
52
|
+
super(new TypeTransformer(transformFn, doneCallback));
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
exports.RespTransformStream = RespTransformStream;
|
|
56
|
+
const extractJson = (str) => {
|
|
57
|
+
const startIndex = str.indexOf('{');
|
|
58
|
+
const endIndex = str.lastIndexOf('}');
|
|
59
|
+
if (startIndex !== -1 && endIndex !== -1) {
|
|
60
|
+
return str.substring(startIndex, endIndex + 1);
|
|
61
|
+
}
|
|
62
|
+
return null;
|
|
63
|
+
};
|
|
64
|
+
//# sourceMappingURL=transform.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"transform.js","sourceRoot":"","sources":["../../../util/transform.ts"],"names":[],"mappings":";;;AAAA,oCAIoB;AAEpB,MAAM,eAAe;IACnB,KAAK,CAAC,SAAS,CACb,GAAW,EACX,UAA+C;QAE/C,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;aACpB,GAAG,CAAC,WAAW,CAAC;aAChB,GAAG,CAAI,CAAC,CAAC,EAAE,EAAE;YACZ,IAAI,CAAC;gBACH,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAC5D,CAAC;YAAC,OAAO,CAAU,EAAE,CAAC;gBACpB,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC,CAAC;aACD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;aAChB,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC;CACF;AAED,gEAAgE;AAChE,MAAa,mBAAuB,SAAQ,qBAA0B;IACpE;QACE,KAAK,CAAC,IAAI,eAAe,EAAK,CAAC,CAAC;IAClC,CAAC;CACF;AAJD,kDAIC;AAED,MAAM,eAAe;IACX,MAAM,CAAO;IACb,YAAY,CAA0C;IACtD,WAAW,CAAiB;IAEpC,YACE,WAA2B,EAC3B,YAAqD;QAErD,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,GAAM,EAAE,UAA+C;QACrE,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QAClC,IAAI,GAAG,EAAE,CAAC;YACR,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YACxB,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,UAA+C;QACzD,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;QAC7C,UAAU,CAAC,SAAS,EAAE,CAAC;IACzB,CAAC;CACF;AAED,gEAAgE;AAChE,MAAa,mBAA0B,SAAQ,qBAAqB;IAClE,YACE,WAA2B,EAC3B,YAAqD;QAErD,KAAK,CAAC,IAAI,eAAe,CAAO,WAAW,EAAE,YAAY,CAAC,CAAC,CAAC;IAC9D,CAAC;CACF;AAPD,kDAOC;AAED,MAAM,WAAW,GAAG,CAAC,GAAW,EAAiB,EAAE;IACjD,MAAM,UAAU,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACpC,MAAM,QAAQ,GAAG,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;IAEtC,IAAI,UAAU,KAAK,CAAC,CAAC,IAAI,QAAQ,KAAK,CAAC,CAAC,EAAE,CAAC;QACzC,OAAO,GAAG,CAAC,SAAS,CAAC,UAAU,EAAE,QAAQ,GAAG,CAAC,CAAC,CAAC;IACjD,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC,CAAC"}
|
package/db/base.d.ts
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
1
|
+
import { type AxSpan, type AxTracer } from '../trace/index.js';
|
|
2
|
+
import type { AxDBQueryRequest, AxDBQueryResponse, AxDBService, AxDBUpsertRequest, AxDBUpsertResponse } from './types.js';
|
|
3
|
+
export interface AxDBBaseArgs {
|
|
4
|
+
fetch?: typeof fetch;
|
|
5
|
+
tracer?: AxTracer;
|
|
6
|
+
}
|
|
7
|
+
export interface AxDBBaseOpOptions {
|
|
8
|
+
span?: AxSpan;
|
|
9
|
+
}
|
|
10
|
+
export declare class AxDBBase implements AxDBService {
|
|
11
|
+
protected name: string;
|
|
12
|
+
protected fetch?: typeof fetch;
|
|
13
|
+
private tracer?;
|
|
14
|
+
_upsert?: (req: Readonly<AxDBUpsertRequest>, update?: boolean, options?: Readonly<AxDBBaseOpOptions>) => Promise<AxDBUpsertResponse>;
|
|
15
|
+
_batchUpsert?: (batchReq: Readonly<AxDBUpsertRequest[]>, update?: boolean, options?: Readonly<AxDBBaseOpOptions>) => Promise<AxDBUpsertResponse>;
|
|
16
|
+
_query?: (req: Readonly<AxDBQueryRequest>, options?: Readonly<AxDBBaseOpOptions>) => Promise<AxDBQueryResponse>;
|
|
17
|
+
constructor({ name, fetch, tracer }: Readonly<AxDBBaseArgs & {
|
|
18
|
+
name: string;
|
|
19
|
+
}>);
|
|
20
|
+
upsert(req: Readonly<AxDBUpsertRequest>, update?: boolean): Promise<AxDBUpsertResponse>;
|
|
21
|
+
batchUpsert(req: Readonly<AxDBUpsertRequest[]>, update?: boolean): Promise<AxDBUpsertResponse>;
|
|
22
|
+
query(req: Readonly<AxDBQueryRequest>): Promise<AxDBQueryResponse>;
|
|
23
|
+
}
|
package/db/base.js
ADDED
|
@@ -0,0 +1,87 @@
|
|
|
1
|
+
import { axSpanAttributes, AxSpanKind } from '../trace/index.js';
|
|
2
|
+
export class AxDBBase {
|
|
3
|
+
name;
|
|
4
|
+
fetch;
|
|
5
|
+
tracer;
|
|
6
|
+
_upsert;
|
|
7
|
+
_batchUpsert;
|
|
8
|
+
_query;
|
|
9
|
+
constructor({ name, fetch, tracer }) {
|
|
10
|
+
this.name = name;
|
|
11
|
+
this.fetch = fetch;
|
|
12
|
+
this.tracer = tracer;
|
|
13
|
+
}
|
|
14
|
+
async upsert(req, update) {
|
|
15
|
+
if (!this._upsert) {
|
|
16
|
+
throw new Error('upsert() not implemented');
|
|
17
|
+
}
|
|
18
|
+
if (!this.tracer) {
|
|
19
|
+
return await this._upsert(req, update);
|
|
20
|
+
}
|
|
21
|
+
return await this.tracer?.startActiveSpan('DB Upsert Request', {
|
|
22
|
+
kind: AxSpanKind.SERVER,
|
|
23
|
+
attributes: {
|
|
24
|
+
[axSpanAttributes.DB_SYSTEM]: this.name,
|
|
25
|
+
[axSpanAttributes.DB_OPERATION_NAME]: 'upsert',
|
|
26
|
+
[axSpanAttributes.DB_TABLE]: req.table,
|
|
27
|
+
[axSpanAttributes.DB_NAMESPACE]: req.namespace,
|
|
28
|
+
[axSpanAttributes.DB_OPERATION_NAME]: update ? 'update' : 'insert'
|
|
29
|
+
}
|
|
30
|
+
}, async (span) => {
|
|
31
|
+
const res = await this._upsert(req, update, { span });
|
|
32
|
+
span.end();
|
|
33
|
+
return res;
|
|
34
|
+
});
|
|
35
|
+
}
|
|
36
|
+
async batchUpsert(req, update) {
|
|
37
|
+
if (!this._batchUpsert) {
|
|
38
|
+
throw new Error('batchUpsert() not implemented');
|
|
39
|
+
}
|
|
40
|
+
if (req.length == 0) {
|
|
41
|
+
throw new Error('Batch request is empty');
|
|
42
|
+
}
|
|
43
|
+
if (!req[0]) {
|
|
44
|
+
throw new Error('Batch request is invalid first element is undefined');
|
|
45
|
+
}
|
|
46
|
+
if (!this.tracer) {
|
|
47
|
+
return await this._batchUpsert(req, update);
|
|
48
|
+
}
|
|
49
|
+
return await this.tracer?.startActiveSpan('DB Batch Upsert Request', {
|
|
50
|
+
kind: AxSpanKind.SERVER,
|
|
51
|
+
attributes: {
|
|
52
|
+
[axSpanAttributes.DB_SYSTEM]: this.name,
|
|
53
|
+
[axSpanAttributes.DB_OPERATION_NAME]: 'upsert',
|
|
54
|
+
[axSpanAttributes.DB_TABLE]: req[0].table,
|
|
55
|
+
[axSpanAttributes.DB_NAMESPACE]: req[0].namespace,
|
|
56
|
+
[axSpanAttributes.DB_OPERATION_NAME]: update ? 'update' : 'insert'
|
|
57
|
+
}
|
|
58
|
+
}, async (span) => {
|
|
59
|
+
const res = await this._batchUpsert(req, update, { span });
|
|
60
|
+
span.end();
|
|
61
|
+
return res;
|
|
62
|
+
});
|
|
63
|
+
}
|
|
64
|
+
async query(req) {
|
|
65
|
+
if (!this._query) {
|
|
66
|
+
throw new Error('query() not implemented');
|
|
67
|
+
}
|
|
68
|
+
if (!this.tracer) {
|
|
69
|
+
return await this._query(req);
|
|
70
|
+
}
|
|
71
|
+
return await this.tracer?.startActiveSpan('DB Query Request', {
|
|
72
|
+
kind: AxSpanKind.SERVER,
|
|
73
|
+
attributes: {
|
|
74
|
+
[axSpanAttributes.DB_SYSTEM]: this.name,
|
|
75
|
+
[axSpanAttributes.DB_OPERATION_NAME]: 'upsert',
|
|
76
|
+
[axSpanAttributes.DB_TABLE]: req.table,
|
|
77
|
+
[axSpanAttributes.DB_NAMESPACE]: req.namespace,
|
|
78
|
+
[axSpanAttributes.DB_OPERATION_NAME]: 'query'
|
|
79
|
+
}
|
|
80
|
+
}, async (span) => {
|
|
81
|
+
const res = await this._query(req, { span });
|
|
82
|
+
span.end();
|
|
83
|
+
return res;
|
|
84
|
+
});
|
|
85
|
+
}
|
|
86
|
+
}
|
|
87
|
+
//# sourceMappingURL=base.js.map
|
package/db/base.js.map
ADDED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base.js","sourceRoot":"","sources":["../../db/base.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,gBAAgB,EAChB,UAAU,EAEX,MAAM,mBAAmB,CAAC;AAmB3B,MAAM,OAAO,QAAQ;IACT,IAAI,CAAS;IACb,KAAK,CAAgB;IACvB,MAAM,CAAY;IAE1B,OAAO,CAI0B;IAEjC,YAAY,CAIqB;IAEjC,MAAM,CAG0B;IAEhC,YAAY,EACV,IAAI,EACJ,KAAK,EACL,MAAM,EACoC;QAC1C,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,MAAM,CACV,GAAgC,EAChC,MAAgB;QAEhB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAC9C,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACzC,CAAC;QAED,OAAO,MAAM,IAAI,CAAC,MAAM,EAAE,eAAe,CACvC,mBAAmB,EACnB;YACE,IAAI,EAAE,UAAU,CAAC,MAAM;YACvB,UAAU,EAAE;gBACV,CAAC,gBAAgB,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,IAAI;gBACvC,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,EAAE,QAAQ;gBAC9C,CAAC,gBAAgB,CAAC,QAAQ,CAAC,EAAE,GAAG,CAAC,KAAK;gBACtC,CAAC,gBAAgB,CAAC,YAAY,CAAC,EAAE,GAAG,CAAC,SAAS;gBAC9C,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ;aACnE;SACF,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAQ,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;YACvD,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,OAAO,GAAG,CAAC;QACb,CAAC,CACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,WAAW,CACf,GAAkC,EAClC,MAAgB;QAEhB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;QACnD,CAAC;QACD,IAAI,GAAG,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAC5C,CAAC;QACD,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;QACzE,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,OAAO,MAAM,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAC9C,CAAC;QAED,OAAO,MAAM,IAAI,CAAC,MAAM,EAAE,eAAe,CACvC,yBAAyB,EACzB;YACE,IAAI,EAAE,UAAU,CAAC,MAAM;YACvB,UAAU,EAAE;gBACV,CAAC,gBAAgB,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,IAAI;gBACvC,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,EAAE,QAAQ;gBAC9C,CAAC,gBAAgB,CAAC,QAAQ,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK;gBACzC,CAAC,gBAAgB,CAAC,YAAY,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS;gBACjD,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ;aACnE;SACF,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,YAAa,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;YAC5D,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,OAAO,GAAG,CAAC;QACb,CAAC,CACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,GAA+B;QACzC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QAC7C,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAChC,CAAC;QAED,OAAO,MAAM,IAAI,CAAC,MAAM,EAAE,eAAe,CACvC,kBAAkB,EAClB;YACE,IAAI,EAAE,UAAU,CAAC,MAAM;YACvB,UAAU,EAAE;gBACV,CAAC,gBAAgB,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,IAAI;gBACvC,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,EAAE,QAAQ;gBAC9C,CAAC,gBAAgB,CAAC,QAAQ,CAAC,EAAE,GAAG,CAAC,KAAK;gBACtC,CAAC,gBAAgB,CAAC,YAAY,CAAC,EAAE,GAAG,CAAC,SAAS;gBAC9C,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,EAAE,OAAO;aAC9C;SACF,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAO,CAAC,GAAG,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;YAC9C,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,OAAO,GAAG,CAAC;QACb,CAAC,CACF,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
import { AxDBBase, type AxDBBaseArgs, type AxDBBaseOpOptions } from './base.js';
|
|
2
|
+
import type { AxDBQueryRequest, AxDBQueryResponse, AxDBUpsertRequest, AxDBUpsertResponse } from './types.js';
|
|
3
|
+
export type AxDBCloudflareOpOptions = AxDBBaseOpOptions;
|
|
4
|
+
export interface AxDBCloudflareArgs extends AxDBBaseArgs {
|
|
5
|
+
name: 'cloudflare';
|
|
6
|
+
apiKey: string;
|
|
7
|
+
accountId: string;
|
|
8
|
+
fetch?: typeof fetch;
|
|
9
|
+
}
|
|
10
|
+
/**
|
|
11
|
+
* Cloudflare: DB Service
|
|
12
|
+
* @export
|
|
13
|
+
*/
|
|
14
|
+
export declare class AxDBCloudflare extends AxDBBase {
|
|
15
|
+
private apiKey;
|
|
16
|
+
private accountId;
|
|
17
|
+
constructor({ apiKey, accountId, fetch, tracer }: Readonly<Omit<AxDBCloudflareArgs, 'name'>>);
|
|
18
|
+
_upsert: (req: Readonly<AxDBUpsertRequest>, update?: boolean, options?: Readonly<AxDBCloudflareOpOptions>) => Promise<AxDBUpsertResponse>;
|
|
19
|
+
batchUpsert: (batchReq: Readonly<AxDBUpsertRequest[]>, update?: boolean, options?: Readonly<AxDBCloudflareOpOptions>) => Promise<AxDBUpsertResponse>;
|
|
20
|
+
query: (req: Readonly<AxDBQueryRequest>, options?: Readonly<AxDBCloudflareOpOptions>) => Promise<AxDBQueryResponse>;
|
|
21
|
+
}
|