Merge remote-tracking branch 'upstream/main' into feature/elasticsearch-connector

This commit is contained in:
Anish Sarkar 2025-10-17 01:02:29 +05:30
commit 8e1e81ebae
12 changed files with 668 additions and 1 deletions

View file

@ -1057,6 +1057,32 @@ async def fetch_relevant_documents(
}
)
elif connector == "BAIDU_SEARCH_API":
(
source_object,
baidu_chunks,
) = await connector_service.search_baidu(
user_query=reformulated_query,
user_id=user_id,
search_space_id=search_space_id,
top_k=top_k,
)
# Add to sources and raw documents
if source_object:
all_sources.append(source_object)
all_raw_documents.extend(baidu_chunks)
# Stream found document count
if streaming_service and writer:
writer(
{
"yield_value": streaming_service.format_terminal_info_delta(
f"🇨🇳 Found {len(baidu_chunks)} Baidu Search results related to your query"
)
}
)
elif connector == "DISCORD_CONNECTOR":
(
source_object,