docling-studio/frontend/src/features/ingestion/api.ts

import { apiFetch } from '../../shared/api/http'

export interface IngestionResult {
  docId: string
  chunksIndexed: number
  embeddingDimension: number
}

export interface IngestionStatus {
  available: boolean
  opensearchConnected: boolean
}

export interface SearchResultItem {
  docId: string
  filename: string
  content: string
  chunkIndex: number
  pageNumber: number
  score: number
  headings: string[]
}

export interface SearchResponse {
  results: SearchResultItem[]
  total: number
  query: string
}

export function ingestAnalysis(jobId: string): Promise<IngestionResult> {
  return apiFetch<IngestionResult>(`/api/ingestion/${jobId}`, {
    method: 'POST',
  })
}

export function deleteIngested(docId: string): Promise<unknown> {
  return apiFetch(`/api/ingestion/${docId}`, { method: 'DELETE' })
}

export function fetchIngestionStatus(): Promise<IngestionStatus> {
  return apiFetch<IngestionStatus>('/api/ingestion/status')
}

export function searchChunks(
  query: string,
  options: { docId?: string; k?: number } = {},
): Promise<SearchResponse> {
  const params = new URLSearchParams({ q: query })
  if (options.docId) params.set('doc_id', options.docId)
  if (options.k) params.set('k', String(options.k))
  return apiFetch<SearchResponse>(`/api/ingestion/search?${params}`)
}