import {
type NewsResult,
type SearchQuery,
type SearchResponse,
SearchResponseSchema,
type WebResult,
} from './search.schemas';
export const fetchSearchResults = async ({
YDC_API_KEY = process.env.YDC_API_KEY,
searchQuery: {
query,
site,
fileType,
language,
exactTerms,
excludeTerms,
...rest
},
}: {
searchQuery: SearchQuery;
YDC_API_KEY?: string;
}) => {
const url = new URL('https://api.ydc-index.io/v1/search');
const searchParams = new URLSearchParams();
// Build Query Param
const searchQuery = [query];
site && searchQuery.push(`site:${site}`);
fileType && searchQuery.push(`fileType:${fileType}`);
language && searchQuery.push(`lang:${language}`);
if (exactTerms && excludeTerms) {
throw new Error(
'Cannot specify both exactTerms and excludeTerms - please use only one',
);
}
exactTerms &&
searchQuery.push(
exactTerms
.split('|')
.map((term) => `+${term}`)
.join(' AND '),
);
excludeTerms &&
searchQuery.push(
excludeTerms
.split('|')
.map((term) => `-${term}`)
.join(' AND '),
);
searchParams.append('query', searchQuery.join(' '));
// Append additional advanced Params
for (const [name, value] of Object.entries(rest)) {
if (value) searchParams.append(name, `${value}`);
}
url.search = searchParams.toString();
const options = {
method: 'GET',
headers: new Headers({
'X-API-Key': YDC_API_KEY || '',
}),
};
const response = await fetch(url, options);
if (!response.ok) {
const errorCode = response.status;
if (errorCode === 429) {
throw new Error('Rate limited by You.com API. Please try again later.');
} else if (errorCode === 403) {
throw new Error('Forbidden. Please check your You.com API key.');
}
throw new Error(`Failed to perform search. Error code: ${errorCode}`);
}
const results = await response.json();
const parsedResults = SearchResponseSchema.parse(results);
return parsedResults;
};
export const formatSearchResults = (response: SearchResponse) => {
let formattedResults = '';
// Format web results
if (response.results.web?.length) {
const webResults = response.results.web
.map(
(hit: WebResult) =>
`Title: ${hit.title}\nURL: ${hit.url}\nDescription: ${
hit.description
}\nSnippets:\n- ${hit.snippets.join('\n- ')}`,
)
.join('\n\n---\n\n');
formattedResults += `WEB RESULTS:\n\n${webResults}`;
}
// Format news results
if (response.results.news?.length) {
const newsResults = response.results.news
.map(
(article: NewsResult) =>
`Title: ${article.title}\nURL: ${article.url}\nDescription: ${article.description}\nPublished: ${article.page_age}`,
)
.join('\n\n---\n\n');
if (formattedResults) {
formattedResults += `\n\n${'='.repeat(50)}\n\n`;
}
formattedResults += `NEWS RESULTS:\n\n${newsResults}`;
}
return {
content: [
{
type: 'text' as const,
text: `Search Results for "${response.metadata.query}":\n\n${formattedResults}`,
},
],
structuredContent: response,
};
};
