feat: add authentication flow in curator client (#606)

Author: kartik4949

src/bespokelabs/curator/client.py

@@ -39,12 +39,21 @@ def __init__(self, hosted: bool = False) -> None:
         self._hosted = self._hosted or hosted
         self.semaphore = asyncio.Semaphore(N_CONCURRENT_VIEWER_REQUESTS)
         self._async_client = None
+        self.api_key = os.environ.get("BESPOKE_API_KEY")
+        if self.api_key:
+            logger.info("Authenticated with Curator API key")
+        self._headers = {"Authorization": f"Bearer {self.api_key}"} if self.api_key else {}
 
     @property
     def session(self):
         """Get the session ID."""
         return self._session
 
+    @property
+    def authenticated(self):
+        """Check if the client is authenticated."""
+        return self._authenticated
+
     @property
     def hosted(self):
         """Check if the client is hosted."""
@@ -67,7 +76,7 @@ def create_session(self, metadata: t.Dict, session_id: str | None = None) -> str
             return self.session
         metadata.update({"status": _SessionStatus.STARTED})
 
-        response = requests.post(f"{BASE_CLIENT_URL}/sessions", json=metadata)
+        response = requests.post(f"{BASE_CLIENT_URL}/sessions", json=metadata, headers=self._headers)
 
         if response.status_code == 200:
             self._session = response.json().get("session_id")
@@ -79,7 +88,7 @@ def create_session(self, metadata: t.Dict, session_id: str | None = None) -> str
 
     async def _update_state(self):
         async with httpx.AsyncClient() as client:
-            response = await client.put(f"{BASE_CLIENT_URL}/sessions/{self.session}", json={"status": self._state})
+            response = await client.put(f"{BASE_CLIENT_URL}/sessions/{self.session}", json={"status": self._state}, headers=self._headers)
         if response.status_code != 200:
             logger.debug(f"Failed to update session status: {response.status_code}, {response.text}")
 
@@ -113,7 +122,7 @@ async def stream_response(self, response_data: str, idx: int):
 
         response_data = json.dumps({"response_data": response_data})
         async with self.semaphore:
-            response = await self._async_client.post(f"{BASE_CLIENT_URL}/sessions/{self.session}/responses/{idx}", data=response_data)
+            response = await self._async_client.post(f"{BASE_CLIENT_URL}/sessions/{self.session}/responses/{idx}", data=response_data, headers=self._headers)
 
         if response.status_code != 200:
             logger.debug(f"Failed to stream response to curator Viewer: {response.status_code}, {response.text}")