#쿼리 명확화 (Query Clarification)

// LLM SDK는 무엇이든 "프롬프트 → 텍스트" 함수로 감싸서 쓴다고 가정합니다.
type Complete = (prompt: string) => Promise<string>;
type Search = (query: string, k: number) => Promise<{ id: string; text: string; score: number }[]>;

interface Clarified {
  standalone_question: string | null;
  needs_clarification: boolean;
  clarifying_question: string | null;
}

const AMBIGUOUS = /(이거|그거|저거|이것|그것|저것|거기|그때)/;

export async function clarifyAndSearch(
  question: string,
  history: string[],
  pageContext: string,
  complete: Complete,
  search: Search,
) {
  // 1) 명확한 질문이면 LLM 호출 없이 바로 검색
  const looksAmbiguous = AMBIGUOUS.test(question) || question.trim().length < 6;
  if (!looksAmbiguous) return { type: "results", results: await search(question, 10) } as const;

  // 2) 맥락으로 재작성
  const raw = await complete(
    PROMPT.replace("{history}", history.slice(-6).join("\n"))
      .replace("{page_context}", pageContext || "(없음)")
      .replace("{question}", question),
  );
  // 실서비스에서는 SDK의 구조화 출력(JSON schema) 기능을 쓰고, 파싱 실패 시 원 질문으로 폴백하세요.
  const parsed = JSON.parse(raw) as Clarified;

  // 3) 확정 불가면 되묻기
  if (parsed.needs_clarification || !parsed.standalone_question) {
    return { type: "ask", message: parsed.clarifying_question ?? "무엇에 대한 질문인지 알려주세요." } as const;
  }

  // 4) 재작성 질문 + 원 질문을 함께 검색 (원 질문은 백스톱)
  const [rewritten, original] = await Promise.all([
    search(parsed.standalone_question, 10),
    search(question, 10),
  ]);
  return { type: "results", query: parsed.standalone_question, results: dedupe([...rewritten, ...original]) } as const;
}

function dedupe<T extends { id: string }>(docs: T[]): T[] {
  const seen = new Set<string>();
  return docs.filter((d) => (seen.has(d.id) ? false : (seen.add(d.id), true)));
}

const PROMPT = `...위 프롬프트 전문 ({history}, {page_context}, {question} 자리표시자 포함)...`;