refactor: update operations that were not using traits

Author: heyjorgedev

src/Operations/QueryDataOperation.php

@@ -6,18 +6,19 @@
 use Upstash\Vector\DataQuery;
 use Upstash\Vector\DataQueryResult;
 use Upstash\Vector\Operations\Concerns\AssertsApiResponseErrors;
+use Upstash\Vector\Operations\Concerns\MapsVectorMatches;
 use Upstash\Vector\Transporter\ContentType;
 use Upstash\Vector\Transporter\Method;
 use Upstash\Vector\Transporter\TransporterRequest;
 use Upstash\Vector\Transporter\TransporterResponse;
-use Upstash\Vector\VectorMatch;
 
 /**
  * @internal
  */
 final readonly class QueryDataOperation
 {
     use AssertsApiResponseErrors;
+    use MapsVectorMatches;
 
     public function __construct(
         private string $namespace,
@@ -49,16 +50,7 @@ public function query(DataQuery $query): DataQueryResult
     private function transformResponse(TransporterResponse $response): DataQueryResult
     {
         $data = json_decode($response->data, true);
-
-        $results = array_map(function (array $result) {
-            return new VectorMatch(
-                id: $result['id'],
-                score: $result['score'],
-                vector: $result['vector'] ?? [],
-                data: $result['data'] ?? '',
-                metadata: $result['metadata'] ?? [],
-            );
-        }, $data['result'] ?? []);
+        $results = array_map(fn (array $result) => $this->mapVectorMatch($result), $data['result'] ?? []);
 
         return new DataQueryResult($results);
     }

src/Operations/UpsertDataOperation.php

@@ -4,17 +4,18 @@
 
 use Upstash\Vector\Contracts\TransporterInterface;
 use Upstash\Vector\DataUpsert;
-use Upstash\Vector\Exceptions\OperationFailedException;
+use Upstash\Vector\Operations\Concerns\AssertsApiResponseErrors;
 use Upstash\Vector\Transporter\ContentType;
 use Upstash\Vector\Transporter\Method;
 use Upstash\Vector\Transporter\TransporterRequest;
-use Upstash\Vector\Transporter\TransporterResponse;
 
 /**
  * @internal
  */
 final readonly class UpsertDataOperation
 {
+    use AssertsApiResponseErrors;
+
     public function __construct(private string $namespace, private TransporterInterface $transporter) {}
 
     public function upsert(DataUpsert $upsert): void
@@ -56,12 +57,4 @@ private function createRequest(array $data): TransporterRequest
             data: $data,
         );
     }
-
-    private function assertResponse(TransporterResponse $response): void
-    {
-        if ($response->statusCode === 422) {
-            $data = json_decode($response->data, true);
-            throw new OperationFailedException($data['error']);
-        }
-    }
 }