[HN-504] feat: add suggested prompts endpoint

hive_agent_client/client.py

@@ -19,6 +19,7 @@
 )
 from hive_agent_client.files import upload_files, list_files, delete_file, rename_file
 from hive_agent_client.tools import install_tools
+from hive_agent_client.prompts import sample_prompts
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -255,6 +256,18 @@ async def install_tools(self, tools: List[Dict[str, Any]]) -> Dict:
             logger.error(f"Failed to install tools: {e}")
             raise Exception(f"Failed to install tools: {e}")
 
+    async def sample_prompts(self) -> Dict:
+        """
+        List all sample prompts configured for the agent.
+
+        :return: A dictionary with a list of the suggested prompts.
+        """
+        try:
+            return await sample_prompts(self.http_client, self.base_url)
+        except Exception as e:
+            logger.error(f"Failed to list the suggested prompts: {e}")
+            raise Exception(f"Failed to list the suggested prompts: {e}")
+
     async def close(self):
         """
         Close the HTTP client session.

hive_agent_client/prompts/__init__.py

@@ -0,0 +1 @@
+from .sample_prompts import sample_prompts

hive_agent_client/prompts/sample_prompts.py

@@ -0,0 +1,39 @@
+import httpx
+import logging
+import os
+import sys
+
+
+def get_log_level():
+    HIVE_AGENT_LOG_LEVEL = os.getenv("HIVE_AGENT_LOG_LEVEL", "INFO").upper()
+    return getattr(logging, HIVE_AGENT_LOG_LEVEL, logging.INFO)
+
+
+logging.basicConfig(stream=sys.stdout, level=get_log_level())
+logging.getLogger().addHandler(logging.StreamHandler(stream=sys.stdout))
+
+logger = logging.getLogger()
+logger.setLevel(get_log_level())
+
+
+async def sample_prompts(http_client: httpx.AsyncClient, base_url: str) -> dict:
+    """
+    Lists all sample prompts for the agent
+
+    :param http_client: An HTTP client for sending requests.
+    :param base_url: The base URL of the Hive Agent API.
+    :return: A dictionary containing the list of all the suggested prompts for starting an interaction with the agent.
+    :raises Exception: If the HTTP request fails or the API returns an error response.
+    """
+    endpoint = "/sample_prompts/"
+    url = f"{base_url}{endpoint}"
+
+    try:
+        logger.debug(f"Listing sample prompts at {url}")
+        response = await http_client.get(url)
+        response.raise_for_status()
+        logger.debug(f"Response for sample prompts at {url}: {response.json()}")
+        return response.json()
+    except httpx.HTTPStatusError as e:
+        logging.error(f"Failed to get sample prompts: {e}")
+        raise Exception(f"Failed to get sample prompts: {e.response.text}") from e