phospho-app / phospho Goto Github PK

def detect_str_from_input(input: RawDataType) -> str:
    # OpenAI inputs: Look for messages list
    if (
        isinstance(input, dict)
        and ("messages" in input.keys())
        and isinstance(input["messages"], list)
        and (len(input["messages"]) > 0)
        and ("content" in input["messages"][-1])
    ):
        role = input["messages"][-1].get("role", None)
        content = input["messages"][-1].get("content", None)
        if role is not None and content is not None:
            return f"{role}: {content}" # Should we really add the role?
        elif content is not None:
            return str(content)

    # Unimplemented. Translate everything to str
    return str(input)

class Message:

    # Typed attributes
    def __init__(self, content: str, payload={}, metadata={}):
        self.content = content # can be empty, even though it doesn't make sense. Cannot be None
        self.payload = payload # Implemented by the dev user, can be empty
        self.metadata = metadata # timestamp, user_id, etc. TODO: define the metadata

class Message:
    content: str
    payload: dict
    metadata: dict

    # Typed attributes
    def __init__(self, content: str, payload={}, metadata={}):
        self.content = content # can be empty, even though it doesn't make sense. Cannot be None
        self.payload = payload # Implemented by the dev user, can be empty
        self.metadata = metadata # timestamp, user_id, etc. TODO: define the metadata

from dataclasses import dataclass, field

@dataclass
class Message:
    content: str # can be empty, even though it doesn't make sense. Cannot be None
    payload: dict = field(default_factory=dict) # Implemented by the dev user, can be empty
    metadata: dict = field(default_factory=dict) # timestamp, user_id, etc. TODO: define the metadata