refac

backend/open_webui/retrieval/web/sougou.py

@@ -1,8 +1,8 @@
-import json
 import logging
 from typing import List, Optional
 
 from open_webui.retrieval.web.main import SearchResult, get_filtered_results
+from open_webui.utils.json_codec import JSONCodec
 
 log = logging.getLogger(__name__)
 
@@ -28,10 +28,10 @@ def search_sougou(
         http_profile.endpoint = 'tms.tencentcloudapi.com'
         client_profile = ClientProfile()
         client_profile.http_profile = http_profile
-        params = json.dumps({'Query': query, 'Cnt': 20})
+        params = JSONCodec.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']
+            JSONCodec.loads(page) for page in common_client.call_json('SearchPro', JSONCodec.loads(params))['Response']['Pages']
         ]
         sorted_results = sorted(results, key=lambda x: x.get('scour', 0.0), reverse=True)
         if filter_list: