feat: add support for semantic search using operand

This commit is contained in:
Jacky Zhao 2022-07-31 12:16:36 -07:00
parent 14b89105dc
commit 5ef9aad501
8 changed files with 69 additions and 17 deletions

View file

@ -56,6 +56,6 @@
}
const allIds = new Set([...getByField("title"), ...getByField("content")])
const finalResults = [...allIds].map(formatForDisplay)
displayResults(finalResults)
displayResults(finalResults, true)
})
})()

View file

@ -0,0 +1,35 @@
const apiKey = "{{$.Site.Data.config.operandApiKey}}"
async function searchContents(query) {
const response = await fetch('https://prod.operand.ai/v3/search/objects', {
method: 'POST',
headers: {
'Content-Type': 'application/json',
Authorization: apiKey,
},
body: JSON.stringify({
query,
max: 10
}),
});
return (await response.json());
}
function debounce(func, timeout = 300) {
let timer;
return (...args) => {
clearTimeout(timer)
timer = setTimeout(() => { func.apply(this, args); }, timeout)
};
}
registerHandlers(debounce((e) => {
term = e.target.value
searchContents(term)
.then((res) => res.results.map(entry => ({
url: entry.object.metadata.url,
content: entry.snippet,
title: entry.object.title
})))
.then(results => displayResults(results))
}))

View file

@ -108,13 +108,11 @@ const highlight = (content, term) => {
}
// Common utilities for search
const resultToHTML = ({ url, title, content, term }) => {
const text = removeMarkdown(content)
const resultTitle = highlight(title, term)
const resultText = highlight(text, term)
const resultToHTML = ({ url, title, content }) => {
const cleaned = removeMarkdown(content)
return `<button class="result-card" id="${url}">
<h3>${resultTitle}</h3>
<p>${resultText}</p>
<h3>${title}</h3>
<p>${cleaned}</p>
</button>`
}
@ -183,7 +181,7 @@ const registerHandlers = (onInputFn) => {
})
}
const displayResults = (finalResults) => {
const displayResults = (finalResults, extractHighlight = false) => {
const results = document.getElementById("results-container")
if (finalResults.length === 0) {
results.innerHTML = `<button class="result-card">
@ -192,11 +190,17 @@ const displayResults = (finalResults) => {
</button>`
} else {
results.innerHTML = finalResults
.map((result) =>
resultToHTML({
...result,
term,
}),
.map((result) => {
if (extractHighlight) {
return resultToHTML({
url: result.url,
title: highlight(result.title, term),
content: highlight(result.content, term)
})
} else {
return resultToHTML(result)
}
}
)
.join("\n")
const anchors = [...document.getElementsByClassName("result-card")]