This commit is contained in:
Timothy J. Baek
2024-05-21 14:18:25 -07:00
parent fe49f4af7e
commit db1262def9
4 changed files with 49 additions and 36 deletions

View File

@@ -1,5 +1,8 @@
import uuid
import time
from typing import List
from schemas import OpenAIChatMessage
from config import MODEL_ID
@@ -18,3 +21,10 @@ def stream_message_template(message: str):
}
],
}
def get_last_user_message(messages: List[OpenAIChatMessage]) -> str:
for message in reversed(messages):
if message.role == "user":
return message.content
return None