Initial commit

skills/json-outputs-implementer/examples/contact-extraction.py

@@ -0,0 +1,138 @@
+"""
+Contact Information Extraction Example
+
+Extracts structured contact information from unstructured text (emails, messages, etc.)
+using JSON outputs mode with Pydantic schema validation.
+"""
+
+from pydantic import BaseModel, Field, EmailStr
+from typing import Optional, List
+from anthropic import Anthropic
+import os
+
+# Initialize client
+client = Anthropic(api_key=os.environ.get("ANTHROPIC_API_KEY"))
+
+
+# Define schema with Pydantic
+class ContactInfo(BaseModel):
+    """Structured contact information extracted from text."""
+
+    name: str = Field(description="Full name of the contact person")
+    email: EmailStr = Field(description="Email address")
+    phone: Optional[str] = Field(
+        None, description="Phone number in any format"
+    )
+    company: Optional[str] = Field(
+        None, description="Company or organization name"
+    )
+    plan_interest: Optional[str] = Field(
+        None, description="Product plan or tier they're interested in"
+    )
+    demo_requested: bool = Field(
+        False, description="Whether they requested a product demo"
+    )
+    tags: List[str] = Field(
+        default_factory=list,
+        description="Relevant tags or categories"
+    )
+
+
+def extract_contact(text: str) -> Optional[ContactInfo]:
+    """
+    Extract contact information from unstructured text.
+
+    Args:
+        text: Unstructured text containing contact information
+
+    Returns:
+        ContactInfo object with extracted data, or None if request refused
+    """
+    try:
+        response = client.beta.messages.parse(
+            model="claude-sonnet-4-5",
+            max_tokens=1024,
+            betas=["structured-outputs-2025-11-13"],
+            messages=[{
+                "role": "user",
+                "content": f"Extract contact information from the following text:\n\n{text}"
+            }],
+            output_format=ContactInfo,
+        )
+
+        # Handle different stop reasons
+        if response.stop_reason == "refusal":
+            print(f"⚠️  Request refused for safety reasons")
+            return None
+
+        if response.stop_reason == "max_tokens":
+            print(f"⚠️  Response truncated - increase max_tokens")
+            return None
+
+        # Return validated contact info
+        return response.parsed_output
+
+    except Exception as e:
+        print(f"❌ Error extracting contact: {e}")
+        raise
+
+
+def main():
+    """Run contact extraction examples."""
+
+    examples = [
+        # Example 1: Complete contact info
+        """
+        Hi, I'm John Smith from Acme Corp. You can reach me at john.smith@acme.com
+        or call me at (555) 123-4567. I'm interested in your Enterprise plan and
+        would love to schedule a demo next week.
+        """,
+
+        # Example 2: Minimal info
+        """
+        Contact: jane.doe@example.com
+        """,
+
+        # Example 3: Informal message
+        """
+        Hey! Bob here. Email me at bob@startup.io if you want to chat about
+        the Pro plan. Thanks!
+        """,
+
+        # Example 4: Multiple contacts (extracts first/primary)
+        """
+        From: alice@company.com
+        CC: support@company.com
+
+        Hi, I'm Alice Johnson, VP of Engineering at TechCo.
+        We're evaluating your platform for our team of 50 developers.
+        """,
+    ]
+
+    print("=" * 70)
+    print("Contact Extraction Examples")
+    print("=" * 70)
+
+    for i, text in enumerate(examples, 1):
+        print(f"\n📧 Example {i}:")
+        print(f"Input: {text.strip()[:100]}...")
+
+        contact = extract_contact(text)
+
+        if contact:
+            print(f"\n✅ Extracted Contact:")
+            print(f"   Name: {contact.name}")
+            print(f"   Email: {contact.email}")
+            print(f"   Phone: {contact.phone or 'N/A'}")
+            print(f"   Company: {contact.company or 'N/A'}")
+            print(f"   Plan Interest: {contact.plan_interest or 'N/A'}")
+            print(f"   Demo Requested: {contact.demo_requested}")
+            print(f"   Tags: {', '.join(contact.tags) if contact.tags else 'None'}")
+        else:
+            print(f"\n❌ No contact extracted")
+
+        print("-" * 70)
+
+
+if __name__ == "__main__":
+    main()