@lobehub/chat
Version:
Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.
125 lines (107 loc) • 3.84 kB
text/typescript
import { TRPCError } from '@trpc/server';
import debug from 'debug';
import urlJoin from 'url-join';
import { SearchParams, UniformSearchResponse, UniformSearchResult } from '@/types/tool/search';
import { SearchServiceImpl } from '../type';
import { BochaSearchParameters, BochaResponse } from './type';
const log = debug('lobe-search:Bocha');
const timeRangeMapping = {
day: 'oneDay',
month: 'oneMonth',
week: 'oneWeek',
year: 'oneYear',
};
/**
* Bocha implementation of the search service
* Primarily used for web crawling
*/
export class BochaImpl implements SearchServiceImpl {
private get apiKey(): string | undefined {
return process.env.BOCHA_API_KEY;
}
private get baseUrl(): string {
// Assuming the base URL is consistent with the crawl endpoint
return 'https://api.bochaai.com/v1';
}
async query(query: string, params: SearchParams = {}): Promise<UniformSearchResponse> {
log('Starting Bocha query with query: "%s", params: %o', query, params);
const endpoint = urlJoin(this.baseUrl, '/web-search');
const defaultQueryParams: BochaSearchParameters = {
count: 15,
query,
summary: true,
};
let body: BochaSearchParameters = {
...defaultQueryParams,
freshness:
params?.searchTimeRange && params.searchTimeRange !== 'anytime'
? timeRangeMapping[params.searchTimeRange as keyof typeof timeRangeMapping] ?? undefined
: undefined,
};
log('Constructed request body: %o', body);
let response: Response;
const startAt = Date.now();
let costTime = 0;
try {
log('Sending request to endpoint: %s', endpoint);
response = await fetch(endpoint, {
body: JSON.stringify(body),
headers: {
'Authorization': this.apiKey ? `Bearer ${this.apiKey}` : '',
'Content-Type': 'application/json',
},
method: 'POST',
});
log('Received response with status: %d', response.status);
costTime = Date.now() - startAt;
} catch (error) {
log.extend('error')('Bocha fetch error: %o', error);
throw new TRPCError({
cause: error,
code: 'SERVICE_UNAVAILABLE',
message: 'Failed to connect to Bocha.',
});
}
if (!response.ok) {
const errorBody = await response.text();
log.extend('error')(
`Bocha request failed with status ${response.status}: %s`,
errorBody.length > 200 ? `${errorBody.slice(0, 200)}...` : errorBody,
);
throw new TRPCError({
cause: errorBody,
code: 'SERVICE_UNAVAILABLE',
message: `Bocha request failed: ${response.statusText}`,
});
}
try {
const bochaResponse = (await response.json()) as BochaResponse;
log('Parsed Bocha response: %o', bochaResponse);
const mappedResults = (bochaResponse.data.webPages.value || []).map(
(result): UniformSearchResult => ({
category: 'general', // Default category
content: result.summary || result.snippet || '', // Prioritize content, fallback to snippet
engines: ['bocha'], // Use 'bocha' as the engine name
parsedUrl: result.url ? new URL(result.url).hostname : '', // Basic URL parsing
score: 1, // Default score to 1
title: result.name || '',
url: result.url,
}),
);
log('Mapped %d results to SearchResult format', mappedResults.length);
return {
costTime,
query: query,
resultNumbers: mappedResults.length,
results: mappedResults,
};
} catch (error) {
log.extend('error')('Error parsing Bocha response: %o', error);
throw new TRPCError({
cause: error,
code: 'INTERNAL_SERVER_ERROR',
message: 'Failed to parse Bocha response.',
});
}
}
}