27 lines
No EOL
816 B
Python
27 lines
No EOL
816 B
Python
#!/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() |