@ax-llm/ax 9.0.3 → 9.0.5

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (145) hide show
  1. package/README.md +21 -8
  2. package/build/module/src/ai/anthropic/api.d.ts +10 -9
  3. package/build/module/src/ai/anthropic/api.js +12 -8
  4. package/build/module/src/ai/anthropic/api.js.map +1 -1
  5. package/build/module/src/ai/anthropic/info.js +12 -6
  6. package/build/module/src/ai/anthropic/info.js.map +1 -1
  7. package/build/module/src/ai/anthropic/types.d.ts +16 -15
  8. package/build/module/src/ai/anthropic/types.js +9 -8
  9. package/build/module/src/ai/anthropic/types.js.map +1 -1
  10. package/build/module/src/ai/azure-openai/api.d.ts +11 -10
  11. package/build/module/src/ai/azure-openai/api.js +11 -7
  12. package/build/module/src/ai/azure-openai/api.js.map +1 -1
  13. package/build/module/src/ai/cohere/api.d.ts +13 -12
  14. package/build/module/src/ai/cohere/api.js +17 -13
  15. package/build/module/src/ai/cohere/api.js.map +1 -1
  16. package/build/module/src/ai/cohere/info.js +9 -9
  17. package/build/module/src/ai/cohere/info.js.map +1 -1
  18. package/build/module/src/ai/cohere/types.d.ts +12 -12
  19. package/build/module/src/ai/cohere/types.js +14 -14
  20. package/build/module/src/ai/cohere/types.js.map +1 -1
  21. package/build/module/src/ai/deepseek/api.d.ts +9 -8
  22. package/build/module/src/ai/deepseek/api.js +13 -9
  23. package/build/module/src/ai/deepseek/api.js.map +1 -1
  24. package/build/module/src/ai/deepseek/info.js +3 -3
  25. package/build/module/src/ai/deepseek/info.js.map +1 -1
  26. package/build/module/src/ai/deepseek/types.d.ts +1 -1
  27. package/build/module/src/ai/deepseek/types.js +5 -5
  28. package/build/module/src/ai/deepseek/types.js.map +1 -1
  29. package/build/module/src/ai/google-gemini/api.d.ts +15 -14
  30. package/build/module/src/ai/google-gemini/api.js +27 -23
  31. package/build/module/src/ai/google-gemini/api.js.map +1 -1
  32. package/build/module/src/ai/google-gemini/info.d.ts +1 -1
  33. package/build/module/src/ai/google-gemini/info.js +5 -5
  34. package/build/module/src/ai/google-gemini/info.js.map +1 -1
  35. package/build/module/src/ai/google-gemini/types.d.ts +32 -32
  36. package/build/module/src/ai/google-gemini/types.js +25 -25
  37. package/build/module/src/ai/google-gemini/types.js.map +1 -1
  38. package/build/module/src/ai/groq/api.d.ts +8 -7
  39. package/build/module/src/ai/groq/api.js +12 -8
  40. package/build/module/src/ai/groq/api.js.map +1 -1
  41. package/build/module/src/ai/groq/types.d.ts +1 -1
  42. package/build/module/src/ai/groq/types.js +7 -7
  43. package/build/module/src/ai/groq/types.js.map +1 -1
  44. package/build/module/src/ai/huggingface/api.d.ts +10 -9
  45. package/build/module/src/ai/huggingface/api.js +14 -10
  46. package/build/module/src/ai/huggingface/api.js.map +1 -1
  47. package/build/module/src/ai/huggingface/types.d.ts +6 -6
  48. package/build/module/src/ai/huggingface/types.js +4 -4
  49. package/build/module/src/ai/huggingface/types.js.map +1 -1
  50. package/build/module/src/ai/index.d.ts +1 -13
  51. package/build/module/src/ai/index.js +1 -39
  52. package/build/module/src/ai/index.js.map +1 -1
  53. package/build/module/src/ai/mistral/api.d.ts +9 -8
  54. package/build/module/src/ai/mistral/api.js +14 -10
  55. package/build/module/src/ai/mistral/api.js.map +1 -1
  56. package/build/module/src/ai/mistral/info.js +6 -6
  57. package/build/module/src/ai/mistral/info.js.map +1 -1
  58. package/build/module/src/ai/mistral/types.d.ts +2 -2
  59. package/build/module/src/ai/mistral/types.js +12 -12
  60. package/build/module/src/ai/mistral/types.js.map +1 -1
  61. package/build/module/src/ai/ollama/api.d.ts +10 -9
  62. package/build/module/src/ai/ollama/api.js +10 -10
  63. package/build/module/src/ai/ollama/api.js.map +1 -1
  64. package/build/module/src/ai/openai/api.d.ts +15 -14
  65. package/build/module/src/ai/openai/api.js +23 -19
  66. package/build/module/src/ai/openai/api.js.map +1 -1
  67. package/build/module/src/ai/openai/info.js +8 -8
  68. package/build/module/src/ai/openai/info.js.map +1 -1
  69. package/build/module/src/ai/openai/types.d.ts +17 -17
  70. package/build/module/src/ai/openai/types.js +17 -17
  71. package/build/module/src/ai/openai/types.js.map +1 -1
  72. package/build/module/src/ai/together/api.d.ts +8 -7
  73. package/build/module/src/ai/together/api.js +9 -5
  74. package/build/module/src/ai/together/api.js.map +1 -1
  75. package/build/module/src/ai/types.d.ts +2 -2
  76. package/build/module/src/ai/wrap.d.ts +47 -0
  77. package/build/module/src/ai/wrap.js +95 -0
  78. package/build/module/src/ai/wrap.js.map +1 -0
  79. package/build/module/src/db/base.d.ts +7 -7
  80. package/build/module/src/db/base.js +1 -1
  81. package/build/module/src/db/cloudflare.d.ts +9 -9
  82. package/build/module/src/db/cloudflare.js +2 -2
  83. package/build/module/src/db/cloudflare.js.map +1 -1
  84. package/build/module/src/db/index.d.ts +1 -6
  85. package/build/module/src/db/index.js +1 -17
  86. package/build/module/src/db/index.js.map +1 -1
  87. package/build/module/src/db/memory.d.ts +9 -9
  88. package/build/module/src/db/memory.js +2 -2
  89. package/build/module/src/db/memory.js.map +1 -1
  90. package/build/module/src/db/pinecone.d.ts +9 -9
  91. package/build/module/src/db/pinecone.js +2 -2
  92. package/build/module/src/db/pinecone.js.map +1 -1
  93. package/build/module/src/db/weaviate.d.ts +9 -9
  94. package/build/module/src/db/weaviate.js +2 -2
  95. package/build/module/src/db/weaviate.js.map +1 -1
  96. package/build/module/src/db/wrap.d.ts +13 -0
  97. package/build/module/src/db/wrap.js +35 -0
  98. package/build/module/src/db/wrap.js.map +1 -0
  99. package/build/module/src/dsp/generate.js +8 -2
  100. package/build/module/src/dsp/generate.js.map +1 -1
  101. package/build/module/src/dsp/program.d.ts +1 -0
  102. package/build/module/src/dsp/program.js.map +1 -1
  103. package/build/module/src/dsp/router.js +2 -2
  104. package/build/module/src/examples/agent.js +3 -2
  105. package/build/module/src/examples/agent.js.map +1 -1
  106. package/build/module/src/examples/balancer.d.ts +1 -0
  107. package/build/module/src/examples/balancer.js +26 -0
  108. package/build/module/src/examples/balancer.js.map +1 -0
  109. package/build/module/src/examples/chain-of-thought.js +3 -2
  110. package/build/module/src/examples/chain-of-thought.js.map +1 -1
  111. package/build/module/src/examples/customer-support.js +3 -2
  112. package/build/module/src/examples/customer-support.js.map +1 -1
  113. package/build/module/src/examples/fibonacci.js +3 -2
  114. package/build/module/src/examples/fibonacci.js.map +1 -1
  115. package/build/module/src/examples/food-search.js +3 -2
  116. package/build/module/src/examples/food-search.js.map +1 -1
  117. package/build/module/src/examples/marketing.js +3 -2
  118. package/build/module/src/examples/marketing.js.map +1 -1
  119. package/build/module/src/examples/multi-modal.js +4 -3
  120. package/build/module/src/examples/multi-modal.js.map +1 -1
  121. package/build/module/src/examples/qna-tune.js +3 -2
  122. package/build/module/src/examples/qna-tune.js.map +1 -1
  123. package/build/module/src/examples/qna-use-tuned.js +3 -2
  124. package/build/module/src/examples/qna-use-tuned.js.map +1 -1
  125. package/build/module/src/examples/rag-docs.js +4 -3
  126. package/build/module/src/examples/rag-docs.js.map +1 -1
  127. package/build/module/src/examples/rag.js +3 -2
  128. package/build/module/src/examples/rag.js.map +1 -1
  129. package/build/module/src/examples/react.js +3 -2
  130. package/build/module/src/examples/react.js.map +1 -1
  131. package/build/module/src/examples/routing.js +3 -2
  132. package/build/module/src/examples/routing.js.map +1 -1
  133. package/build/module/src/examples/smart-home.js +3 -2
  134. package/build/module/src/examples/smart-home.js.map +1 -1
  135. package/build/module/src/examples/streaming1.js +8 -6
  136. package/build/module/src/examples/streaming1.js.map +1 -1
  137. package/build/module/src/examples/streaming2.js +7 -5
  138. package/build/module/src/examples/streaming2.js.map +1 -1
  139. package/build/module/src/examples/summarize.js +6 -4
  140. package/build/module/src/examples/summarize.js.map +1 -1
  141. package/build/module/src/examples/vectordb.js +4 -4
  142. package/build/module/src/examples/vectordb.js.map +1 -1
  143. package/build/module/src/prompts/prompts.test.js +7 -3
  144. package/build/module/src/prompts/prompts.test.js.map +1 -1
  145. package/package.json +1 -1
