github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-06 11:31:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(build): Fix typo and new pre-commit config (#987)

Browse Source
This commit is contained in:
Fangyin Cheng
2023-12-28 14:14:20 +08:00
committed by GitHub
parent b13d3f6d92
commit fd3a5d2bfa
77 changed files with 432 additions and 446 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
dbgpt/model/adapter/base.py
View File

@@ -158,17 +158,19 @@ class LLMModelAdapter(ABC):
Default is the OpenAI format, example:
.. code-block:: python
return_messages = [
{"role": "system", "content": "You are a helpful assistant"},
{"role": "user", "content": "Hello"},
{"role": "assistant", "content": "Hi"}
{"role": "assistant", "content": "Hi"},
]
But some model may need to transform the messages to other format(e.g. There is no system message), such as:
.. code-block:: python
return_messages = [
{"role": "user", "content": "Hello"},
{"role": "assistant", "content": "Hi"}
{"role": "assistant", "content": "Hi"},
]
Args:
messages (List[ModelMessage]): The model messages
@@ -191,10 +193,11 @@ class LLMModelAdapter(ABC):
Merge the system messages to the last user message, example:
.. code-block:: python
return_messages = [
{"role": "system", "content": "You are a helpful assistant"},
{"role": "user", "content": "Hello"},
{"role": "assistant", "content": "Hi"}
{"role": "assistant", "content": "Hi"},
]
=>
return_messages = [