feat: tesou

src/blackbox/blackbox_factory.py

@@ -14,11 +14,11 @@ class BlackboxFactory:
         self.tts = TTS()
         self.asr = ASR(".env.yaml")
         self.sentiment = Sentiment()
-        #self.sum = SUM()
-        #self.calculator = Calculator()
-        #self.audio_to_text = AudioToText()
-        #self.text_to_audio = TextToAudio()
-        #self.tesou = Tesou()
+        self.sum = SUM()
+        self.calculator = Calculator()
+        self.audio_to_text = AudioToText()
+        self.text_to_audio = TextToAudio()
+        self.tesou = Tesou()
 
     def __call__(self, *args, **kwargs):
         return self.processing(*args, **kwargs)

src/blackbox/tesou.py

@@ -23,16 +23,16 @@ class Tesou(Blackbox):
             "user_id": id,
             "prompt": prompt,
         }
-
+        print(message)
         response = requests.post(url, json=message)
-        return response
+        return response.json()
     
     async def fast_api_handler(self, request: Request) -> Response:
         try:
             data = await request.json()
         except:
             return JSONResponse(content={"error": "json parse error"}, status_code=status.HTTP_400_BAD_REQUEST)
-        user_id = data.get("id")
+        user_id = data.get("user_id")
         user_prompt = data.get("prompt")    
         if user_prompt is None:
             return JSONResponse(content={"error": "question is required"}, status_code=status.HTTP_400_BAD_REQUEST)