feat: filter outlet & libretranslate example

This commit is contained in:
Timothy J. Baek
2024-05-30 02:04:07 -07:00
parent 95ed193ef7
commit 2e934304cf
3 changed files with 158 additions and 3 deletions

View File

@@ -24,6 +24,21 @@ def stream_message_template(model: str, message: str):
def get_last_user_message(messages: List[dict]) -> str:
for message in reversed(messages):
if message.role == "user":
return message.content
if message["role"] == "user":
if isinstance(message["content"], list):
for item in message["content"]:
if item["type"] == "text":
return item["text"]
return message["content"]
return None
def get_last_assistant_message(messages: List[dict]) -> str:
for message in reversed(messages):
if message["role"] == "assistant":
if isinstance(message["content"], list):
for item in message["content"]:
if item["type"] == "text":
return item["text"]
return message["content"]
return None