update_config

.gitignore

@@ -14,6 +14,7 @@ rag/chroma/
 # log
 *.log
 .env
+*.mp3
 
 # ---> VisualStudioCode
 .vscode/*

configs/agent_cfg.yaml

@@ -0,0 +1,14 @@
+# agent config
+repo_root: /root/wulewule
+data_source_dir: ${repo_root}/data  # txt data dir
+
+# local llm
+llm_model: ${repo_root}/models/wulewule_v1_7b  # wulewule model path
+agent_embeddings_model: /root/model/sentence-transformer   # agent used embeddings model
+
+# remote llm
+use_remote: True   ## use SiliconFlow api llm
+SiliconFlow_api: ""   ## SiliconFlow api key
+remote_llm: "Qwen/Qwen2.5-72B-Instruct"  # SiliconFlow llm
+remote_embeddings_model: "BAAI/bge-m3"   # SiliconFlow embeddings model
+

configs/model_cfg.yaml

@@ -3,10 +3,12 @@
 defaults:
   - rag_cfg
   - _self_
+  - agent_cfg
 
 llm_default_dir:  ./models
 llm_model: ${llm_default_dir}/wulewule_v1_1_8b-w4a16-4bit  # wulewule model path, w4a16-4bit must turn on "use_lmdepoly"
 # llm_model: ${repo_root}/models/wulewule_v1_1_8b  # wulewule model path
+agent_mode: True
 use_lmdepoly: True
 use_rag: True  ## 是否开启rag
 stream_response: True # whether use streaming output