langchain 0.0.4

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 (217) hide show
  1. package/dist/cjs/agents/agent.d.ts +42 -0
  2. package/dist/cjs/agents/agent.js +129 -0
  3. package/dist/cjs/agents/agent.js.map +1 -0
  4. package/dist/cjs/agents/executor.d.ts +27 -0
  5. package/dist/cjs/agents/executor.js +105 -0
  6. package/dist/cjs/agents/executor.js.map +1 -0
  7. package/dist/cjs/agents/helpers.d.ts +3 -0
  8. package/dist/cjs/agents/helpers.js +30 -0
  9. package/dist/cjs/agents/helpers.js.map +1 -0
  10. package/dist/cjs/agents/index.d.ts +6 -0
  11. package/dist/cjs/agents/index.js +13 -0
  12. package/dist/cjs/agents/index.js.map +1 -0
  13. package/dist/cjs/agents/load.d.ts +6 -0
  14. package/dist/cjs/agents/load.js +28 -0
  15. package/dist/cjs/agents/load.js.map +1 -0
  16. package/dist/cjs/agents/mrkl/index.d.ts +33 -0
  17. package/dist/cjs/agents/mrkl/index.js +109 -0
  18. package/dist/cjs/agents/mrkl/index.js.map +1 -0
  19. package/dist/cjs/agents/mrkl/prompt.d.ts +3 -0
  20. package/dist/cjs/agents/mrkl/prompt.js +21 -0
  21. package/dist/cjs/agents/mrkl/prompt.js.map +1 -0
  22. package/dist/cjs/agents/tests/agent.test.d.ts +1 -0
  23. package/dist/cjs/agents/tests/agent.test.js +31 -0
  24. package/dist/cjs/agents/tests/agent.test.js.map +1 -0
  25. package/dist/cjs/agents/tools/index.d.ts +7 -0
  26. package/dist/cjs/agents/tools/index.js +6 -0
  27. package/dist/cjs/agents/tools/index.js.map +1 -0
  28. package/dist/cjs/agents/tools/llm-math.d.ts +0 -0
  29. package/dist/cjs/agents/tools/llm-math.js +2 -0
  30. package/dist/cjs/agents/tools/llm-math.js.map +1 -0
  31. package/dist/cjs/agents/tools/serpapi.d.ts +3 -0
  32. package/dist/cjs/agents/tools/serpapi.js +50 -0
  33. package/dist/cjs/agents/tools/serpapi.js.map +1 -0
  34. package/dist/cjs/agents/types.d.ts +25 -0
  35. package/dist/cjs/agents/types.js +3 -0
  36. package/dist/cjs/agents/types.js.map +1 -0
  37. package/dist/cjs/cache.d.ts +11 -0
  38. package/dist/cjs/cache.js +27 -0
  39. package/dist/cjs/cache.js.map +1 -0
  40. package/dist/cjs/chains/base.d.ts +12 -0
  41. package/dist/cjs/chains/base.js +23 -0
  42. package/dist/cjs/chains/base.js.map +1 -0
  43. package/dist/cjs/chains/index.d.ts +3 -0
  44. package/dist/cjs/chains/index.js +10 -0
  45. package/dist/cjs/chains/index.js.map +1 -0
  46. package/dist/cjs/chains/llm_chain.d.ts +30 -0
  47. package/dist/cjs/chains/llm_chain.js +83 -0
  48. package/dist/cjs/chains/llm_chain.js.map +1 -0
  49. package/dist/cjs/chains/llm_math_chain.d.ts +0 -0
  50. package/dist/cjs/chains/llm_math_chain.js +129 -0
  51. package/dist/cjs/chains/llm_math_chain.js.map +1 -0
  52. package/dist/cjs/chains/load.d.ts +2 -0
  53. package/dist/cjs/chains/load.js +29 -0
  54. package/dist/cjs/chains/load.js.map +1 -0
  55. package/dist/cjs/chains/tests/llm_chain.test.d.ts +1 -0
  56. package/dist/cjs/chains/tests/llm_chain.test.js +32 -0
  57. package/dist/cjs/chains/tests/llm_chain.test.js.map +1 -0
  58. package/dist/cjs/index.d.ts +3 -0
  59. package/dist/cjs/index.js +12 -0
  60. package/dist/cjs/index.js.map +1 -0
  61. package/dist/cjs/llms/base.d.ts +23 -0
  62. package/dist/cjs/llms/base.js +159 -0
  63. package/dist/cjs/llms/base.js.map +1 -0
  64. package/dist/cjs/llms/index.d.ts +18 -0
  65. package/dist/cjs/llms/index.js +11 -0
  66. package/dist/cjs/llms/index.js.map +1 -0
  67. package/dist/cjs/llms/load.d.ts +3 -0
  68. package/dist/cjs/llms/load.js +9 -0
  69. package/dist/cjs/llms/load.js.map +1 -0
  70. package/dist/cjs/llms/openai.d.ts +63 -0
  71. package/dist/cjs/llms/openai.js +202 -0
  72. package/dist/cjs/llms/openai.js.map +1 -0
  73. package/dist/cjs/llms/tests/openai.test.d.ts +1 -0
  74. package/dist/cjs/llms/tests/openai.test.js +19 -0
  75. package/dist/cjs/llms/tests/openai.test.js.map +1 -0
  76. package/dist/cjs/prompt/base.d.ts +19 -0
  77. package/dist/cjs/prompt/base.js +54 -0
  78. package/dist/cjs/prompt/base.js.map +1 -0
  79. package/dist/cjs/prompt/few_shot.d.ts +47 -0
  80. package/dist/cjs/prompt/few_shot.js +143 -0
  81. package/dist/cjs/prompt/few_shot.js.map +1 -0
  82. package/dist/cjs/prompt/index.d.ts +3 -0
  83. package/dist/cjs/prompt/index.js +10 -0
  84. package/dist/cjs/prompt/index.js.map +1 -0
  85. package/dist/cjs/prompt/load.d.ts +2 -0
  86. package/dist/cjs/prompt/load.js +25 -0
  87. package/dist/cjs/prompt/load.js.map +1 -0
  88. package/dist/cjs/prompt/parser.d.ts +40 -0
  89. package/dist/cjs/prompt/parser.js +93 -0
  90. package/dist/cjs/prompt/parser.js.map +1 -0
  91. package/dist/cjs/prompt/prompt.d.ts +28 -0
  92. package/dist/cjs/prompt/prompt.js +91 -0
  93. package/dist/cjs/prompt/prompt.js.map +1 -0
  94. package/dist/cjs/prompt/template.d.ts +16 -0
  95. package/dist/cjs/prompt/template.js +87 -0
  96. package/dist/cjs/prompt/template.js.map +1 -0
  97. package/dist/cjs/prompt/tests/load.test.d.ts +1 -0
  98. package/dist/cjs/prompt/tests/load.test.js +30 -0
  99. package/dist/cjs/prompt/tests/load.test.js.map +1 -0
  100. package/dist/cjs/prompt/tests/template.test.d.ts +1 -0
  101. package/dist/cjs/prompt/tests/template.test.js +27 -0
  102. package/dist/cjs/prompt/tests/template.test.js.map +1 -0
  103. package/dist/cjs/util/hub.d.ts +1 -0
  104. package/dist/cjs/util/hub.js +50 -0
  105. package/dist/cjs/util/hub.js.map +1 -0
  106. package/dist/cjs/util/index.d.ts +12 -0
  107. package/dist/cjs/util/index.js +107 -0
  108. package/dist/cjs/util/index.js.map +1 -0
  109. package/dist/esm/agents/agent.d.ts +42 -0
  110. package/dist/esm/agents/agent.js +124 -0
  111. package/dist/esm/agents/agent.js.map +1 -0
  112. package/dist/esm/agents/executor.d.ts +27 -0
  113. package/dist/esm/agents/executor.js +101 -0
  114. package/dist/esm/agents/executor.js.map +1 -0
  115. package/dist/esm/agents/helpers.d.ts +3 -0
  116. package/dist/esm/agents/helpers.js +26 -0
  117. package/dist/esm/agents/helpers.js.map +1 -0
  118. package/dist/esm/agents/index.d.ts +6 -0
  119. package/dist/esm/agents/index.js +5 -0
  120. package/dist/esm/agents/index.js.map +1 -0
  121. package/dist/esm/agents/load.d.ts +6 -0
  122. package/dist/esm/agents/load.js +24 -0
  123. package/dist/esm/agents/load.js.map +1 -0
  124. package/dist/esm/agents/mrkl/index.d.ts +33 -0
  125. package/dist/esm/agents/mrkl/index.js +106 -0
  126. package/dist/esm/agents/mrkl/index.js.map +1 -0
  127. package/dist/esm/agents/mrkl/prompt.d.ts +3 -0
  128. package/dist/esm/agents/mrkl/prompt.js +17 -0
  129. package/dist/esm/agents/mrkl/prompt.js.map +1 -0
  130. package/dist/esm/agents/tests/agent.test.d.ts +1 -0
  131. package/dist/esm/agents/tests/agent.test.js +29 -0
  132. package/dist/esm/agents/tests/agent.test.js.map +1 -0
  133. package/dist/esm/agents/tools/index.d.ts +7 -0
  134. package/dist/esm/agents/tools/index.js +2 -0
  135. package/dist/esm/agents/tools/index.js.map +1 -0
  136. package/dist/esm/agents/tools/llm-math.d.ts +0 -0
  137. package/dist/esm/agents/tools/llm-math.js +2 -0
  138. package/dist/esm/agents/tools/llm-math.js.map +1 -0
  139. package/dist/esm/agents/tools/serpapi.d.ts +3 -0
  140. package/dist/esm/agents/tools/serpapi.js +46 -0
  141. package/dist/esm/agents/tools/serpapi.js.map +1 -0
  142. package/dist/esm/agents/types.d.ts +25 -0
  143. package/dist/esm/agents/types.js +2 -0
  144. package/dist/esm/agents/types.js.map +1 -0
  145. package/dist/esm/cache.d.ts +11 -0
  146. package/dist/esm/cache.js +22 -0
  147. package/dist/esm/cache.js.map +1 -0
  148. package/dist/esm/chains/base.d.ts +12 -0
  149. package/dist/esm/chains/base.js +19 -0
  150. package/dist/esm/chains/base.js.map +1 -0
  151. package/dist/esm/chains/index.d.ts +3 -0
  152. package/dist/esm/chains/index.js +4 -0
  153. package/dist/esm/chains/index.js.map +1 -0
  154. package/dist/esm/chains/llm_chain.d.ts +30 -0
  155. package/dist/esm/chains/llm_chain.js +79 -0
  156. package/dist/esm/chains/llm_chain.js.map +1 -0
  157. package/dist/esm/chains/llm_math_chain.d.ts +0 -0
  158. package/dist/esm/chains/llm_math_chain.js +129 -0
  159. package/dist/esm/chains/llm_math_chain.js.map +1 -0
  160. package/dist/esm/chains/load.d.ts +2 -0
  161. package/dist/esm/chains/load.js +25 -0
  162. package/dist/esm/chains/load.js.map +1 -0
  163. package/dist/esm/chains/tests/llm_chain.test.d.ts +1 -0
  164. package/dist/esm/chains/tests/llm_chain.test.js +30 -0
  165. package/dist/esm/chains/tests/llm_chain.test.js.map +1 -0
  166. package/dist/esm/index.d.ts +3 -0
  167. package/dist/esm/index.js +4 -0
  168. package/dist/esm/index.js.map +1 -0
  169. package/dist/esm/llms/base.d.ts +23 -0
  170. package/dist/esm/llms/base.js +154 -0
  171. package/dist/esm/llms/base.js.map +1 -0
  172. package/dist/esm/llms/index.d.ts +18 -0
  173. package/dist/esm/llms/index.js +4 -0
  174. package/dist/esm/llms/index.js.map +1 -0
  175. package/dist/esm/llms/load.d.ts +3 -0
  176. package/dist/esm/llms/load.js +5 -0
  177. package/dist/esm/llms/load.js.map +1 -0
  178. package/dist/esm/llms/openai.d.ts +63 -0
  179. package/dist/esm/llms/openai.js +198 -0
  180. package/dist/esm/llms/openai.js.map +1 -0
  181. package/dist/esm/llms/tests/openai.test.d.ts +1 -0
  182. package/dist/esm/llms/tests/openai.test.js +17 -0
  183. package/dist/esm/llms/tests/openai.test.js.map +1 -0
  184. package/dist/esm/prompt/base.d.ts +19 -0
  185. package/dist/esm/prompt/base.js +50 -0
  186. package/dist/esm/prompt/base.js.map +1 -0
  187. package/dist/esm/prompt/few_shot.d.ts +47 -0
  188. package/dist/esm/prompt/few_shot.js +139 -0
  189. package/dist/esm/prompt/few_shot.js.map +1 -0
  190. package/dist/esm/prompt/index.d.ts +3 -0
  191. package/dist/esm/prompt/index.js +4 -0
  192. package/dist/esm/prompt/index.js.map +1 -0
  193. package/dist/esm/prompt/load.d.ts +2 -0
  194. package/dist/esm/prompt/load.js +21 -0
  195. package/dist/esm/prompt/load.js.map +1 -0
  196. package/dist/esm/prompt/parser.d.ts +40 -0
  197. package/dist/esm/prompt/parser.js +86 -0
  198. package/dist/esm/prompt/parser.js.map +1 -0
  199. package/dist/esm/prompt/prompt.d.ts +28 -0
  200. package/dist/esm/prompt/prompt.js +87 -0
  201. package/dist/esm/prompt/prompt.js.map +1 -0
  202. package/dist/esm/prompt/template.d.ts +16 -0
  203. package/dist/esm/prompt/template.js +80 -0
  204. package/dist/esm/prompt/template.js.map +1 -0
  205. package/dist/esm/prompt/tests/load.test.d.ts +1 -0
  206. package/dist/esm/prompt/tests/load.test.js +25 -0
  207. package/dist/esm/prompt/tests/load.test.js.map +1 -0
  208. package/dist/esm/prompt/tests/template.test.d.ts +1 -0
  209. package/dist/esm/prompt/tests/template.test.js +25 -0
  210. package/dist/esm/prompt/tests/template.test.js.map +1 -0
  211. package/dist/esm/util/hub.d.ts +1 -0
  212. package/dist/esm/util/hub.js +43 -0
  213. package/dist/esm/util/hub.js.map +1 -0
  214. package/dist/esm/util/index.d.ts +12 -0
  215. package/dist/esm/util/index.js +74 -0
  216. package/dist/esm/util/index.js.map +1 -0
  217. package/package.json +97 -0
