Spaces:
Sleeping
Sleeping
| import fetch from 'node-fetch'; | |
| import { d as private_env } from './shared-server-49TKSBDM.js'; | |
| import dns from 'node:dns'; | |
| dns.setDefaultResultOrder("ipv4first"); | |
| class FullTextSearchService { | |
| url = ""; | |
| constructor(url) { | |
| this.url = url; | |
| } | |
| async health() { | |
| try { | |
| const r = await fetch(`${this.url}/health`, { | |
| method: "GET", | |
| headers: { | |
| "Content-Type": "application/json" | |
| } | |
| }); | |
| const data = await r.json(); | |
| if (data.status === "ok") { | |
| return data.status; | |
| } | |
| } catch (error) { | |
| console.log(JSON.parse(JSON.stringify(error))); | |
| } | |
| return "unavailable"; | |
| } | |
| async search(query, use_llm_for_teasers, do_multistep_search, use_accounting_docs, use_reranking, disable_thresholds, use_llm_for_filtering, find_transaction_maps_by_question, find_transaction_maps_by_operation, use_keywords, { abortController }) { | |
| const timeoutId = setTimeout(() => { | |
| abortController.abort(); | |
| }, 6e4 * Number(private_env.SEARCH_API_REQUEST_TIMEOUT)); | |
| const startTime = Date.now(); | |
| try { | |
| const r = await fetch(`${this.url}/search`, { | |
| method: "POST", | |
| headers: { | |
| "Content-Type": "application/json" | |
| }, | |
| body: JSON.stringify({ | |
| "query": query, | |
| "top": 30, | |
| "use_llm_for_teasers": use_llm_for_teasers, | |
| "do_multistep_search": do_multistep_search, | |
| "use_accounting_docs": use_accounting_docs, | |
| "use_reranking": use_reranking, | |
| "disable_thresholds": disable_thresholds, | |
| "use_llm_for_filtering": use_llm_for_filtering, | |
| "find_transaction_maps_by_question": find_transaction_maps_by_question, | |
| "find_transaction_maps_by_operation": find_transaction_maps_by_operation, | |
| "use_keywords": use_keywords | |
| }), | |
| signal: abortController.signal | |
| }); | |
| if (!r.ok) { | |
| console.error(`Failed to load search results: ${await r.text()}`); | |
| throw new Error(`Не удалось загрузить результаты поиска. Возможно, поиск ещё не запустился. Попробуйте отправить запрос ещё раз.`); | |
| } | |
| const response = await r.json(); | |
| if (find_transaction_maps_by_question || find_transaction_maps_by_operation) { | |
| return { | |
| transactionMapResults: response.transaction_maps_results, | |
| searchResults: null, | |
| consultations: null | |
| }; | |
| } else { | |
| return { | |
| transactionMapResults: null, | |
| searchResults: response.results, | |
| consultations: response.consultations | |
| }; | |
| } | |
| } catch (error) { | |
| if (abortController.signal.aborted || error.name === "AbortError") { | |
| throw new Error(`Сервис поиска не отвечает. Попробуйте изменить и повторить запрос'`); | |
| } | |
| console.log(JSON.parse(JSON.stringify(error))); | |
| throw error; | |
| } finally { | |
| const endTime = Date.now(); | |
| const difference = Math.abs(endTime - startTime) / 1e3; | |
| console.log("Search response time:", difference); | |
| clearTimeout(timeoutId); | |
| } | |
| } | |
| } | |
| export { FullTextSearchService as F }; | |
| //# sourceMappingURL=FullTextSearchService-0xoNRjpt.js.map | |