diff --git a/src/blackbox/blackbox_factory.py b/src/blackbox/blackbox_factory.py index ff0b4ea..d2828ca 100644 --- a/src/blackbox/blackbox_factory.py +++ b/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) diff --git a/src/blackbox/tesou.py b/src/blackbox/tesou.py index a81fe9b..d07d898 100755 --- a/src/blackbox/tesou.py +++ b/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)