@lota-sdk/ui 0.1.5

package/src/tools/user-questions.ts

@@ -0,0 +1,77 @@
+import type { UserQuestionItem, UserQuestionsArgs } from '@lota-sdk/shared/schemas/tools'
+
+export type UserQuestionAnswer = { selected: string[]; customText: string }
+export type UserQuestionsSubmitPayload = {
+  text: string
+  questions: UserQuestionsArgs['questions']
+  answers: UserQuestionAnswer[]
+}
+export type UserQuestionsResolution = 'pending' | 'answered' | 'skipped'
+
+const ANSWER_PREFIX = 'Answers:'
+
+export function formatUserQuestionsAnswerText(
+  questions: UserQuestionsArgs['questions'],
+  answers: UserQuestionAnswer[],
+): string {
+  const lines = [ANSWER_PREFIX]
+
+  for (let index = 0; index < questions.length; index += 1) {
+    const question = questions[index]
+    const answer = answers.at(index)
+    const parts: string[] = []
+
+    if (answer) {
+      for (const selected of answer.selected) {
+        parts.push(selected)
+      }
+
+      if (answer.customText.trim()) {
+        parts.push(`custom: "${answer.customText.trim()}"`)
+      }
+    }
+
+    lines.push(`${index + 1}. ${question.question}: ${parts.join(', ') || '(no answer)'}`)
+  }
+
+  return lines.join('\n')
+}
+
+export function resolveUserQuestionsResolution<
+  TMessage extends { role: string; parts: Array<{ type: string; text?: string }> },
+>(messages: TMessage[] | undefined, currentMessageIndex: number | undefined): UserQuestionsResolution {
+  if (!messages || typeof currentMessageIndex !== 'number') return 'pending'
+  if (currentMessageIndex < 0 || currentMessageIndex >= messages.length - 1) return 'pending'
+
+  const nextMessage = messages[currentMessageIndex + 1]
+  if (nextMessage.role === 'user') {
+    const textPart = nextMessage.parts.find(
+      (part) => part.type === 'text' && typeof part.text === 'string' && part.text.length > 0,
+    )
+    if (textPart?.text?.startsWith(ANSWER_PREFIX)) return 'answered'
+  }
+
+  return 'skipped'
+}
+
+export function createInitialUserQuestionAnswers(questions: UserQuestionsArgs['questions']): UserQuestionAnswer[] {
+  return questions.map(() => ({ selected: [], customText: '' }))
+}
+
+export function toUserQuestionKey(question: UserQuestionItem): string {
+  const options = question.options?.join('|') ?? ''
+  return `${question.type}:${question.question}:${options}:${question.allowCustomAnswer === false ? 'fixed' : 'custom'}`
+}
+
+export function withDedupedQuestionValues(values: string[]): Array<{ key: string; value: string }> {
+  const seen = new Map<string, number>()
+  return values.map((value) => {
+    const count = (seen.get(value) ?? 0) + 1
+    seen.set(value, count)
+    return { key: count === 1 ? value : `${value}__${count}`, value }
+  })
+}
+
+export function toUserQuestionsResetKey(questions: UserQuestionsArgs['questions']): string {
+  return questions.map((question) => toUserQuestionKey(question)).join('||')
+}