fix chroma_bug

This commit is contained in:
HengZhang 2026-03-01 11:25:38 +08:00
parent 187b8c8756
commit 3788f8b68e
23 changed files with 142 additions and 8 deletions

2
.gitignore vendored
View File

@ -1,2 +1,4 @@
.git .git
.idea .idea
.__pycache__/
.chroma_db/

Binary file not shown.

Binary file not shown.

BIN
chroma_db/chroma.sqlite3 Normal file

Binary file not shown.

View File

@ -1,2 +0,0 @@
import os

View File

@ -26,3 +26,135 @@
2026-03-01 11:08:25,493 - Agent - INFO - middleware.py:23 - [tool minitor]工具rag_summarize调用成功 2026-03-01 11:08:25,493 - Agent - INFO - middleware.py:23 - [tool minitor]工具rag_summarize调用成功
2026-03-01 11:08:25,494 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有9条消息 2026-03-01 11:08:25,494 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有9条消息
2026-03-01 11:08:25,494 - Agent - DEBUG - middleware.py:40 - [log_before_model]ToolMessage | 消息内容:定期清理滚刷和边刷上的缠绕物,清空尘盒,擦拭传感器和充电触点,检查并清理滤网,避免在潮湿或有水区域使用,及时更换磨损配件。 2026-03-01 11:08:25,494 - Agent - DEBUG - middleware.py:40 - [log_before_model]ToolMessage | 消息内容:定期清理滚刷和边刷上的缠绕物,清空尘盒,擦拭传感器和充电触点,检查并清理滤网,避免在潮湿或有水区域使用,及时更换磨损配件。
2026-03-01 11:11:51,192 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有1条消息
2026-03-01 11:11:51,193 - Agent - DEBUG - middleware.py:40 - [log_before_model]HumanMessage | 消息内容:扫地机器人在我所在地的气温下如何保养
2026-03-01 11:11:55,178 - Agent - INFO - middleware.py:18 - [tool monitor]执行工具get_user_location
2026-03-01 11:11:55,178 - Agent - INFO - middleware.py:19 - [tool monitor]传入参数:{}
2026-03-01 11:11:55,179 - Agent - INFO - middleware.py:23 - [tool minitor]工具get_user_location调用成功
2026-03-01 11:11:55,180 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有3条消息
2026-03-01 11:11:55,180 - Agent - DEBUG - middleware.py:40 - [log_before_model]ToolMessage | 消息内容:深圳
2026-03-01 11:11:58,399 - Agent - INFO - middleware.py:18 - [tool monitor]执行工具get_weather
2026-03-01 11:11:58,399 - Agent - INFO - middleware.py:19 - [tool monitor]传入参数:{'city': '深圳'}
2026-03-01 11:11:58,400 - Agent - INFO - middleware.py:23 - [tool minitor]工具get_weather调用成功
2026-03-01 11:11:58,401 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有5条消息
2026-03-01 11:11:58,401 - Agent - DEBUG - middleware.py:40 - [log_before_model]ToolMessage | 消息内容城市深圳天气为晴天温度为26摄氏度湿度50%南风1级AQI21最近6小时降雨概率极低
2026-03-01 11:12:02,884 - Agent - INFO - middleware.py:18 - [tool monitor]执行工具rag_summarize
2026-03-01 11:12:02,886 - Agent - INFO - middleware.py:19 - [tool monitor]传入参数:{'query': '扫地机器人在26摄氏度、湿度50%的环境下如何保养'}
2026-03-01 11:12:05,104 - Agent - INFO - middleware.py:23 - [tool minitor]工具rag_summarize调用成功
2026-03-01 11:12:05,106 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有7条消息
2026-03-01 11:12:05,106 - Agent - DEBUG - middleware.py:40 - [log_before_model]ToolMessage | 消息内容参考资料中未提及扫地机器人在26摄氏度、湿度50%环境下的具体保养方法。
2026-03-01 11:12:09,956 - Agent - INFO - middleware.py:18 - [tool monitor]执行工具rag_summarize
2026-03-01 11:12:09,956 - Agent - INFO - middleware.py:19 - [tool monitor]传入参数:{'query': '扫地机器人日常保养方法'}
2026-03-01 11:12:14,599 - Agent - INFO - middleware.py:23 - [tool minitor]工具rag_summarize调用成功
2026-03-01 11:12:14,600 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有9条消息
2026-03-01 11:12:14,600 - Agent - DEBUG - middleware.py:40 - [log_before_model]ToolMessage | 消息内容:定期清理滚刷和边刷上的缠绕物,清空尘盒,擦拭传感器和充电触点,检查并清理滤网,避免在潮湿或有大量液体的地面上使用。
2026-03-01 11:15:00,856 - Agent - INFO - vector_store.py:88 - [加载知识库]/Users/czzhangheng/edu/agent_proj/data/选购指南.txt 加载成功
2026-03-01 11:15:04,872 - Agent - INFO - vector_store.py:88 - [加载知识库]/Users/czzhangheng/edu/agent_proj/data/扫地机器人100问2.txt 加载成功
2026-03-01 11:15:14,181 - Agent - INFO - vector_store.py:88 - [加载知识库]/Users/czzhangheng/edu/agent_proj/data/故障排除.txt 加载成功
2026-03-01 11:15:19,352 - Agent - INFO - vector_store.py:88 - [加载知识库]/Users/czzhangheng/edu/agent_proj/data/扫拖一体机器人100问.txt 加载成功
2026-03-01 11:15:26,670 - Agent - INFO - vector_store.py:88 - [加载知识库]/Users/czzhangheng/edu/agent_proj/data/维护保养.txt 加载成功
2026-03-01 11:15:26,672 - Agent - INFO - vector_store.py:70 - [加载知识库]: /Users/czzhangheng/edu/agent_proj/data/扫地机器人100问.pdf 的 md5已存在, 跳过
2026-03-01 11:16:06,714 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有1条消息
2026-03-01 11:16:06,714 - Agent - DEBUG - middleware.py:40 - [log_before_model]HumanMessage | 消息内容:扫地机器人在我所在地的气温下如何保养
2026-03-01 11:16:09,907 - Agent - INFO - middleware.py:18 - [tool monitor]执行工具get_user_location
2026-03-01 11:16:09,908 - Agent - INFO - middleware.py:19 - [tool monitor]传入参数:{}
2026-03-01 11:16:09,909 - Agent - INFO - middleware.py:23 - [tool minitor]工具get_user_location调用成功
2026-03-01 11:16:09,911 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有3条消息
2026-03-01 11:16:09,911 - Agent - DEBUG - middleware.py:40 - [log_before_model]ToolMessage | 消息内容:深圳
2026-03-01 11:16:13,189 - Agent - INFO - middleware.py:18 - [tool monitor]执行工具get_weather
2026-03-01 11:16:13,190 - Agent - INFO - middleware.py:19 - [tool monitor]传入参数:{'city': '深圳'}
2026-03-01 11:16:13,191 - Agent - INFO - middleware.py:23 - [tool minitor]工具get_weather调用成功
2026-03-01 11:16:13,192 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有5条消息
2026-03-01 11:16:13,193 - Agent - DEBUG - middleware.py:40 - [log_before_model]ToolMessage | 消息内容城市深圳天气为晴天温度为26摄氏度湿度50%南风1级AQI21最近6小时降雨概率极低
2026-03-01 11:16:17,561 - Agent - INFO - middleware.py:18 - [tool monitor]执行工具rag_summarize
2026-03-01 11:16:17,561 - Agent - INFO - middleware.py:19 - [tool monitor]传入参数:{'query': '扫地机器人在26摄氏度50%湿度环境下的保养方法'}
2026-03-01 11:16:23,493 - Agent - INFO - middleware.py:23 - [tool minitor]工具rag_summarize调用成功
2026-03-01 11:16:23,495 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有7条消息
2026-03-01 11:16:23,495 - Agent - DEBUG - middleware.py:40 - [log_before_model]ToolMessage | 消息内容参考资料中未提及扫地机器人在26摄氏度50%湿度环境下的具体保养方法。
2026-03-01 11:16:27,178 - Agent - INFO - middleware.py:18 - [tool monitor]执行工具rag_summarize
2026-03-01 11:16:27,179 - Agent - INFO - middleware.py:19 - [tool monitor]传入参数:{'query': '扫地机器人日常保养方法和注意事项'}
2026-03-01 11:16:31,076 - Agent - INFO - middleware.py:23 - [tool minitor]工具rag_summarize调用成功
2026-03-01 11:16:31,077 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有9条消息
2026-03-01 11:16:31,078 - Agent - DEBUG - middleware.py:40 - [log_before_model]ToolMessage | 消息内容:扫地机器人日常保养需定期清理尘盒、滤网和滚刷,避免毛发缠绕;使用后及时倒掉垃圾并擦拭传感器;充电时确保电量充足但不过充;避免在潮湿或有水区域使用;定期检查边刷和主刷磨损情况,必要时更换;长时间不用应断电存放于干燥处。
2026-03-01 11:17:25,207 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有1条消息
2026-03-01 11:17:25,207 - Agent - DEBUG - middleware.py:40 - [log_before_model]HumanMessage | 消息内容:扫地机器人在我所在地的气温下如何保养
2026-03-01 11:17:28,108 - Agent - INFO - middleware.py:18 - [tool monitor]执行工具get_user_location
2026-03-01 11:17:28,108 - Agent - INFO - middleware.py:19 - [tool monitor]传入参数:{}
2026-03-01 11:17:28,109 - Agent - INFO - middleware.py:23 - [tool minitor]工具get_user_location调用成功
2026-03-01 11:17:28,110 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有3条消息
2026-03-01 11:17:28,110 - Agent - DEBUG - middleware.py:40 - [log_before_model]ToolMessage | 消息内容:合肥
2026-03-01 11:17:31,177 - Agent - INFO - middleware.py:18 - [tool monitor]执行工具get_weather
2026-03-01 11:17:31,178 - Agent - INFO - middleware.py:19 - [tool monitor]传入参数:{'city': '合肥'}
2026-03-01 11:17:31,179 - Agent - INFO - middleware.py:23 - [tool minitor]工具get_weather调用成功
2026-03-01 11:17:31,180 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有5条消息
2026-03-01 11:17:31,181 - Agent - DEBUG - middleware.py:40 - [log_before_model]ToolMessage | 消息内容城市合肥天气为晴天温度为26摄氏度湿度50%南风1级AQI21最近6小时降雨概率极低
2026-03-01 11:17:34,854 - Agent - INFO - middleware.py:18 - [tool monitor]执行工具rag_summarize
2026-03-01 11:17:34,855 - Agent - INFO - middleware.py:19 - [tool monitor]传入参数:{'query': '扫地机器人在26摄氏度温度和50%湿度环境下的保养方法'}
2026-03-01 11:17:36,814 - Agent - INFO - middleware.py:23 - [tool minitor]工具rag_summarize调用成功
2026-03-01 11:17:36,816 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有7条消息
2026-03-01 11:17:36,816 - Agent - DEBUG - middleware.py:40 - [log_before_model]ToolMessage | 消息内容参考资料中未提及扫地机器人在26摄氏度温度和50%湿度环境下的具体保养方法。
2026-03-01 11:17:39,332 - Agent - INFO - middleware.py:18 - [tool monitor]执行工具rag_summarize
2026-03-01 11:17:39,333 - Agent - INFO - middleware.py:19 - [tool monitor]传入参数:{'query': '扫地机器人日常保养方法和注意事项'}
2026-03-01 11:17:43,139 - Agent - INFO - middleware.py:23 - [tool minitor]工具rag_summarize调用成功
2026-03-01 11:17:43,141 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有9条消息
2026-03-01 11:17:43,141 - Agent - DEBUG - middleware.py:40 - [log_before_model]ToolMessage | 消息内容:日常保养方法包括:定期清理滚刷和边刷上的缠绕物,清空尘盒并清洗滤网(若为可水洗型),检查并清洁传感器和充电触点。注意事项有:避免在潮湿地面使用,防止进水;勿在高温或阳光直射处存放;定期更换滤网和滚刷以维持吸力;使用后及时充电,避免电池过度放电。
2026-03-01 11:19:46,949 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有1条消息
2026-03-01 11:19:46,949 - Agent - DEBUG - middleware.py:40 - [log_before_model]HumanMessage | 消息内容:扫地机器人在我所在地的气温下如何保养
2026-03-01 11:19:49,795 - Agent - INFO - middleware.py:18 - [tool monitor]执行工具get_user_location
2026-03-01 11:19:49,795 - Agent - INFO - middleware.py:19 - [tool monitor]传入参数:{}
2026-03-01 11:19:49,797 - Agent - INFO - middleware.py:23 - [tool minitor]工具get_user_location调用成功
2026-03-01 11:19:49,798 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有3条消息
2026-03-01 11:19:49,798 - Agent - DEBUG - middleware.py:40 - [log_before_model]ToolMessage | 消息内容:深圳
2026-03-01 11:19:53,523 - Agent - INFO - middleware.py:18 - [tool monitor]执行工具get_weather
2026-03-01 11:19:53,524 - Agent - INFO - middleware.py:19 - [tool monitor]传入参数:{'city': '深圳'}
2026-03-01 11:19:53,525 - Agent - INFO - middleware.py:23 - [tool minitor]工具get_weather调用成功
2026-03-01 11:19:53,527 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有5条消息
2026-03-01 11:19:53,527 - Agent - DEBUG - middleware.py:40 - [log_before_model]ToolMessage | 消息内容城市深圳天气为晴天温度为26摄氏度湿度50%南风1级AQI21最近6小时降雨概率极低
2026-03-01 11:19:58,971 - Agent - INFO - middleware.py:18 - [tool monitor]执行工具rag_summarize
2026-03-01 11:19:58,972 - Agent - INFO - middleware.py:19 - [tool monitor]传入参数:{'query': '扫地机器人在26摄氏度50%湿度环境下的保养方法'}
2026-03-01 11:19:59,393 - Agent - INFO - rag_service.py:33 - [rag_summarize]: 召回了0条参考资料
2026-03-01 11:20:01,300 - Agent - INFO - middleware.py:23 - [tool minitor]工具rag_summarize调用成功
2026-03-01 11:20:01,302 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有7条消息
2026-03-01 11:20:01,302 - Agent - DEBUG - middleware.py:40 - [log_before_model]ToolMessage | 消息内容参考资料中未提及扫地机器人在26摄氏度50%湿度环境下的具体保养方法。
2026-03-01 11:20:04,682 - Agent - INFO - middleware.py:18 - [tool monitor]执行工具rag_summarize
2026-03-01 11:20:04,682 - Agent - INFO - middleware.py:19 - [tool monitor]传入参数:{'query': '扫地机器人日常保养方法和注意事项'}
2026-03-01 11:20:05,086 - Agent - INFO - rag_service.py:33 - [rag_summarize]: 召回了0条参考资料
2026-03-01 11:20:08,042 - Agent - INFO - middleware.py:23 - [tool minitor]工具rag_summarize调用成功
2026-03-01 11:20:08,043 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有9条消息
2026-03-01 11:20:08,043 - Agent - DEBUG - middleware.py:40 - [log_before_model]ToolMessage | 消息内容:扫地机器人日常保养需定期清理尘盒、滤网和滚刷,避免毛发缠绕;使用后及时倒空尘盒并清洗可水洗部件;注意检查边刷和传感器是否被灰尘遮挡;避免在潮湿或有水区域使用;长时间不用时应断电并存放于干燥处。
2026-03-01 11:20:59,704 - Agent - INFO - vector_store.py:88 - [加载知识库]/Users/czzhangheng/edu/agent_proj/data/选购指南.txt 加载成功
2026-03-01 11:21:04,963 - Agent - INFO - vector_store.py:88 - [加载知识库]/Users/czzhangheng/edu/agent_proj/data/扫地机器人100问2.txt 加载成功
2026-03-01 11:21:14,369 - Agent - INFO - vector_store.py:88 - [加载知识库]/Users/czzhangheng/edu/agent_proj/data/故障排除.txt 加载成功
2026-03-01 11:21:19,648 - Agent - INFO - vector_store.py:88 - [加载知识库]/Users/czzhangheng/edu/agent_proj/data/扫拖一体机器人100问.txt 加载成功
2026-03-01 11:21:24,742 - Agent - INFO - vector_store.py:88 - [加载知识库]/Users/czzhangheng/edu/agent_proj/data/维护保养.txt 加载成功
2026-03-01 11:21:28,778 - Agent - INFO - vector_store.py:88 - [加载知识库]/Users/czzhangheng/edu/agent_proj/data/扫地机器人100问.pdf 加载成功
2026-03-01 11:23:38,382 - Agent - INFO - vector_store.py:70 - [加载知识库]: /Users/czzhangheng/edu/agent_proj/data/选购指南.txt 的 md5已存在, 跳过
2026-03-01 11:23:38,382 - Agent - INFO - vector_store.py:70 - [加载知识库]: /Users/czzhangheng/edu/agent_proj/data/扫地机器人100问2.txt 的 md5已存在, 跳过
2026-03-01 11:23:38,383 - Agent - INFO - vector_store.py:70 - [加载知识库]: /Users/czzhangheng/edu/agent_proj/data/故障排除.txt 的 md5已存在, 跳过
2026-03-01 11:23:38,383 - Agent - INFO - vector_store.py:70 - [加载知识库]: /Users/czzhangheng/edu/agent_proj/data/扫拖一体机器人100问.txt 的 md5已存在, 跳过
2026-03-01 11:23:38,383 - Agent - INFO - vector_store.py:70 - [加载知识库]: /Users/czzhangheng/edu/agent_proj/data/维护保养.txt 的 md5已存在, 跳过
2026-03-01 11:23:38,384 - Agent - INFO - vector_store.py:70 - [加载知识库]: /Users/czzhangheng/edu/agent_proj/data/扫地机器人100问.pdf 的 md5已存在, 跳过
2026-03-01 11:23:52,632 - Agent - INFO - vector_store.py:88 - [加载知识库]/Users/czzhangheng/edu/agent_proj/data/选购指南.txt 加载成功
2026-03-01 11:23:56,731 - Agent - INFO - vector_store.py:88 - [加载知识库]/Users/czzhangheng/edu/agent_proj/data/扫地机器人100问2.txt 加载成功
2026-03-01 11:24:05,696 - Agent - INFO - vector_store.py:88 - [加载知识库]/Users/czzhangheng/edu/agent_proj/data/故障排除.txt 加载成功
2026-03-01 11:24:10,675 - Agent - INFO - vector_store.py:88 - [加载知识库]/Users/czzhangheng/edu/agent_proj/data/扫拖一体机器人100问.txt 加载成功
2026-03-01 11:24:15,997 - Agent - INFO - vector_store.py:88 - [加载知识库]/Users/czzhangheng/edu/agent_proj/data/维护保养.txt 加载成功
2026-03-01 11:24:19,368 - Agent - INFO - vector_store.py:88 - [加载知识库]/Users/czzhangheng/edu/agent_proj/data/扫地机器人100问.pdf 加载成功
2026-03-01 11:24:31,603 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有1条消息
2026-03-01 11:24:31,603 - Agent - DEBUG - middleware.py:40 - [log_before_model]HumanMessage | 消息内容:扫地机器人在我所在地的气温下如何保养
2026-03-01 11:24:34,912 - Agent - INFO - middleware.py:18 - [tool monitor]执行工具get_user_location
2026-03-01 11:24:34,912 - Agent - INFO - middleware.py:19 - [tool monitor]传入参数:{}
2026-03-01 11:24:34,913 - Agent - INFO - middleware.py:23 - [tool minitor]工具get_user_location调用成功
2026-03-01 11:24:34,915 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有3条消息
2026-03-01 11:24:34,915 - Agent - DEBUG - middleware.py:40 - [log_before_model]ToolMessage | 消息内容:杭州
2026-03-01 11:24:37,778 - Agent - INFO - middleware.py:18 - [tool monitor]执行工具get_weather
2026-03-01 11:24:37,779 - Agent - INFO - middleware.py:19 - [tool monitor]传入参数:{'city': '杭州'}
2026-03-01 11:24:37,780 - Agent - INFO - middleware.py:23 - [tool minitor]工具get_weather调用成功
2026-03-01 11:24:37,781 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有5条消息
2026-03-01 11:24:37,781 - Agent - DEBUG - middleware.py:40 - [log_before_model]ToolMessage | 消息内容城市杭州天气为晴天温度为26摄氏度湿度50%南风1级AQI21最近6小时降雨概率极低
2026-03-01 11:24:42,103 - Agent - INFO - middleware.py:18 - [tool monitor]执行工具rag_summarize
2026-03-01 11:24:42,103 - Agent - INFO - middleware.py:19 - [tool monitor]传入参数:{'query': '扫地机器人在26摄氏度50%湿度环境下的保养方法'}
2026-03-01 11:24:42,574 - Agent - INFO - rag_service.py:33 - [rag_summarize]: 召回了3条参考资料
2026-03-01 11:24:47,101 - Agent - INFO - middleware.py:23 - [tool minitor]工具rag_summarize调用成功
2026-03-01 11:24:47,103 - Agent - INFO - middleware.py:39 - [log_before_model]即将调用模型带有7条消息
2026-03-01 11:24:47,103 - Agent - DEBUG - middleware.py:40 - [log_before_model]ToolMessage | 消息内容在26摄氏度、50%湿度环境下,扫地机器人保养应遵循通用基础维护方法:每日使用后用干软布擦拭机身外壳,去除灰尘和水渍;每次清扫完成后及时清理防撞条缝隙中的毛发和线头,防止卡扣卡顿。该环境未在参考资料中被特别归类,因此无需额外特殊处理。

