diff options
| author | Ty Dunn <ty@tydunn.com> | 2023-06-20 17:57:17 -0700 |
|---|---|---|
| committer | Ty Dunn <ty@tydunn.com> | 2023-06-20 17:57:17 -0700 |
| commit | 61b278510cf8ab05976d77e1c6a32d0ac6068a0f (patch) | |
| tree | f7881ef96799b6cf75c63ea1cdf8d0f554b41843 /continuedev/src/continuedev/steps | |
| parent | e8c96911812a79892f8531731ae98fa3473b35ab (diff) | |
| download | sncontinue-61b278510cf8ab05976d77e1c6a32d0ac6068a0f.tar.gz sncontinue-61b278510cf8ab05976d77e1c6a32d0ac6068a0f.tar.bz2 sncontinue-61b278510cf8ab05976d77e1c6a32d0ac6068a0f.zip | |
fixing hardcoded stream chat
Diffstat (limited to 'continuedev/src/continuedev/steps')
| -rw-r--r-- | continuedev/src/continuedev/steps/core/core.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/continuedev/src/continuedev/steps/core/core.py b/continuedev/src/continuedev/steps/core/core.py index 09f9facc..71a5b5b2 100644 --- a/continuedev/src/continuedev/steps/core/core.py +++ b/continuedev/src/continuedev/steps/core/core.py @@ -213,7 +213,7 @@ class DefaultModelEditCodeStep(Step): if model_to_use.name == "gpt-4": - total_tokens = model_to_use.count_tokens(full_file_contents) + total_tokens = model_to_use.count_tokens(full_file_contents + self._prompt) cur_start_line, cur_end_line = cut_context( model_to_use, total_tokens, cur_start_line, cur_end_line) @@ -223,7 +223,7 @@ class DefaultModelEditCodeStep(Step): model_to_use = sdk.models.gpt3516k total_tokens = model_to_use.count_tokens( - full_file_contents) + full_file_contents + self._prompt) cur_start_line, cur_end_line = cut_context( model_to_use, total_tokens, cur_start_line, cur_end_line) |