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 | f9a06f3eed89b9d89cda3060be1deb9de93299d7 (patch) | |
| tree | 0e888519cf432dc6e2f421d4519005e64773edb6 /continuedev/src | |
| parent | 640f267e5ee179e15285c827c9326b8c5e09f9b0 (diff) | |
| download | sncontinue-f9a06f3eed89b9d89cda3060be1deb9de93299d7.tar.gz sncontinue-f9a06f3eed89b9d89cda3060be1deb9de93299d7.tar.bz2 sncontinue-f9a06f3eed89b9d89cda3060be1deb9de93299d7.zip | |
fix renaming user messages to assistant
Diffstat (limited to 'continuedev/src')
| -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 |