fix(lightrag): 修复只有实体没有关系的chunk处理逻辑 #443

magicyuan876 · 2024-12-10T06:16:21Z

- 只有实体没有关系时，继续处理，而不是直接return - 当只有实体而没有关系的图片在高阶查询关系时会返回空，这里优化返回，当没有关系时降级为local查询

- 在 llm.py 中添加了日志输出，用于调试和记录 LLM 查询输入 - 在 prompt.py 中增加了 "category" 实体类型，扩展了实体提取的范围

- 简化了评分流程，要求直接返回数字 - 期望通过这些改动提高评估的准确性和一致性

- 在全局配置中添加 entity_types 参数，用于自定义实体类型 - 在生成实体提取和关系提取的提示时，使用配置的实体类型替代默认值 - 优化了提示生成逻辑，提高了代码的可配置性和灵活性

magicyuan876 added 7 commits December 10, 2024 14:13

fix(lightrag): 修复只有实体没有关系的chunk处理逻辑

58c0f94

- 只有实体没有关系时，继续处理，而不是直接return - 当只有实体而没有关系的图片在高阶查询关系时会返回空，这里优化返回，当没有关系时降级为local查询

更新日志描述

316c4df

feat(llm, prompt):增加日志输出并扩展实体类型

0a41cc8

- 在 llm.py 中添加了日志输出，用于调试和记录 LLM 查询输入 - 在 prompt.py 中增加了 "category" 实体类型，扩展了实体提取的范围

refactor(prompt): 优化提示模板以提高相似度评估的准确性- 明确了相似度评分的评判标准，包括不同情况下的评分依据

a09e1ba

- 简化了评分流程，要求直接返回数字 - 期望通过这些改动提高评估的准确性和一致性

style(lightrag): 调整代码格式

9a2afc9

style(lightrag): 调整ReadMe,加入自定义实体类型参数配置示例

b63c615

LarFii merged commit 65e0e67 into HKUDS:main Dec 11, 2024
1 check passed

Provide feedback