move authorization to use Bearer with api key

Author: matank-vendi

src/vendi/completions/completions.py

@@ -97,7 +97,7 @@ def create(
             data["stop"] = stop
 
         _res = self.__client.post(
-            uri=f"/v1/chat/completions/",
+            uri=f"/v1/chat/completions",
             json_data=data,
             headers=extra_headers
         )
@@ -154,7 +154,7 @@ async def acreate(
             data["stop"] = stop
 
         _res = await self.__aclient.post(
-            path=f"/v1/chat/completions/",
+            path=f"/v1/chat/completions",
             json=data,
             headers=extra_headers
         )
@@ -263,7 +263,7 @@ async def arun_batch_job(
         poll_interval: int = 5,
     ) -> BatchInference:
         _res = await self.__aclient.post(
-            path="/platform/v1/inference/batch/",
+            path="/platform/v1/inference/batch",
             json={
                 "dataset_id": str(dataset_id),
                 "model_parameters": [{**i.model_dump(), **i.model_extra} for i in model_parameters]

src/vendi/core/ahttp_client.py

@@ -9,11 +9,8 @@ def __init__(self, base_url: str, timeout: int = 60):
             "content-type": "application/json",
         }
 
-    def set_auth_header(self, access_key: str | None = None, jwt: str | None = None):
-        if access_key:
-            self.headers["x-api-key"] = access_key
-        if jwt:
-            self.headers["Authorization"] = f"Bearer {jwt}"
+    def set_auth_header(self, api_key: str):
+        self.headers["Authorization"] = f"Bearer {api_key}"
 
     async def get(self, path: str, params: dict = None) -> dict:
         url = self.base_url + path

src/vendi/core/http_client.py

@@ -46,7 +46,8 @@ def post(
 
     def _set_headers(self, headers: Dict[str, str]) -> Dict[str, str]:
         _headers = {
-            "x-api-key": self.__api_key,
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {self.__api_key}",
         }
         if headers:
             _headers.update(headers)