From 68444ad7ff2037183360562f0956b6b0e6285de4 Mon Sep 17 00:00:00 2001
From: glowz <24627181@qq.com>
Date: Sat, 5 Jul 2025 11:43:50 +0800
Subject: [PATCH] =?UTF-8?q?=E9=87=8D=E6=9E=84=E6=B7=BB=E5=8A=A0=E6=96=87?=
 =?UTF-8?q?=E6=A1=A3=E5=9D=97=E5=8A=9F=E8=83=BD=EF=BC=8C=E6=94=B9=E8=BF=9B?=
 =?UTF-8?q?=E6=96=87=E4=BB=B6=E8=AF=BB=E5=8F=96=E5=92=8C=E9=94=99=E8=AF=AF?=
 =?UTF-8?q?=E5=A4=84=E7=90=86=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/add_chunk.py | 60 +++++++++++++++++++++++++++++++++++++-----------
 1 file changed, 47 insertions(+), 13 deletions(-)

diff --git a/src/add_chunk.py b/src/add_chunk.py
index c167d5a..d5ea5bc 100644
--- a/src/add_chunk.py
+++ b/src/add_chunk.py
@@ -1,20 +1,50 @@
 from tkinter import Tk, StringVar, Label, OptionMenu, Button, filedialog
 from ragflow_sdk import RAGFlow
 
-api_key = "ragflow-I5ZDNjMWNhNTdlMjExZjBiOTEwMzI0ZT"
-base_url = "http://192.168.107.165:8099"
 
+base_url = "http://127.0.0.1"
+api_key = "ragflow-MyMjM2ODE2NThlMTExZjBiMzJlNzY5Mj"
 rag_object = RAGFlow(api_key=api_key, base_url=base_url)
 
+
 def add_chunk_to_document():
-    dataset_id = dataset_var.get()
-    document_id = document_var.get()
-    file_path = filedialog.askopenfilename(filetypes=[("Text files", "*.txt")])
-    
-    if file_path:
-        with open(file_path, 'r') as file:
+    try:
+        dataset_name = dataset_var.get()
+        document_name = document_var.get()
+        if not dataset_name or not document_name:
+            print("请选择数据集和文档！")
+            return
+
+        file_path = filedialog.askopenfilename(filetypes=[("Text files", "*.txt")])
+        if not file_path:
+            print("未选择文件")
+            return
+
+        with open(file_path, 'r', encoding='utf-8') as file:
             content = file.read()
-            rag_object.add_chunk(dataset_id, document_id, content)
+        
+        datasets = rag_object.list_datasets(name=dataset_name)
+        if not datasets:
+            print(f"数据集 '{dataset_name}' 不存在!")
+            return
+        dataset = datasets[0]
+        
+        documents = dataset.list_documents(name=document_name)
+        if not documents:
+            print(f"文档 '{document_name}' 不存在!")
+            return
+        document = documents[0]
+
+        content="test chunk"
+        chunk = document.add_chunk(content=content)
+        print(f"Chunk添加成功! ID: {chunk.id}")
+        
+    except Exception as e:
+        print(f"添加chunk时发生错误: {e}")
+        import traceback
+        traceback.print_exc()
+
+
 
 def update_documents(*args):
     dataset_name = dataset_var.get()
@@ -28,11 +58,15 @@ def update_documents(*args):
 root = Tk()
 root.title("Add Chunk to Document")
 
-dataset_var = StringVar(root)
-document_var = StringVar(root)
-
+# 初始化变量时设置默认值
 datasets = rag_object.list_datasets()
-dataset_menu = OptionMenu(root, dataset_var, *[ds.name for ds in datasets], command=update_documents)
+dataset_names = [ds.name for ds in datasets]
+dataset_var = StringVar(root)
+dataset_var.set(dataset_names[0] if dataset_names else "")
+document_var = StringVar(root)
+document_var.set("")
+
+dataset_menu = OptionMenu(root, dataset_var, *dataset_names, command=update_documents)
 dataset_menu.pack()