github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-05 19:11:52 +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 = [

2
dbgpt/model/proxy/llms/gemini.py
View File

@@ -95,7 +95,7 @@ def _transform_to_gemini_messages(
assert user_prompt == "How are you?"
assert gemini_hist == [
{"role": "user", "parts": {"text": "Hello"}},
{"role": "model", "parts": {"text": "Hi there!"}}
{"role": "model", "parts": {"text": "Hi there!"}},
]
"""
user_prompt, system_messages, history_messages = parse_model_messages(messages)