|
@@ -2,7 +2,7 @@ from pathlib import Path
|
|
|
import hashlib
|
|
|
import re
|
|
|
from datetime import timedelta
|
|
|
-from typing import Optional, List, Tuple, Callable
|
|
|
+from typing import Optional, Callable
|
|
|
import uuid
|
|
|
import time
|
|
|
|
|
@@ -52,7 +52,7 @@ def remove_system_message(messages: list[dict]) -> list[dict]:
|
|
|
return [message for message in messages if message["role"] != "system"]
|
|
|
|
|
|
|
|
|
-def pop_system_message(messages: list[dict]) -> Tuple[Optional[dict], list[dict]]:
|
|
|
+def pop_system_message(messages: list[dict]) -> tuple[Optional[dict], list[dict]]:
|
|
|
return get_system_message(messages), remove_system_message(messages)
|
|
|
|
|
|
|