소스 검색

Train default

donzito 2 년 전
부모
커밋
b56d1d63d4
2개의 변경된 파일11개의 추가작업 그리고 8개의 파일을 삭제
  1. 1 0
      .gitignore
  2. 10 8
      adm/management/commands/trainChatbot.py

+ 1 - 0
.gitignore

@@ -1,3 +1,4 @@
 /storage
 /storage/*
 .DS_Store
+.vscode/launch.json

+ 10 - 8
adm/management/commands/trainChatbot.py

@@ -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)