# type: ignore

api/core/indexing_runner.py

@@ -672,7 +672,7 @@ class IndexingRunner:
 
         if extra_update_params:
             update_params.update(extra_update_params)
-
+        # type: ignore
         db.session.query(DatasetDocument).filter_by(id=document_id).update(update_params)
         db.session.commit()
 

api/core/rag/extractor/notion_extractor.py

@@ -331,9 +331,10 @@ class NotionExtractor(BaseExtractor):
         last_edited_time = self.get_notion_last_edited_time()
         data_source_info = document_model.data_source_info_dict
         data_source_info["last_edited_time"] = last_edited_time
-        update_params = {DocumentModel.data_source_info: json.dumps(data_source_info)}
+        # type: ignore
-
+        db.session.query(DocumentModel).filter_by(id=document_model.id).update(
-        db.session.query(DocumentModel).filter_by(id=document_model.id).update(update_params)
+            {DocumentModel.data_source_info: json.dumps(data_source_info)}
+        )
         db.session.commit()
 
     def get_notion_last_edited_time(self) -> str:

api/models/account.py

@@ -144,7 +144,7 @@ class Account(UserMixin, Base):
             return
 
         tenant, join = tenant_account_join
-        self.role = join.role
+        self.role = TenantAccountRole(join.role)
         self._current_tenant = tenant
 
     @property

api/schedule/clean_unused_datasets_task.py

@@ -99,9 +99,8 @@ def clean_unused_datasets_task():
                     index_processor.clean(dataset, None)
 
                     # update document
-                    update_params = {Document.enabled: False}
+                    # type: ignore
-
+                    db.session.query(Document).filter_by(dataset_id=dataset.id).update({Document.enabled: False})
-                    db.session.query(Document).filter_by(dataset_id=dataset.id).update(update_params)
                     db.session.commit()
                     click.echo(click.style("Cleaned unused dataset {} from db success!".format(dataset.id), fg="green"))
                 except Exception as e:
@@ -176,9 +175,8 @@ def clean_unused_datasets_task():
                         index_processor.clean(dataset, None)
 
                         # update document
-                        update_params = {Document.enabled: False}
+                        # type: ignore
-
+                        db.session.query(Document).filter_by(dataset_id=dataset.id).update({Document.enabled: False})
-                        db.session.query(Document).filter_by(dataset_id=dataset.id).update(update_params)
                         db.session.commit()
                         click.echo(
                             click.style("Cleaned unused dataset {} from db success!".format(dataset.id), fg="green")

api/services/account_service.py

@@ -997,7 +997,7 @@ class TenantService:
             .filter(TenantAccountJoin.tenant_id == tenant.id, TenantAccountJoin.account_id == account.id)
             .first()
         )
-        return join.role if join else None
+        return TenantAccountRole(join.role) if join else None
 
     @staticmethod
     def get_tenant_count() -> int:

api/services/billing_service.py

@@ -82,7 +82,7 @@ class BillingService:
         if not join:
             raise ValueError("Tenant account join not found")
 
-        if not TenantAccountRole.is_privileged_role(join.role):
+        if not TenantAccountRole.is_privileged_role(TenantAccountRole(join.role)):
             raise ValueError("Only team owner or team admin can perform this action")
 
     @classmethod

api/services/dataset_service.py

@@ -214,7 +214,9 @@ class DatasetService:
         dataset.created_by = account.id
         dataset.updated_by = account.id
         dataset.tenant_id = tenant_id
+        # type: ignore
         dataset.embedding_model_provider = embedding_model.provider if embedding_model else None
+        # type: ignore
         dataset.embedding_model = embedding_model.model if embedding_model else None
         dataset.retrieval_model = retrieval_model.model_dump() if retrieval_model else None
         dataset.permission = permission or DatasetPermissionEnum.ONLY_ME
@@ -1539,6 +1541,7 @@ class DocumentService:
         db.session.add(document)
         db.session.commit()
         # update document segment
+        # type: ignore
         update_params = {DocumentSegment.status: "re_segment"}
         db.session.query(DocumentSegment).filter_by(document_id=document.id).update(update_params)
         db.session.commit()