Explorar el Código

fix: query empty not allow (#255)

John Wang hace 2 años
padre
commit
ccd80653ff
Se han modificado 1 ficheros con 4 adiciones y 0 borrados
  1. 4 0
      api/services/completion_service.py

+ 4 - 0
api/services/completion_service.py

@@ -33,6 +33,10 @@ class CompletionService:
         # is streaming mode
         inputs = args['inputs']
         query = args['query']
+
+        if not query:
+            raise ValueError('query is required')
+
         conversation_id = args['conversation_id'] if 'conversation_id' in args else None
 
         conversation = None