You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
swarms/examples/misc/conversation_test.py

23 lines
514 B

from swarms.structs.conversation import Conversation
# Create a conversation object
conversation = Conversation(backend="in-memory")
# Add a message to the conversation
conversation.add(
role="user", content="Hello, how are you?", category="input"
)
# Add a message to the conversation
conversation.add(
role="assistant",
content="I'm good, thank you!",
category="output",
)
print(
conversation.export_and_count_categories(
tokenizer_model_name="claude-3-5-sonnet-20240620"
)
)