添加初始文件和功能,包括数据集和文档类的定义,以及添加文档块的用户界面
This commit is contained in:
36
chunk.py
Normal file
36
chunk.py
Normal 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)
|
||||
|
||||
|
Reference in New Issue
Block a user