Commit Graph

919 Commits

Author SHA1 Message Date
zrguo
c840a053ac Merge pull request #477 from theClassLight/GG/pr
refactor(lightrag): 优化hashing_kv初始化逻辑
2024-12-17 15:09:26 +08:00
GG
9893e93dd1 refactor(lightrag): 优化hashing_kv初始化逻辑
- 修改了 llm_model_func 和 query 方法中的hashing_kv初始化逻辑
- 当 self.llm_response_cache 不存在或没有 global_config 属性时,会创建一个新的hashing_kv实例
- 所有基于hashing_kv获取global_config属性的对象可以正常运行
2024-12-17 13:45:39 +08:00
Saifeddine ALOUI
b9c75dc2cd Merge branch 'main' of https://github.com/ParisNeo/LightRAG 2024-12-16 21:49:51 +01:00
Saifeddine ALOUI
c67ec3a740 updazted 2024-12-16 21:49:50 +01:00
Saifeddine ALOUI
c0f4d13240 Update README.md 2024-12-16 20:59:55 +01:00
Saifeddine ALOUI
9a41912b88 Update README.md 2024-12-16 20:54:24 +01:00
Saifeddine ALOUI
883a67903c Update README.md 2024-12-16 20:41:05 +01:00
Saifeddine ALOUI
190694c769 Update README.md 2024-12-16 13:06:25 +01:00
Saifeddine ALOUI
d6d535234d Merge branch 'HKUDS:main' into main 2024-12-16 13:05:42 +01:00
zrguo
a3bf3a58f0 Merge pull request #465 from alllexx88/age_graph
Add Apache AGE graph storage
2024-12-16 16:21:47 +08:00
Saifeddine ALOUI
d8f4f3eede Added a fastapi service 2024-12-16 01:05:49 +01:00
Alex Potapenko
a44aceab1e AGEStorage: fix has_node() and has_edge() 2024-12-14 13:37:34 +01:00
Alex Potapenko
25d6939ca3 AGEStorage: all queries as multiline strings for consistency 2024-12-14 12:49:39 +01:00
Alex Potapenko
05c73bcfd1 AGEStorage: fix debug logging, and some cosmetics 2024-12-14 12:47:29 +01:00
Alex Potapenko
eb2e6d0d87 AGEStorage: fix has_edge(): AGE always returns field names in lower case 2024-12-14 12:24:45 +01:00
Alex Potapenko
7564841450 Add Apache AGE graph storage 2024-12-13 20:41:38 +01:00
LarFii
b7a2d336e6 Update __version__ 2024-12-13 20:15:49 +08:00
zrguo
9cac3b0ed7 Merge pull request #462 from JasonGuoo/main
Supporting Zhipu AI API
2024-12-13 20:09:11 +08:00
Jason Guo
e64cf5068f Fix import 2024-12-13 19:57:25 +08:00
zrguo
ae0c43b120 Merge pull request #464 from billvsme/fix/asyncio.as_completed
Maybe very important!!! Fix embedding error
2024-12-13 17:17:37 +08:00
billvsme
a788c78197 keep tqdm_async work 2024-12-13 16:48:22 +08:00
Jason Guo
6a0e9c6c77 Modify the chat_complete method to support keywords extraction. 2024-12-13 16:18:33 +08:00
billvsme
cf0278ca43 The asyncio.as_completed() function does not guarantee that the results are ordered 2024-12-13 15:42:41 +08:00
zrguo
b7552f35aa Merge pull request #461 from tjyiiuan/main
fix: update operate.py
2024-12-13 15:10:53 +08:00
zrguo
3de02457ff Merge pull request #458 from ChenZiHong-Gavin/fix-variable-name
fix: fix variable name(entitiy->entity)
2024-12-13 15:08:44 +08:00
Jiyu Tian
aac26b086e fix: update operate.py
1. 避免变量在赋值之前就被引用
2. 解决未找到entity返回None导致的unpack问题
2024-12-12 15:47:57 -05:00
chenzihong
e9107a67c3 fix: fix variable name(entitiy->entity) 2024-12-12 23:59:40 +08:00
zrguo
3e33cfc20e Merge pull request #452 from Weaxs/main
support TiDB: add TiDBKVStorage, TiDBVectorDBStorage
2024-12-12 11:35:25 +08:00
Weaxs
288985eab4 pre-commit fix tidb 2024-12-12 10:22:31 +08:00
Weaxs
8ef5a6b8cd support TiDB: add TiDBKVStorage, TiDBVectorDBStorage 2024-12-11 16:23:50 +08:00
zrguo
aee622f4e1 Merge pull request #447 from spo0nman/pkaushal/vectordb-chroma
feat: Add ChromaDB integration for vector storage
2024-12-11 14:22:08 +08:00
zrguo
504a3c233b Merge branch 'main' into pkaushal/vectordb-chroma 2024-12-11 14:21:36 +08:00
zrguo
7fbd9aa3e0 Merge pull request #444 from davidleon/fix/lazy_import
Fix/lazy import
2024-12-11 14:19:48 +08:00
zrguo
65e0e675cc Merge pull request #443 from magicyuan876/main
fix(lightrag): 修复只有实体没有关系的chunk处理逻辑
2024-12-11 14:14:59 +08:00
Magic_yuan
b63c6155ee style(lightrag): 调整ReadMe,加入自定义实体类型参数配置示例 2024-12-11 14:10:27 +08:00
Magic_yuan
9a2afc9484 style(lightrag): 调整代码格式 2024-12-11 14:06:55 +08:00
Magic_yuan
b89041b5b3 feat(operate): 添加实体类型配置并优化提示生成
- 在全局配置中添加 entity_types 参数,用于自定义实体类型
- 在生成实体提取和关系提取的提示时,使用配置的实体类型替代默认值
- 优化了提示生成逻辑,提高了代码的可配置性和灵活性
2024-12-11 13:53:05 +08:00
Magic_yuan
a09e1ba190 refactor(prompt): 优化提示模板以提高相似度评估的准确性- 明确了相似度评分的评判标准,包括不同情况下的评分依据
- 简化了评分流程,要求直接返回数字
- 期望通过这些改动提高评估的准确性和一致性
2024-12-11 12:57:58 +08:00
Magic_yuan
0a41cc8a9a feat(llm, prompt):增加日志输出并扩展实体类型
- 在 llm.py 中添加了日志输出,用于调试和记录 LLM 查询输入
- 在 prompt.py 中增加了 "category" 实体类型,扩展了实体提取的范围
2024-12-11 12:45:10 +08:00
Pankaj Kaushal
ca788463cc feat: Add ChromaDB integration for vector storage
- Implemented `ChromaVectorDBStorage` class in `lightrag/kg/chroma_impl.py` to support ChromaDB as a vector storage backend.
- Updated `lightrag.py` to include `ChromaVectorDBStorage` in the storage class mapping.
- Added a test script `test_chromadb.py` to demonstrate the usage of ChromaDB with LightRAG, including configuration for embedding functions and ChromaDB connection settings.
- fix lazy import function to support package context for dynamic class loading.
  288d4b8355
