ivanwu/jarvis-models
1
0
Fork 0
mirror of https://github.com/BoardWare-Genius/jarvis-models.git synced 2025-12-13 16:53:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: tesou

Browse Source
This commit is contained in:
superobk
2024-03-27 16:36:04 +08:00
parent 2a0c0e0477
commit 5e807a7e56
2 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/blackbox/blackbox_factory.py
View File

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

6
src/blackbox/tesou.py
View File

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