fix(storage): fix some more issues

Author: o-santi

src/storage/src/storage3/_async/client.py

@@ -84,7 +84,7 @@ def from_(self, id: str) -> AsyncBucketProxy:
 
     def vectors(self) -> AsyncStorageVectorsClient:
         return AsyncStorageVectorsClient(
-            url=self._base_url,
+            url=self._base_url.joinpath("v1", "vector"),
             headers=self._headers,
             session=self.session,
         )

src/storage/src/storage3/_async/request.py

@@ -19,10 +19,12 @@ async def send(
         body: JSON = None,
         query_params: Optional[QueryParams] = None,
     ) -> Response:
-        return await self._session.request(
+        data = await self._session.request(
             method=http_method,
             json=body,
             url=str(self._base_url.joinpath(*path)),
             headers=self.headers,
             params=query_params or QueryParams(),
         )
+        data.raise_for_status()
+        return data

src/storage/src/storage3/_async/vectors.py

@@ -32,12 +32,14 @@ def with_metadata(self, **data: JSON) -> JSON:
 
     async def create_index(
         self,
+        index_name: str,
         dimension: int,
         distance_metric: DistanceMetric,
         data_type: str,
         metadata: Optional[MetadataConfiguration] = None,
     ) -> None:
         body = self.with_metadata(
+            indexName=index_name,
             dimension=dimension,
             distanceMetric=distance_metric,
             dataType=data_type,
@@ -90,7 +92,7 @@ def with_metadata(self, **data: JSON) -> JSON:
         }
 
     async def put(self, vectors: List[VectorObject]) -> None:
-        body = self.with_metadata(vectors=list(dict(v) for v in vectors))
+        body = self.with_metadata(vectors=[v.as_json() for v in vectors])
         await self._request.send(http_method="POST", path=["PutVectors"], body=body)
 
     async def get(
@@ -159,3 +161,14 @@ def __init__(self, url: URL, headers: Headers, session: AsyncClient) -> None:
 
     def from_(self, bucket_name: str) -> AsyncVectorBucketScope:
         return AsyncVectorBucketScope(self._request, bucket_name)
+
+    async def create_bucket(self, bucket_name: str) -> None:
+        body = {"vectorBucketName": bucket_name}
+        await self._request.send(
+            http_method="POST", path=["CreateVectorBucket"], body=body
+        )
+
+    # async def get_bucket(self, bucket_name: str) -> GetBucketResponse:
+    #     body = { 'vectorBucketName': bucket_name }
+    #     data = await self._request.send(http_method='POST', path=['GetVectorBucket'], body=body)
+    #     return GetVectorsResponse.model_validate(data.content)

src/storage/src/storage3/types.py

@@ -207,6 +207,9 @@ class VectorObject(BaseModel):
     data: VectorData
     metadata: Optional[dict[str, Any]] = None
 
+    def as_json(self) -> JSON:
+        return {"key": self.key, "data": dict(self.data), "metadata": self.metadata}
+
 
 class VectorMatch(BaseModel):
     key: str