[PY] feat: augmentations

[lilyydu]

Linked issues

closes: #1068

Details

Implemented the logic and tests for augmentations for Python, based on JS.

• Renamed the DO command's entities to parameters (from Steve's previous breaking changes)
• Type for Schema is Dict[str, Any] to follow previous conventions
• Renamed _token in ActionAugmentationSection to _token_list, as this conflicted with the super class' token property
• Decomposed InnerMonologue into separate Thought and Action data classes
• Added __test__ = False to TestablePromptSectionBase as this is apart of setup for the tests and a warning appeared that the class was being ran as a test instead
• Follow-up PR to be added, to add augmentation logic to PromptManager and PromptTemplate

More specific:

• For both MonologueAugmentation and SequenceAugmentation, in create_plan_from_response and validate_response, I removed the union for undefined parameters/return types, since these are required.
• I also removed it for the Augmentation class type.
• For MonologueAugmentation, I used the dataclass_json library to decode from dict to the class type, as our validator was throwing lint errors.
• For SequenceAugmentation, I used our internal static method from_dict from the Plan class.

Attestation Checklist

• My code follows the style guidelines of this project

• I have checked for/fixed spelling, linting, and other errors

• I have commented my code for clarity

• I have made corresponding changes to the documentation (updating the doc strings in the code is sufficient)

• My changes generate no new warnings

• I have added tests that validates my changes, and provides sufficient test coverage. I have tested with:

  • Local testing
  • E2E testing in Teams

• New and existing unit tests pass locally with my changes