@@ -1,6 +1,6 @@
1
1
  import { Readable } from 'stream';
2
2
  import test from 'ava';
3
- import { axAI } from '../ai/index.js';
3
+ import { AxAI } from '../ai/index.js';
4
4
  import { AxSignature } from '../dsp/sig.js';
5
5
  import { AxChainOfThought } from './cot.js';
6
6
  const text = `The technological singularity—or simply the singularity[1]—is a hypothetical future point in time at which technological growth becomes uncontrollable and irreversible.`;
@@ -36,8 +36,12 @@ const mockFetch = async () => {
36
36
  };
37
37
  test('generate prompt', async (t) => {
38
38
  const options = { fetch: mockFetch };
39
- const ai = axAI('openai', { apiKey: 'no-key', options });
40
- // const ai = axAI('ollama', { model: 'nous-hermes2' });
39
+ const ai = new AxAI({
40
+ name: 'openai',
41
+ apiKey: 'no-key',
42
+ options
43
+ });
44
+ // const ai = new AxAI({ name: 'ollama', config: { model: 'nous-hermes2' } });
41
45
  const gen = new AxChainOfThought(ai, `text -> shortSummary "summarize in 5 to 10 words"`);
42
46
  gen.setExamples(examples);
43
47
  const res = await gen.forward({ text });
@@ -1 +1 @@
1
- {"version":3,"file":"prompts.test.js","sourceRoot":"","sources":["../../../../src/prompts/prompts.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAC;AAElC,OAAO,IAAI,MAAM,KAAK,CAAC;AAEvB,OAAO,EAAE,IAAI,EAAqB,MAAM,gBAAgB,CAAC;AACzD,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAE5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAE5C,MAAM,IAAI,GAAG,0KAA0K,CAAC;AAExL,MAAM,QAAQ,GAAG;IACf;QACE,IAAI,EAAE,0TAA0T;QAChU,YAAY,EACV,+GAA+G;KAClH;IACD;QACE,IAAI,EAAE,gZAAgZ;QACtZ,YAAY,EACV,+HAA+H;KAClI;CACF,CAAC;AAEF,MAAM,SAAS,GAAG,KAAK,IAAuB,EAAE;IAC9C,MAAM,OAAO,GAAG;QACd,OAAO,EAAE;YACP;gBACE,OAAO,EAAE;oBACP,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,4DAA4D;iBACtE;aACF;SACF;KACF,CAAC;IAEF,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,OAAO,CAAC;YACN,EAAE,EAAE,IAAI;YACR,MAAM,EAAE,GAAG;YACX,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YAC5C,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;SACtC,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AAEF,IAAI,CAAC,iBAAiB,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;IAClC,MAAM,OAAO,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;IACrC,MAAM,EAAE,GAAG,IAAI,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAkB,CAAC,CAAC;IAEzE,wDAAwD;IAExD,MAAM,GAAG,GAAG,IAAI,gBAAgB,CAC9B,EAAE,EACF,mDAAmD,CACpD,CAAC;IACF,GAAG,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IAE1B,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;IAExC,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE;QACf,MAAM,EAAE,gBAAgB;QACxB,YAAY,EAAE,qBAAqB;KACpC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,oCAAoC,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;IACrD,CAAC,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,WAAW,CAAC,sBAAsB,CAAC,CAAC,CAAC;AAC1D,CAAC,CAAC,CAAC"}
1
+ {"version":3,"file":"prompts.test.js","sourceRoot":"","sources":["../../../../src/prompts/prompts.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAC;AAElC,OAAO,IAAI,MAAM,KAAK,CAAC;AAEvB,OAAO,EAAE,IAAI,EAAE,MAAM,gBAAgB,CAAC;AACtC,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAE5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAE5C,MAAM,IAAI,GAAG,0KAA0K,CAAC;AAExL,MAAM,QAAQ,GAAG;IACf;QACE,IAAI,EAAE,0TAA0T;QAChU,YAAY,EACV,+GAA+G;KAClH;IACD;QACE,IAAI,EAAE,gZAAgZ;QACtZ,YAAY,EACV,+HAA+H;KAClI;CACF,CAAC;AAEF,MAAM,SAAS,GAAG,KAAK,IAAuB,EAAE;IAC9C,MAAM,OAAO,GAAG;QACd,OAAO,EAAE;YACP;gBACE,OAAO,EAAE;oBACP,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,4DAA4D;iBACtE;aACF;SACF;KACF,CAAC;IAEF,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,OAAO,CAAC;YACN,EAAE,EAAE,IAAI;YACR,MAAM,EAAE,GAAG;YACX,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YAC5C,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;SACtC,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AAEF,IAAI,CAAC,iBAAiB,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;IAClC,MAAM,OAAO,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;IACrC,MAAM,EAAE,GAAG,IAAI,IAAI,CAAC;QAClB,IAAI,EAAE,QAAQ;QACd,MAAM,EAAE,QAAQ;QAChB,OAAO;KACR,CAAC,CAAC;IAEH,8EAA8E;IAE9E,MAAM,GAAG,GAAG,IAAI,gBAAgB,CAC9B,EAAE,EACF,mDAAmD,CACpD,CAAC;IACF,GAAG,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IAE1B,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;IAExC,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE;QACf,MAAM,EAAE,gBAAgB;QACxB,YAAY,EAAE,qBAAqB;KACpC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,oCAAoC,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;IACrD,CAAC,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,WAAW,CAAC,sBAAsB,CAAC,CAAC,CAAC;AAC1D,CAAC,CAAC,CAAC"}
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@ax-llm/ax",
3
- "version": "9.0.3",
3
+ "version": "9.0.5",
4
4
  "type": "module",
5
5
  "description": "The best library to work with LLMs",
6
6
  "typings": "build/module/src/index.d.ts",