ragflow-sdk-myfbee-test 0.27.2.dev0__tar.gz → 0.27.2.dev2__tar.gz

{ragflow_sdk_myfbee_test-0.27.2.dev0 → ragflow_sdk_myfbee_test-0.27.2.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragflow-sdk-myfbee-test
-Version: 0.27.2.dev0
+Version: 0.27.2.dev2
 Summary: Python client sdk of [RAGFlow](https://github.com/infiniflow/ragflow). RAGFlow is an open-source RAG (Retrieval-Augmented Generation) engine based on deep document understanding.
 Author-email: Zhichang Yu <yuzhichang@gmail.com>
 License: Apache License, Version 2.0

{ragflow_sdk_myfbee_test-0.27.2.dev0 → ragflow_sdk_myfbee_test-0.27.2.dev2}/pyproject.toml

@@ -1,6 +1,7 @@
 [project]
+#name = "ragflow-sdk"
 name = "ragflow-sdk-myfbee-test"
-version = "0.27.2-dev"
+version = "0.27.2.dev2"
 description = "Python client sdk of [RAGFlow](https://github.com/infiniflow/ragflow). RAGFlow is an open-source RAG (Retrieval-Augmented Generation) engine based on deep document understanding."
 authors = [{ name = "Zhichang Yu", email = "yuzhichang@gmail.com" }]
 license = { text = "Apache License, Version 2.0" }

{ragflow_sdk_myfbee_test-0.27.2.dev0 → ragflow_sdk_myfbee_test-0.27.2.dev2}/ragflow_sdk/modules/dataset.py

@@ -119,17 +119,10 @@ class DataSet(Base):
         finished = []
         while pending:
             for doc_id in list(pending):
-
-                def fetch_doc(doc_id: str) -> Document | None:
-                    try:
-                        docs = self.list_documents(id=doc_id)
-                        return docs[0] if docs else None
-                    except Exception:
-                        return None
-
-                doc = fetch_doc(doc_id)
-                if doc is None:
-                    continue
+                docs = self.list_documents(id=doc_id)
+                if not docs:
+                    raise RuntimeError(f"Document {doc_id} not found while waiting for parsing to finish")
+                doc = docs[0]
                 if isinstance(doc.run, str) and doc.run.upper() in terminal_states:
                     finished.append((doc_id, doc.run, doc.chunk_count, doc.token_count))
                     pending.discard(doc_id)
@@ -149,7 +142,7 @@ class DataSet(Base):
     def parse_documents(self, document_ids):
         try:
             self.async_parse_documents(document_ids)
-            self._get_documents_status(document_ids)
+            return self._get_documents_status(document_ids)
         except KeyboardInterrupt:
             self.async_cancel_parse_documents(document_ids)
 

{ragflow_sdk_myfbee_test-0.27.2.dev0 → ragflow_sdk_myfbee_test-0.27.2.dev2}/ragflow_sdk/modules/session.py

@@ -149,14 +149,17 @@ class Session(Base):
     def _structure_answer(self, json_data):
         answer = ""
         event = None
+        message_id = None
         if self.__session_type == "agent":
             event = json_data.get("event")
+            message_id = json_data.get("message_id")
             json_data = json_data["data"]
             answer = json_data.get("content", "")
         elif self.__session_type == "chat":
             answer = json_data["answer"]
+            message_id = json_data.get("id")
         reference = json_data.get("reference", {})
-        temp_dict = {"content": answer, "role": "assistant"}
+        temp_dict = {"id": message_id, "content": answer, "role": "assistant"}
         if reference and "chunks" in reference:
             chunks = reference["chunks"]
             if isinstance(chunks, dict):

{ragflow_sdk_myfbee_test-0.27.2.dev0 → ragflow_sdk_myfbee_test-0.27.2.dev2}/ragflow_sdk_myfbee_test.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragflow-sdk-myfbee-test
-Version: 0.27.2.dev0
+Version: 0.27.2.dev2
 Summary: Python client sdk of [RAGFlow](https://github.com/infiniflow/ragflow). RAGFlow is an open-source RAG (Retrieval-Augmented Generation) engine based on deep document understanding.
 Author-email: Zhichang Yu <yuzhichang@gmail.com>
 License: Apache License, Version 2.0