Update CodeWriterFlow.py
Browse files- CodeWriterFlow.py +4 -0
CodeWriterFlow.py
CHANGED
|
@@ -6,6 +6,10 @@ from aiflows.base_flows import CircularFlow
|
|
| 6 |
|
| 7 |
|
| 8 |
class CodeWriterFlow(ContentWriterFlow):
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
def _on_reach_max_round(self):
|
| 10 |
self._state_update_dict({
|
| 11 |
"code": "The maximum amount of rounds was reached before the model generated the code.",
|
|
|
|
| 6 |
|
| 7 |
|
| 8 |
class CodeWriterFlow(ContentWriterFlow):
|
| 9 |
+
"""
|
| 10 |
+
This flow inherits from ContentWriterFlow, it is used to write code in an interactive way.
|
| 11 |
+
In the subflow of the executor, we specify an InteractiveCodeGenFlow (https://huggingface.co/Tachi67/InteractiveCodeGenFlowModule)
|
| 12 |
+
"""
|
| 13 |
def _on_reach_max_round(self):
|
| 14 |
self._state_update_dict({
|
| 15 |
"code": "The maximum amount of rounds was reached before the model generated the code.",
|