Throw better exceptions when connecting to non-existent chroma server

Author: CodeWithKyrian

src/Generated/ChromaApiClient.php

@@ -5,6 +5,7 @@
 
 namespace Codewithkyrian\ChromaDB\Generated;
 
+use Codewithkyrian\ChromaDB\Generated\Exceptions\ChromaConnectionException;
 use Codewithkyrian\ChromaDB\Generated\Exceptions\ChromaException;
 use Codewithkyrian\ChromaDB\Generated\Models\Collection;
 use Codewithkyrian\ChromaDB\Generated\Models\Database;
@@ -21,6 +22,7 @@
 use Codewithkyrian\ChromaDB\Generated\Responses\GetItemsResponse;
 use Codewithkyrian\ChromaDB\Generated\Responses\QueryItemsResponse;
 use GuzzleHttp\Client;
+use GuzzleHttp\Exception\ConnectException;
 use GuzzleHttp\Exception\RequestException;
 use Psr\Http\Client\ClientExceptionInterface;
 
@@ -318,6 +320,13 @@ public function reset(): bool
 
     private function handleChromaApiException(\Exception|ClientExceptionInterface $e): void
     {
+        if ($e instanceof ConnectException) {
+            $context = $e->getHandlerContext();
+            $message = $context['error'] ?? $e->getMessage();
+            $code = $context['errno'] ?? $e->getCode();
+            throw new ChromaConnectionException($message, $code);
+        }
+
         if ($e instanceof RequestException) {
             $errorString = $e->getResponse()->getBody()->getContents();
 

src/Generated/Exceptions/ChromaConnectionException.php

@@ -0,0 +1,11 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Codewithkyrian\ChromaDB\Generated\Exceptions;
+
+
+class ChromaConnectionException extends ChromaException
+{
+
+}

tests/ChromaDB.php

@@ -5,6 +5,7 @@
 use Codewithkyrian\ChromaDB\Client;
 use Codewithkyrian\ChromaDB\ChromaDB;
 use Codewithkyrian\ChromaDB\Generated\Exceptions\ChromaAuthorizationException;
+use Codewithkyrian\ChromaDB\Generated\Exceptions\ChromaConnectionException;
 
 it('can connect to a normal chroma server', function () {
     $client = ChromaDB::client();
@@ -42,3 +43,10 @@
         ->withPort(8001)
         ->connect();
 })->throws(ChromaAuthorizationException::class);
+
+it('throws a connection exception when connecting to a non-existent chroma server', function () {
+    ChromaDB::factory()
+        ->withHost('http://localhost')
+        ->withPort(8002)
+        ->connect();
+})->throws(ChromaConnectionException::class);