@agent-smith/task 0.1.5 → 0.1.7
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/interfaces.d.ts +3 -4
- package/dist/task.js +7 -2
- package/dist/variables.js +29 -0
- package/package.json +9 -8
package/dist/interfaces.d.ts
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import { InferenceParams, InferenceResult, InferenceOptions, ToolSpec, HistoryTurn, ToolCallSpec } from "@locallm/types";
|
|
2
|
-
import { PromptTemplate } from "modprompt";
|
|
1
|
+
import type { InferenceParams, InferenceResult, InferenceOptions, ToolSpec, HistoryTurn, ToolCallSpec } from "@locallm/types";
|
|
2
|
+
import type { PromptTemplate } from "modprompt";
|
|
3
3
|
interface ModelSpec {
|
|
4
4
|
name: string;
|
|
5
5
|
template?: string;
|
|
@@ -83,7 +83,6 @@ interface TaskDef {
|
|
|
83
83
|
}
|
|
84
84
|
interface TaskOutput {
|
|
85
85
|
answer: InferenceResult;
|
|
86
|
-
|
|
87
|
-
template?: PromptTemplate;
|
|
86
|
+
template: PromptTemplate;
|
|
88
87
|
}
|
|
89
88
|
export { ModelSpec, TaskInput, TaskDef, TemplateSpec, TaskConf, TaskOutput, TaskVariableDef, TaskOptionalVariableDef, TaskVariables, };
|
package/dist/task.js
CHANGED
|
@@ -73,7 +73,7 @@ class Task {
|
|
|
73
73
|
console.log("----------------------------------------------");
|
|
74
74
|
console.log("Infer params:", this.def.inferParams);
|
|
75
75
|
console.log("----------------------------------------------");
|
|
76
|
-
options.debug = true
|
|
76
|
+
//options.debug = true
|
|
77
77
|
}
|
|
78
78
|
if (this.agent.lm.providerType == "ollama") {
|
|
79
79
|
if (!this.def.inferParams?.extra) {
|
|
@@ -83,6 +83,10 @@ class Task {
|
|
|
83
83
|
this.def.inferParams["extra"]["raw"] = true;
|
|
84
84
|
}
|
|
85
85
|
let answer;
|
|
86
|
+
if (options?.debug) {
|
|
87
|
+
// cut debug here. TODO: debug log levels
|
|
88
|
+
options.debug = false;
|
|
89
|
+
}
|
|
86
90
|
if (!useTemplates) {
|
|
87
91
|
if (this.def.template?.system) {
|
|
88
92
|
options.system = this.def.template.system;
|
|
@@ -98,10 +102,11 @@ class Task {
|
|
|
98
102
|
//console.log("RUN AGENT (TASK) params:", this.def.inferParams);
|
|
99
103
|
//console.log("RUN AGENT (TASK) options:", options);
|
|
100
104
|
answer = await this.agent.run(finalPrompt, this.def.inferParams, options, tpl);
|
|
105
|
+
tpl.history = this.agent.history;
|
|
101
106
|
//console.log("RAW ANSWER", answer);
|
|
102
107
|
}
|
|
103
108
|
//console.log("TASK: ANSWER FINAL:", { answer: answer.result, errors: {}, template: answer.template })
|
|
104
|
-
return { answer,
|
|
109
|
+
return { answer, template: tpl };
|
|
105
110
|
}
|
|
106
111
|
addTools(tools) {
|
|
107
112
|
tools.forEach(t => this.agent.tools[t.name] = t);
|
package/dist/variables.js
CHANGED
|
@@ -11,9 +11,24 @@ function applyVariables(taskDef, taskInput) {
|
|
|
11
11
|
}
|
|
12
12
|
if (taskDef.variables?.optional) {
|
|
13
13
|
for (const name of Object.keys(taskDef.variables.optional)) {
|
|
14
|
+
// cleanup unused optional variables
|
|
14
15
|
if (!(name in taskInput)) {
|
|
15
16
|
const v = taskDef.variables.optional[name]?.default ?? "";
|
|
16
17
|
taskDef.prompt = taskDef.prompt.replaceAll(`{${name}}`, v);
|
|
18
|
+
if (taskDef.template?.system) {
|
|
19
|
+
taskDef.template.system = taskDef.template.system.replaceAll(`{${name}}`, v);
|
|
20
|
+
}
|
|
21
|
+
if (taskDef?.shots) {
|
|
22
|
+
const nshots = new Array();
|
|
23
|
+
taskDef.shots.forEach(s => {
|
|
24
|
+
let nshot = s;
|
|
25
|
+
if (s?.user) {
|
|
26
|
+
nshot.user = s.user.replaceAll(`{${name}}`, v);
|
|
27
|
+
}
|
|
28
|
+
nshots.push(nshot);
|
|
29
|
+
});
|
|
30
|
+
taskDef.shots = nshots;
|
|
31
|
+
}
|
|
17
32
|
}
|
|
18
33
|
}
|
|
19
34
|
}
|
|
@@ -21,6 +36,20 @@ function applyVariables(taskDef, taskInput) {
|
|
|
21
36
|
// apply variables
|
|
22
37
|
for (const [k, v] of Object.entries(taskInput)) {
|
|
23
38
|
taskDef.prompt = taskDef.prompt.replaceAll(`{${k}}`, v);
|
|
39
|
+
if (taskDef.template?.system) {
|
|
40
|
+
taskDef.template.system = taskDef.template.system.replaceAll(`{${k}}`, v);
|
|
41
|
+
}
|
|
42
|
+
if (taskDef?.shots) {
|
|
43
|
+
const nshots = new Array();
|
|
44
|
+
taskDef.shots.forEach(s => {
|
|
45
|
+
let nshot = s;
|
|
46
|
+
if (s?.user) {
|
|
47
|
+
nshot.user = s.user.replaceAll(`{${k}}`, v);
|
|
48
|
+
}
|
|
49
|
+
nshots.push(nshot);
|
|
50
|
+
});
|
|
51
|
+
taskDef.shots = nshots;
|
|
52
|
+
}
|
|
24
53
|
}
|
|
25
54
|
return taskDef;
|
|
26
55
|
}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@agent-smith/task",
|
|
3
|
-
"version": "0.1.
|
|
3
|
+
"version": "0.1.7",
|
|
4
4
|
"description": "A toolkit to create human friendly agents: the language model tasks module",
|
|
5
5
|
"repository": {
|
|
6
6
|
"type": "git",
|
|
@@ -14,23 +14,24 @@
|
|
|
14
14
|
"docs": "typedoc --entryPointStrategy expand"
|
|
15
15
|
},
|
|
16
16
|
"dependencies": {
|
|
17
|
-
"@agent-smith/agent": "^0.1.
|
|
17
|
+
"@agent-smith/agent": "^0.1.5",
|
|
18
18
|
"@agent-smith/tfm": "^0.2.0",
|
|
19
|
-
"modprompt": "^0.12.
|
|
20
|
-
"yaml": "^2.8.
|
|
19
|
+
"modprompt": "^0.12.6",
|
|
20
|
+
"yaml": "^2.8.2"
|
|
21
21
|
},
|
|
22
22
|
"devDependencies": {
|
|
23
|
-
"
|
|
24
|
-
"
|
|
23
|
+
"@locallm/api": "^0.7.2",
|
|
24
|
+
"openai": "6.10.0",
|
|
25
|
+
"@locallm/types": "^0.6.4",
|
|
25
26
|
"@rollup/plugin-node-resolve": "^16.0.3",
|
|
26
27
|
"@rollup/plugin-terser": "^0.4.4",
|
|
27
28
|
"@rollup/plugin-typescript": "^12.3.0",
|
|
28
|
-
"@types/node": "^
|
|
29
|
+
"@types/node": "^25.0.1",
|
|
29
30
|
"markdown-it-replace-link": "^1.2.2",
|
|
30
31
|
"restmix": "^0.6.1",
|
|
31
32
|
"rollup": "^4.53.3",
|
|
32
33
|
"tslib": "^2.8.1",
|
|
33
|
-
"typedoc": "^0.28.
|
|
34
|
+
"typedoc": "^0.28.15",
|
|
34
35
|
"typedoc-plugin-markdown": "^4.9.0",
|
|
35
36
|
"typedoc-plugin-rename-defaults": "^0.7.3",
|
|
36
37
|
"typescript": "^5.9.3"
|