refac

backend/open_webui/retrieval/web/yandex.py

@@ -1,6 +1,5 @@
 import base64
 import io
-import json
 import logging
 import os
 from typing import List, Optional
@@ -12,6 +11,7 @@ from fastapi import Request
 from open_webui.env import FORWARD_SESSION_INFO_HEADER_CHAT_ID
 from open_webui.retrieval.web.main import SearchResult, get_filtered_results
 from open_webui.utils.headers import include_user_info_headers
+from open_webui.utils.json_codec import JSONCodec
 
 log = logging.getLogger(__name__)
 
@@ -55,7 +55,7 @@ def search_yandex(
         if chat_id:
             headers[FORWARD_SESSION_INFO_HEADER_CHAT_ID] = str(chat_id)
 
-        payload = {} if yandex_search_config == '' else json.loads(yandex_search_config)
+        payload = {} if yandex_search_config == '' else JSONCodec.loads(yandex_search_config)
 
         if type(payload.get('query', None)) != dict:
             payload['query'] = {}