Compare commits

...

2 Commits

Author SHA1 Message Date
Thuc Pham 51008e81dc remove extract text 2024-05-16 14:53:58 +07:00
Thuc Pham 56d83ceead fix: allow custom task query for agents 2024-05-16 12:09:17 +07:00
+2 -3
View File
@@ -21,7 +21,6 @@ import type {
LLM,
MessageContent,
} from "../llm/index.js";
import { extractText } from "../llm/utils.js";
import type { BaseToolWithCall, ToolOutput } from "../types.js";
import type {
AgentTaskContext,
@@ -169,7 +168,7 @@ export abstract class AgentWorker<
abstract taskHandler: TaskHandler<AI, Store, AdditionalMessageOptions>;
public createTask(
query: string,
query: MessageContent,
context: AgentTaskContext<AI, Store, AdditionalMessageOptions>,
): ReadableStream<TaskStepOutput<AI, Store, AdditionalMessageOptions>> {
context.store.messages.push({
@@ -305,7 +304,7 @@ export abstract class AgentRunner<
});
}
}
return this.#runner.createTask(extractText(message), {
return this.#runner.createTask(message, {
stream,
toolCallCount: 0,
llm: this.#llm,