diff --git a/src/blackbox/emotion.py b/src/blackbox/emotion.py index 4d2d060..80b6966 100644 --- a/src/blackbox/emotion.py +++ b/src/blackbox/emotion.py @@ -29,8 +29,7 @@ class Emotion(Blackbox): data = args[0] return isinstance(data, str) - def processing(self, *args, **kwargs) -> int: - text = args[0] + def processing(self, text: str) -> int: text = "Please use one word to infer the emotion of the following passage:\n" + text + "\nJust print out that signle word pls." text = [{'role': 'user', 'content': text}] return self.model.stream_chat(text) @@ -47,4 +46,4 @@ class Emotion(Blackbox): text = [{'role': 'user', 'content': text}] sentiment = self.processing(text) return JSONResponse(content={"sentiment": sentiment }, status_code=status.HTTP_200_OK) - \ No newline at end of file +