fix error message

Author: Davidyz

src/vectorcode/database/chroma0.py

@@ -183,11 +183,15 @@ async def get_client(self, configs: Config, need_lock: bool = True):
                 process = await _start_server(configs)
                 is_bundled = True
 
-            self.__clients[project_root] = _Chroma0ClientModel(
-                client=await self._create_client(configs),
-                is_bundled=is_bundled,
-                process=process,
-            )
+            try:
+                self.__clients[project_root] = _Chroma0ClientModel(
+                    client=await self._create_client(configs),
+                    is_bundled=is_bundled,
+                    process=process,
+                )
+            except httpx.RemoteProtocolError as e:  # pragma: nocover
+                e.add_note(f"Please verify that {url} is a working chromadb server.")
+                raise
         lock = None
         if self.__clients[project_root].is_bundled and need_lock:
             lock = LockManager().get_lock(str(db_path))

src/vectorcode/main.py

@@ -4,8 +4,6 @@
 import sys
 import traceback
 
-import httpx
-
 from vectorcode import __version__
 from vectorcode.cli_utils import (
     CliAction,
@@ -107,12 +105,8 @@ async def async_main():
                 from vectorcode.subcommands import files
 
                 return_val = await files(final_configs)
-    except Exception as e:
+    except Exception:
         return_val = 1
-        if isinstance(e, httpx.RemoteProtocolError):  # pragma: nocover
-            e.add_note(
-                f"Please verify that {final_configs.db_url} is a working chromadb server."
-            )
         logger.error(traceback.format_exc())
     finally:
         return return_val