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)