@@ -0,0 +1,143 @@
1
+ "use strict";
2
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
3
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
4
+ return new (P || (P = Promise))(function (resolve, reject) {
5
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
6
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
7
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
8
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
9
+ });
10
+ };
11
+ Object.defineProperty(exports, "__esModule", { value: true });
12
+ exports.FewShotPromptTemplate = void 0;
13
+ const index_1 = require("./index");
14
+ const template_1 = require("./template");
15
+ const util_1 = require("../util");
16
+ const prompt_1 = require("./prompt");
17
+ const parser_1 = require("./parser");
18
+ class FewShotPromptTemplate extends index_1.BasePromptTemplate {
19
+ constructor(input) {
20
+ super(input);
21
+ Object.defineProperty(this, "examples", {
22
+ enumerable: true,
23
+ configurable: true,
24
+ writable: true,
25
+ value: void 0
26
+ });
27
+ Object.defineProperty(this, "exampleSelector", {
28
+ enumerable: true,
29
+ configurable: true,
30
+ writable: true,
31
+ value: void 0
32
+ });
33
+ Object.defineProperty(this, "examplePrompt", {
34
+ enumerable: true,
35
+ configurable: true,
36
+ writable: true,
37
+ value: void 0
38
+ });
39
+ Object.defineProperty(this, "suffix", {
40
+ enumerable: true,
41
+ configurable: true,
42
+ writable: true,
43
+ value: void 0
44
+ });
45
+ Object.defineProperty(this, "exampleSeparator", {
46
+ enumerable: true,
47
+ configurable: true,
48
+ writable: true,
49
+ value: void 0
50
+ });
51
+ Object.defineProperty(this, "prefix", {
52
+ enumerable: true,
53
+ configurable: true,
54
+ writable: true,
55
+ value: void 0
56
+ });
57
+ Object.defineProperty(this, "templateFormat", {
58
+ enumerable: true,
59
+ configurable: true,
60
+ writable: true,
61
+ value: "f-string"
62
+ });
63
+ Object.defineProperty(this, "validateTemplate", {
64
+ enumerable: true,
65
+ configurable: true,
66
+ writable: true,
67
+ value: true
68
+ });
69
+ Object.assign(this, input);
70
+ if (this.examples !== undefined && this.exampleSelector !== undefined) {
71
+ throw new Error("Only one of 'examples' and 'example_selector' should be provided");
72
+ }
73
+ if (this.examples === undefined && this.exampleSelector === undefined) {
74
+ throw new Error("One of 'examples' and 'example_selector' should be provided");
75
+ }
76
+ if (this.validateTemplate) {
77
+ (0, template_1.checkValidTemplate)(this.prefix + this.suffix, this.templateFormat, this.inputVariables);
78
+ }
79
+ }
80
+ _getPromptType() {
81
+ return "few_shot";
82
+ }
83
+ getExamples(_) {
84
+ if (this.examples !== undefined) {
85
+ return this.examples;
86
+ }
87
+ if (this.exampleSelector !== undefined) {
88
+ throw new Error("Example selectors are not yet supported.");
89
+ }
90
+ throw new Error("One of 'examples' and 'example_selector' should be provided");
91
+ }
92
+ format(values) {
93
+ const examples = this.getExamples(values);
94
+ const exampleStrings = examples.map((example) => this.examplePrompt.format(example));
95
+ const template = [this.prefix, ...exampleStrings, this.suffix].join(this.exampleSeparator);
96
+ return (0, template_1.renderTemplate)(template, this.templateFormat, values);
97
+ }
98
+ serialize() {
99
+ var _a;
100
+ if (this.exampleSelector || !this.examples) {
101
+ throw new Error("Serializing an example selector is not currently supported");
102
+ }
103
+ return {
104
+ _type: this._getPromptType(),
105
+ input_variables: this.inputVariables,
106
+ output_parser: (_a = this.outputParser) === null || _a === void 0 ? void 0 : _a.serialize(),
107
+ example_prompt: this.examplePrompt.serialize(),
108
+ example_separator: this.exampleSeparator,
109
+ suffix: this.suffix,
110
+ prefix: this.prefix,
111
+ template_format: this.templateFormat,
112
+ examples: this.examples,
113
+ };
114
+ }
115
+ static deserialize(data) {
116
+ return __awaiter(this, void 0, void 0, function* () {
117
+ const serializedPrompt = (0, util_1.resolveConfigFromFile)("example_prompt", data);
118
+ const examplePrompt = yield prompt_1.PromptTemplate.deserialize(serializedPrompt);
119
+ let examples;
120
+ if (typeof data.examples === "string") {
121
+ examples = (0, util_1.parseFileConfig)(data.examples, [".json", ".yml", ".yaml"]);
122
+ }
123
+ else if (Array.isArray(data.examples)) {
124
+ examples = data.examples;
125
+ }
126
+ else {
127
+ throw new Error("Invalid examples format. Only list or string are supported.");
128
+ }
129
+ return new FewShotPromptTemplate({
130
+ inputVariables: data.input_variables,
131
+ outputParser: data.output_parser && parser_1.BaseOutputParser.deserialize(data.output_parser),
132
+ examplePrompt,
133
+ examples,
134
+ exampleSeparator: data.example_separator,
135
+ prefix: (0, util_1.resolveTemplateFromFile)("prefix", data),
136
+ suffix: (0, util_1.resolveTemplateFromFile)("suffix", data),
137
+ templateFormat: data.template_format,
138
+ });
139
+ });
140
+ }
141
+ }
142
+ exports.FewShotPromptTemplate = FewShotPromptTemplate;
143
+ //# sourceMappingURL=few_shot.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"few_shot.js","sourceRoot":"","sources":["../../../prompt/few_shot.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,mCAIiB;AACjB,yCAAgF;AAChF,kCAIiB;AACjB,qCAAoE;AACpE,qCAAoE;AAkCpE,MAAa,qBACX,SAAQ,0BAAkB;IAmB1B,YAAY,KAAiC;QAC3C,KAAK,CAAC,KAAK,CAAC,CAAC;QAjBf;;;;;WAAyB;QAEzB;;;;;WAAkC;QAElC;;;;;WAA8B;QAE9B;;;;;WAAe;QAEf;;;;;WAAyB;QAEzB;;;;;WAAe;QAEf;;;;mBAAiC,UAAU;WAAC;QAE5C;;;;mBAAmB,IAAI;WAAC;QAItB,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAE3B,IAAI,IAAI,CAAC,QAAQ,KAAK,SAAS,IAAI,IAAI,CAAC,eAAe,KAAK,SAAS,EAAE;YACrE,MAAM,IAAI,KAAK,CACb,kEAAkE,CACnE,CAAC;SACH;QAED,IAAI,IAAI,CAAC,QAAQ,KAAK,SAAS,IAAI,IAAI,CAAC,eAAe,KAAK,SAAS,EAAE;YACrE,MAAM,IAAI,KAAK,CACb,6DAA6D,CAC9D,CAAC;SACH;QAED,IAAI,IAAI,CAAC,gBAAgB,EAAE;YACzB,IAAA,6BAAkB,EAChB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,EACzB,IAAI,CAAC,cAAc,EACnB,IAAI,CAAC,cAAc,CACpB,CAAC;SACH;IACH,CAAC;IAED,cAAc;QACZ,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,WAAW,CAAC,CAAc;QAChC,IAAI,IAAI,CAAC,QAAQ,KAAK,SAAS,EAAE;YAC/B,OAAO,IAAI,CAAC,QAAQ,CAAC;SACtB;QACD,IAAI,IAAI,CAAC,eAAe,KAAK,SAAS,EAAE;YACtC,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;SAC7D;QAED,MAAM,IAAI,KAAK,CACb,6DAA6D,CAC9D,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,MAAmB;QACxB,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QAE1C,MAAM,cAAc,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAC9C,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,OAAO,CAAC,CACnC,CAAC;QACF,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CACjE,IAAI,CAAC,gBAAgB,CACtB,CAAC;QACF,OAAO,IAAA,yBAAc,EAAC,QAAQ,EAAE,IAAI,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;IAC/D,CAAC;IAED,SAAS;;QACP,IAAI,IAAI,CAAC,eAAe,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE;YAC1C,MAAM,IAAI,KAAK,CACb,4DAA4D,CAC7D,CAAC;SACH;QACD,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,cAAc,EAAE;YAC5B,eAAe,EAAE,IAAI,CAAC,cAAc;YACpC,aAAa,EAAE,MAAA,IAAI,CAAC,YAAY,0CAAE,SAAS,EAAE;YAC7C,cAAc,EAAE,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE;YAC9C,iBAAiB,EAAE,IAAI,CAAC,gBAAgB;YACxC,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,eAAe,EAAE,IAAI,CAAC,cAAc;YACpC,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACxB,CAAC;IACJ,CAAC;IAED,MAAM,CAAO,WAAW,CACtB,IAA+B;;YAE/B,MAAM,gBAAgB,GAAG,IAAA,4BAAqB,EAG5C,gBAAgB,EAAE,IAAI,CAAC,CAAC;YAC1B,MAAM,aAAa,GAAG,MAAM,uBAAc,CAAC,WAAW,CAAC,gBAAgB,CAAC,CAAC;YAEzE,IAAI,QAAmB,CAAC;YAExB,IAAI,OAAO,IAAI,CAAC,QAAQ,KAAK,QAAQ,EAAE;gBACrC,QAAQ,GAAG,IAAA,sBAAe,EAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;aACvE;iBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE;gBACvC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;aAC1B;iBAAM;gBACL,MAAM,IAAI,KAAK,CACb,6DAA6D,CAC9D,CAAC;aACH;YAED,OAAO,IAAI,qBAAqB,CAAC;gBAC/B,cAAc,EAAE,IAAI,CAAC,eAAe;gBACpC,YAAY,EACV,IAAI,CAAC,aAAa,IAAI,yBAAgB,CAAC,WAAW,CAAC,IAAI,CAAC,aAAa,CAAC;gBACxE,aAAa;gBACb,QAAQ;gBACR,gBAAgB,EAAE,IAAI,CAAC,iBAAiB;gBACxC,MAAM,EAAE,IAAA,8BAAuB,EAAC,QAAQ,EAAE,IAAI,CAAC;gBAC/C,MAAM,EAAE,IAAA,8BAAuB,EAAC,QAAQ,EAAE,IAAI,CAAC;gBAC/C,cAAc,EAAE,IAAI,CAAC,eAAe;aACrC,CAAC,CAAC;QACL,CAAC;KAAA;CACF;AA9HD,sDA8HC"}
@@ -0,0 +1,3 @@
1
+ export { BasePromptTemplate, BasePromptTemplateInput, SerializedBasePromptTemplate, InputValues, } from "./base";
2
+ export { PromptTemplate, PromptTemplateInput, SerializedPromptTemplate, } from "./prompt";
3
+ export { FewShotPromptTemplate, FewShotPromptTemplateInput, SerializedFewShotTemplate, } from "./few_shot";
@@ -0,0 +1,10 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.FewShotPromptTemplate = exports.PromptTemplate = exports.BasePromptTemplate = void 0;
4
+ var base_1 = require("./base");
5
+ Object.defineProperty(exports, "BasePromptTemplate", { enumerable: true, get: function () { return base_1.BasePromptTemplate; } });
6
+ var prompt_1 = require("./prompt");
7
+ Object.defineProperty(exports, "PromptTemplate", { enumerable: true, get: function () { return prompt_1.PromptTemplate; } });
8
+ var few_shot_1 = require("./few_shot");
9
+ Object.defineProperty(exports, "FewShotPromptTemplate", { enumerable: true, get: function () { return few_shot_1.FewShotPromptTemplate; } });
10
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../../prompt/index.ts"],"names":[],"mappings":";;;AAAA,+BAKgB;AAJd,0GAAA,kBAAkB,OAAA;AAKpB,mCAIkB;AAHhB,wGAAA,cAAc,OAAA;AAIhB,uCAIoB;AAHlB,iHAAA,qBAAqB,OAAA"}
@@ -0,0 +1,2 @@
1
+ import { BasePromptTemplate } from ".";
2
+ export declare const loadPrompt: (uri: string) => Promise<BasePromptTemplate>;
@@ -0,0 +1,25 @@
1
+ "use strict";
2
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
3
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
4
+ return new (P || (P = Promise))(function (resolve, reject) {
5
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
6
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
7
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
8
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
9
+ });
10
+ };
11
+ Object.defineProperty(exports, "__esModule", { value: true });
12
+ exports.loadPrompt = void 0;
13
+ const _1 = require(".");
14
+ const hub_1 = require("../util/hub");
15
+ const util_1 = require("../util");
16
+ const loadPromptFromFile = (file) => __awaiter(void 0, void 0, void 0, function* () { return _1.BasePromptTemplate.deserialize((0, util_1.parseFileConfig)(file)); });
17
+ const loadPrompt = (uri) => __awaiter(void 0, void 0, void 0, function* () {
18
+ const hubResult = yield (0, hub_1.loadFromHub)(uri, loadPromptFromFile, "prompts", new Set(["py", "json", "yaml"]));
19
+ if (hubResult) {
20
+ return hubResult;
21
+ }
22
+ return loadPromptFromFile(uri);
23
+ });
24
+ exports.loadPrompt = loadPrompt;
25
+ //# sourceMappingURL=load.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"load.js","sourceRoot":"","sources":["../../../prompt/load.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,wBAAuC;AACvC,qCAA0C;AAC1C,kCAA0C;AAE1C,MAAM,kBAAkB,GAAG,CAAO,IAAY,EAAE,EAAE,kDAChD,OAAA,qBAAkB,CAAC,WAAW,CAAC,IAAA,sBAAe,EAAC,IAAI,CAAC,CAAC,CAAA,GAAA,CAAC;AAEjD,MAAM,UAAU,GAAG,CAAO,GAAW,EAA+B,EAAE;IAC3E,MAAM,SAAS,GAAG,MAAM,IAAA,iBAAW,EACjC,GAAG,EACH,kBAAkB,EAClB,SAAS,EACT,IAAI,GAAG,CAAC,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAChC,CAAC;IACF,IAAI,SAAS,EAAE;QACb,OAAO,SAAS,CAAC;KAClB;IAED,OAAO,kBAAkB,CAAC,GAAG,CAAC,CAAC;AACjC,CAAC,CAAA,CAAC;AAZW,QAAA,UAAU,cAYrB"}
@@ -0,0 +1,40 @@
1
+ export type SerializedOutputParser = SerializedRegexParser | SerializedCommaSeparatedListOutputParser;
2
+ export declare abstract class BaseOutputParser {
3
+ abstract parse(text: string): string | string[] | Record<string, string>;
4
+ _type(): string;
5
+ abstract serialize(): SerializedOutputParser;
6
+ static deserialize(data: SerializedOutputParser): BaseOutputParser;
7
+ }
8
+ export declare abstract class ListOutputParser extends BaseOutputParser {
9
+ abstract parse(text: string): string[];
10
+ }
11
+ type SerializedCommaSeparatedListOutputParser = {
12
+ _type: "comma_separated_list";
13
+ };
14
+ export declare class CommaSeparatedListOutputParser extends ListOutputParser {
15
+ parse(text: string): string[];
16
+ serialize(): SerializedCommaSeparatedListOutputParser;
17
+ static deserialize(_: SerializedCommaSeparatedListOutputParser): CommaSeparatedListOutputParser;
18
+ }
19
+ type SerializedRegexParser = {
20
+ _type: "regex_parser";
21
+ regex: string;
22
+ output_keys: string[];
23
+ default_output_key?: string;
24
+ };
25
+ export declare class RegexParser extends BaseOutputParser {
26
+ regex: string | RegExp;
27
+ outputKeys: string[];
28
+ defaultOutputKey?: string;
29
+ constructor(regex: string | RegExp, outputKeys: string[], defaultOutputKey?: string);
30
+ _type(): string;
31
+ parse(text: string): Record<string, string>;
32
+ serialize(): {
33
+ _type: "regex_parser";
34
+ regex: string;
35
+ output_keys: string[];
36
+ default_output_key: string | undefined;
37
+ };
38
+ static deserialize(data: SerializedRegexParser): RegexParser;
39
+ }
40
+ export {};
@@ -0,0 +1,93 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.RegexParser = exports.CommaSeparatedListOutputParser = exports.ListOutputParser = exports.BaseOutputParser = void 0;
4
+ class BaseOutputParser {
5
+ _type() {
6
+ throw new Error("_type not implemented");
7
+ }
8
+ static deserialize(data) {
9
+ switch (data._type) {
10
+ case "regex_parser":
11
+ // eslint-disable-next-line @typescript-eslint/no-use-before-define
12
+ return RegexParser.deserialize(data);
13
+ default:
14
+ throw new Error(`Unknown parser type: ${data._type}`);
15
+ }
16
+ }
17
+ }
18
+ exports.BaseOutputParser = BaseOutputParser;
19
+ class ListOutputParser extends BaseOutputParser {
20
+ }
21
+ exports.ListOutputParser = ListOutputParser;
22
+ class CommaSeparatedListOutputParser extends ListOutputParser {
23
+ parse(text) {
24
+ return text.trim().split(", ");
25
+ }
26
+ serialize() {
27
+ return {
28
+ _type: "comma_separated_list",
29
+ };
30
+ }
31
+ static deserialize(_) {
32
+ return new CommaSeparatedListOutputParser();
33
+ }
34
+ }
35
+ exports.CommaSeparatedListOutputParser = CommaSeparatedListOutputParser;
36
+ class RegexParser extends BaseOutputParser {
37
+ constructor(regex, outputKeys, defaultOutputKey) {
38
+ super();
39
+ Object.defineProperty(this, "regex", {
40
+ enumerable: true,
41
+ configurable: true,
42
+ writable: true,
43
+ value: void 0
44
+ });
45
+ Object.defineProperty(this, "outputKeys", {
46
+ enumerable: true,
47
+ configurable: true,
48
+ writable: true,
49
+ value: void 0
50
+ });
51
+ Object.defineProperty(this, "defaultOutputKey", {
52
+ enumerable: true,
53
+ configurable: true,
54
+ writable: true,
55
+ value: void 0
56
+ });
57
+ this.regex = regex;
58
+ this.outputKeys = outputKeys;
59
+ this.defaultOutputKey = defaultOutputKey;
60
+ }
61
+ _type() {
62
+ return "regex_parser";
63
+ }
64
+ parse(text) {
65
+ const match = text.match(this.regex);
66
+ if (match) {
67
+ return this.outputKeys.reduce((acc, key, index) => {
68
+ acc[key] = match[index + 1];
69
+ return acc;
70
+ }, {});
71
+ }
72
+ if (this.defaultOutputKey === undefined) {
73
+ throw new Error(`Could not parse output: ${text}`);
74
+ }
75
+ return this.outputKeys.reduce((acc, key) => {
76
+ acc[key] = key === this.defaultOutputKey ? text : "";
77
+ return acc;
78
+ }, {});
79
+ }
80
+ serialize() {
81
+ return {
82
+ _type: "regex_parser",
83
+ regex: typeof this.regex === "string" ? this.regex : this.regex.source,
84
+ output_keys: this.outputKeys,
85
+ default_output_key: this.defaultOutputKey,
86
+ };
87
+ }
88
+ static deserialize(data) {
89
+ return new RegexParser(data.regex, data.output_keys, data.default_output_key);
90
+ }
91
+ }
92
+ exports.RegexParser = RegexParser;
93
+ //# sourceMappingURL=parser.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"parser.js","sourceRoot":"","sources":["../../../prompt/parser.ts"],"names":[],"mappings":";;;AAIA,MAAsB,gBAAgB;IAGpC,KAAK;QACH,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;IAC3C,CAAC;IAID,MAAM,CAAC,WAAW,CAAC,IAA4B;QAC7C,QAAQ,IAAI,CAAC,KAAK,EAAE;YAClB,KAAK,cAAc;gBACjB,mEAAmE;gBACnE,OAAO,WAAW,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YACvC;gBACE,MAAM,IAAI,KAAK,CAAC,wBAAwB,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;SACzD;IACH,CAAC;CACF;AAlBD,4CAkBC;AAED,MAAsB,gBAAiB,SAAQ,gBAAgB;CAE9D;AAFD,4CAEC;AAMD,MAAa,8BAA+B,SAAQ,gBAAgB;IAClE,KAAK,CAAC,IAAY;QAChB,OAAO,IAAI,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACjC,CAAC;IAED,SAAS;QACP,OAAO;YACL,KAAK,EAAE,sBAAsB;SAC9B,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,WAAW,CAChB,CAA2C;QAE3C,OAAO,IAAI,8BAA8B,EAAE,CAAC;IAC9C,CAAC;CACF;AAhBD,wEAgBC;AASD,MAAa,WAAY,SAAQ,gBAAgB;IAO/C,YACE,KAAsB,EACtB,UAAoB,EACpB,gBAAyB;QAEzB,KAAK,EAAE,CAAC;QAXV;;;;;WAAuB;QAEvB;;;;;WAAqB;QAErB;;;;;WAA0B;QAQxB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;IAC3C,CAAC;IAED,KAAK;QACH,OAAO,cAAc,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,IAAY;QAChB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,KAAK,EAAE;YACT,OAAO,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE;gBAChD,GAAG,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;gBAC5B,OAAO,GAAG,CAAC;YACb,CAAC,EAAE,EAA4B,CAAC,CAAC;SAClC;QAED,IAAI,IAAI,CAAC,gBAAgB,KAAK,SAAS,EAAE;YACvC,MAAM,IAAI,KAAK,CAAC,2BAA2B,IAAI,EAAE,CAAC,CAAC;SACpD;QAED,OAAO,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YACzC,GAAG,CAAC,GAAG,CAAC,GAAG,GAAG,KAAK,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;YACrD,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAA4B,CAAC,CAAC;IACnC,CAAC;IAED,SAAS;QACP,OAAO;YACL,KAAK,EAAE,cAAuB;YAC9B,KAAK,EAAE,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM;YACtE,WAAW,EAAE,IAAI,CAAC,UAAU;YAC5B,kBAAkB,EAAE,IAAI,CAAC,gBAAgB;SAC1C,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,WAAW,CAAC,IAA2B;QAC5C,OAAO,IAAI,WAAW,CACpB,IAAI,CAAC,KAAK,EACV,IAAI,CAAC,WAAW,EAChB,IAAI,CAAC,kBAAkB,CACxB,CAAC;IACJ,CAAC;CACF;AAzDD,kCAyDC"}
@@ -0,0 +1,28 @@
1
+ import { BasePromptTemplate, BasePromptTemplateInput, InputValues } from "./index";
2
+ import { TemplateFormat } from "./template";
3
+ import { SerializedOutputParser } from "./parser";
4
+ export type SerializedPromptTemplate = {
5
+ _type?: "prompt";
6
+ input_variables: string[];
7
+ output_parser?: SerializedOutputParser;
8
+ template?: string;
9
+ template_path?: string;
10
+ template_format?: TemplateFormat;
11
+ };
12
+ export interface PromptTemplateInput extends BasePromptTemplateInput {
13
+ template: string;
14
+ templateFormat?: TemplateFormat;
15
+ validateTemplate?: boolean;
16
+ }
17
+ export declare class PromptTemplate extends BasePromptTemplate implements PromptTemplateInput {
18
+ template: string;
19
+ templateFormat: TemplateFormat;
20
+ validateTemplate: boolean;
21
+ constructor(input: PromptTemplateInput);
22
+ _getPromptType(): "prompt";
23
+ format(values: InputValues): string;
24
+ static fromExamples(examples: string[], suffix: string, inputVariables: string[], exampleSeparator?: string, prefix?: string): PromptTemplate;
25
+ static fromTemplate(template: string): PromptTemplate;
26
+ serialize(): SerializedPromptTemplate;
27
+ static deserialize(data: SerializedPromptTemplate): Promise<PromptTemplate>;
28
+ }
@@ -0,0 +1,91 @@
1
+ "use strict";
2
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
3
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
4
+ return new (P || (P = Promise))(function (resolve, reject) {
5
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
6
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
7
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
8
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
9
+ });
10
+ };
11
+ Object.defineProperty(exports, "__esModule", { value: true });
12
+ exports.PromptTemplate = void 0;
13
+ const index_1 = require("./index");
14
+ const template_1 = require("./template");
15
+ const util_1 = require("../util");
16
+ const parser_1 = require("./parser");
17
+ class PromptTemplate extends index_1.BasePromptTemplate {
18
+ constructor(input) {
19
+ super(input);
20
+ Object.defineProperty(this, "template", {
21
+ enumerable: true,
22
+ configurable: true,
23
+ writable: true,
24
+ value: void 0
25
+ });
26
+ Object.defineProperty(this, "templateFormat", {
27
+ enumerable: true,
28
+ configurable: true,
29
+ writable: true,
30
+ value: "f-string"
31
+ });
32
+ Object.defineProperty(this, "validateTemplate", {
33
+ enumerable: true,
34
+ configurable: true,
35
+ writable: true,
36
+ value: true
37
+ });
38
+ Object.assign(this, input);
39
+ if (this.validateTemplate) {
40
+ (0, template_1.checkValidTemplate)(this.template, this.templateFormat, this.inputVariables);
41
+ }
42
+ }
43
+ _getPromptType() {
44
+ return "prompt";
45
+ }
46
+ format(values) {
47
+ return (0, template_1.renderTemplate)(this.template, this.templateFormat, values);
48
+ }
49
+ static fromExamples(examples, suffix, inputVariables, exampleSeparator = "\n\n", prefix = "") {
50
+ const template = [prefix, ...examples, suffix].join(exampleSeparator);
51
+ return new PromptTemplate({
52
+ inputVariables,
53
+ template,
54
+ });
55
+ }
56
+ static fromTemplate(template) {
57
+ const names = new Set();
58
+ (0, template_1.parseFString)(template).forEach((node) => {
59
+ if (node.type === "variable") {
60
+ names.add(node.name);
61
+ }
62
+ });
63
+ return new PromptTemplate({
64
+ inputVariables: [...names],
65
+ template,
66
+ });
67
+ }
68
+ serialize() {
69
+ var _a;
70
+ return {
71
+ _type: this._getPromptType(),
72
+ input_variables: this.inputVariables,
73
+ output_parser: (_a = this.outputParser) === null || _a === void 0 ? void 0 : _a.serialize(),
74
+ template: this.template,
75
+ template_format: this.templateFormat,
76
+ };
77
+ }
78
+ static deserialize(data) {
79
+ return __awaiter(this, void 0, void 0, function* () {
80
+ const res = new PromptTemplate({
81
+ inputVariables: data.input_variables,
82
+ outputParser: data.output_parser && parser_1.BaseOutputParser.deserialize(data.output_parser),
83
+ template: (0, util_1.resolveTemplateFromFile)("template", data),
84
+ templateFormat: data.template_format,
85
+ });
86
+ return res;
87
+ });
88
+ }
89
+ }
90
+ exports.PromptTemplate = PromptTemplate;
91
+ //# sourceMappingURL=prompt.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"prompt.js","sourceRoot":"","sources":["../../../prompt/prompt.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,mCAIiB;AACjB,yCAKoB;AACpB,kCAAkD;AAClD,qCAAoE;AAiBpE,MAAa,cACX,SAAQ,0BAAkB;IAS1B,YAAY,KAA0B;QACpC,KAAK,CAAC,KAAK,CAAC,CAAC;QAPf;;;;;WAAiB;QAEjB;;;;mBAAiC,UAAU;WAAC;QAE5C;;;;mBAAmB,IAAI;WAAC;QAItB,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAE3B,IAAI,IAAI,CAAC,gBAAgB,EAAE;YACzB,IAAA,6BAAkB,EAChB,IAAI,CAAC,QAAQ,EACb,IAAI,CAAC,cAAc,EACnB,IAAI,CAAC,cAAc,CACpB,CAAC;SACH;IACH,CAAC;IAED,cAAc;QACZ,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,MAAM,CAAC,MAAmB;QACxB,OAAO,IAAA,yBAAc,EAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;IACpE,CAAC;IAED,MAAM,CAAC,YAAY,CACjB,QAAkB,EAClB,MAAc,EACd,cAAwB,EACxB,gBAAgB,GAAG,MAAM,EACzB,MAAM,GAAG,EAAE;QAEX,MAAM,QAAQ,GAAG,CAAC,MAAM,EAAE,GAAG,QAAQ,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACtE,OAAO,IAAI,cAAc,CAAC;YACxB,cAAc;YACd,QAAQ;SACT,CAAC,CAAC;IACL,CAAC;IAED,MAAM,CAAC,YAAY,CAAC,QAAgB;QAClC,MAAM,KAAK,GAAG,IAAI,GAAG,EAAU,CAAC;QAChC,IAAA,uBAAY,EAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;YACtC,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE;gBAC5B,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;aACtB;QACH,CAAC,CAAC,CAAC;QAEH,OAAO,IAAI,cAAc,CAAC;YACxB,cAAc,EAAE,CAAC,GAAG,KAAK,CAAC;YAC1B,QAAQ;SACT,CAAC,CAAC;IACL,CAAC;IAED,SAAS;;QACP,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,cAAc,EAAE;YAC5B,eAAe,EAAE,IAAI,CAAC,cAAc;YACpC,aAAa,EAAE,MAAA,IAAI,CAAC,YAAY,0CAAE,SAAS,EAAE;YAC7C,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,eAAe,EAAE,IAAI,CAAC,cAAc;SACrC,CAAC;IACJ,CAAC;IAED,MAAM,CAAO,WAAW,CACtB,IAA8B;;YAE9B,MAAM,GAAG,GAAG,IAAI,cAAc,CAAC;gBAC7B,cAAc,EAAE,IAAI,CAAC,eAAe;gBACpC,YAAY,EACV,IAAI,CAAC,aAAa,IAAI,yBAAgB,CAAC,WAAW,CAAC,IAAI,CAAC,aAAa,CAAC;gBACxE,QAAQ,EAAE,IAAA,8BAAuB,EAAC,UAAU,EAAE,IAAI,CAAC;gBACnD,cAAc,EAAE,IAAI,CAAC,eAAe;aACrC,CAAC,CAAC;YACH,OAAO,GAAG,CAAC;QACb,CAAC;KAAA;CAGF;AAnFD,wCAmFC"}
@@ -0,0 +1,16 @@
1
+ import { InputValues } from "./index";
2
+ export type TemplateFormat = "f-string" | "jinja2";
3
+ type ParsedFStringNode = {
4
+ type: "literal";
5
+ text: string;
6
+ } | {
7
+ type: "variable";
8
+ name: string;
9
+ };
10
+ export declare const parseFString: (template: string) => ParsedFStringNode[];
11
+ export declare const interpolateFString: (template: string, values: InputValues) => string;
12
+ type Interpolator = (template: string, values: InputValues) => string;
13
+ export declare const DEFAULT_FORMATTER_MAPPING: Record<TemplateFormat, Interpolator>;
14
+ export declare const renderTemplate: (template: string, templateFormat: TemplateFormat, inputValues: InputValues) => string;
15
+ export declare const checkValidTemplate: (template: string, templateFormat: TemplateFormat, inputVariables: string[]) => void;
16
+ export {};
@@ -0,0 +1,87 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.checkValidTemplate = exports.renderTemplate = exports.DEFAULT_FORMATTER_MAPPING = exports.interpolateFString = exports.parseFString = void 0;
4
+ const parseFString = (template) => {
5
+ // Core logic replicated from internals of pythons built in Formatter class.
6
+ // https://github.com/python/cpython/blob/135ec7cefbaffd516b77362ad2b2ad1025af462e/Objects/stringlib/unicode_format.h#L700-L706
7
+ const chars = template.split("");
8
+ const nodes = [];
9
+ const nextBracket = (bracket, start) => {
10
+ for (let i = start; i < chars.length; i += 1) {
11
+ if (bracket.includes(chars[i])) {
12
+ return i;
13
+ }
14
+ }
15
+ return -1;
16
+ };
17
+ let i = 0;
18
+ while (i < chars.length) {
19
+ if (chars[i] === "{" && i + 1 < chars.length && chars[i + 1] === "{") {
20
+ nodes.push({ type: "literal", text: "{" });
21
+ i += 2;
22
+ }
23
+ else if (chars[i] === "}" &&
24
+ i + 1 < chars.length &&
25
+ chars[i + 1] === "}") {
26
+ nodes.push({ type: "literal", text: "}" });
27
+ i += 2;
28
+ }
29
+ else if (chars[i] === "{") {
30
+ const j = nextBracket("}", i);
31
+ if (j < 0) {
32
+ throw new Error("Unclosed '{' in template.");
33
+ }
34
+ nodes.push({
35
+ type: "variable",
36
+ name: chars.slice(i + 1, j).join(""),
37
+ });
38
+ i = j + 1;
39
+ }
40
+ else if (chars[i] === "}") {
41
+ throw new Error("Single '}' in template.");
42
+ }
43
+ else {
44
+ const next = nextBracket("{}", i);
45
+ const text = (next < 0 ? chars.slice(i) : chars.slice(i, next)).join("");
46
+ nodes.push({ type: "literal", text });
47
+ i = next < 0 ? chars.length : next;
48
+ }
49
+ }
50
+ return nodes;
51
+ };
52
+ exports.parseFString = parseFString;
53
+ const interpolateFString = (template, values) => (0, exports.parseFString)(template).reduce((res, node) => {
54
+ if (node.type === "variable") {
55
+ if (node.name in values) {
56
+ return res + values[node.name];
57
+ }
58
+ throw new Error(`Missing value for input ${node.name}`);
59
+ }
60
+ return res + node.text;
61
+ }, "");
62
+ exports.interpolateFString = interpolateFString;
63
+ exports.DEFAULT_FORMATTER_MAPPING = {
64
+ "f-string": exports.interpolateFString,
65
+ jinja2: (_, __) => "",
66
+ };
67
+ const renderTemplate = (template, templateFormat, inputValues) => exports.DEFAULT_FORMATTER_MAPPING[templateFormat](template, inputValues);
68
+ exports.renderTemplate = renderTemplate;
69
+ const checkValidTemplate = (template, templateFormat, inputVariables) => {
70
+ if (!(templateFormat in exports.DEFAULT_FORMATTER_MAPPING)) {
71
+ const validFormats = Object.keys(exports.DEFAULT_FORMATTER_MAPPING);
72
+ throw new Error(`Invalid template format. Got \`${templateFormat}\`;
73
+ should be one of ${validFormats}`);
74
+ }
75
+ try {
76
+ const dummyInputs = inputVariables.reduce((acc, v) => {
77
+ acc[v] = "foo";
78
+ return acc;
79
+ }, {});
80
+ (0, exports.renderTemplate)(template, templateFormat, dummyInputs);
81
+ }
82
+ catch (_a) {
83
+ throw new Error("Invalid prompt schema.");
84
+ }
85
+ };
86
+ exports.checkValidTemplate = checkValidTemplate;
87
+ //# sourceMappingURL=template.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"template.js","sourceRoot":"","sources":["../../../prompt/template.ts"],"names":[],"mappings":";;;AAQO,MAAM,YAAY,GAAG,CAAC,QAAgB,EAAuB,EAAE;IACpE,4EAA4E;IAC5E,+HAA+H;IAC/H,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACjC,MAAM,KAAK,GAAwB,EAAE,CAAC;IAEtC,MAAM,WAAW,GAAG,CAAC,OAAyB,EAAE,KAAa,EAAE,EAAE;QAC/D,KAAK,IAAI,CAAC,GAAG,KAAK,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE;YAC5C,IAAI,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE;gBAC9B,OAAO,CAAC,CAAC;aACV;SACF;QACD,OAAO,CAAC,CAAC,CAAC;IACZ,CAAC,CAAC;IAEF,IAAI,CAAC,GAAG,CAAC,CAAC;IACV,OAAO,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE;QACvB,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,GAAG,EAAE;YACpE,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;YAC3C,CAAC,IAAI,CAAC,CAAC;SACR;aAAM,IACL,KAAK,CAAC,CAAC,CAAC,KAAK,GAAG;YAChB,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,MAAM;YACpB,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,GAAG,EACpB;YACA,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;YAC3C,CAAC,IAAI,CAAC,CAAC;SACR;aAAM,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,GAAG,EAAE;YAC3B,MAAM,CAAC,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;YAC9B,IAAI,CAAC,GAAG,CAAC,EAAE;gBACT,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;aAC9C;YAED,KAAK,CAAC,IAAI,CAAC;gBACT,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;aACrC,CAAC,CAAC;YACH,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;SACX;aAAM,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,GAAG,EAAE;YAC3B,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;SAC5C;aAAM;YACL,MAAM,IAAI,GAAG,WAAW,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;YAClC,MAAM,IAAI,GAAG,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACzE,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACtC,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC;SACpC;KACF;IAED,OAAO,KAAK,CAAC;AACf,CAAC,CAAC;AAjDW,QAAA,YAAY,gBAiDvB;AAEK,MAAM,kBAAkB,GAAG,CAAC,QAAgB,EAAE,MAAmB,EAAE,EAAE,CAC1E,IAAA,oBAAY,EAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE;IAC1C,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE;QAC5B,IAAI,IAAI,CAAC,IAAI,IAAI,MAAM,EAAE;YACvB,OAAO,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;SAChC;QACD,MAAM,IAAI,KAAK,CAAC,2BAA2B,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;KACzD;IAED,OAAO,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC;AACzB,CAAC,EAAE,EAAE,CAAC,CAAC;AAVI,QAAA,kBAAkB,sBAUtB;AAII,QAAA,yBAAyB,GAAyC;IAC7E,UAAU,EAAE,0BAAkB;IAC9B,MAAM,EAAE,CAAC,CAAS,EAAE,EAAe,EAAE,EAAE,CAAC,EAAE;CAC3C,CAAC;AAEK,MAAM,cAAc,GAAG,CAC5B,QAAgB,EAChB,cAA8B,EAC9B,WAAwB,EACxB,EAAE,CAAC,iCAAyB,CAAC,cAAc,CAAC,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;AAJzD,QAAA,cAAc,kBAI2C;AAE/D,MAAM,kBAAkB,GAAG,CAChC,QAAgB,EAChB,cAA8B,EAC9B,cAAwB,EACxB,EAAE;IACF,IAAI,CAAC,CAAC,cAAc,IAAI,iCAAyB,CAAC,EAAE;QAClD,MAAM,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC,iCAAyB,CAAC,CAAC;QAC5D,MAAM,IAAI,KAAK,CAAC,kCAAkC,cAAc;4CACxB,YAAY,EAAE,CAAC,CAAC;KACzD;IACD,IAAI;QACF,MAAM,WAAW,GAAgB,cAAc,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE;YAChE,GAAG,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC;YACf,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAA4B,CAAC,CAAC;QACjC,IAAA,sBAAc,EAAC,QAAQ,EAAE,cAAc,EAAE,WAAW,CAAC,CAAC;KACvD;IAAC,WAAM;QACN,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;KAC3C;AACH,CAAC,CAAC;AAnBW,QAAA,kBAAkB,sBAmB7B"}
@@ -0,0 +1 @@
1
+ export {};
@@ -0,0 +1,30 @@
1
+ "use strict";
2
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
3
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
4
+ return new (P || (P = Promise))(function (resolve, reject) {
5
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
6
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
7
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
8
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
9
+ });
10
+ };
11
+ var __importDefault = (this && this.__importDefault) || function (mod) {
12
+ return (mod && mod.__esModule) ? mod : { "default": mod };
13
+ };
14
+ Object.defineProperty(exports, "__esModule", { value: true });
15
+ const globals_1 = require("@jest/globals");
16
+ const path_1 = __importDefault(require("path"));
17
+ const load_1 = require("../load");
18
+ const PROMPTS_DIR = path_1.default.join(__dirname, "prompts");
19
+ (0, globals_1.test)("Load Hello World Prompt", () => __awaiter(void 0, void 0, void 0, function* () {
20
+ const helloWorld = path_1.default.join(PROMPTS_DIR, "hello_world.yaml");
21
+ const prompt = yield (0, load_1.loadPrompt)(helloWorld);
22
+ (0, globals_1.expect)(prompt._getPromptType()).toBe("prompt");
23
+ (0, globals_1.expect)(prompt.format({})).toBe("Say hello world.");
24
+ }));
25
+ (0, globals_1.test)("Load hub prompt", () => __awaiter(void 0, void 0, void 0, function* () {
26
+ const prompt = yield (0, load_1.loadPrompt)("lc@abb92d8://prompts/hello-world/prompt.yaml");
27
+ (0, globals_1.expect)(prompt._getPromptType()).toBe("prompt");
28
+ (0, globals_1.expect)(prompt.format({})).toBe("Say hello world.");
29
+ }));
30
+ //# sourceMappingURL=load.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"load.test.js","sourceRoot":"","sources":["../../../../prompt/tests/load.test.ts"],"names":[],"mappings":";;;;;;;;;;;;;;AAAA,2CAA6C;AAC7C,gDAAwB;AACxB,kCAAqC;AAErC,MAAM,WAAW,GAAG,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;AAEpD,IAAA,cAAI,EAAC,yBAAyB,EAAE,GAAS,EAAE;IACzC,MAAM,UAAU,GAAG,cAAI,CAAC,IAAI,CAAC,WAAW,EAAE,kBAAkB,CAAC,CAAC;IAC9D,MAAM,MAAM,GAAG,MAAM,IAAA,iBAAU,EAAC,UAAU,CAAC,CAAC;IAC5C,IAAA,gBAAM,EAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC/C,IAAA,gBAAM,EAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;AACrD,CAAC,CAAA,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,iBAAiB,EAAE,GAAS,EAAE;IACjC,MAAM,MAAM,GAAG,MAAM,IAAA,iBAAU,EAC7B,8CAA8C,CAC/C,CAAC;IACF,IAAA,gBAAM,EAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC/C,IAAA,gBAAM,EAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;AACrD,CAAC,CAAA,CAAC,CAAC"}
@@ -0,0 +1 @@
1
+ export {};