github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-04 18:40:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(awel): New MessageConverter and more AWEL operators (#1039)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-01-08 09:40:05 +08:00
committed by GitHub
parent 765fb181f6
commit e8861bd8fa
48 changed files with 2333 additions and 719 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
dbgpt/model/proxy/llms/bard.py
View File

@@ -13,6 +13,8 @@ def bard_generate_stream(
proxy_api_key = model_params.proxy_api_key
proxy_server_url = model_params.proxy_server_url
convert_to_compatible_format = params.get("convert_to_compatible_format", False)
history = []
messages: List[ModelMessage] = params["messages"]
for message in messages:
@@ -25,14 +27,15 @@ def bard_generate_stream(
else:
pass
last_user_input_index = None
for i in range(len(history) - 1, -1, -1):
if history[i]["role"] == "user":
last_user_input_index = i
break
if last_user_input_index:
last_user_input = history.pop(last_user_input_index)
history.append(last_user_input)
if convert_to_compatible_format:
last_user_input_index = None
for i in range(len(history) - 1, -1, -1):
if history[i]["role"] == "user":
last_user_input_index = i
break
if last_user_input_index:
last_user_input = history.pop(last_user_input_index)
history.append(last_user_input)
msgs = []
for msg in history: