fix: IntegrityError import wrong (#1868)

2024-01-03 09:35:03 +08:00
parent 4aed1fe8a8
commit 3b668c0bb1
1 changed files with 1 additions and 1 deletions
--- a/api/core/provider_manager.py
+++ b/api/core/provider_manager.py
@@ -3,7 +3,7 @@ from collections import defaultdict
 from json import JSONDecodeError
 from typing import Optional

-from psycopg2 import IntegrityError
+from sqlalchemy.exc import IntegrityError

 from core.entities.model_entities import DefaultModelEntity, DefaultModelProviderEntity
 from core.entities.provider_configuration import ProviderConfigurations, ProviderConfiguration, ProviderModelBundle