| 1234567891011121314151617181920 |
- from django.core.management.base import BaseCommand, CommandError
- from chatbot.main import instance as Chatbot
- from adm.constants import CTS
- class Command(BaseCommand):
- def add_arguments(self, parser):
- parser.add_argument('interaction', type=str, help='Chatbot interaction to train')
-
- def handle(self, *args, **options):
- interaction = options['interaction']
- interaction = interaction.split('>>>')
- chat = {
- 'chat': ['Driver: ' + interaction[0]]
- }
- for content in interaction[1:]:
- text = 'Support: '+ content
- chat['chat'].append(text)
-
- Chatbot.train(chat)
|
