37 lines
1005 B
Python
37 lines
1005 B
Python
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)
|
|
|
|
|