Skip to content

Commit 0f6c286

Browse files
committed
fix:merge error
1 parent c103b32 commit 0f6c286

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

examples/mem_os/simple_vllm_memos.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
from memos.configs.llm import VLLMLLMConfig
88
from memos.llms.vllm import VLLMLLM
9-
from memos.types import MessageList
9+
from memos.types import MessageDict
1010

1111
def main():
1212
"""Main function demonstrating VLLMLLM usage."""
@@ -28,7 +28,7 @@ def main():
2828

2929
# Test messages for KV cache building
3030
print("\nBuilding KV cache for system messages...")
31-
system_messages: MessageList = [
31+
system_messages: list[MessageDict] = [
3232
{"role": "system", "content": "You are a helpful AI assistant."},
3333
{"role": "user", "content": "Hello! Can you tell me about vLLM?"}
3434
]
@@ -40,8 +40,8 @@ def main():
4040

4141
# Test with different messages for generation
4242
print("\nGenerating response...")
43-
user_messages: MessageList = [
44-
{"role": "system", "content": "You are a helpful AI assistant."},
43+
user_messages: list[MessageDict] = [
44+
{"role": "system", "content": "You are a helpful AI assistant. Please Introduce yourself "},
4545
{"role": "user", "content": "What are the benefits of using vLLM?"}
4646
]
4747
try:

0 commit comments

Comments
 (0)