2024-12-10 16:23:05 +01:00
david
288d4b8355 fix lazy import 2024-12-10 17:16:21 +08:00
Magic_yuan
316c4df949 更新日志描述 2024-12-10 14:15:43 +08:00
Magic_yuan
58c0f94346 fix(lightrag): 修复只有实体没有关系的chunk处理逻辑
- 只有实体没有关系时,继续处理,而不是直接return
- 当只有实体而没有关系的图片在高阶查询关系时会返回空,这里优化返回,当没有关系时降级为local查询
2024-12-10 14:13:11 +08:00
david
21a3992e39 fix extra keyword_extraction. 2024-12-10 09:52:27 +08:00
david
f6eeedb050 add concurrent embedding limit 2024-12-10 09:01:21 +08:00
david
d0a4ef252e fix: rare embedding issue. 2024-12-10 09:00:22 +08:00
zrguo
11d8440393 Merge pull request #436 from eltociear/patch-1
chore: update llm.py
2024-12-09 22:24:13 +08:00
Ikko Eltociear Ashimine
b8cddb6c72 chore: update llm.py
intialize -> initialize
2024-12-09 22:08:06 +09:00
zrguo
e5cc43ad3c Merge pull request #434 from magicyuan876/main
提取关键词的prompt优化
2024-12-09 20:55:27 +08:00
zrguo
6282abcb42 Merge pull request #435 from davidleon/fix/unicode_escape
fix unicode escape. for the case of "/utils" in the response.
2024-12-09 20:54:58 +08:00