Adubatl/model fetching poc

agentstack/_tools/agentmail/__init__.py

@@ -0,0 +1,184 @@
+from agentmail import AgentMail
+from typing import Optional, List
+
+
+client = AgentMail()
+
+
+def list_inboxes(limit: Optional[int] = None, last_key: Optional[str] = None):
+    """
+    List inboxes.
+
+    Args:
+        limit: The maximum number of inboxes to return.
+        last_key: The last key returned from the previous page.
+
+    Returns:
+        A list of inboxes.
+    """
+    return client.inboxes.list(limit=limit, last_key=last_key)
+
+
+def get_inbox(inbox_id: str):
+    """
+    Get an inbox by ID.
+
+    Args:
+        inbox_id: The ID of the inbox to get.
+
+    Returns:
+        An inbox.
+    """
+    return client.inboxes.get(inbox_id)
+
+
+def create_inbox(
+    username: Optional[str] = None,
+    domain: Optional[str] = None,
+    display_name: Optional[str] = None,
+):
+    """
+    Create an inbox.
+
+    Args:
+        username: The username of the inbox.
+        domain: The domain of the inbox.
+        display_name: The display name of the inbox.
+
+    Returns:
+        An inbox.
+    """
+    return client.inboxes.create(
+        username=username, domain=domain, display_name=display_name
+    )
+
+
+def list_threads(
+    inbox_id: str, limit: Optional[int] = None, last_key: Optional[str] = None
+):
+    """
+    List threads in an inbox.
+
+    Args:
+        inbox_id: The ID of the inbox to list threads for.
+        limit: The maximum number of threads to return.
+        last_key: The last key returned from the previous page.
+
+    Returns:
+        A list of threads.
+    """
+    return client.threads.list(inbox_id=inbox_id, limit=limit, last_key=last_key)
+
+
+def get_thread(inbox_id: str, thread_id: str):
+    """
+    Get a thread by ID.
+
+    Args:
+        inbox_id: The ID of the inbox to get the thread for.
+        thread_id: The ID of the thread to get.
+
+    Returns:
+        A thread.
+    """
+    return client.threads.get(inbox_id=inbox_id, thread_id=thread_id)
+
+
+def list_messages(
+    inbox_id: str, limit: Optional[int] = None, last_key: Optional[str] = None
+):
+    """
+    List messages in an inbox.
+
+    Args:
+        inbox_id: The ID of the inbox to list messages for.
+        limit: The maximum number of messages to return.
+        last_key: The last key returned from the previous page.
+
+    Returns:
+        A list of messages.
+    """
+    return client.messages.list(inbox_id=inbox_id, limit=limit, last_key=last_key)
+
+
+def get_message(inbox_id: str, message_id: str):
+    """
+    Get a message by ID.
+
+    Args:
+        inbox_id: The ID of the inbox to get the message for.
+        message_id: The ID of the message to get.
+
+    Returns:
+        A message.
+    """
+    return client.messages.get(inbox_id=inbox_id, message_id=message_id)
+
+
+def get_attachment(inbox_id: str, message_id: str, attachment_id: str):
+    """
+    Get an attachment by message ID and attachment ID.
+
+    Args:
+        inbox_id: The ID of the inbox to get the attachment for.
+        message_id: The ID of the message to get the attachment for.
+        attachment_id: The ID of the attachment to get.
+
+    Returns:
+        An attachment.
+    """
+    return client.messages.get_attachment(
+        inbox_id=inbox_id, message_id=message_id, attachment_id=attachment_id
+    )
+
+
+def send_message(
+    inbox_id: str,
+    to: List[str],
+    cc: Optional[List[str]] = None,
+    bcc: Optional[List[str]] = None,
+    subject: Optional[str] = None,
+    text: Optional[str] = None,
+    html: Optional[str] = None,
+):
+    """
+    Send a message.
+
+    Args:
+        inbox_id: The ID of the inbox to send the message from.
+        to: The list of recipients.
+        cc: The list of CC recipients.
+        bcc: The list of BCC recipients.
+        subject: The subject of the message.
+        text: The plain text body of the message.
+        html: The HTML body of the message.
+
+    Returns:
+        A message.
+    """
+    return client.messages.send(
+        inbox_id=inbox_id, to=to, cc=cc, bcc=bcc, subject=subject, text=text, html=html
+    )
+
+
+def reply_to_message(
+    inbox_id: str,
+    message_id: str,
+    text: Optional[str] = None,
+    html: Optional[str] = None,
+):
+    """
+    Reply to a message.
+
+    Args:
+        inbox_id: The ID of the inbox to reply to the message in.
+        message_id: The ID of the message to reply to.
+        text: The plain text body of the reply.
+        html: The HTML body of the reply.
+
+    Returns:
+        A message.
+    """
+    return client.messages.reply(
+        inbox_id=inbox_id, message_id=message_id, text=text, html=html
+    )

