golf-mcp 0.1.11__py3-none-any.whl → 0.1.13__py3-none-any.whl

golf/examples/basic/tools/payments/charge.py

@@ -1,40 +1,50 @@
 """Charge payment tool"""
 
 from typing import Annotated
+
 from pydantic import BaseModel, Field
+
 from .common import payment_client
 
 
 class Output(BaseModel):
     """Response from the charge payment tool."""
-    
+
     success: bool
     charge_id: str
     message: str
 
 
 async def charge(
-    amount: Annotated[float, Field(
-        description="Amount to charge in USD",
-        gt=0,  # Must be greater than 0
-        le=10000  # Maximum charge limit
-    )],
-    card_token: Annotated[str, Field(
-        description="Tokenized payment card identifier",
-        pattern=r"^tok_[a-zA-Z0-9]+$"  # Validate token format
-    )],
-    description: Annotated[str, Field(
-        description="Optional payment description for the charge",
-        max_length=200
-    )] = ""
+    amount: Annotated[
+        float,
+        Field(
+            description="Amount to charge in USD",
+            gt=0,  # Must be greater than 0
+            le=10000,  # Maximum charge limit
+        ),
+    ],
+    card_token: Annotated[
+        str,
+        Field(
+            description="Tokenized payment card identifier",
+            pattern=r"^tok_[a-zA-Z0-9]+$",  # Validate token format
+        ),
+    ],
+    description: Annotated[
+        str,
+        Field(
+            description="Optional payment description for the charge", max_length=200
+        ),
+    ] = "",
 ) -> Output:
     """Process a payment charge.
-    
+
     This example demonstrates nested directory organization where related tools
     are grouped in subdirectories (tools/payments/charge.py).
-    
+
     The resulting tool ID will be: charge-payments
-    
+
     Args:
         amount: Amount to charge in USD
         card_token: Tokenized payment card
@@ -43,19 +53,18 @@ async def charge(
     # The framework will add a context object automatically
     # You can log using regular print during development
     print(f"Processing charge for ${amount:.2f}...")
-    
+
     # Use the shared payment client from common.py
     charge_result = await payment_client.create_charge(
-        amount=amount,
-        token=card_token,
-        description=description
+        amount=amount, token=card_token, description=description
     )
-    
+
     # Create and return the response
     return Output(
         success=True,
         charge_id=charge_result["id"],
-        message=f"Successfully charged ${amount:.2f}"
-    ) 
+        message=f"Successfully charged ${amount:.2f}",
+    )
+
 
-export = charge
+export = charge

golf/examples/basic/tools/payments/common.py

@@ -1,6 +1,6 @@
 """Payments shared functionality.
 
-This common.py file demonstrates the recommended pattern for 
+This common.py file demonstrates the recommended pattern for
 sharing functionality across multiple tools in a directory.
 """
 
@@ -13,17 +13,19 @@ PAYMENT_API_URL = os.environ.get("PAYMENT_API_URL", "https://api.example.com/pay
 
 class PaymentClient:
     """Mock payment provider client."""
-    
-    def __init__(self, api_key: str = PAYMENT_API_KEY, api_url: str = PAYMENT_API_URL):
+
+    def __init__(
+        self, api_key: str = PAYMENT_API_KEY, api_url: str = PAYMENT_API_URL
+    ) -> None:
         self.api_key = api_key
         self.api_url = api_url
-    
+
     async def create_charge(self, amount: float, token: str, **kwargs):
         """Create a charge (mock implementation)."""
         # In a real implementation, this would make an API request
         # using the configured API key and URL
         return {"id": f"ch_{int(amount * 100)}_{hash(token) % 10000:04d}"}
-    
+
     async def create_refund(self, charge_id: str, amount: float, **kwargs):
         """Create a refund (mock implementation)."""
         # In a real implementation, this would make an API request
@@ -31,4 +33,4 @@ class PaymentClient:
 
 
 # Create a shared payment client that can be imported by all tools in this directory
-payment_client = PaymentClient()
+payment_client = PaymentClient()

golf/examples/basic/tools/payments/refund.py

@@ -1,57 +1,61 @@
 """Refund payment tool"""
 
-from typing import Annotated, Optional
+from typing import Annotated
+
 from pydantic import BaseModel, Field
+
 from .common import payment_client
 
 
 class Output(BaseModel):
     """Response from the refund payment tool."""
-    
+
     success: bool
     refund_id: str
     message: str
 
 
 async def refund(
-    charge_id: Annotated[str, Field(
-        description="The ID of the charge to refund",
-        pattern=r"^ch_[a-zA-Z0-9]+$"
-    )],
-    amount: Annotated[Optional[float], Field(
-        description="Amount to refund in USD. If not specified, refunds the full charge amount",
-        gt=0,
-        default=None
-    )] = None,
-    reason: Annotated[str, Field(
-        description="Reason for the refund",
-        min_length=3,
-        max_length=200
-    )] = "Customer request"
+    charge_id: Annotated[
+        str,
+        Field(
+            description="The ID of the charge to refund", pattern=r"^ch_[a-zA-Z0-9]+$"
+        ),
+    ],
+    amount: Annotated[
+        float | None,
+        Field(
+            description="Amount to refund in USD. If not specified, refunds the full charge amount",
+            gt=0,
+            default=None,
+        ),
+    ] = None,
+    reason: Annotated[
+        str, Field(description="Reason for the refund", min_length=3, max_length=200)
+    ] = "Customer request",
 ) -> Output:
     """Process a payment refund.
-    
+
     This example demonstrates nested directory organization where related tools
     are grouped in subdirectories (tools/payments/refund.py).
-    
+
     The resulting tool ID will be: refund-payments
     """
     # The framework will add a context object automatically
     # You can log using regular print during development
     print(f"Processing refund for charge {charge_id}...")
-    
+
     # Use the shared payment client from common.py
     refund_result = await payment_client.create_refund(
-        charge_id=charge_id,
-        amount=amount,
-        reason=reason
+        charge_id=charge_id, amount=amount, reason=reason
     )
-    
+
     # Create and return the response
     return Output(
         success=True,
         refund_id=refund_result["id"],
-        message=f"Successfully refunded charge {charge_id}"
+        message=f"Successfully refunded charge {charge_id}",
     )
 
-export = refund
+
+export = refund

golf/telemetry/__init__.py

@@ -1,19 +1,19 @@
 """Golf telemetry module for OpenTelemetry instrumentation."""
 
 from golf.telemetry.instrumentation import (
-    instrument_tool,
-    instrument_resource,
+    get_tracer,
+    init_telemetry,
     instrument_prompt,
+    instrument_resource,
+    instrument_tool,
     telemetry_lifespan,
-    init_telemetry,
-    get_tracer,
 )
 
 __all__ = [
     "instrument_tool",
-    "instrument_resource", 
+    "instrument_resource",
     "instrument_prompt",
     "telemetry_lifespan",
     "init_telemetry",
     "get_tracer",
-] 
+]