|
|
@@ -5,16 +5,18 @@ from adm.constants import CTS
|
|
|
class Command(BaseCommand):
|
|
|
|
|
|
def add_arguments(self, parser):
|
|
|
- parser.add_argument('interaction', type=str, help='Chatbot interaction to train')
|
|
|
+ parser.add_argument('interaction', type=str, nargs='?', default=None, 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)
|
|
|
+ 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)
|