Mark SDK disconnected (#46)

.gitignore

@@ -155,3 +155,4 @@ cython_debug/
 # Other
 
 test*
+local*

.vscode/settings.json

@@ -0,0 +1,15 @@
+{
+    "[python]": {
+        "editor.defaultFormatter": "ms-python.black-formatter",
+        "editor.formatOnSave": true,
+        "editor.codeActionsOnSave": {
+            "source.organizeImports": "explicit"
+        }
+    },
+    "black-formatter.args": [],
+    "isort.args": [
+        "--profile",
+        "black"
+    ],
+    "python.analysis.typeCheckingMode": "basic",
+}

aiochroma/aiochroma.py

@@ -466,3 +466,9 @@ def identity(self) -> dict[str, str]:
         """Chroma versions"""
 
         return self._connection.identity
+
+    @property
+    def connected(self) -> bool:
+        """Connection state"""
+
+        return self._connection.connected

aiochroma/connection.py

@@ -8,8 +8,15 @@
 
 import aiohttp
 
-from .const import CREDENTIALS, DEFAULT_PORT, DEFAULT_SLEEP, HEADERS, URL, URL_MAIN
-from .error import ChromaResultError
+from aiochroma.const import (
+    CREDENTIALS,
+    DEFAULT_PORT,
+    DEFAULT_SLEEP,
+    HEADERS,
+    URL,
+    URL_MAIN,
+)
+from aiochroma.error import ChromaError, ChromaResultError
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -35,6 +42,19 @@ def __init__(
         self._connected: bool = False
         self._sid: int | None = None
 
+    ### ------------------------
+    ### Service methods -->
+    ### ------------------------
+
+    def _mark_disconnected(self) -> None:
+        """Mark connection as disconnected."""
+
+        self._connected = False
+
+    ### ------------------------
+    ### <-- Service methods
+    ### ------------------------
+
     ### SEND REQUESTS ->
 
     async def async_request(
@@ -46,7 +66,10 @@ async def async_request(
     ) -> dict[str, Any]:
         """Send a request"""
 
-        json_body = {}
+        # Check that we are connected
+        if not self._connected and endpoint != URL_MAIN:
+            if not await self.async_connect():
+                raise ChromaError("Cannot connect to Chroma SDK")
 
         url = URL.format(
             self._host,
@@ -55,17 +78,29 @@ async def async_request(
         )
 
         try:
-            async with method(url=url, headers=HEADERS, data=payload, ssl=True) as r:
-                json_body = await r.json()
+            async with method(
+                url=url, headers=HEADERS, data=payload, ssl=True
+            ) as response:
+                responce_status = response.status
+                if responce_status == 404:
+                    raise ChromaError("Chroma SDK is not available")
+
+                json_body = await response.json()
+
+                if "result" in json_body and json_body["result"] != 0:
+                    raise ChromaResultError(json_body["result"])
+
+                await self.async_sleep(interval)
 
-            if "result" in json_body and json_body["result"] != 0:
-                raise ChromaResultError(json_body["result"])
+                return json_body
 
-            await self.async_sleep(interval)
+        except aiohttp.ClientConnectorError as ex:
+            self._mark_disconnected()
+            raise ChromaError("Cannot connect to the Chroma SDK") from ex
 
-            return json_body
         except Exception as ex:
-            raise ex
+            self._mark_disconnected()
+            raise ChromaError("Error communicating with the Chroma SDK") from ex
 
     async def async_delete(
         self,