add Annotated for the function argument docs

Author: HarshCasper

.dagger/src/localstack/main.py

@@ -1,7 +1,13 @@
+"""LocalStack module for managing LocalStack instances and state.
+
+This module provides functions to start LocalStack containers, manage state through Cloud Pods,
+and handle ephemeral LocalStack instances in the cloud.
+"""
+
 import os
 import dagger
-from dagger import dag, function, object_type
-from typing import Optional
+from dagger import dag, function, object_type, Doc
+from typing import Optional, Annotated
 import base64
 from datetime import datetime
 import requests
@@ -10,13 +16,15 @@
 
 @object_type
 class Localstack:
+    """LocalStack service management functions."""
+
     @function
     def start(
         self, 
-        auth_token: Optional[dagger.Secret] = None,
-        configuration: Optional[str] = None,
-        docker_sock: Optional[dagger.Socket] = None,
-        image_name: Optional[str] = None
+        auth_token: Annotated[Optional[dagger.Secret], Doc("LocalStack Pro Auth Token for authentication")] = None,
+        configuration: Annotated[Optional[str], Doc("Configuration variables in format 'KEY1=value1,KEY2=value2'")] = None,
+        docker_sock: Annotated[Optional[dagger.Socket], Doc("Docker socket for container interactions")] = None,
+        image_name: Annotated[Optional[str], Doc("Custom LocalStack image name to use")] = None
     ) -> dagger.Service:
         """Start a LocalStack service with appropriate configuration.
         
@@ -25,7 +33,7 @@ def start(
         Otherwise starts LocalStack Community edition.
         
         Args:
-            auth_token: Optional secret containing LocalStack Pro auth token
+            auth_token: Optional secret containing LocalStack Pro Auth Token
             configuration: Optional string of configuration variables in format "KEY1=value1,KEY2=value2"
                          Example: "DEBUG=1,LS_LOG=trace"
             docker_sock: Optional Docker socket for container interactions
@@ -47,7 +55,7 @@ def start(
         if docker_sock:
             container = container.with_unix_socket("/var/run/docker.sock", docker_sock)
 
-        # Add auth token if provided
+        # Add Auth Token if provided
         if auth_token:
             container = container.with_secret_variable("LOCALSTACK_AUTH_TOKEN", auth_token)
 
@@ -74,16 +82,16 @@ def start(
     @function
     async def state(
         self,
-        auth_token: Optional[dagger.Secret] = None,
-        load: Optional[str] = None,
-        save: Optional[str] = None,
-        endpoint: Optional[str] = None,
-        reset: bool = False
+        auth_token: Annotated[Optional[dagger.Secret], Doc("LocalStack Auth Token (required for save/load)")] = None,
+        load: Annotated[Optional[str], Doc("Name of the Cloud Pod to load")] = None,
+        save: Annotated[Optional[str], Doc("Name of the Cloud Pod to save")] = None,
+        endpoint: Annotated[Optional[str], Doc("LocalStack endpoint (defaults to host.docker.internal:4566)")] = None,
+        reset: Annotated[bool, Doc("Reset the LocalStack state")] = False
     ) -> str:
         """Load, save, or reset LocalStack state.
         
         Args:
-            auth_token: Secret containing LocalStack auth token (required for save/load)
+            auth_token: Secret containing LocalStack Auth Token (required for save/load)
             load: Name of the Cloud Pod to load
             save: Name of the Cloud Pod to save
             reset: Reset the LocalStack state
@@ -140,7 +148,7 @@ async def state(
                     save_response.raise_for_status()
                     return save_response.text
                 except requests.RequestException:
-                    return f"Error: Failed to save pod '{save}'. Please check the pod name and your auth token."
+                    return f"Error: Failed to save pod '{save}'. Please check the pod name and your Auth Token."
             elif load:
                 try:
                     load_response = requests.put(
@@ -151,24 +159,24 @@ async def state(
                     load_response.raise_for_status()
                     return load_response.text
                 except requests.RequestException:
-                    return f"Error: Failed to load pod '{load}'. Please check the pod name and your auth token."
+                    return f"Error: Failed to load pod '{load}'. Please check the pod name and your Auth Token."
 
         return "No operation specified. Please provide either --load, --save, or --reset parameter."
 
     @function
     async def ephemeral(
         self,
-        auth_token: dagger.Secret,
-        operation: str,
-        name: Optional[str] = None,
-        lifetime: Optional[int] = None,
-        auto_load_pod: Optional[str] = None,
-        extension_auto_install: Optional[str] = None
+        auth_token: Annotated[dagger.Secret, Doc("LocalStack Auth Token (required)")],
+        operation: Annotated[str, Doc("Operation to perform (create, list, delete, logs)")],
+        name: Annotated[Optional[str], Doc("Name of the ephemeral instance (required for create, delete, logs)")] = None,
+        lifetime: Annotated[Optional[int], Doc("Lifetime of the instance in minutes (default: 60)")] = None,
+        auto_load_pod: Annotated[Optional[str], Doc("Auto load pod configuration")] = None,
+        extension_auto_install: Annotated[Optional[str], Doc("Extension auto install configuration")] = None
     ) -> str:
         """Manage ephemeral LocalStack instances in the cloud.
         
         Args:
-            auth_token: LocalStack auth token (required)
+            auth_token: LocalStack Auth Token (required)
             operation: Operation to perform (create, list, delete, logs)
             name: Name of the ephemeral instance (required for create, delete, logs)
             lifetime: Lifetime of the instance in minutes (optional, default: 60)
@@ -184,7 +192,7 @@ async def ephemeral(
         # Base API endpoint
         api_endpoint = "https://api.localstack.cloud/v1"
 
-        # Get auth token value from secret
+        # Get Auth Token value from secret
         auth_token_value = await auth_token.plaintext()
 
         # Common headers