| import json | |
| from handler import EndpointHandler | |
| # Instantiate the handler with a dummy model_dir (replace with actual path if needed) | |
| # Since we are testing the handler logic and not loading a real model here, | |
| # we can initialize it with a placeholder. If your __init__ requires a valid path, | |
| # you might need to adjust this or mock the AutoTokenizer and AutoModelForSeq2SeqLM calls. | |
| # For a basic test of the __call__ method's structure, this might suffice. | |
| # If model loading is essential for your test, you'll need to handle that. | |
| handler = EndpointHandler(model_dir=".") | |
| # Example function without a docstring | |
| def multiply(a, b): | |
| return a * b | |
| # Prepare the data in the expected format for the handler | |
| test_data = {"inputs": ["def multiply(a, b):\n return a * b"]} | |
| # Call the handler | |
| try: | |
| response = handler(test_data) | |
| print(json.dumps(response, indent=2)) | |
| except Exception as e: | |
| print(f"An error occurred: {e}") | |