更新文档处理逻辑,添加从文件读取文本块并上传至知识库的功能
This commit is contained in:
40
chunk.py
40
chunk.py
@@ -1,28 +1,48 @@
|
||||
from ragflow_sdk import RAGFlow
|
||||
|
||||
api_key = "ragflow-I5ZDNjMWNhNTdlMjExZjBiOTEwMzI0ZT"
|
||||
base_url = "http://192.168.107.165:8099"
|
||||
#api_key = "ragflow-I5ZDNjMWNhNTdlMjExZjBiOTEwMzI0ZT"
|
||||
#base_url = "http://192.168.107.165:8099"
|
||||
base_url = "http://localhost"
|
||||
api_key = "ragflow-MyMjM2ODE2NThlMTExZjBiMzJlNzY5Mj"
|
||||
|
||||
rag_object = RAGFlow(api_key=api_key, base_url=base_url)
|
||||
#dataset = rag_object.create_dataset(name="kb_1")
|
||||
|
||||
datasets = rag_object.list_datasets()
|
||||
#dataset = rag_object.list_datasets(name="kb_1")
|
||||
dataset = rag_object.list_datasets(name="制度")
|
||||
|
||||
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("=========================================")
|
||||
# for doc in dataset.list_documents( page=0, page_size=12):
|
||||
# print(doc)
|
||||
# print("=========================================")
|
||||
|
||||
|
||||
doc = dataset.list_documents(name= 'ragflow.txt')
|
||||
doc = dataset.list_documents(name= '科技创新管理办法(试行).pdf')
|
||||
doc = doc[0]
|
||||
doc.update({"parser_config": {"chunk_token_count": 256}})
|
||||
chunk = doc.add_chunk(content="xxxxxxx")
|
||||
print(doc)
|
||||
# doc.update({"parser_config": {"chunk_token_count": 256}})
|
||||
file_path ="G:\\11\\ragflow_api_test\\1.txt"
|
||||
with open(file_path, 'r', encoding='utf-8') as file:
|
||||
file_content = file.read()
|
||||
for num,txt_chunk in enumerate(file_content.split('\n\n')):
|
||||
print(f"处理文本块: {txt_chunk[:30]}...") # 打印前30个字符以示例
|
||||
|
||||
if txt_chunk.strip(): # 确保不是空行
|
||||
chunk = doc.add_chunk(content=txt_chunk)
|
||||
print(f"第{num+1} Chunk添加成功! ID: {chunk.id}")
|
||||
|
||||
|
||||
|
||||
# content = '''
|
||||
# 第二章 部门职责>第六条 【财务管理部】
|
||||
# (一) 配合投标相关费用的办理工作;
|
||||
# (二) 负责提供投标所需的相关财务资料。'''
|
||||
# chunk = doc.add_chunk(content=content)
|
||||
# print(f"Chunk添加成功! ID: {chunk.id}")
|
||||
#dataset.upload_documents([{"display_name": "1.txt", "blob": open('1.txt',"rb").read()}])
|
||||
|
||||
# 查询所有知识库
|
||||
|
Reference in New Issue
Block a user