From 063474ad8b01de1f2f27711d4a39d6b9119ace0b Mon Sep 17 00:00:00 2001
From: Hetavi Shah <hetavi.shah@bluexkye.com>
Date: Mon, 1 Dec 2025 17:37:36 +0530
Subject: [PATCH] [OND211-2329]: Updated Cryto imports.

---
 admin/client/admin_client.py                                | 6 +++---
 api/utils/crypt.py                                          | 4 ++--
 test/testcases/conftest.py                                  | 4 ++--
 .../test_http_api/test_user_management/conftest.py          | 4 ++--
 4 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/admin/client/admin_client.py b/admin/client/admin_client.py
index cfbd5ff78..4b7880090 100644
--- a/admin/client/admin_client.py
+++ b/admin/client/admin_client.py
@@ -18,10 +18,10 @@ import argparse
 import base64
 from cmd import Cmd
 
-from Cryptodome.PublicKey import RSA  # type: ignore
-from Cryptodome.Cipher import PKCS1_v1_5 as Cipher_pkcs1_v1_5  # type: ignore
+from Crypto.PublicKey import RSA
+from Crypto.Cipher import PKCS1_v1_5 as Cipher_pkcs1_v1_5
 from typing import Dict, List, Any
-from lark import Lark, Transformer, Tree  # type: ignore
+from lark import Lark, Transformer, Tree
 import requests
 import getpass
 
diff --git a/api/utils/crypt.py b/api/utils/crypt.py
index 454d29298..174ca3568 100644
--- a/api/utils/crypt.py
+++ b/api/utils/crypt.py
@@ -17,8 +17,8 @@
 import base64
 import os
 import sys
-from Cryptodome.PublicKey import RSA  # type: ignore
-from Cryptodome.Cipher import PKCS1_v1_5 as Cipher_pkcs1_v1_5  # type: ignore
+from Cryptodome.PublicKey import RSA
+from Cryptodome.Cipher import PKCS1_v1_5 as Cipher_pkcs1_v1_5
 from common.file_utils import get_project_base_directory
 
 
diff --git a/test/testcases/conftest.py b/test/testcases/conftest.py
index ef378c79b..931bed069 100644
--- a/test/testcases/conftest.py
+++ b/test/testcases/conftest.py
@@ -21,8 +21,8 @@ from typing import Any, Dict, Optional
 
 import pytest
 import requests
-from Cryptodome.Cipher import PKCS1_v1_5 as Cipher_pkcs1_v1_5  # type: ignore
-from Cryptodome.PublicKey import RSA  # type: ignore
+from Crypto.Cipher import PKCS1_v1_5 as Cipher_pkcs1_v1_5
+from Crypto.PublicKey import RSA
 
 from configs import EMAIL, HOST_ADDRESS, VERSION, ZHIPU_AI_API_KEY
 from libs.auth import RAGFlowWebApiAuth
diff --git a/test/testcases/test_http_api/test_user_management/conftest.py b/test/testcases/test_http_api/test_user_management/conftest.py
index b0e3398a8..eb242de0c 100644
--- a/test/testcases/test_http_api/test_user_management/conftest.py
+++ b/test/testcases/test_http_api/test_user_management/conftest.py
@@ -25,8 +25,8 @@ from pathlib import Path
 from typing import Any
 
 import pytest
-from Cryptodome.Cipher import PKCS1_v1_5 as Cipher_pkcs1_v1_5  # type: ignore
-from Cryptodome.PublicKey import RSA  # type: ignore
+from Crypto.Cipher import PKCS1_v1_5 as Cipher_pkcs1_v1_5
+from Crypto.PublicKey import RSA
 
 from common import get_user_info
 from libs.auth import RAGFlowWebApiAuth