| 12345678910111213141516171819202122 |
- 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, nargs='?', default=None, help='Chatbot interaction to train')
-
- def handle(self, *args, **options):
- interaction = options['interaction']
- chat = None
- if interaction:
- interaction = interaction.split('>>>')
- chat = {
- 'chat': ['Driver: ' + interaction[0]]
- }
- for content in interaction[1:]:
- text = 'Support: '+ content
- chat['chat'].append(text)
-
- Chatbot.train(chat)
|
