def charge_customer(customer_id: str, amount: int) -> str:
    response = stripe.Charge.create(
        customer=customer_id,
        amount=amount,
        currency="usd"
    )
    return response["id"]  # AI needs to know this field exists

from mokra import mockworld

# AI generates integration code
generated_code = ai.generate("""
    Write a function that:
    1. Creates a Stripe customer
    2. Charges them $50
    3. Returns the charge ID
""")

# Test the generated code in a MockWorld
world = mockworld(name="AI code test", services=["stripe"])

with world.run():
    # Execute the AI-generated code
    exec(generated_code)

# See what API calls the code made
world.observe()
# => "POST stripe/v1/customers - Created customer cus_mock_123"
# => "POST stripe/v1/charges - Created charge for $50.00"

# AI-generated code calls Stripe
response = stripe.Customer.create(email="test@example.com")

# Gets back real structure:
{
  "id": "cus_mock_abc123",
  "object": "customer",
  "email": "test@example.com",
  "created": 1678901234,
  "livemode": false,
  "metadata": {},
  ...
}

world.observe()

POST stripe/v1/customers
  → Created customer cus_mock_abc123

POST stripe/v1/charges
  → Created charge ch_mock_xyz789 for $50.00

GET stripe/v1/charges/ch_mock_xyz789
  → Retrieved charge details

def generate_and_verify(task: str, max_attempts: int = 3) -> str:
    for attempt in range(max_attempts):
        # AI generates code
        code = ai.generate(task)

        # Test in MockWorld
        world = mockworld(name="Test", services=["stripe"])

        try:
            with world.run():
                result = exec(code)

            # Check results
            observations = world.observe(print=False)

            if "error" not in observations.lower():
                print(f"Success on attempt {attempt + 1}")
                return code

            # AI learns from failure
            ai.feedback(f"Code produced errors: {observations}")

        except Exception as e:
            ai.feedback(f"Code crashed: {e}")

    raise Exception("Could not generate working code")


# Usage
working_code = generate_and_verify("""
    Create a Stripe checkout session for a $99 product
""")

from langchain.agents import tool
from mokra import mockworld

@tool
def test_stripe_code(code: str) -> str:
    """Test generated Stripe integration code"""
    world = mockworld(name="Code test", services=["stripe"])

    try:
        with world.run():
            exec(code)
        return world.observe(print=False)
    except Exception as e:
        return f"Error: {e}"

from crewai import Task

code_testing_task = Task(
    description="Test the generated code against Mokra mock servers",
    expected_output="Verification that the code makes correct API calls"
)