apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-06-17 15:25:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

Update ArchBaseHandler

Browse source
This commit is contained in:
Shuguang Chen 2024-12-06 14:07:01 -08:00
parent 2fd8a5a06d
commit 79eafc0241
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
model_server/app/model_handler/base_handler.py
View file

@ -106,7 +106,7 @@ class ArchBaseHandler:
self,
messages: List[Message],
tools: List[Dict[str, Any]] = None,
extra_instructions: str = None,
extra_instruction: str = None,
):
"""
Processes a list of messages and formats them appropriately.
@ -114,7 +114,7 @@ class ArchBaseHandler:
Args:
messages (List[Message]): A list of message objects.
tools (List[Dict[str, Any]], optional): A list of tools to include in the system prompt.
extra_instructions (str, optional): Additional instructions to append to the last user message.
extra_instruction (str, optional): Additional instructions to append to the last user message.
Returns:
List[Dict[str, Any]]: A list of processed message dictionaries.
@ -148,8 +148,8 @@ class ArchBaseHandler:
assert processed_messages[-1]["role"] == "user"
if extra_instructions:
processed_messages[-1]["content"] += extra_instructions
if extra_instruction:
processed_messages[-1]["content"] += extra_instruction
return processed_messages