10
md5.txt
View File

@ -1,6 +1,6 @@
94bd6c0b4283b8524f4e8dfc5b72f8ab
a29a4ba5a539a2a462def8f91eb98ccc
928c24dbdeeb1c0a109f5620c8bac429
d294281a5769e0a3d1455fbf637ee572
fc539588792a89ff842cf241cf822de1
6d726f4604d9e7ebcf59a932ae9e9c00 6d726f4604d9e7ebcf59a932ae9e9c00
4ef62dc4189785055436124361296d05
76d81647e3901472e01ad9859425506f
85fd3b64ff87fe08fdc5c4a34998b555
96a099807d1ad7c26e37b53d130f26d2
1ce3fc50860e15d1c106900b63ef42a2

Binary file not shown.

View File

@ -7,6 +7,7 @@ from langchain_core.prompts import PromptTemplate
from model.factory import chat_model from model.factory import chat_model
from langchain_community.docstore.document import Document from langchain_community.docstore.document import Document
from typing import List from typing import List
from utils.logger_handler import logger
class RagSummarizeService: class RagSummarizeService:
def __init__(self): def __init__(self):
@ -29,6 +30,7 @@ class RagSummarizeService:
context_docs = self.retriever_docs(query) context_docs = self.retriever_docs(query)
context = "" context = ""
counter = 0 counter = 0
logger.info(f"[rag_summarize]: 召回了{len(context_docs)}条参考资料")
for doc in context_docs: for doc in context_docs:
counter += 1 counter += 1
context += f"[参考资料{counter}]: 参考资料:{doc.page_content} | 参考源: {doc.metadata} \n" context += f"[参考资料{counter}]: 参考资料:{doc.page_content} | 参考源: {doc.metadata} \n"

View File

@ -16,7 +16,7 @@ class VectorStoreSerivce:
self.vector_store = Chroma( self.vector_store = Chroma(
collection_name=chroma_conf["collection_name"], collection_name=chroma_conf["collection_name"],
embedding_function=embedding_model, embedding_function=embedding_model,
persist_directory=chroma_conf["persist_directory"], persist_directory=get_abs_path(chroma_conf["persist_directory"]),
) )
self.spliter = RecursiveCharacterTextSplitter( self.spliter = RecursiveCharacterTextSplitter(
chunk_size=chroma_conf["chunk_size"], chunk_size=chroma_conf["chunk_size"],

View File

View File