Function queryKnowledgeBase

apps/sim/lib/guardrails/validate_hallucination.ts:50–96

* Query knowledge base to get relevant context chunks using the search API

(
  knowledgeBaseId: string,
  query: string,
  topK: number,
  requestId: string,
  workflowId?: string,
  authHeaders?: { cookie?: string; authorization?: string }
)

 * Query knowledge base to get relevant context chunks using the search API
 */
async function queryKnowledgeBase(
  knowledgeBaseId: string,
  query: string,
  topK: number,
  requestId: string,
  workflowId?: string,
  authHeaders?: { cookie?: string; authorization?: string }
): Promise<string[]> {
  try {
    // Call the knowledge base search API directly
    const searchUrl = `${getInternalApiBaseUrl()}/api/knowledge/search`

    const response = await fetch(searchUrl, {
      method: 'POST',
      headers: {
        'Content-Type': 'application/json',
        ...(authHeaders?.cookie ? { Cookie: authHeaders.cookie } : {}),
        ...(authHeaders?.authorization ? { Authorization: authHeaders.authorization } : {}),
      },
      body: JSON.stringify({
        knowledgeBaseIds: [knowledgeBaseId],
        query,
        topK,
        workflowId,
      }),
    })

    if (!response.ok) {
      logger.error(`[${requestId}] Knowledge base query failed`, {
        status: response.status,
      })
      return []
    }

    const result = await response.json()
    const results = result.data?.results || []

    const chunks = results.map((r: any) => r.content || '').filter((c: string) => c.length > 0)

    return chunks
  } catch (error: any) {
    logger.error(`[${requestId}] Error querying knowledge base`, {
      error: error.message,
    })
    return []
  }
}

/**
 * Use an LLM to score confidence based on RAG context