@ax-llm/ax 9.0.0 → 9.0.3

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 (287) hide show
  1. package/README.md +106 -99
  2. package/build/module/docs/assets/navigation.js +1 -1
  3. package/build/module/docs/assets/navigation.js.map +1 -1
  4. package/build/module/docs/assets/search.js +1 -1
  5. package/build/module/docs/assets/search.js.map +1 -1
  6. package/build/module/src/ai/anthropic/api.d.ts +13 -19
  7. package/build/module/src/ai/anthropic/api.js +165 -72
  8. package/build/module/src/ai/anthropic/api.js.map +1 -1
  9. package/build/module/src/ai/anthropic/info.d.ts +2 -2
  10. package/build/module/src/ai/anthropic/info.js +7 -7
  11. package/build/module/src/ai/anthropic/info.js.map +1 -1
  12. package/build/module/src/ai/anthropic/types.d.ts +63 -52
  13. package/build/module/src/ai/anthropic/types.js +8 -12
  14. package/build/module/src/ai/anthropic/types.js.map +1 -1
  15. package/build/module/src/ai/azure-openai/api.d.ts +12 -28
  16. package/build/module/src/ai/azure-openai/api.js +6 -22
  17. package/build/module/src/ai/azure-openai/api.js.map +1 -1
  18. package/build/module/src/ai/balance.d.ts +9 -12
  19. package/build/module/src/ai/balance.js +1 -1
  20. package/build/module/src/ai/balance.js.map +1 -1
  21. package/build/module/src/ai/base.d.ts +26 -29
  22. package/build/module/src/ai/base.js +37 -55
  23. package/build/module/src/ai/base.js.map +1 -1
  24. package/build/module/src/ai/cohere/api.d.ts +16 -30
  25. package/build/module/src/ai/cohere/api.js +26 -32
  26. package/build/module/src/ai/cohere/api.js.map +1 -1
  27. package/build/module/src/ai/cohere/info.d.ts +2 -2
  28. package/build/module/src/ai/cohere/info.js +10 -10
  29. package/build/module/src/ai/cohere/info.js.map +1 -1
  30. package/build/module/src/ai/cohere/types.d.ts +13 -13
  31. package/build/module/src/ai/cohere/types.js +14 -14
  32. package/build/module/src/ai/cohere/types.js.map +1 -1
  33. package/build/module/src/ai/deepseek/api.d.ts +10 -22
  34. package/build/module/src/ai/deepseek/api.js +14 -24
  35. package/build/module/src/ai/deepseek/api.js.map +1 -1
  36. package/build/module/src/ai/deepseek/info.d.ts +2 -2
  37. package/build/module/src/ai/deepseek/info.js +4 -4
  38. package/build/module/src/ai/deepseek/info.js.map +1 -1
  39. package/build/module/src/ai/deepseek/types.d.ts +1 -1
  40. package/build/module/src/ai/deepseek/types.js +5 -5
  41. package/build/module/src/ai/deepseek/types.js.map +1 -1
  42. package/build/module/src/ai/google-gemini/api.d.ts +18 -20
  43. package/build/module/src/ai/google-gemini/api.js +93 -67
  44. package/build/module/src/ai/google-gemini/api.js.map +1 -1
  45. package/build/module/src/ai/google-gemini/info.d.ts +3 -3
  46. package/build/module/src/ai/google-gemini/info.js +6 -6
  47. package/build/module/src/ai/google-gemini/info.js.map +1 -1
  48. package/build/module/src/ai/google-gemini/types.d.ts +41 -48
  49. package/build/module/src/ai/google-gemini/types.js +25 -37
  50. package/build/module/src/ai/google-gemini/types.js.map +1 -1
  51. package/build/module/src/ai/groq/api.d.ts +9 -18
  52. package/build/module/src/ai/groq/api.js +11 -17
  53. package/build/module/src/ai/groq/api.js.map +1 -1
  54. package/build/module/src/ai/groq/types.d.ts +6 -0
  55. package/build/module/src/ai/groq/types.js +8 -0
  56. package/build/module/src/ai/groq/types.js.map +1 -0
  57. package/build/module/src/ai/huggingface/api.d.ts +13 -27
  58. package/build/module/src/ai/huggingface/api.js +12 -24
  59. package/build/module/src/ai/huggingface/api.js.map +1 -1
  60. package/build/module/src/ai/huggingface/info.d.ts +2 -2
  61. package/build/module/src/ai/huggingface/info.js +1 -1
  62. package/build/module/src/ai/huggingface/info.js.map +1 -1
  63. package/build/module/src/ai/huggingface/types.d.ts +7 -15
  64. package/build/module/src/ai/huggingface/types.js +4 -8
  65. package/build/module/src/ai/huggingface/types.js.map +1 -1
  66. package/build/module/src/ai/index.d.ts +13 -13
  67. package/build/module/src/ai/index.js +23 -23
  68. package/build/module/src/ai/index.js.map +1 -1
  69. package/build/module/src/ai/mistral/api.d.ts +10 -22
  70. package/build/module/src/ai/mistral/api.js +14 -24
  71. package/build/module/src/ai/mistral/api.js.map +1 -1
  72. package/build/module/src/ai/mistral/info.d.ts +2 -2
  73. package/build/module/src/ai/mistral/info.js +7 -7
  74. package/build/module/src/ai/mistral/info.js.map +1 -1
  75. package/build/module/src/ai/mistral/types.d.ts +2 -10
  76. package/build/module/src/ai/mistral/types.js +12 -20
  77. package/build/module/src/ai/mistral/types.js.map +1 -1
  78. package/build/module/src/ai/ollama/api.d.ts +13 -16
  79. package/build/module/src/ai/ollama/api.js +17 -13
  80. package/build/module/src/ai/ollama/api.js.map +1 -1
  81. package/build/module/src/ai/openai/api.d.ts +19 -40
  82. package/build/module/src/ai/openai/api.js +90 -135
  83. package/build/module/src/ai/openai/api.js.map +1 -1
  84. package/build/module/src/ai/openai/info.d.ts +2 -2
  85. package/build/module/src/ai/openai/info.js +9 -9
  86. package/build/module/src/ai/openai/info.js.map +1 -1
  87. package/build/module/src/ai/openai/types.d.ts +29 -56
  88. package/build/module/src/ai/openai/types.js +17 -34
  89. package/build/module/src/ai/openai/types.js.map +1 -1
  90. package/build/module/src/ai/together/api.d.ts +9 -17
  91. package/build/module/src/ai/together/api.js +9 -15
  92. package/build/module/src/ai/together/api.js.map +1 -1
  93. package/build/module/src/ai/together/info.d.ts +2 -2
  94. package/build/module/src/ai/together/info.js +1 -1
  95. package/build/module/src/ai/together/info.js.map +1 -1
  96. package/build/module/src/ai/types.d.ts +112 -25
  97. package/build/module/src/ai/util.d.ts +4 -4
  98. package/build/module/src/ai/util.js +6 -3
  99. package/build/module/src/ai/util.js.map +1 -1
  100. package/build/module/src/db/base.d.ts +14 -15
  101. package/build/module/src/db/base.js +26 -29
  102. package/build/module/src/db/base.js.map +1 -1
  103. package/build/module/src/db/cloudflare.d.ts +10 -9
  104. package/build/module/src/db/cloudflare.js +2 -2
  105. package/build/module/src/db/cloudflare.js.map +1 -1
  106. package/build/module/src/db/index.d.ts +6 -7
  107. package/build/module/src/db/index.js +9 -9
  108. package/build/module/src/db/index.js.map +1 -1
  109. package/build/module/src/db/memory.d.ts +13 -11
  110. package/build/module/src/db/memory.js +2 -2
  111. package/build/module/src/db/memory.js.map +1 -1
  112. package/build/module/src/db/pinecone.d.ts +10 -9
  113. package/build/module/src/db/pinecone.js +2 -2
  114. package/build/module/src/db/pinecone.js.map +1 -1
  115. package/build/module/src/db/types.d.ts +9 -9
  116. package/build/module/src/db/weaviate.d.ts +10 -9
  117. package/build/module/src/db/weaviate.js +2 -2
  118. package/build/module/src/db/weaviate.js.map +1 -1
  119. package/build/module/src/docs/manager.d.ts +16 -16
  120. package/build/module/src/docs/manager.js +1 -1
  121. package/build/module/src/docs/manager.js.map +1 -1
  122. package/build/module/src/docs/reranker.d.ts +6 -6
  123. package/build/module/src/docs/reranker.js +3 -3
  124. package/build/module/src/docs/reranker.js.map +1 -1
  125. package/build/module/src/docs/rewriter.d.ts +3 -3
  126. package/build/module/src/docs/rewriter.js +2 -2
  127. package/build/module/src/docs/rewriter.js.map +1 -1
  128. package/build/module/src/docs/tika.d.ts +5 -8
  129. package/build/module/src/docs/tika.js +2 -2
  130. package/build/module/src/docs/tika.js.map +1 -1
  131. package/build/module/src/dsp/asserts.d.ts +8 -8
  132. package/build/module/src/dsp/asserts.js +9 -6
  133. package/build/module/src/dsp/asserts.js.map +1 -1
  134. package/build/module/src/dsp/eval.d.ts +9 -28
  135. package/build/module/src/dsp/eval.js +10 -5
  136. package/build/module/src/dsp/eval.js.map +1 -1
  137. package/build/module/src/dsp/evaluate.d.ts +8 -8
  138. package/build/module/src/dsp/evaluate.js +1 -1
  139. package/build/module/src/dsp/evaluate.js.map +1 -1
  140. package/build/module/src/dsp/extract.d.ts +6 -6
  141. package/build/module/src/dsp/extract.js +1 -0
  142. package/build/module/src/dsp/extract.js.map +1 -1
  143. package/build/module/src/dsp/generate.d.ts +24 -25
  144. package/build/module/src/dsp/generate.js +24 -23
  145. package/build/module/src/dsp/generate.js.map +1 -1
  146. package/build/module/src/dsp/generate.test.js +4 -4
  147. package/build/module/src/dsp/generate.test.js.map +1 -1
  148. package/build/module/src/dsp/index.d.ts +4 -4
  149. package/build/module/src/dsp/loader.d.ts +5 -5
  150. package/build/module/src/dsp/loader.js +1 -1
  151. package/build/module/src/dsp/loader.js.map +1 -1
  152. package/build/module/src/dsp/optimize.d.ts +12 -12
  153. package/build/module/src/dsp/optimize.js +5 -5
  154. package/build/module/src/dsp/optimize.js.map +1 -1
  155. package/build/module/src/dsp/parser.d.ts +2 -1
  156. package/build/module/src/dsp/parser.js +178 -73
  157. package/build/module/src/dsp/parser.js.map +1 -1
  158. package/build/module/src/dsp/program.d.ts +38 -37
  159. package/build/module/src/dsp/program.js +4 -53
  160. package/build/module/src/dsp/program.js.map +1 -1
  161. package/build/module/src/dsp/prompt.d.ts +36 -13
  162. package/build/module/src/dsp/prompt.js +123 -51
  163. package/build/module/src/dsp/prompt.js.map +1 -1
  164. package/build/module/src/dsp/registry.d.ts +1 -1
  165. package/build/module/src/dsp/registry.js +1 -1
  166. package/build/module/src/dsp/registry.js.map +1 -1
  167. package/build/module/src/dsp/router.d.ts +7 -7
  168. package/build/module/src/dsp/router.js +4 -4
  169. package/build/module/src/dsp/router.js.map +1 -1
  170. package/build/module/src/dsp/sig.d.ts +13 -13
  171. package/build/module/src/dsp/sig.js +7 -3
  172. package/build/module/src/dsp/sig.js.map +1 -1
  173. package/build/module/src/dsp/sig.test.js +4 -1
  174. package/build/module/src/dsp/sig.test.js.map +1 -1
  175. package/build/module/src/dsp/stopwords.js +1 -0
  176. package/build/module/src/dsp/stopwords.js.map +1 -1
  177. package/build/module/src/dsp/strutil.d.ts +10 -8
  178. package/build/module/src/dsp/strutil.js +14 -6
  179. package/build/module/src/dsp/strutil.js.map +1 -1
  180. package/build/module/src/dsp/util.d.ts +4 -0
  181. package/build/module/src/dsp/util.js +62 -0
  182. package/build/module/src/dsp/util.js.map +1 -1
  183. package/build/module/src/examples/agent.js +7 -5
  184. package/build/module/src/examples/agent.js.map +1 -1
  185. package/build/module/src/examples/chain-of-thought.js +5 -3
  186. package/build/module/src/examples/chain-of-thought.js.map +1 -1
  187. package/build/module/src/examples/customer-support.js +6 -4
  188. package/build/module/src/examples/customer-support.js.map +1 -1
  189. package/build/module/src/examples/fibonacci.js +6 -4
  190. package/build/module/src/examples/fibonacci.js.map +1 -1
  191. package/build/module/src/examples/food-search.js +8 -8
  192. package/build/module/src/examples/food-search.js.map +1 -1
  193. package/build/module/src/examples/marketing.js +5 -3
  194. package/build/module/src/examples/marketing.js.map +1 -1
  195. package/build/module/src/examples/multi-modal.d.ts +1 -0
  196. package/build/module/src/examples/multi-modal.js +16 -0
  197. package/build/module/src/examples/multi-modal.js.map +1 -0
  198. package/build/module/src/examples/qna-tune.js +9 -7
  199. package/build/module/src/examples/qna-tune.js.map +1 -1
  200. package/build/module/src/examples/qna-use-tuned.js +8 -6
  201. package/build/module/src/examples/qna-use-tuned.js.map +1 -1
  202. package/build/module/src/examples/rag-docs.js +7 -5
  203. package/build/module/src/examples/rag-docs.js.map +1 -1
  204. package/build/module/src/examples/rag.js +6 -4
  205. package/build/module/src/examples/rag.js.map +1 -1
  206. package/build/module/src/examples/react.js +5 -3
  207. package/build/module/src/examples/react.js.map +1 -1
  208. package/build/module/src/examples/routing.js +9 -7
  209. package/build/module/src/examples/routing.js.map +1 -1
  210. package/build/module/src/examples/smart-home.d.ts +10 -0
  211. package/build/module/src/examples/smart-home.js +106 -0
  212. package/build/module/src/examples/smart-home.js.map +1 -0
  213. package/build/module/src/examples/streaming1.js +7 -5
  214. package/build/module/src/examples/streaming1.js.map +1 -1
  215. package/build/module/src/examples/streaming2.js +9 -7
  216. package/build/module/src/examples/streaming2.js.map +1 -1
  217. package/build/module/src/examples/summarize.js +5 -5
  218. package/build/module/src/examples/summarize.js.map +1 -1
  219. package/build/module/src/examples/vectordb.js +6 -4
  220. package/build/module/src/examples/vectordb.js.map +1 -1
  221. package/build/module/src/funcs/code.d.ts +3 -4
  222. package/build/module/src/funcs/code.js +15 -15
  223. package/build/module/src/funcs/code.js.map +1 -1
  224. package/build/module/src/funcs/embed.d.ts +3 -5
  225. package/build/module/src/funcs/embed.js +2 -2
  226. package/build/module/src/funcs/embed.js.map +1 -1
  227. package/build/module/src/funcs/functions.d.ts +18 -0
  228. package/build/module/src/funcs/functions.js +55 -0
  229. package/build/module/src/funcs/functions.js.map +1 -0
  230. package/build/module/src/funcs/index.d.ts +1 -0
  231. package/build/module/src/funcs/index.js +1 -0
  232. package/build/module/src/funcs/index.js.map +1 -1
  233. package/build/module/src/funcs/jsonschema.d.ts +2 -0
  234. package/build/module/src/funcs/jsonschema.js +97 -0
  235. package/build/module/src/funcs/jsonschema.js.map +1 -0
  236. package/build/module/src/index.d.ts +1 -2
  237. package/build/module/src/index.js +1 -2
  238. package/build/module/src/index.js.map +1 -1
  239. package/build/module/src/mem/index.d.ts +2 -0
  240. package/build/module/src/mem/index.js +3 -0
  241. package/build/module/src/mem/index.js.map +1 -0
  242. package/build/module/src/mem/memory.d.ts +16 -0
  243. package/build/module/src/mem/memory.js +74 -0
  244. package/build/module/src/mem/memory.js.map +1 -0
  245. package/build/module/src/mem/types.d.ts +10 -0
  246. package/build/module/src/mem/types.js +2 -0
  247. package/build/module/src/mem/types.js.map +1 -0
  248. package/build/module/src/prompts/agent.d.ts +14 -15
  249. package/build/module/src/prompts/agent.js +10 -11
  250. package/build/module/src/prompts/agent.js.map +1 -1
  251. package/build/module/src/prompts/cot.d.ts +6 -6
  252. package/build/module/src/prompts/cot.js +4 -4
  253. package/build/module/src/prompts/cot.js.map +1 -1
  254. package/build/module/src/prompts/prompts.test.js +9 -5
  255. package/build/module/src/prompts/prompts.test.js.map +1 -1
  256. package/build/module/src/prompts/rag.d.ts +7 -7
  257. package/build/module/src/prompts/rag.js +6 -6
  258. package/build/module/src/prompts/rag.js.map +1 -1
  259. package/build/module/src/prompts/react.d.ts +7 -7
  260. package/build/module/src/prompts/react.js +4 -4
  261. package/build/module/src/prompts/react.js.map +1 -1
  262. package/build/module/src/text/functions.d.ts +3 -20
  263. package/build/module/src/text/functions.js.map +1 -1
  264. package/build/module/src/text/jsonschema.d.ts +1 -2
  265. package/build/module/src/text/memory.d.ts +8 -8
  266. package/build/module/src/text/memory.js.map +1 -1
  267. package/build/module/src/text/types.d.ts +14 -14
  268. package/build/module/src/trace/index.d.ts +50 -50
  269. package/build/module/src/trace/index.js +30 -30
  270. package/build/module/src/trace/index.js.map +1 -1
  271. package/build/module/src/types/index.d.ts +9 -39
  272. package/build/module/src/types/index.js +1 -1
  273. package/build/module/src/types/index.js.map +1 -1
  274. package/build/module/src/util/apicall.d.ts +2 -5
  275. package/build/module/src/util/apicall.js +4 -3
  276. package/build/module/src/util/apicall.js.map +1 -1
  277. package/build/module/src/util/log.js +1 -0
  278. package/build/module/src/util/log.js.map +1 -1
  279. package/build/module/src/util/other.js +1 -0
  280. package/build/module/src/util/other.js.map +1 -1
  281. package/build/module/src/util/stream.d.ts +0 -1
  282. package/build/module/src/util/stream.js +1 -0
  283. package/build/module/src/util/stream.js.map +1 -1
  284. package/build/module/src/util/transform.d.ts +0 -1
  285. package/build/module/src/util/transform.js +2 -0
  286. package/build/module/src/util/transform.js.map +1 -1
  287. package/package.json +2 -4
