| 123456789101112131415161718192021222324252627282930313233343536373839404142434445 |
- import z from "zod"
- import { Effect } from "effect"
- import { Tool } from "./tool"
- import { Question } from "../question"
- import DESCRIPTION from "./question.txt"
- const parameters = z.object({
- questions: z.array(Question.Info.omit({ custom: true })).describe("Questions to ask"),
- })
- type Metadata = {
- answers: Question.Answer[]
- }
- export const QuestionTool = Tool.defineEffect<typeof parameters, Metadata, Question.Service>(
- "question",
- Effect.gen(function* () {
- const question = yield* Question.Service
- return {
- description: DESCRIPTION,
- parameters,
- execute: (params: z.infer<typeof parameters>, ctx: Tool.Context<Metadata>) =>
- Effect.gen(function* () {
- const answers = yield* question.ask({
- sessionID: ctx.sessionID,
- questions: params.questions,
- tool: ctx.callID ? { messageID: ctx.messageID, callID: ctx.callID } : undefined,
- })
- const formatted = params.questions
- .map((q, i) => `"${q.question}"="${answers[i]?.length ? answers[i].join(", ") : "Unanswered"}"`)
- .join(", ")
- return {
- title: `Asked ${params.questions.length} question${params.questions.length > 1 ? "s" : ""}`,
- output: `User has answered your questions: ${formatted}. You can now continue with the user's answers in mind.`,
- metadata: {
- answers,
- },
- }
- }).pipe(Effect.runPromise),
- }
- }),
- )
|