Update LaasApiClient class to include the chat_completions method for making API calls to the chat completions endpoint. This allows users to interact with the chat completions API and receive responses. Also, remove the unused 'hash' parameter from the call_llm_preset method.

libs/laas.py

@@ -1,5 +1,5 @@
 import os
-from typing import Any, Dict, Optional
+from typing import Any, Dict, List, Optional
 
 import requests
 
@@ -11,7 +11,6 @@ class LaasApiClient:
     Usage:
     client = LaasApiClient()
     response = client.call_llm_preset(
-        hash="your_hash",
         params={},
         model="your_model",
         messages=[],
@@ -22,6 +21,10 @@ class LaasApiClient:
         source_count=0
     )
     print(response)
+
+    # For chat completions:
+    chat_response = client.chat_completions(messages=[{"role": "user", "content": "Hello!"}])
+    print(chat_response)
     """
 
     BASE_URL = "https://api-laas.wanted.co.kr"
@@ -31,22 +34,28 @@ class LaasApiClient:
         base_url: str = BASE_URL,
         api_key: Optional[str] = None,
         project: Optional[str] = None,
+        hash: Optional[str] = None,
     ):
         self.base_url = base_url
+        self.hash = hash or os.environ.get("LAAS_HASH")
         self.headers = {
             "Content-Type": "application/json",
             "apiKey": api_key or os.environ.get("LAAS_API_KEY"),
             "project": project or os.environ.get("LAAS_PROJECT"),
         }
         if not self.headers["apiKey"]:
-            raise ValueError("API key is required")
+            raise ValueError("API key is required to use the LAAS API.")
         if not self.headers["project"]:
-            raise ValueError("Project is required")
+            raise ValueError("Project is required to use the LAAS API.")
+        if not self.hash:
+            raise ValueError("Hash is required to use the LAAS API.")
 
     def _make_api_call(self, endpoint: str, payload: Dict[str, Any]) -> Dict[str, Any]:
         """Make an API call to the specified endpoint with the given payload."""
         url = f"{self.base_url}/api/{endpoint}"
-        response = requests.post(url, headers=self.headers, json=payload)
+        response = requests.post(
+            url, headers=self.headers, json={"hash": self.hash, **payload}
+        )
         response.raise_for_status()  # Raise an exception for HTTP errors
         return response.json()
 
@@ -63,3 +72,22 @@ class LaasApiClient:
         """
         endpoint = "preset/complete" if complete else "preset"
         return self._make_api_call(endpoint, kwargs)
+
+    def chat_completions(
+        self, messages: List[Dict[str, str]], **kwargs
+    ) -> Dict[str, Any]:
+        """
+        Call the chat completions API.
+
+        Args:
+            messages (List[Dict[str, str]]): List of message dictionaries.
+            **kwargs: Additional keyword arguments for the API call.
+
+        Returns:
+            Dict[str, Any]: The JSON response from the API.
+        """
+        payload = {
+            "messages": messages,
+            **kwargs,
+        }
+        return self._make_api_call("preset/chat/completions", payload)