shihao
16263710d9
Some checks failed
Create and publish Docker images with specific build args / build-main-image (linux/amd64, ubuntu-latest) (push) Has been cancelled
Create and publish Docker images with specific build args / build-main-image (linux/arm64, ubuntu-24.04-arm) (push) Has been cancelled
Create and publish Docker images with specific build args / build-cuda-image (linux/amd64, ubuntu-latest) (push) Has been cancelled
Create and publish Docker images with specific build args / build-cuda-image (linux/arm64, ubuntu-24.04-arm) (push) Has been cancelled
Create and publish Docker images with specific build args / build-cuda126-image (linux/amd64, ubuntu-latest) (push) Has been cancelled
Create and publish Docker images with specific build args / build-cuda126-image (linux/arm64, ubuntu-24.04-arm) (push) Has been cancelled
Create and publish Docker images with specific build args / build-ollama-image (linux/amd64, ubuntu-latest) (push) Has been cancelled
Create and publish Docker images with specific build args / build-ollama-image (linux/arm64, ubuntu-24.04-arm) (push) Has been cancelled
Create and publish Docker images with specific build args / build-slim-image (linux/amd64, ubuntu-latest) (push) Has been cancelled
Create and publish Docker images with specific build args / build-slim-image (linux/arm64, ubuntu-24.04-arm) (push) Has been cancelled
Create and publish Docker images with specific build args / merge-main-images (push) Has been cancelled
Create and publish Docker images with specific build args / merge-cuda-images (push) Has been cancelled
Create and publish Docker images with specific build args / merge-cuda126-images (push) Has been cancelled
Create and publish Docker images with specific build args / merge-ollama-images (push) Has been cancelled
Create and publish Docker images with specific build args / merge-slim-images (push) Has been cancelled
Python CI / Format Backend (3.11.x) (push) Has been cancelled
Python CI / Format Backend (3.12.x) (push) Has been cancelled
Frontend Build / Format & Build Frontend (push) Has been cancelled
Frontend Build / Frontend Unit Tests (push) Has been cancelled
Close inactive issues / close-issues (push) Has been cancelled
59 lines
1.9 KiB
Python
59 lines
1.9 KiB
Python
import logging
|
|
import json
|
|
from typing import Optional, List
|
|
|
|
|
|
from open_webui.retrieval.web.main import SearchResult, get_filtered_results
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def search_sougou(
|
|
sougou_api_sid: str,
|
|
sougou_api_sk: str,
|
|
query: str,
|
|
count: int,
|
|
filter_list: Optional[List[str]] = None,
|
|
) -> List[SearchResult]:
|
|
from tencentcloud.common.common_client import CommonClient
|
|
from tencentcloud.common import credential
|
|
from tencentcloud.common.exception.tencent_cloud_sdk_exception import (
|
|
TencentCloudSDKException,
|
|
)
|
|
from tencentcloud.common.profile.client_profile import ClientProfile
|
|
from tencentcloud.common.profile.http_profile import HttpProfile
|
|
|
|
try:
|
|
cred = credential.Credential(sougou_api_sid, sougou_api_sk)
|
|
http_profile = HttpProfile()
|
|
http_profile.endpoint = "tms.tencentcloudapi.com"
|
|
client_profile = ClientProfile()
|
|
client_profile.http_profile = http_profile
|
|
params = json.dumps({"Query": query, "Cnt": 20})
|
|
common_client = CommonClient(
|
|
"tms", "2020-12-29", cred, "", profile=client_profile
|
|
)
|
|
results = [
|
|
json.loads(page)
|
|
for page in common_client.call_json("SearchPro", json.loads(params))[
|
|
"Response"
|
|
]["Pages"]
|
|
]
|
|
sorted_results = sorted(
|
|
results, key=lambda x: x.get("scour", 0.0), reverse=True
|
|
)
|
|
if filter_list:
|
|
sorted_results = get_filtered_results(sorted_results, filter_list)
|
|
|
|
return [
|
|
SearchResult(
|
|
link=result.get("url"),
|
|
title=result.get("title"),
|
|
snippet=result.get("passage"),
|
|
)
|
|
for result in sorted_results[:count]
|
|
]
|
|
except TencentCloudSDKException as err:
|
|
log.error(f"Error in Sougou search: {err}")
|
|
return []
|