Update test cases for intent-classification endpoint

app.py

@@ -18,6 +18,7 @@ from pydantic import BaseModel
 from mathtext_fastapi.logging import prepare_message_data_for_logging
 from mathtext_fastapi.conversation_manager import manage_conversation_response
 from mathtext_fastapi.nlu import evaluate_message_with_nlu
+from mathtext_fastapi.nlu import run_intent_classification
 
 app = FastAPI()
 
@@ -88,6 +89,13 @@ async def programmatic_message_manager(request: Request):
     return JSONResponse(context)
 
 
+@app.post("/intent-classification")
+def intent_classification_ep(content: Text = None):
+    ml_response = run_intent_classification(content.content)
+    content = {"message": ml_response}
+    return JSONResponse(content=content)
+
+
 @app.post("/nlu")
 async def evaluate_user_message_with_nlu_api(request: Request):
     """ Calls nlu evaluation and returns the nlu_response

mathtext_fastapi/nlu.py

@@ -145,9 +145,6 @@ def evaluate_message_with_nlu(message_data):
     number_api_resp = text2int(message_text.lower())
 
     if number_api_resp == 32202:
-        print("MESSAGE TEXT")
-        print(message_text)
-        print("============")
         intent_api_response = run_intent_classification(message_text)
         if intent_api_response['data']:
             return intent_api_response

scripts/make_request.py

@@ -44,7 +44,7 @@ def run_simulated_request(endpoint, sample_answer, context=None):
     print(f"Case: {sample_answer}")
     b_string = add_message_text_to_sample_object(sample_answer)
 
-    if endpoint == 'sentiment-analysis' or endpoint == 'text2int':
+    if endpoint == 'sentiment-analysis' or endpoint == 'text2int' or endpoint =='intent-classification':
         request = requests.post(
             url=f'http://localhost:7860/{endpoint}',
             json={'content': sample_answer}
@@ -58,6 +58,7 @@ def run_simulated_request(endpoint, sample_answer, context=None):
     print(request)
 
 
+run_simulated_request('intent-classification', 'exit')
 run_simulated_request('sentiment-analysis', 'I reject it')
 run_simulated_request('text2int', 'seven thousand nine hundred fifty seven')
 run_simulated_request('nlu', 'test message')