fix: update thrown exception

vertexcover-io · Dec 26, 2024 · 64ff02f · 64ff02f
1 parent d69c6bd
commit 64ff02f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 8 deletions.
diff --git a/python_client/locatr/_locatr.py b/python_client/locatr/_locatr.py
@@ -21,7 +21,6 @@
 from locatr.exceptions import (
     FailedToRetrieveLocatr,
     LocatrInitialHandshakeFailed,
-    LocatrOutputMessageValidationFailed,
     LocatrSocketIsNone,
     SocketInitializationError,
 )
@@ -132,8 +131,8 @@ def get_locatr(self, user_req: str):
             if not output.status == OutputStatus.OK:
                 raise FailedToRetrieveLocatr(output.error)
             return output.output
-        except ValidationError as e:
-            raise LocatrOutputMessageValidationFailed(str(e))
+        except ValidationError:
+            raise FailedToRetrieveLocatr("Internal message validation failed.")
 
     async def get_locatr_async(self, user_req: str) -> str:
         return await asyncio.to_thread(self.get_locatr, user_req)

diff --git a/python_client/locatr/exceptions.py b/python_client/locatr/exceptions.py
@@ -27,11 +27,6 @@ def __init__(self, msg) -> None:
         super().__init__(msg)
 
 
-class LocatrOutputMessageValidationFailed(Exception):
-    def __init__(self, msg: str) -> None:
-        super().__init__(f"Failed to validate message locatr output: {msg}")
-
-
 class LocatrSocketError(Exception):
     def __init__(self, msg: str) -> None:
         super().__init__(msg)