feat: mypy for all type check (#10921)

2024-12-24 18:38:51 +08:00 · 2024-12-24 18:38:51 +08:00 · 56e15d09a9
commit 56e15d09a9
parent c91e8b1737
584 changed files with 3975 additions and 2826 deletions
--- a/api/core/memory/token_buffer_memory.py
+++ b/api/core/memory/token_buffer_memory.py
@ -68,7 +68,7 @@ class TokenBufferMemory:

        messages = list(reversed(thread_messages))

-        prompt_messages = []
+        prompt_messages: list[PromptMessage] = []
        for message in messages:
            files = db.session.query(MessageFile).filter(MessageFile.message_id == message.id).all()
            if files: