#!/usr/bin/env python3

from modules.workflow.methodBase import MethodBase

class TestMethod(MethodBase):
    pass

def test_parameter_extraction():
    test = TestMethod(None)
    test.name = 'test'
    
    docstring = """Call AI service with document content

Parameters:
    prompt (str): The prompt to send to the AI service
    documents (List[Dict[str, Any]], optional): List of documents to include in context
        Each document should have: documentReference (str), contentExtractionPrompt (str, optional)"""
    
    print("Docstring:")
    print(docstring)
    print("\nExtracted descriptions:")
    descriptions = test._extractParameterDescriptions(docstring)
    for param, desc in descriptions.items():
        print(f"  {param}: {desc}")

if __name__ == "__main__":
    test_parameter_extraction()