@@ -0,0 +1,16 @@
1
+ import type { AxChatRequest, AxChatResponseResult } from '../ai/types.js';
2
+ import type { AxAIMemory } from './types.js';
3
+ export declare class AxMemory implements AxAIMemory {
4
+ private data;
5
+ private sdata;
6
+ private limit;
7
+ constructor(limit?: number);
8
+ add(value: Readonly<AxChatRequest['chatPrompt'][0] | AxChatRequest['chatPrompt']>, sessionId?: string): void;
9
+ addResult({ content, name, functionCalls }: Readonly<AxChatResponseResult>, sessionId?: string): void;
10
+ updateResult({ content, name, functionCalls }: Readonly<AxChatResponseResult>, sessionId?: string): void;
11
+ history(sessionId?: string): AxChatRequest['chatPrompt'];
12
+ peek(sessionId?: string): AxChatRequest['chatPrompt'];
13
+ getLast(sessionId?: string): AxChatRequest['chatPrompt'][0] | undefined;
14
+ reset(sessionId?: string): void;
15
+ private get;
16
+ }
@@ -0,0 +1,74 @@
1
+ export class AxMemory {
2
+ data = [];
3
+ sdata = new Map();
4
+ limit;
5
+ constructor(limit = 50) {
6
+ if (limit <= 0) {
7
+ throw Error("argument 'last' must be greater than 0");
8
+ }
9
+ this.limit = limit;
10
+ }
11
+ add(value, sessionId) {
12
+ const d = this.get(sessionId);
13
+ let n = 0;
14
+ if (Array.isArray(value)) {
15
+ n = d.push(...structuredClone(value));
16
+ }
17
+ else {
18
+ n = d.push({
19
+ ...structuredClone(value)
20
+ });
21
+ }
22
+ if (d.length > this.limit) {
23
+ d.splice(0, this.limit + n - this.limit);
24
+ }
25
+ }
26
+ addResult({ content, name, functionCalls }, sessionId) {
27
+ this.add({ content, name, role: 'assistant', functionCalls }, sessionId);
28
+ }
29
+ updateResult({ content, name, functionCalls }, sessionId) {
30
+ const item = this.get(sessionId);
31
+ const lastItem = item.at(-1);
32
+ if (!lastItem || lastItem.role !== 'assistant') {
33
+ this.addResult({ content, name, functionCalls }, sessionId);
34
+ return;
35
+ }
36
+ if ('content' in lastItem && content) {
37
+ lastItem.content = content;
38
+ }
39
+ if ('name' in lastItem && name) {
40
+ lastItem.name = name;
41
+ }
42
+ if ('functionCalls' in lastItem && functionCalls) {
43
+ lastItem.functionCalls = functionCalls;
44
+ }
45
+ }
46
+ history(sessionId) {
47
+ return this.get(sessionId);
48
+ }
49
+ peek(sessionId) {
50
+ return this.get(sessionId);
51
+ }
52
+ getLast(sessionId) {
53
+ const d = this.get(sessionId);
54
+ return d.at(-1);
55
+ }
56
+ reset(sessionId) {
57
+ if (!sessionId) {
58
+ this.data = [];
59
+ }
60
+ else {
61
+ this.sdata.set(sessionId, []);
62
+ }
63
+ }
64
+ get(sessionId) {
65
+ if (!sessionId) {
66
+ return this.data;
67
+ }
68
+ if (!this.sdata.has(sessionId)) {
69
+ this.sdata.set(sessionId, []);
70
+ }
71
+ return this.sdata.get(sessionId) || [];
72
+ }
73
+ }
74
+ //# sourceMappingURL=memory.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"memory.js","sourceRoot":"","sources":["../../../../src/mem/memory.ts"],"names":[],"mappings":"AAMA,MAAM,OAAO,QAAQ;IACX,IAAI,GAAgC,EAAE,CAAC;IACvC,KAAK,GAAG,IAAI,GAAG,EAAuC,CAAC;IACvD,KAAK,CAAS;IAEtB,YAAY,KAAK,GAAG,EAAE;QACpB,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;YACf,MAAM,KAAK,CAAC,wCAAwC,CAAC,CAAC;QACxD,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,GAAG,CACD,KAEC,EACD,SAAkB;QAElB,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC9B,IAAI,CAAC,GAAG,CAAC,CAAC;QAEV,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACzB,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC,CAAC;QACxC,CAAC;aAAM,CAAC;YACN,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;gBACT,GAAG,eAAe,CAAC,KAAK,CAAC;aACQ,CAAC,CAAC;QACvC,CAAC;QACD,IAAI,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;YAC1B,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IAED,SAAS,CACP,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAkC,EAChE,SAAkB;QAElB,IAAI,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,aAAa,EAAE,EAAE,SAAS,CAAC,CAAC;IAC3E,CAAC;IAED,YAAY,CACV,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAkC,EAChE,SAAkB;QAElB,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAEjC,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAE1B,CAAC;QACF,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC/C,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,SAAS,CAAC,CAAC;YAC5D,OAAO;QACT,CAAC;QAED,IAAI,SAAS,IAAI,QAAQ,IAAI,OAAO,EAAE,CAAC;YACrC,QAAQ,CAAC,OAAO,GAAG,OAAO,CAAC;QAC7B,CAAC;QACD,IAAI,MAAM,IAAI,QAAQ,IAAI,IAAI,EAAE,CAAC;YAC/B,QAAQ,CAAC,IAAI,GAAG,IAAI,CAAC;QACvB,CAAC;QACD,IAAI,eAAe,IAAI,QAAQ,IAAI,aAAa,EAAE,CAAC;YACjD,QAAQ,CAAC,aAAa,GAAG,aAAa,CAAC;QACzC,CAAC;IACH,CAAC;IAED,OAAO,CAAC,SAAkB;QACxB,OAAO,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC7B,CAAC;IAED,IAAI,CAAC,SAAkB;QACrB,OAAO,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC7B,CAAC;IAED,OAAO,CAAC,SAAkB;QACxB,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC9B,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,SAAkB;QACtB,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,IAAI,CAAC,IAAI,GAAG,EAAE,CAAC;QACjB,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAEO,GAAG,CAAC,SAAkB;QAC5B,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,OAAO,IAAI,CAAC,IAAI,CAAC;QACnB,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YAC/B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAChC,CAAC;QAED,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;IACzC,CAAC;CACF"}
@@ -0,0 +1,10 @@
1
+ import type { AxChatRequest, AxChatResponseResult } from '../ai/index.js';
2
+ export interface AxAIMemory {
3
+ add(result: Readonly<AxChatRequest['chatPrompt'] | AxChatRequest['chatPrompt'][0]>, sessionId?: string): void;
4
+ addResult(result: Readonly<AxChatResponseResult>, sessionId?: string): void;
5
+ updateResult(result: Readonly<AxChatResponseResult>, sessionId?: string): void;
6
+ history(sessionId?: string): AxChatRequest['chatPrompt'];
7
+ peek(sessionId?: string): AxChatRequest['chatPrompt'];
8
+ reset(sessionId?: string): void;
9
+ getLast(sessionId?: string): AxChatRequest['chatPrompt'][0] | undefined;
10
+ }
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=types.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../src/mem/types.ts"],"names":[],"mappings":""}
@@ -1,24 +1,23 @@
1
- import { type GenerateOptions, Signature } from '../dsp/index.js';
2
- import { type GenIn, type GenOut, type ITunable, type IUsable, Program, type ProgramForwardOptions } from '../dsp/program.js';
3
- import { type AITextFunction } from '../text/index.js';
4
- import type { AIService } from '../text/types.js';
5
- export interface AgentI extends ITunable, IUsable {
6
- getFunction(): AITextFunction;
1
+ import type { AxAIService, AxFunction } from '../ai/types.js';
2
+ import { type AxGenerateOptions, AxSignature } from '../dsp/index.js';
3
+ import { type AxGenIn, type AxGenOut, AxProgram, type AxProgramForwardOptions, type AxTunable, type AxUsable } from '../dsp/program.js';
4
+ export interface AxAgentic extends AxTunable, AxUsable {
5
+ getFunction(): AxFunction;
7
6
  }
8
- export type AgentOptions = Omit<GenerateOptions, 'functions' | 'functionCall'>;
9
- export declare class Agent<IN extends GenIn, OUT extends GenOut> extends Program<IN, OUT> implements AgentI {
7
+ export type AxAgentOptions = Omit<AxGenerateOptions, 'functions' | 'functionCall'>;
8
+ export declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut> extends AxProgram<IN, OUT> implements AxAgentic {
10
9
  private gen;
11
10
  private name;
12
11
  private description;
13
12
  private subAgentList?;
14
13
  private func;
15
- constructor(ai: AIService, { name, description, signature, agents, functions }: Readonly<{
14
+ constructor(ai: AxAIService, { name, description, signature, agents, functions }: Readonly<{
16
15
  name: string;
17
16
  description: string;
18
- signature: Signature | string;
19
- agents?: AgentI[];
20
- functions?: AITextFunction[];
21
- }>, options?: Readonly<AgentOptions>);
22
- getFunction(): AITextFunction;
23
- forward(values: IN, options?: Readonly<ProgramForwardOptions>): Promise<OUT>;
17
+ signature: AxSignature | string;
18
+ agents?: AxAgentic[];
19
+ functions?: AxFunction[];
20
+ }>, options?: Readonly<AxAgentOptions>);
21
+ getFunction(): AxFunction;
22
+ forward(values: IN, options?: Readonly<AxProgramForwardOptions>): Promise<OUT>;
24
23
  }
@@ -1,10 +1,9 @@
1
- import { Signature } from '../dsp/index.js';
2
- import { Program } from '../dsp/program.js';
3
- import {} from '../text/index.js';
4
- import { SpanKind } from '../trace/index.js';
5
- import { ChainOfThought } from './cot.js';
6
- import { ReAct } from './react.js';
7
- export class Agent extends Program {
1
+ import { AxSignature } from '../dsp/index.js';
2
+ import { AxProgram } from '../dsp/program.js';
3
+ import { AxSpanKind } from '../trace/index.js';
4
+ import { AxChainOfThought } from './cot.js';
5
+ import { AxReAct } from './react.js';
6
+ export class AxAgent extends AxProgram {
8
7
  gen;
9
8
  name;
10
9
  description;
@@ -12,7 +11,7 @@ export class Agent extends Program {
12
11
  func;
13
12
  constructor(ai, { name, description, signature, agents, functions }, options) {
14
13
  super();
15
- const sig = new Signature(signature);
14
+ const sig = new AxSignature(signature);
16
15
  const funcs = [
17
16
  ...(functions ?? []),
18
17
  ...(agents?.map((a) => a.getFunction()) ?? [])
@@ -24,8 +23,8 @@ export class Agent extends Program {
24
23
  };
25
24
  this.gen =
26
25
  funcs.length > 0
27
- ? new ReAct(ai, sig, opt)
28
- : new ChainOfThought(ai, sig, opt);
26
+ ? new AxReAct(ai, sig, opt)
27
+ : new AxChainOfThought(ai, sig, opt);
29
28
  this.name = name;
30
29
  this.description = description;
31
30
  this.subAgentList = agents?.map((a) => a.getFunction().name).join(', ');
@@ -53,7 +52,7 @@ export class Agent extends Program {
53
52
  ['agent.subAgents']: this.subAgentList ?? 'none'
54
53
  };
55
54
  return await options?.tracer.startActiveSpan('Agent', {
56
- kind: SpanKind.SERVER,
55
+ kind: AxSpanKind.SERVER,
57
56
  attributes
58
57
  }, async (span) => {
59
58
  const res = await this.gen.forward(values, options);
@@ -1 +1 @@
1
- {"version":3,"file":"agent.js","sourceRoot":"","sources":["../../../../src/prompts/agent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAwB,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAClE,OAAO,EAKL,OAAO,EAER,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAuB,MAAM,kBAAkB,CAAC;AAEvD,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAC1C,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAQnC,MAAM,OAAO,KACX,SAAQ,OAAgB;IAGhB,GAAG,CAAmB;IAEtB,IAAI,CAAS;IACb,WAAW,CAAS;IACpB,YAAY,CAAU;IACtB,IAAI,CAAiB;IAE7B,YACE,EAAa,EACb,EACE,IAAI,EACJ,WAAW,EACX,SAAS,EACT,MAAM,EACN,SAAS,EAOT,EACF,OAAgC;QAEhC,KAAK,EAAE,CAAC;QAER,MAAM,GAAG,GAAG,IAAI,SAAS,CAAC,SAAS,CAAC,CAAC;QAErC,MAAM,KAAK,GAAqB;YAC9B,GAAG,CAAC,SAAS,IAAI,EAAE,CAAC;YACpB,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC;SAC/C,CAAC;QAEF,MAAM,GAAG,GAAG;YACV,cAAc,EAAE,OAAO,EAAE,cAAc;YACvC,OAAO,EAAE,OAAO,EAAE,OAAO;YACzB,SAAS,EAAE,KAAK;SACjB,CAAC;QAEF,IAAI,CAAC,GAAG;YACN,KAAK,CAAC,MAAM,GAAG,CAAC;gBACd,CAAC,CAAC,IAAI,KAAK,CAAU,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC;gBAClC,CAAC,CAAC,IAAI,cAAc,CAAU,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QAEhD,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,YAAY,GAAG,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxE,IAAI,CAAC,IAAI,GAAG;YACV,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAE,GAAG,CAAC,YAAY,EAAE;YAC9B,IAAI,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO;SACzB,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAExB,KAAK,MAAM,KAAK,IAAI,MAAM,IAAI,EAAE,EAAE,CAAC;YACjC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACvB,CAAC;IACH,CAAC;IAEM,WAAW;QAChB,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAEe,KAAK,CAAC,OAAO,CAC3B,MAAU,EACV,OAAyC;QAEzC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;YACrB,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACjD,CAAC;QAED,MAAM,UAAU,GAAG;YACjB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,IAAI;YACzB,CAAC,mBAAmB,CAAC,EAAE,IAAI,CAAC,WAAW;YACvC,CAAC,iBAAiB,CAAC,EAAE,IAAI,CAAC,YAAY,IAAI,MAAM;SACjD,CAAC;QAEF,OAAO,MAAM,OAAO,EAAE,MAAM,CAAC,eAAe,CAC1C,OAAO,EACP;YACE,IAAI,EAAE,QAAQ,CAAC,MAAM;YACrB,UAAU;SACX,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACpD,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,OAAO,GAAG,CAAC;QACb,CAAC,CACF,CAAC;IACJ,CAAC;CACF"}
1
+ {"version":3,"file":"agent.js","sourceRoot":"","sources":["../../../../src/prompts/agent.ts"],"names":[],"mappings":"AACA,OAAO,EAA0B,WAAW,EAAE,MAAM,iBAAiB,CAAC;AACtE,OAAO,EAGL,SAAS,EAIV,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAE/C,OAAO,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAC5C,OAAO,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AAWrC,MAAM,OAAO,OACX,SAAQ,SAAkB;IAGlB,GAAG,CAAqB;IAExB,IAAI,CAAS;IACb,WAAW,CAAS;IACpB,YAAY,CAAU;IACtB,IAAI,CAAa;IAEzB,YACE,EAAe,EACf,EACE,IAAI,EACJ,WAAW,EACX,SAAS,EACT,MAAM,EACN,SAAS,EAOT,EACF,OAAkC;QAElC,KAAK,EAAE,CAAC;QAER,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC,SAAS,CAAC,CAAC;QAEvC,MAAM,KAAK,GAAiB;YAC1B,GAAG,CAAC,SAAS,IAAI,EAAE,CAAC;YACpB,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC;SAC/C,CAAC;QAEF,MAAM,GAAG,GAAG;YACV,cAAc,EAAE,OAAO,EAAE,cAAc;YACvC,OAAO,EAAE,OAAO,EAAE,OAAO;YACzB,SAAS,EAAE,KAAK;SACjB,CAAC;QAEF,IAAI,CAAC,GAAG;YACN,KAAK,CAAC,MAAM,GAAG,CAAC;gBACd,CAAC,CAAC,IAAI,OAAO,CAAU,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC;gBACpC,CAAC,CAAC,IAAI,gBAAgB,CAAU,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QAElD,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,YAAY,GAAG,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxE,IAAI,CAAC,IAAI,GAAG;YACV,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAE,GAAG,CAAC,YAAY,EAAE;YAC9B,IAAI,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO;SACzB,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAExB,KAAK,MAAM,KAAK,IAAI,MAAM,IAAI,EAAE,EAAE,CAAC;YACjC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACvB,CAAC;IACH,CAAC;IAEM,WAAW;QAChB,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAEe,KAAK,CAAC,OAAO,CAC3B,MAAU,EACV,OAA2C;QAE3C,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;YACrB,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACjD,CAAC;QAED,MAAM,UAAU,GAAG;YACjB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,IAAI;YACzB,CAAC,mBAAmB,CAAC,EAAE,IAAI,CAAC,WAAW;YACvC,CAAC,iBAAiB,CAAC,EAAE,IAAI,CAAC,YAAY,IAAI,MAAM;SACjD,CAAC;QAEF,OAAO,MAAM,OAAO,EAAE,MAAM,CAAC,eAAe,CAC1C,OAAO,EACP;YACE,IAAI,EAAE,UAAU,CAAC,MAAM;YACvB,UAAU;SACX,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACpD,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,OAAO,GAAG,CAAC;QACb,CAAC,CACF,CAAC;IACJ,CAAC;CACF"}
@@ -1,9 +1,9 @@
1
- import { Generate, type GenerateOptions } from '../dsp/generate.js';
2
- import type { GenIn, GenOut } from '../dsp/program.js';
3
- import { Signature } from '../dsp/sig.js';
4
- import type { AIService } from '../text/types.js';
5
- export declare class ChainOfThought<IN extends GenIn = GenIn, OUT extends GenOut = GenOut> extends Generate<IN, OUT & {
1
+ import type { AxAIService } from '../ai/types.js';
2
+ import { AxGenerate, type AxGenerateOptions } from '../dsp/generate.js';
3
+ import type { AxGenIn, AxGenOut } from '../dsp/program.js';
4
+ import { AxSignature } from '../dsp/sig.js';
5
+ export declare class AxChainOfThought<IN extends AxGenIn = AxGenIn, OUT extends AxGenOut = AxGenOut> extends AxGenerate<IN, OUT & {
6
6
  reason: string;
7
7
  }> {
8
- constructor(ai: AIService, signature: Readonly<Signature | string>, options?: Readonly<GenerateOptions>);
8
+ constructor(ai: AxAIService, signature: Readonly<AxSignature | string>, options?: Readonly<AxGenerateOptions>);
9
9
  }
@@ -1,8 +1,8 @@
1
- import { Generate } from '../dsp/generate.js';
2
- import { Signature } from '../dsp/sig.js';
3
- export class ChainOfThought extends Generate {
1
+ import { AxGenerate } from '../dsp/generate.js';
2
+ import { AxSignature } from '../dsp/sig.js';
3
+ export class AxChainOfThought extends AxGenerate {
4
4
  constructor(ai, signature, options) {
5
- const sig = new Signature(signature);
5
+ const sig = new AxSignature(signature);
6
6
  const description = `Let's work this out in a step by step way in order to ensure we have the right answer.`;
7
7
  sig.setOutputFields([
8
8
  {
@@ -1 +1 @@
1
- {"version":3,"file":"cot.js","sourceRoot":"","sources":["../../../../src/prompts/cot.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAwB,MAAM,oBAAoB,CAAC;AAEpE,OAAO,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAG1C,MAAM,OAAO,cAGX,SAAQ,QAAsC;IAC9C,YACE,EAAa,EACb,SAAuC,EACvC,OAAmC;QAEnC,MAAM,GAAG,GAAG,IAAI,SAAS,CAAC,SAAS,CAAC,CAAC;QACrC,MAAM,WAAW,GAAG,wFAAwF,CAAC;QAE7G,GAAG,CAAC,eAAe,CAAC;YAClB;gBACE,IAAI,EAAE,QAAQ;gBACd,WAAW;aACZ;YACD,GAAG,GAAG,CAAC,eAAe,EAAE;SACzB,CAAC,CAAC;QAEH,KAAK,CAAC,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;IAC1B,CAAC;CACF"}
1
+ {"version":3,"file":"cot.js","sourceRoot":"","sources":["../../../../src/prompts/cot.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAA0B,MAAM,oBAAoB,CAAC;AAExE,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAE5C,MAAM,OAAO,gBAGX,SAAQ,UAAwC;IAChD,YACE,EAAe,EACf,SAAyC,EACzC,OAAqC;QAErC,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC,SAAS,CAAC,CAAC;QACvC,MAAM,WAAW,GAAG,wFAAwF,CAAC;QAE7G,GAAG,CAAC,eAAe,CAAC;YAClB;gBACE,IAAI,EAAE,QAAQ;gBACd,WAAW;aACZ;YACD,GAAG,GAAG,CAAC,eAAe,EAAE;SACzB,CAAC,CAAC;QAEH,KAAK,CAAC,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;IAC1B,CAAC;CACF"}
@@ -1,7 +1,8 @@
1
1
  import { Readable } from 'stream';
2
2
  import test from 'ava';
3
- import { AI } from '../ai/index.js';
4
- import { ChainOfThought } from './cot.js';
3
+ import { axAI } from '../ai/index.js';
4
+ import { AxSignature } from '../dsp/sig.js';
5
+ import { AxChainOfThought } from './cot.js';
5
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.`;
6
7
  const examples = [
7
8
  {
@@ -35,9 +36,9 @@ const mockFetch = async () => {
35
36
  };
36
37
  test('generate prompt', async (t) => {
37
38
  const options = { fetch: mockFetch };
38
- const ai = AI('openai', { apiKey: 'no-key', options });
39
- // const ai = AI('ollama', { model: 'nous-hermes2' });
40
- const gen = new ChainOfThought(ai, `text -> shortSummary "summarize in 5 to 10 words"`);
39
+ const ai = axAI('openai', { apiKey: 'no-key', options });
40
+ // const ai = axAI('ollama', { model: 'nous-hermes2' });
41
+ const gen = new AxChainOfThought(ai, `text -> shortSummary "summarize in 5 to 10 words"`);
41
42
  gen.setExamples(examples);
42
43
  const res = await gen.forward({ text });
43
44
  t.deepEqual(res, {
@@ -45,4 +46,7 @@ test('generate prompt', async (t) => {
45
46
  shortSummary: 'More blah blah blah'
46
47
  });
47
48
  });
49
+ test('generate prompt: invalid signature', async (t) => {
50
+ t.throws(() => new AxSignature(`text -> output:image`));
51
+ });
48
52
  //# sourceMappingURL=prompts.test.js.map
@@ -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,EAAE,EAAmB,MAAM,gBAAgB,CAAC;AAErD,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAE1C,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,EAAE,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAgB,CAAC,CAAC;IAErE,sDAAsD;IAEtD,MAAM,GAAG,GAAG,IAAI,cAAc,CAC5B,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"}
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,8 +1,8 @@
1
- import { type GenerateOptions } from '../dsp/index.js';
2
- import { type ProgramForwardOptions } from '../dsp/program.js';
3
- import type { AIService } from '../text/types.js';
4
- import { ChainOfThought } from './cot.js';
5
- export declare class RAG extends ChainOfThought<{
1
+ import type { AxAIService } from '../ai/types.js';
2
+ import { type AxGenerateOptions } from '../dsp/index.js';
3
+ import { type AxProgramForwardOptions } from '../dsp/program.js';
4
+ import { AxChainOfThought } from './cot.js';
5
+ export declare class AxRAG extends AxChainOfThought<{
6
6
  context: string[];
7
7
  question: string;
8
8
  }, {
@@ -11,12 +11,12 @@ export declare class RAG extends ChainOfThought<{
11
11
  private genQuery;
12
12
  private queryFn;
13
13
  private maxHops;
14
- constructor(ai: AIService, queryFn: (query: string) => Promise<string>, options: Readonly<GenerateOptions & {
14
+ constructor(ai: AxAIService, queryFn: (query: string) => Promise<string>, options: Readonly<AxGenerateOptions & {
15
15
  maxHops?: number;
16
16
  }>);
17
17
  forward({ question }: Readonly<{
18
18
  question: string;
19
- }>, options?: Readonly<ProgramForwardOptions>): Promise<{
19
+ }>, options?: Readonly<AxProgramForwardOptions>): Promise<{
20
20
  answer: string;
21
21
  reason: string;
22
22
  }>;
@@ -1,7 +1,7 @@
1
- import { dedup, Generate, Signature } from '../dsp/index.js';
1
+ import { AxGenerate, AxSignature, axStringUtil } from '../dsp/index.js';
2
2
  import {} from '../dsp/program.js';
3
- import { ChainOfThought } from './cot.js';
4
- export class RAG extends ChainOfThought {
3
+ import { AxChainOfThought } from './cot.js';
4
+ export class AxRAG extends AxChainOfThought {
5
5
  genQuery;
6
6
  queryFn;
7
7
  maxHops;
@@ -9,8 +9,8 @@ export class RAG extends ChainOfThought {
9
9
  const sig = '"Answer questions with short factoid answers." context:string[] "may contain relevant facts", question -> answer';
10
10
  super(ai, sig, options);
11
11
  this.maxHops = options?.maxHops ?? 3;
12
- const qsig = new Signature('"Write a simple search query that will help answer a complex question." context?:string[] "may contain relevant facts", question -> query "question to further our understanding"');
13
- this.genQuery = new Generate(ai, qsig);
12
+ const qsig = new AxSignature('"Write a simple search query that will help answer a complex question." context?:string[] "may contain relevant facts", question -> query "question to further our understanding"');
13
+ this.genQuery = new AxGenerate(ai, qsig);
14
14
  this.queryFn = queryFn;
15
15
  this.register(this.genQuery);
16
16
  }
@@ -22,7 +22,7 @@ export class RAG extends ChainOfThought {
22
22
  question
23
23
  }, options);
24
24
  const val = await this.queryFn(query);
25
- context = dedup([...context, val]);
25
+ context = axStringUtil.dedup([...context, val]);
26
26
  }
27
27
  return super.forward({ context, question }, options);
28
28
  }
@@ -1 +1 @@
1
- {"version":3,"file":"rag.js","sourceRoot":"","sources":["../../../../src/prompts/rag.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,EACL,QAAQ,EAER,SAAS,EACV,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAA8B,MAAM,mBAAmB,CAAC;AAG/D,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAE1C,MAAM,OAAO,GAAI,SAAQ,cAGxB;IACS,QAAQ,CAGd;IACM,OAAO,CAAqC;IAC5C,OAAO,CAAS;IAExB,YACE,EAAa,EACb,OAA2C,EAC3C,OAAyD;QAEzD,MAAM,GAAG,GACP,kHAAkH,CAAC;QACrH,KAAK,CAAC,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;QAExB,IAAI,CAAC,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,CAAC,CAAC;QAErC,MAAM,IAAI,GAAG,IAAI,SAAS,CACxB,mLAAmL,CACpL,CAAC;QACF,IAAI,CAAC,QAAQ,GAAG,IAAI,QAAQ,CAG1B,EAAE,EAAE,IAAI,CAAC,CAAC;QACZ,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC/B,CAAC;IAEe,KAAK,CAAC,OAAO,CAC3B,EAAE,QAAQ,EAAkC,EAC5C,OAAyC;QAEzC,IAAI,OAAO,GAAa,EAAE,CAAC;QAE3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC;YACtC,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAC3C;gBACE,OAAO;gBACP,QAAQ;aACT,EACD,OAAO,CACR,CAAC;YACF,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YACtC,OAAO,GAAG,KAAK,CAAC,CAAC,GAAG,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;QACrC,CAAC;QAED,OAAO,KAAK,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,OAAO,CAAC,CAAC;IACvD,CAAC;CACF"}
1
+ {"version":3,"file":"rag.js","sourceRoot":"","sources":["../../../../src/prompts/rag.ts"],"names":[],"mappings":"AACA,OAAO,EACL,UAAU,EAEV,WAAW,EACX,YAAY,EACb,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAgC,MAAM,mBAAmB,CAAC;AAEjE,OAAO,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAE5C,MAAM,OAAO,KAAM,SAAQ,gBAG1B;IACS,QAAQ,CAGd;IACM,OAAO,CAAqC;IAC5C,OAAO,CAAS;IAExB,YACE,EAAe,EACf,OAA2C,EAC3C,OAA2D;QAE3D,MAAM,GAAG,GACP,kHAAkH,CAAC;QACrH,KAAK,CAAC,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;QAExB,IAAI,CAAC,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,CAAC,CAAC;QAErC,MAAM,IAAI,GAAG,IAAI,WAAW,CAC1B,mLAAmL,CACpL,CAAC;QACF,IAAI,CAAC,QAAQ,GAAG,IAAI,UAAU,CAG5B,EAAE,EAAE,IAAI,CAAC,CAAC;QACZ,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC/B,CAAC;IAEe,KAAK,CAAC,OAAO,CAC3B,EAAE,QAAQ,EAAkC,EAC5C,OAA2C;QAE3C,IAAI,OAAO,GAAa,EAAE,CAAC;QAE3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC;YACtC,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAC3C;gBACE,OAAO;gBACP,QAAQ;aACT,EACD,OAAO,CACR,CAAC;YACF,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YACtC,OAAO,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,GAAG,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;QAClD,CAAC;QAED,OAAO,KAAK,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,OAAO,CAAC,CAAC;IACvD,CAAC;CACF"}
@@ -1,8 +1,8 @@
1
- import type { GenerateOptions } from '../dsp/generate.js';
2
- import type { GenIn, GenOut } from '../dsp/program.js';
3
- import { Signature } from '../dsp/sig.js';
4
- import type { AIService } from '../text/types.js';
5
- import { ChainOfThought } from './cot.js';
6
- export declare class ReAct<IN extends GenIn = GenIn, OUT extends GenOut = GenOut> extends ChainOfThought<IN, OUT> {
7
- constructor(ai: AIService, signature: Readonly<Signature | string>, options: Readonly<GenerateOptions>);
1
+ import type { AxAIService } from '../ai/index.js';
2
+ import type { AxGenerateOptions } from '../dsp/generate.js';
3
+ import type { AxGenIn, AxGenOut } from '../dsp/program.js';
4
+ import { AxSignature } from '../dsp/sig.js';
5
+ import { AxChainOfThought } from './cot.js';
6
+ export declare class AxReAct<IN extends AxGenIn = AxGenIn, OUT extends AxGenOut = AxGenOut> extends AxChainOfThought<IN, OUT> {
7
+ constructor(ai: AxAIService, signature: Readonly<AxSignature | string>, options: Readonly<AxGenerateOptions>);
8
8
  }
@@ -1,13 +1,13 @@
1
- import { Signature } from '../dsp/sig.js';
2
- import { ChainOfThought } from './cot.js';
3
- export class ReAct extends ChainOfThought {
1
+ import { AxSignature } from '../dsp/sig.js';
2
+ import { AxChainOfThought } from './cot.js';
3
+ export class AxReAct extends AxChainOfThought {
4
4
  constructor(ai, signature, options) {
5
5
  if (!options?.functions || options.functions.length === 0) {
6
6
  throw new Error('No functions provided');
7
7
  }
8
8
  const functions = [...options.functions];
9
9
  const funcList = functions.map((f) => `'${f.name}'`).join(', ');
10
- const sig = new Signature(signature);
10
+ const sig = new AxSignature(signature);
11
11
  sig.setDescription(`Use the provided functions ${funcList} to complete the task and return the result if any.`);
12
12
  // sig.addInputField({
13
13
  // name: 'observation',
@@ -1 +1 @@
1
- {"version":3,"file":"react.js","sourceRoot":"","sources":["../../../../src/prompts/react.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAG1C,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAE1C,MAAM,OAAO,KAGX,SAAQ,cAAuB;IAC/B,YACE,EAAa,EACb,SAAuC,EACvC,OAAkC;QAElC,IAAI,CAAC,OAAO,EAAE,SAAS,IAAI,OAAO,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1D,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;QAED,MAAM,SAAS,GAAG,CAAC,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;QACzC,MAAM,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEhE,MAAM,GAAG,GAAG,IAAI,SAAS,CAAC,SAAS,CAAC,CAAC;QACrC,GAAG,CAAC,cAAc,CAChB,8BAA8B,QAAQ,qDAAqD,CAC5F,CAAC;QAEF,sBAAsB;QACtB,yBAAyB;QACzB,wDAAwD;QACxD,sBAAsB;QACtB,4CAA4C;QAC5C,MAAM;QAEN,GAAG,CAAC,eAAe,CACjB,GAAG,CAAC,eAAe,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAC/D,CAAC;QAEF,KAAK,CAAC,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;IAC1B,CAAC;CACF"}
1
+ {"version":3,"file":"react.js","sourceRoot":"","sources":["../../../../src/prompts/react.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAE5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAE5C,MAAM,OAAO,OAGX,SAAQ,gBAAyB;IACjC,YACE,EAAe,EACf,SAAyC,EACzC,OAAoC;QAEpC,IAAI,CAAC,OAAO,EAAE,SAAS,IAAI,OAAO,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1D,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;QAED,MAAM,SAAS,GAAG,CAAC,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;QACzC,MAAM,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEhE,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC,SAAS,CAAC,CAAC;QACvC,GAAG,CAAC,cAAc,CAChB,8BAA8B,QAAQ,qDAAqD,CAC5F,CAAC;QAEF,sBAAsB;QACtB,yBAAyB;QACzB,wDAAwD;QACxD,sBAAsB;QACtB,4CAA4C;QAC5C,MAAM;QAEN,GAAG,CAAC,eAAe,CACjB,GAAG,CAAC,eAAe,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAC/D,CAAC;QAEF,KAAK,CAAC,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;IAC1B,CAAC;CACF"}
@@ -1,25 +1,8 @@
1
- import type { TextResponseFunctionCall } from '../ai/types.js';
1
+ import type { ChatResponseFunctionCall } from '../ai/types.js';
2
2
  import type { AIServiceActionOptions, FunctionExec } from './types.js';
3
- export type AITextFunctionHandler = (args?: any, extra?: Readonly<AIServiceActionOptions>) => unknown;
4
- export type FunctionJSONSchema = {
5
- type: string;
6
- properties?: Record<string, FunctionJSONSchema & {
7
- enum?: string[];
8
- default?: unknown;
9
- description: string;
10
- }>;
11
- required?: string[];
12
- items?: FunctionJSONSchema;
13
- };
14
- export type AITextFunction = {
15
- name: string;
16
- description: string;
17
- parameters?: FunctionJSONSchema;
18
- func?: AITextFunctionHandler;
19
- };
20
3
  export declare class FunctionProcessor {
21
4
  private funcList;
22
- constructor(funcList: readonly AITextFunction[]);
5
+ constructor(funcList: readonly Function[]);
23
6
  private executeFunction;
24
- execute: (func: Readonly<TextResponseFunctionCall>, options?: Readonly<AIServiceActionOptions>) => Promise<FunctionExec>;
7
+ execute: (func: Readonly<ChatResponseFunctionCall>, options?: Readonly<AIServiceActionOptions>) => Promise<FunctionExec>;
25
8
  }
@@ -1 +1 @@
1
- {"version":3,"file":"functions.js","sourceRoot":"","sources":["../../../../src/text/functions.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAI1B,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AA8BrD,MAAM,OAAO,iBAAiB;IACpB,QAAQ,CAA4B;IAE5C,YAAY,QAAmC;QAC7C,QAAQ;aACL,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC;aAC3B,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,CAAC,UAAW,CAAC,CAAC,CAAC;QACrD,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAEO,eAAe,GAAG,KAAK,EAC7B,MAAgC,EAChC,IAAwC,EACxC,OAA0C,EACnB,EAAE;QACzB,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,wBAAwB,MAAM,CAAC,IAAI,kBAAkB,CAAC,CAAC;QACzE,CAAC;QAED,IAAI,MAAM,CAAC;QACX,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtC,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;YACvB,MAAM,GAAG,GACP,MAAM,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;gBACtB,CAAC,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC;gBAC5B,CAAC,CAAC,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAE1B,OAAO;gBACL,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC;aACrC,CAAC;QACJ,CAAC;QAED,MAAM,GAAG,GACP,MAAM,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;YACtB,CAAC,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC;YACpC,CAAC,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEhC,OAAO;YACL,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC;SACrC,CAAC;IACJ,CAAC,CAAC;IAEK,OAAO,GAAG,KAAK,EACpB,IAAwC,EACxC,OAA0C,EACnB,EAAE;QACzB,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAC/B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAC7C,CAAC;QACF,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,sBAAsB,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC;QACtD,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,2BAA2B,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3D,CAAC;QAED,+BAA+B;QAC/B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;QAEnE,sCAAsC;QACtC,0DAA0D;QAC1D,sDAAsD;QACtD,IAAI;QACJ,OAAO,QAAQ,CAAC;IAClB,CAAC,CAAC;CACH"}
1
+ {"version":3,"file":"functions.js","sourceRoot":"","sources":["../../../../src/text/functions.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAI1B,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AAGrD,MAAM,OAAO,iBAAiB;IACpB,QAAQ,CAAsB;IAEtC,YAAY,QAA6B;QACvC,QAAQ;aACL,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC;aAC3B,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,CAAC,UAAW,CAAC,CAAC,CAAC;QACrD,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAEO,eAAe,GAAG,KAAK,EAC7B,MAA0B,EAC1B,IAAwC,EACxC,OAA0C,EACnB,EAAE;QACzB,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,wBAAwB,MAAM,CAAC,IAAI,kBAAkB,CAAC,CAAC;QACzE,CAAC;QAED,IAAI,MAAM,CAAC;QACX,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtC,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;YACvB,MAAM,GAAG,GACP,MAAM,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;gBACtB,CAAC,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC;gBAC5B,CAAC,CAAC,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAE1B,OAAO;gBACL,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC;aACrC,CAAC;QACJ,CAAC;QAED,MAAM,GAAG,GACP,MAAM,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;YACtB,CAAC,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC;YACpC,CAAC,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEhC,OAAO;YACL,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC;SACrC,CAAC;IACJ,CAAC,CAAC;IAEK,OAAO,GAAG,KAAK,EACpB,IAAwC,EACxC,OAA0C,EACnB,EAAE;QACzB,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAC/B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAC7C,CAAC;QACF,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,sBAAsB,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC;QACtD,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,2BAA2B,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3D,CAAC;QAED,+BAA+B;QAC/B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;QAEnE,sCAAsC;QACtC,0DAA0D;QAC1D,sDAAsD;QACtD,IAAI;QACJ,OAAO,QAAQ,CAAC;IAClB,CAAC,CAAC;CACH"}
@@ -1,2 +1 @@
1
- import type { FunctionJSONSchema } from './functions.js';
2
- export declare const validateJSONSchema: (schema: Readonly<FunctionJSONSchema>) => void;
1
+ export declare const validateJSONSchema: (schema: FunctionJSONSchema) => void;
@@ -1,5 +1,5 @@
1
- import type { TextResponseResult } from '../ai/types.js';
2
- import type { AITextChatRequest } from '../types/index.js';
1
+ import type { ChatResponseResult } from '../ai/types.js';
2
+ import type { ChatRequest } from '../types/index.js';
3
3
  import type { AIMemory } from './types.js';
4
4
  /**
5
5
  * A memory class to store ai interactions
@@ -10,12 +10,12 @@ export declare class Memory implements AIMemory {
10
10
  private sdata;
11
11
  private limit;
12
12
  constructor(limit?: number);
13
- add(value: Readonly<AITextChatRequest['chatPrompt'][0] | AITextChatRequest['chatPrompt']>, sessionId?: string): void;
14
- addResult({ content, name, functionCalls }: Readonly<TextResponseResult>, sessionId?: string): void;
15
- updateResult({ content, name, functionCalls }: Readonly<TextResponseResult>, sessionId?: string): void;
16
- history(sessionId?: string): AITextChatRequest['chatPrompt'];
17
- peek(sessionId?: string): AITextChatRequest['chatPrompt'];
18
- getLast(sessionId?: string): AITextChatRequest['chatPrompt'][0] | undefined;
13
+ add(value: Readonly<ChatRequest['chatPrompt'][0] | ChatRequest['chatPrompt']>, sessionId?: string): void;
14
+ addResult({ content, name, functionCalls }: Readonly<ChatResponseResult>, sessionId?: string): void;
15
+ updateResult({ content, name, functionCalls }: Readonly<ChatResponseResult>, sessionId?: string): void;
16
+ history(sessionId?: string): ChatRequest['chatPrompt'];
17
+ peek(sessionId?: string): ChatRequest['chatPrompt'];
18
+ getLast(sessionId?: string): ChatRequest['chatPrompt'][0] | undefined;
19
19
  reset(sessionId?: string): void;
20
20
  private get;
21
21
  }
@@ -1 +1 @@
1
- {"version":3,"file":"memory.js","sourceRoot":"","sources":["../../../../src/text/memory.ts"],"names":[],"mappings":"AAKA;;;GAGG;AACH,MAAM,OAAO,MAAM;IACT,IAAI,GAAoC,EAAE,CAAC;IAC3C,KAAK,GAAG,IAAI,GAAG,EAA2C,CAAC;IAC3D,KAAK,CAAS;IAEtB,YAAY,KAAK,GAAG,EAAE;QACpB,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;YACf,MAAM,KAAK,CAAC,wCAAwC,CAAC,CAAC;QACxD,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,GAAG,CACD,KAEC,EACD,SAAkB;QAElB,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC9B,IAAI,CAAC,GAAG,CAAC,CAAC;QACV,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACzB,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC,CAAC;QACxC,CAAC;aAAM,CAAC;YACN,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;gBACT,GAAG,eAAe,CAAC,KAAK,CAAC;aACY,CAAC,CAAC;QAC3C,CAAC;QACD,IAAI,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;YAC1B,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IAED,SAAS,CACP,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAgC,EAC9D,SAAkB;QAElB,IAAI,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,aAAa,EAAE,EAAE,SAAS,CAAC,CAAC;IAC3E,CAAC;IAED,YAAY,CACV,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAgC,EAC9D,SAAkB;QAElB,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACjC,IAAI,SAAS,IAAI,IAAI,IAAI,OAAO,EAAE,CAAC;YACjC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACzB,CAAC;QACD,IAAI,MAAM,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;YAC3B,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACnB,CAAC;QACD,IAAI,eAAe,IAAI,IAAI,IAAI,aAAa,EAAE,CAAC;YAC7C,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACrC,CAAC;IACH,CAAC;IAED,OAAO,CAAC,SAAkB;QACxB,OAAO,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC7B,CAAC;IAED,IAAI,CAAC,SAAkB;QACrB,OAAO,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC7B,CAAC;IAED,OAAO,CAAC,SAAkB;QACxB,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC9B,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,SAAkB;QACtB,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,IAAI,CAAC,IAAI,GAAG,EAAE,CAAC;QACjB,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAEO,GAAG,CAAC,SAAkB;QAC5B,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,OAAO,IAAI,CAAC,IAAI,CAAC;QACnB,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YAC/B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAChC,CAAC;QAED,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;IACzC,CAAC;CACF"}
1
+ {"version":3,"file":"memory.js","sourceRoot":"","sources":["../../../../src/text/memory.ts"],"names":[],"mappings":"AAKA;;;GAGG;AACH,MAAM,OAAO,MAAM;IACT,IAAI,GAA8B,EAAE,CAAC;IACrC,KAAK,GAAG,IAAI,GAAG,EAAqC,CAAC;IACrD,KAAK,CAAS;IAEtB,YAAY,KAAK,GAAG,EAAE;QACpB,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;YACf,MAAM,KAAK,CAAC,wCAAwC,CAAC,CAAC;QACxD,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,GAAG,CACD,KAAyE,EACzE,SAAkB;QAElB,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC9B,IAAI,CAAC,GAAG,CAAC,CAAC;QACV,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACzB,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC,CAAC;QACxC,CAAC;aAAM,CAAC;YACN,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;gBACT,GAAG,eAAe,CAAC,KAAK,CAAC;aACM,CAAC,CAAC;QACrC,CAAC;QACD,IAAI,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;YAC1B,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IAED,SAAS,CACP,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAgC,EAC9D,SAAkB;QAElB,IAAI,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,aAAa,EAAE,EAAE,SAAS,CAAC,CAAC;IAC3E,CAAC;IAED,YAAY,CACV,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAgC,EAC9D,SAAkB;QAElB,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACjC,IAAI,SAAS,IAAI,IAAI,IAAI,OAAO,EAAE,CAAC;YACjC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACzB,CAAC;QACD,IAAI,MAAM,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;YAC3B,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACnB,CAAC;QACD,IAAI,eAAe,IAAI,IAAI,IAAI,aAAa,EAAE,CAAC;YAC7C,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACrC,CAAC;IACH,CAAC;IAED,OAAO,CAAC,SAAkB;QACxB,OAAO,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC7B,CAAC;IAED,IAAI,CAAC,SAAkB;QACrB,OAAO,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC7B,CAAC;IAED,OAAO,CAAC,SAAkB;QACxB,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC9B,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,SAAkB;QACtB,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,IAAI,CAAC,IAAI,GAAG,EAAE,CAAC;QACjB,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAEO,GAAG,CAAC,SAAkB;QAC5B,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,OAAO,IAAI,CAAC,IAAI,CAAC;QACnB,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YAC/B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAChC,CAAC;QAED,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;IACzC,CAAC;CACF"}
@@ -1,28 +1,28 @@
1
1
  /// <reference types="node" />
2
2
  /// <reference types="node" />
3
3
  import type { ReadableStream } from 'stream/web';
4
- import type { EmbedResponse, RateLimiterFunction, TextModelConfig, TextModelInfo, TextResponse, TextResponseResult } from '../ai/types.js';
4
+ import type { ChatResponse, ChatResponseResult, EmbedResponse, ModelConfig, ModelInfo, RateLimiterFunction } from '../ai/types.js';
5
5
  import type { Tracer } from '../trace/index.js';
6
- import type { AITextChatRequest, AITextEmbedRequest } from '../types/index.js';
6
+ import type { ChatRequest, EmbedRequest } from '../types/index.js';
7
7
  export type FunctionExec = {
8
8
  id?: string;
9
9
  name: string;
10
10
  args?: any;
11
11
  result?: string;
12
12
  };
13
- export type AITextResponse<T> = {
13
+ export type AIChatResponse<T> = {
14
14
  prompt: string;
15
15
  sessionId?: string;
16
16
  value(): T;
17
17
  };
18
18
  export interface AIMemory {
19
- add(result: Readonly<AITextChatRequest['chatPrompt'] | AITextChatRequest['chatPrompt'][0]>, sessionId?: string): void;
20
- addResult(result: Readonly<TextResponseResult>, sessionId?: string): void;
21
- updateResult(result: Readonly<TextResponseResult>, sessionId?: string): void;
22
- history(sessionId?: string): AITextChatRequest['chatPrompt'];
23
- peek(sessionId?: string): AITextChatRequest['chatPrompt'];
19
+ add(result: Readonly<ChatRequest['chatPrompt'] | ChatRequest['chatPrompt'][0]>, sessionId?: string): void;
20
+ addResult(result: Readonly<ChatResponseResult>, sessionId?: string): void;
21
+ updateResult(result: Readonly<ChatResponseResult>, sessionId?: string): void;
22
+ history(sessionId?: string): ChatRequest['chatPrompt'];
23
+ peek(sessionId?: string): ChatRequest['chatPrompt'];
24
24
  reset(sessionId?: string): void;
25
- getLast(sessionId?: string): AITextChatRequest['chatPrompt'][0] | undefined;
25
+ getLast(sessionId?: string): ChatRequest['chatPrompt'][0] | undefined;
26
26
  }
27
27
  export type AIPromptConfig = {
28
28
  stream?: boolean;
@@ -42,16 +42,16 @@ export type AIServiceActionOptions = {
42
42
  };
43
43
  export interface AIService {
44
44
  getName(): string;
45
- getModelInfo(): Readonly<TextModelInfo & {
45
+ getModelInfo(): Readonly<ModelInfo & {
46
46
  provider: string;
47
47
  }>;
48
- getEmbedModelInfo(): Readonly<TextModelInfo> | undefined;
49
- getModelConfig(): Readonly<TextModelConfig>;
48
+ getEmbedModelInfo(): Readonly<ModelInfo> | undefined;
49
+ getModelConfig(): Readonly<ModelConfig>;
50
50
  getFeatures(): {
51
51
  functions: boolean;
52
52
  streaming: boolean;
53
53
  };
54
- chat(req: Readonly<AITextChatRequest>, options?: Readonly<AIPromptConfig & AIServiceActionOptions>): Promise<TextResponse | ReadableStream<TextResponse>>;
55
- embed(req: Readonly<AITextEmbedRequest>, options?: Readonly<AIServiceActionOptions & AIServiceActionOptions>): Promise<EmbedResponse>;
54
+ chat(req: Readonly<ChatRequest>, options?: Readonly<AIPromptConfig & AIServiceActionOptions>): Promise<ChatResponse | ReadableStream<ChatResponse>>;
55
+ embed(req: Readonly<EmbedRequest>, options?: Readonly<AIServiceActionOptions & AIServiceActionOptions>): Promise<EmbedResponse>;
56
56
  setOptions(options: Readonly<AIServiceOptions>): void;
57
57
  }