diff options
author | Nate Sesti <sestinj@gmail.com> | 2023-07-31 00:05:30 -0700 |
---|---|---|
committer | Nate Sesti <sestinj@gmail.com> | 2023-07-31 00:05:30 -0700 |
commit | 396679009fef21e13c1a6095212d1bd68e7f2a86 (patch) | |
tree | bb1ff71675d1175ff75250f04ac1b11720ba9dd8 | |
parent | c6a12550ffca1ffe35630e7aa9af6913ddbe0675 (diff) | |
download | sncontinue-396679009fef21e13c1a6095212d1bd68e7f2a86.tar.gz sncontinue-396679009fef21e13c1a6095212d1bd68e7f2a86.tar.bz2 sncontinue-396679009fef21e13c1a6095212d1bd68e7f2a86.zip |
feat: :technologist: bit of customization for DefaultPolicy
-rw-r--r-- | continuedev/src/continuedev/plugins/policies/default.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/continuedev/src/continuedev/plugins/policies/default.py b/continuedev/src/continuedev/plugins/policies/default.py index f479b758..2e7573f3 100644 --- a/continuedev/src/continuedev/plugins/policies/default.py +++ b/continuedev/src/continuedev/plugins/policies/default.py @@ -45,7 +45,8 @@ def parse_custom_command(inp: str, config: ContinueConfig) -> Union[None, Step]: class DefaultPolicy(Policy): - ran_code_last: bool = False + + default_step: Step = SimpleChatStep() def next(self, config: ContinueConfig, history: History) -> Step: # At the very start, run initial Steps spcecified in the config @@ -74,6 +75,6 @@ class DefaultPolicy(Policy): if user_input.startswith("/edit"): return EditHighlightedCodeStep(user_input=user_input[5:]) - return SimpleChatStep() + return self.default_step return None |