diff --git a/.gitignore b/.gitignore index 723ef36..255b249 100644 --- a/.gitignore +++ b/.gitignore @@ -1 +1,3 @@ -.idea \ No newline at end of file +.idea +moka-ai/* +logs \ No newline at end of file diff --git a/README.md b/README.md new file mode 100644 index 0000000..ce09db2 --- /dev/null +++ b/README.md @@ -0,0 +1,37 @@ +# 上传的参数params +```text +knowledge_base_name: ceshi +score_threshold: 0.6 +top_k:2 +appid:d597331d +api_secret:ZDUwNDAxNWI2OWQwZjA2OTQ3ZTJhNGE1 +api_key:7f91f70f4384805483d948b9aa1317e6 +``` + +# 发送消息格式 +与讯飞星火完全一致。https://www.xfyun.cn/doc/spark/Web.html +```text +{ + "header": { + "app_id": "d597331d", + "uid": "1234" + }, + "parameter": { + "chat": { + "domain": "general", + "temperature": 0.5, + "max_tokens": 2048 + } + }, + "payload": { + "message": { + "text": [ + { + "role": "user", + "content": "你是谁" + } + ] + } + } +} +``` \ No newline at end of file diff --git a/knowledge_base/info.db b/knowledge_base/info.db index 21cdaf5..56e68b8 100644 Binary files a/knowledge_base/info.db and b/knowledge_base/info.db differ diff --git a/requirements.txt b/requirements.txt index 05b999c..48e66b9 100755 --- a/requirements.txt +++ b/requirements.txt @@ -24,15 +24,15 @@ spacy PyMuPDF rapidocr_onnxruntime -requests~=2.31.0 -pathlib~=1.0.1 -pytest~=7.4.3 -numexpr~=2.8.7 -strsimpy~=0.2.1 -markdownify~=0.11.6 +requests +pathlib +pytest +numexpr +strsimpy +markdownify tiktoken -tqdm~=4.66.1 -websockets~=11.0.3 +tqdm +websockets numpy~=1.24.4 pandas~=2.0.3 einops @@ -41,32 +41,12 @@ transformers_stream_generator==0.0.4 vllm>=0.2.0; sys_platform == "linux" # online api libs -# zhipuai -# dashscope>=1.10.0 # qwen -# qianfan +zhipuai +dashscope>=1.10.0 # qwen +qianfan # volcengine>=1.0.106 # fangzhou # uncomment libs if you want to use corresponding vector store # pymilvus==2.1.3 # requires milvus==2.1.3 # psycopg2 # pgvector - -# WebUI requirements - -streamlit~=1.27.0 -streamlit-option-menu>=0.3.6 -streamlit-antd-components>=0.1.11 -streamlit-chatbox>=1.1.11 -streamlit-aggrid>=0.3.4.post3 -httpx[brotli,http2,socks]>=0.25.0 -watchdog - -sentencepiece~=0.1.99 -cachetools~=5.3.2 -chardet~=5.2.0 -python-dateutil~=2.8.2 -safetensors~=0.4.0 -readline~=8.2 -colorama~=0.4.6 -loguru~=0.7.2 -pyyaml~=6.0.1 \ No newline at end of file