Skip to content

Commit 5b5430b

Browse files
authored
fix: pydantic
1 parent d81eaf3 commit 5b5430b

1 file changed

Lines changed: 11 additions & 8 deletions

File tree

reai_toolkit/__init__.py

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -5,13 +5,16 @@
55
import importlib
66

77
def delete_module(module_name):
8-
to_delete = []
9-
for module in sys.modules:
10-
if module.startswith(module_name):
11-
to_delete.append(module)
12-
for module_to_delete in to_delete:
13-
log_info(f"RevEng.AI | Deleting module: {module_to_delete}")
14-
del sys.modules[module_to_delete]
8+
try:
9+
to_delete = []
10+
for module in sys.modules:
11+
if module.startswith(module_name):
12+
to_delete.append(module)
13+
for module_to_delete in to_delete:
14+
log_info(f"RevEng.AI | Deleting module: {module_to_delete}")
15+
del sys.modules[module_to_delete]
16+
except Exception as e:
17+
log_error(f"RevEng.AI | Error deleting module: {e}")
1518

1619
def import_module(module_name):
1720
try:
@@ -46,7 +49,7 @@ def import_module(module_name):
4649
log_info(f"RevEng.AI | Vendor directory contains {len(contents)} items")
4750
log_info(f"RevEng.AI | Sample contents: {', '.join(contents[:5])}")
4851

49-
modules = ["urllib3", "certifi", "revengai"]
52+
modules = ["urllib3", "certifi", "revengai", "pydantic", "pydantic_core"]
5053

5154
for module in modules:
5255
delete_module(module)

0 commit comments

Comments
 (0)