Commit 1f793c2

@@ -11,6 +11,8 @@ from networking import hx_req
 
 
 async def get_online_search_result(query: str) -> list[dict]:
+    if GPT.PRIMARY_SEARCH_ENGINE == "google":
+        return await google_search_result(query)
     try:
         client = AsyncOpenAI(
             api_key=GPT.SEARCH_API_KEY,

@@ -146,6 +146,7 @@ class GPT:
     SEARCH_API_KEY = os.getenv("GPT_SEARCH_API_KEY", "")  # online search (currently, we use GLM)
     SEARCH_BASE_URL = os.getenv("GPT_SEARCH_BASE_URL", "https://open.bigmodel.cn/api/paas/v4")
     SEARCH_MODEL = os.getenv("GPT_SEARCH_MODEL", "web-search-pro")
+    PRIMARY_SEARCH_ENGINE = os.getenv("GPT_PRIMARY_SEARCH_ENGINE", "google")  # google or glm
     TEXT_TIMEOUT = os.getenv("GPT_TEXT_TIMEOUT", "120")
     IMAGE_TIMEOUT = os.getenv("GPT_IMAGE_TIMEOUT", "120")
     VIDEO_TIMEOUT = os.getenv("GPT_VIDEO_TIMEOUT", "120")