diff options
| author | Nate Sesti <sestinj@gmail.com> | 2023-07-07 14:36:14 -0700 |
|---|---|---|
| committer | Nate Sesti <sestinj@gmail.com> | 2023-07-07 14:36:14 -0700 |
| commit | 4559e123b9d1834cab73451ce37c46e3ad56ba06 (patch) | |
| tree | 54f84cc624509314848337c170f748848b2cfd79 /continuedev | |
| parent | 307ad0a2daadfaffd34203cd0230f56b78e43752 (diff) | |
| download | sncontinue-4559e123b9d1834cab73451ce37c46e3ad56ba06.tar.gz sncontinue-4559e123b9d1834cab73451ce37c46e3ad56ba06.tar.bz2 sncontinue-4559e123b9d1834cab73451ce37c46e3ad56ba06.zip | |
fix renaming user messages to assistant
Diffstat (limited to 'continuedev')
| -rw-r--r-- | continuedev/src/continuedev/core/main.py | 3 | ||||
| -rw-r--r-- | continuedev/src/continuedev/steps/chat.py | 2 |
2 files changed, 3 insertions, 2 deletions
diff --git a/continuedev/src/continuedev/core/main.py b/continuedev/src/continuedev/core/main.py index 403e5417..4ea17f20 100644 --- a/continuedev/src/continuedev/core/main.py +++ b/continuedev/src/continuedev/core/main.py @@ -38,7 +38,8 @@ class ChatMessage(ContinueBaseModel): del d[key] if not with_functions: - d["role"] = "assistant" + if d["role"] == "function": + d["role"] = "assistant" if "name" in d: del d["name"] if "function_call" in d: diff --git a/continuedev/src/continuedev/steps/chat.py b/continuedev/src/continuedev/steps/chat.py index 37d1b1cc..a10319d8 100644 --- a/continuedev/src/continuedev/steps/chat.py +++ b/continuedev/src/continuedev/steps/chat.py @@ -27,7 +27,7 @@ class SimpleChatStep(Step): async def run(self, sdk: ContinueSDK): completion = "" messages = self.messages or await sdk.get_chat_context() - async for chunk in sdk.models.gpt4.stream_chat(messages): + async for chunk in sdk.models.gpt4.stream_chat(messages, temperature=0.5): if sdk.current_step_was_deleted(): return |