agentstack/_tools/agentmail/config.json

@@ -0,0 +1,13 @@
+{
+    "name": "agentmail",
+    "category": "email",
+    "tools": ["list_inboxes", "get_inbox", "create_inbox", "list_threads", "get_thread", "list_messages", "get_message", "get_attachment", "send_message", "reply_to_message"],
+    "url": "https://agentmail.to",
+    "cta": "Get your AgentMail API key at https://agentmail.to",
+    "env": {
+        "AGENTMAIL_API_KEY": null
+    },
+    "dependencies": [
+        "agentmail>=0.0.19"
+    ]
+}

agentstack/_tools/agentql/__init__.py

@@ -8,12 +8,18 @@
 
 API_KEY = os.getenv("AGENTQL_API_KEY")
 
-def query_data(url: str, query: Optional[str], prompt: Optional[str]) -> dict:
+
+def extract_data(
+    url: str,
+    query: Optional[str],
+    prompt: Optional[str],
+    is_stealth_mode_enabled: bool = False,
+) -> dict:
     """
     url: url of website to scrape
     query: described below
     prompt: Natural language description of the data you want to scrape
-
+    is_stealth_mode_enabled: Enable stealth mode for web scraping (default: False)
 
     AgentQL query to scrape the url.
 
@@ -47,34 +53,44 @@ def query_data(url: str, query: Optional[str], prompt: Optional[str]) -> dict:
     payload = {
         "url": url,
         "query": query,
-        "prompt": prompt
+        "prompt": prompt,
+        "metadata": {
+            "experimental_stealth_mode_enabled": is_stealth_mode_enabled,
+        },
     }
 
     headers = {
         "X-API-Key": f"{API_KEY}",
-        "Content-Type": "application/json"
+        "Content-Type": "application/json",
+        "X-TF-Request-Origin": "agentstack",
     }
 
     try:
         response = httpx.post(
             QUERY_DATA_ENDPOINT,
             headers=headers,
             json=payload,
-            timeout=API_TIMEOUT_SECONDS
+            timeout=API_TIMEOUT_SECONDS,
         )
         response.raise_for_status()
 
     except httpx.HTTPStatusError as e:
         response = e.response
         if response.status_code in [401, 403]:
-            raise ValueError("Please, provide a valid API Key. You can create one at https://dev.agentql.com.") from e
+            raise ValueError(
+                "Please, provide a valid API Key. You can create one at https://dev.agentql.com."
+            ) from e
         else:
             try:
                 error_json = response.json()
-                msg = error_json["error_info"] if "error_info" in error_json else error_json["detail"]
+                msg = (
+                    error_json["error_info"]
+                    if "error_info" in error_json
+                    else error_json["detail"]
+                )
             except (ValueError, TypeError):
                 msg = f"HTTP {e}."
             raise ValueError(msg) from e
     else:
         json = response.json()
-        return json["data"]
+        return json["data"]

agentstack/_tools/agentql/config.json

@@ -6,6 +6,6 @@
     "env": {
         "AGENTQL_API_KEY": "..."
     },
-    "tools": ["query_data"],
+    "tools": ["extract_data"],
     "cta": "Create your AgentQL API key at https://dev.agentql.com"
 }