chore: Add stream method to LaasApiClient for streaming chat completions API response

libs/laas.py

@@ -1,5 +1,5 @@
 import os
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List, Optional, Generator
 
 import requests
 
@@ -25,6 +25,10 @@ class LaasApiClient:
     # For chat completions:
     chat_response = client.chat_completions(messages=[{"role": "user", "content": "Hello!"}])
     print(chat_response)
+
+    # For streaming:
+    for chunk in client.stream(messages=[{"role": "user", "content": "Tell me a story"}]):
+        print(chunk, end="", flush=True)
     """
 
     BASE_URL = "https://api-laas.wanted.co.kr"
@@ -91,3 +95,32 @@ class LaasApiClient:
             **kwargs,
         }
         return self._make_api_call("preset/chat/completions", payload)
+
+    def stream(
+        self, messages: List[Dict[str, str]], **kwargs
+    ) -> Generator[str, None, None]:
+        """
+        Stream the chat completions API response.
+
+        Args:
+            messages (List[Dict[str, str]]): List of message dictionaries.
+            **kwargs: Additional keyword arguments for the API call.
+
+        Yields:
+            str: Chunks of the streaming response.
+        """
+        url = f"{self.base_url}/api/preset/chat/completions"
+        payload = {
+            "hash": self.hash,
+            "messages": messages,
+            "stream": True,
+            **kwargs,
+        }
+
+        with requests.post(
+            url, headers=self.headers, json=payload, stream=True
+        ) as response:
+            response.raise_for_status()
+            for line in response.iter_lines():
+                if line:
+                    yield line.decode("utf-8")