添加初始文件和功能,包括数据集和文档类的定义,以及添加文档块的用户界面

This commit is contained in:
2025-07-04 18:26:36 +08:00
parent db6f3b865f
commit 587305f070
7 changed files with 133 additions and 0 deletions

36
chunk.py Normal file
View File

@@ -0,0 +1,36 @@
from ragflow_sdk import RAGFlow
api_key = "ragflow-I5ZDNjMWNhNTdlMjExZjBiOTEwMzI0ZT"
base_url = "http://192.168.107.165:8099"
rag_object = RAGFlow(api_key=api_key, base_url=base_url)
#dataset = rag_object.create_dataset(name="kb_1")
dataset = rag_object.list_datasets(name="kb_1")
dataset = dataset[0]
# filename1 = "ragflow.txt"
# blob = open(filename1 , "rb").read()
# dataset.upload_documents([{"display_name":filename1,"blob":blob}])
for doc in dataset.list_documents( page=0, page_size=12):
print(doc)
print("=========================================")
doc = dataset.list_documents(name= 'ragflow.txt')
doc = doc[0]
doc.update({"parser_config": {"chunk_token_count": 256}})
chunk = doc.add_chunk(content="xxxxxxx")
print(doc)
#dataset.upload_documents([{"display_name": "1.txt", "blob": open('1.txt',"rb").read()}])
# 查询所有知识库
# for dataset in rag_object.list_datasets():
# print(dataset)
# dataset = rag_object.list_datasets(name = "制度")
#print(dataset)