Skip to content

Jiaxi-Huang/HackerLLM

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

HackerLLM

简介

本项目基于vue(前端)以及fastapi(后端),通过调用bce-embedding模型进行向量数据库文本嵌入,结合到prompt调用通义前文qwen-long长对话模型,实现特定领域的问答。同时vue引入vuetify组件实现对https://www.hackthebox.com 网站的分析结果进行可视化展示

运行

适用于ubuntu linux(也适用于windows本地运行)

  • 安装anaconda/miniconda并初始化
mkdir -p ~/miniconda3
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda3/miniconda.sh
bash ~/miniconda3/miniconda.sh -b -u -p ~/miniconda3
rm -rf ~/miniconda3/miniconda.sh

~/miniconda3/bin/conda init bash
~/miniconda3/bin/conda init zsh
  • 下载项目需要的python库,主要是dashscope,langchain,fastapi,chromadb等,有报错缺少xx,直接pip install xx就好
  • 安装node.js(不必要)
sudo apt-get update 
sudo apt-get install nodejs

并且在main.py中配置api_key,详见: https://help.aliyun.com/zh/dashscope/developer-reference/acquisition-and-configuration-of-api-key?spm=a2c4g.11186623.0.0.42124937yljUMp

model_name = r"/root/huggingface_cache/bce-embedding-base_v1"
model_kwargs = {'device': 'cpu'}
encode_kwargs = {'normalize_embeddings': False}
hfembedding = HuggingFaceEmbeddings(
    model_name=model_name,
    model_kwargs=model_kwargs,
    encode_kwargs=encode_kwargs
)

将model_name改为你下载好后放的路径

贡献

SWS3023 Web Mining Group1

参考资料

https://github.com/datawhalechina/llm-universe 😊鸣谢

Welcome pull request