diff options
Diffstat (limited to 'continuedev/src/continuedev/steps/react.py')
-rw-r--r-- | continuedev/src/continuedev/steps/react.py | 37 |
1 files changed, 37 insertions, 0 deletions
diff --git a/continuedev/src/continuedev/steps/react.py b/continuedev/src/continuedev/steps/react.py new file mode 100644 index 00000000..411adc87 --- /dev/null +++ b/continuedev/src/continuedev/steps/react.py @@ -0,0 +1,37 @@ +from textwrap import dedent +from typing import List +from ..core.main import Step +from ..core.sdk import ContinueSDK +from .main import MessageStep + + +class NLDecisionStep(Step): + user_input: str + steps: List[Step] + + async def run(self, sdk: ContinueSDK): + step_descriptions = "\n".join([ + f"- {step.name}: {step.description}" + for step in self.steps + ]) + prompt = dedent(f"""\ + The following steps are available, in the format "- [step name]: [step description]": + {step_descriptions} + + The user gave the following input: + + {self.user_input} + + Select the step which should be taken next. Say only the name of the selected step:""") + + resp = (await sdk.models.gpt35.complete(prompt)).lower() + + step_to_run = None + for step in self.steps: + if step.name in resp: + step_to_run = step + + step_to_run = step_to_run or MessageStep( + message="Unable to decide the next step") + + await sdk.run_step(step_to_run) |