? Papel | Demonstração
Inglês | 简体中文
git clone https://github.com/InternLM/MindSearch
cd MindSearch
pip install -r requirements.txt
Antes de configurar a API, você precisa configurar variáveis de ambiente. Renomeie o arquivo .env.example
para .env
e preencha os valores necessários.
mv .env.example .env
# Open .env and add your keys and model configurations
Configure o servidor FastAPI.
python -m mindsearch.app --lang en --model_format internlm_server --search_engine DuckDuckGoSearch --asy
--lang
: idioma do modelo, en
para inglês e cn
para chinês.
--model_format
: formato do modelo.
internlm_server
para InternLM2.5-7b-chat com servidor local. (InternLM2.5-7b-chat foi melhor otimizado para chinês.)gpt4
para GPT4. se você quiser usar outros modelos, modifique os modelos --search_engine
: mecanismo de pesquisa.
DuckDuckGoSearch
o mecanismo de pesquisa do DuckDuckGo.BingSearch
para mecanismo de pesquisa do Bing.BraveSearch
para o mecanismo de API da web de pesquisa Brave.GoogleSearch
para mecanismo de API de pesquisa na web Google Serper.TencentSearch
para mecanismo de API de pesquisa Tencent. Defina a chave da API do seu mecanismo de pesquisa na Web como a variável de ambiente WEB_SEARCH_API_KEY
, a menos que você esteja usando DuckDuckGo
ou TencentSearch
que requer um ID secreto como TENCENT_SEARCH_SECRET_ID
e uma chave secreta como TENCENT_SEARCH_SECRET_KEY
.
--asy
: implanta agentes assíncronos.
Fornecendo as seguintes interfaces de front-end,
Primeiro configure o URL de back-end para o proxy Vite.
HOST= " 127.0.0.1 " # modify as you need
PORT=8002
sed -i -r " s/target:s* "" /target: " ${HOST} : ${PORT} " / " frontend/React/vite.config.ts
# Install Node.js and npm
# for Ubuntu
sudo apt install nodejs npm
# for windows
# download from https://nodejs.org/zh-cn/download/prebuilt-installer
# Install dependencies
cd frontend/React
npm install
npm start
Detalhes podem ser encontrados em React
python frontend/mindsearch_gradio.py
streamlit run frontend/mindsearch_streamlit.py
Para usar um tipo diferente de API de pesquisa na web, modifique o atributo searcher_type
em searcher_cfg
localizado em mindsearch/agent/__init__.py
. As APIs de pesquisa na web atualmente suportadas incluem:
GoogleSearch
DuckDuckGoSearch
BraveSearch
BingSearch
TencentSearch
Por exemplo, para mudar para a API Brave Search, você deve configurá-la da seguinte forma:
BingBrowser (
searcher_type = 'BraveSearch' ,
topk = 2 ,
api_key = os . environ . get ( 'BRAVE_API_KEY' , 'YOUR BRAVE API' )
)
Para usuários que preferem interagir diretamente com o back-end, use o script backend_example.py
. Este script demonstra como enviar uma consulta ao backend e processar a resposta.
python backend_example.py
Certifique-se de ter configurado as variáveis de ambiente e de que o back-end esteja em execução antes de executar o script.
python -m mindsearch.terminal
Este projeto é lançado sob a licença Apache 2.0.
Se você achar este projeto útil em sua pesquisa, considere citar:
@article{chen2024mindsearch,
title={MindSearch: Mimicking Human Minds Elicits Deep AI Searcher},
author={Chen, Zehui and Liu, Kuikun and Wang, Qiuchen and Liu, Jiangning and Zhang, Wenwei and Chen, Kai and Zhao, Feng},
journal={arXiv preprint arXiv:2407.20183},
year={2024}
}
Explore nossa pesquisa adicional sobre grandes modelos de linguagem, com foco em agentes LLM.