from langchain.load.serializable import Serializable


class DBOrigin(Serializable):
    """Class for storing a db_type and db_path: dict"""
    db_type: str
    db_path: dict

    def to_dict(self) -> dict:
        return {
            "db_type": self.db_type,
            "db_path": self.db_path
        }
