File size: 1,113 Bytes
acb3380 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
from typing import Dict, List, Union
from ctm.messengers.messenger_base import BaseMessenger
@BaseMessenger.register_messenger("roberta_text_sentiment_messenger") # type: ignore[no-untyped-call] # FIX ME
class RobertaTextSentimentMessenger(BaseMessenger):
def __init__(self, role=None, content=None, *args, **kwargs): # type: ignore[no-untyped-def] # FIX ME
self.init_messenger(role, content)
def init_messenger( # type: ignore[no-untyped-def] # FIX ME
self, role: str = None, content: Union[str, Dict, List] = None # type: ignore[assignment, type-arg] # FIX ME
):
self.messages = ""
if content and role:
self.update_messages(role, content) # type: ignore[attr-defined] # FIX ME
def update_message(self, role: str, content: Union[str, Dict, List]): # type: ignore[no-untyped-def, type-arg] # FIX ME
# should replace with updated message
self.messages = content # type: ignore[assignment] # FIX ME
def check_iter_round_num(self): # type: ignore[no-untyped-def] # FIX ME
return 1 if len(self.messages) > 0 else 0
|