liuguancen
0c3a393d04
搜索接口：
- duckduckgo_search.py / ZhipuSearchAPI.py 切换到内网 searxng (原 43.251.225.121 / 134.122.191.214 已失效)
启动脚本清理：
- 删除废弃 backend/ 目录 (与 chat_web_backend/ 编译产物 jar MD5 相同，仅是改名副本)
- 删除 start_all.sh 与 langchain-chat/{start,stop,stop_quick,shutdown_all,restart}.sh (被 scripts/*-restart.sh 覆盖)
- 删除 chat_web_backend/{start,test_mysql}.sh
修复：
- scripts/backend-restart.sh 对齐当前实际在跑的 chat_web_backend.jar (profile=dev)
- scripts/log-trim-daemon.sh 把 LOCK 移到 /tmp 按用户命名，修复非首次用户跑时的 Permission denied
新增：
- scripts/start-all.sh：一键启动入口，串联 mysql/redis/milvus/langchain/backend/frontend，含端口自检
- chat_web_backend/application-local.yml.archived：原 backend/ 下 yj profile 覆盖配置的归档备份
其他：
- .gitignore 忽略 scripts/pptist-deploy/PPTist/ (323M 第三方源码树)
119 lines
4.2 KiB
Python
119 lines
4.2 KiB
Python
from datetime import datetime
|
|
import logging
|
|
import requests
|
|
# import uuid
|
|
|
|
from configs.model_config import LLM_MODELS
|
|
from server.chat.policy_fun_iast import get_llm_model_response
|
|
|
|
api_key = "2b35424a76188ea96558f9631890ecd3.Wl9stbNr8TJ9L5PJ"
|
|
|
|
class ZhipuSearchAPIWrapper:
|
|
# def zhipu_search(self, origin_query):
|
|
# search_query = get_llm_model_response(
|
|
# strategy_name="zhipu_search_rewrite",
|
|
# llm_model_name=LLM_MODELS[0],
|
|
# template_prompt_name="zhipu_search_rewrite",
|
|
# prompt_param_dict={"input": origin_query, "year": datetime.now().strftime("%Y")},
|
|
# temperature=0.3,
|
|
# max_tokens=512
|
|
# )
|
|
# logging.info(f"Zhipu检索内容:{search_query}")
|
|
# msg = [
|
|
# {
|
|
# "role": "user",
|
|
# "content": search_query
|
|
# }
|
|
# ]
|
|
# tool = "web-search-pro"
|
|
# url = "https://open.bigmodel.cn/api/paas/v4/tools"
|
|
# request_id = str(uuid.uuid4())
|
|
# data = {
|
|
# "request_id": request_id,
|
|
# "tool": tool,
|
|
# "stream": False,
|
|
# "messages": msg
|
|
# }
|
|
|
|
# try:
|
|
# resp = requests.post(
|
|
# url,
|
|
# json=data,
|
|
# headers={'Authorization': api_key},
|
|
# timeout=300
|
|
# )
|
|
# resp.raise_for_status() # 检查请求是否成功
|
|
# resp_json = resp.json()
|
|
# except requests.exceptions.RequestException as e:
|
|
# print(f"请求错误: {e}")
|
|
# return []
|
|
|
|
# # 解析响应以提取 search_result
|
|
# search_results = []
|
|
# choices = resp_json.get('choices', [])
|
|
# for choice in choices:
|
|
# message = choice.get('message', {})
|
|
# tool_calls = message.get('tool_calls', [])
|
|
# for tool_call in tool_calls:
|
|
# if 'search_result' in tool_call:
|
|
# search_results.extend(tool_call['search_result'])
|
|
|
|
# return search_results[:7]
|
|
def zhipu_search(self, origin_query):
|
|
search_query = get_llm_model_response(
|
|
strategy_name="zhipu_search_rewrite",
|
|
llm_model_name=LLM_MODELS[0],
|
|
template_prompt_name="zhipu_search_rewrite",
|
|
prompt_param_dict={"input": origin_query, "year": datetime.now().strftime("%Y")},
|
|
temperature=0.3,
|
|
max_tokens=512
|
|
)
|
|
|
|
logging.info(f"Zhipu检索内容:{search_query}")
|
|
url = "http://118.196.92.255/searxng/search"
|
|
engines = "duckduckgo,bing"
|
|
data = {
|
|
"format":"json",
|
|
"q":search_query,
|
|
"engines":engines,
|
|
"limit":10
|
|
}
|
|
|
|
try:
|
|
# resp = requests.post(
|
|
# url,
|
|
# json=data,
|
|
# timeout=5000
|
|
# )
|
|
resp = requests.get(
|
|
url,
|
|
params=data,
|
|
timeout=5000
|
|
)
|
|
resp.raise_for_status() # 检查请求是否成功
|
|
resp_json = resp.json()
|
|
except requests.exceptions.RequestException as e:
|
|
print(f"请求错误: {e}")
|
|
return []
|
|
|
|
# 解析响应以提取 search_result
|
|
search_results = []
|
|
choices = resp_json.get('results', [])
|
|
for choice in choices:
|
|
if 'publishedDate' in choice:
|
|
choice["content"] +=f"发布于{choice['publishedDate']}"
|
|
search_results.append(choice)
|
|
else:
|
|
search_results.append(choice)
|
|
# search_results.extend(choices)
|
|
return search_results[:7]
|
|
|
|
if __name__ == "__main__":
|
|
zhipu_search = ZhipuSearchAPIWrapper()
|
|
query = "粉末冶金产业技术创新战略联盟 粉末冶金领域关键词"
|
|
results = zhipu_search.zhipu_search(query)
|
|
if results:
|
|
print(results)
|
|
print(f"成功获取到 {len(results)} 个搜索结果")
|
|
else:
|
|
logging.info("没有找到任何搜索结果") |