| import base64 | |
| from typing import Any | |
| from Brain.src.common.utils import ( | |
| OPENAI_API_KEY, | |
| PINECONE_KEY, | |
| PINECONE_ENV, | |
| FIREBASE_ENV, | |
| ) | |
| class ReqModel: | |
| class Settings: | |
| def __init__(self): | |
| self.temperature: float = 0.6 | |
| openai_key: str | |
| pinecone_key: str | |
| pinecone_env: str | |
| firebase_key: str | |
| token: str = "" | |
| uuid: str = "" | |
| settings: Settings | |
| def __init__(self, data: dict): | |
| self.openai_key = ( | |
| OPENAI_API_KEY if data["openai_key"] == "" else data["openai_key"] | |
| ) | |
| self.pinecone_key = ( | |
| PINECONE_KEY if data["pinecone_key"] == "" else data["pinecone_key"] | |
| ) | |
| self.pinecone_env = ( | |
| PINECONE_ENV if data["pinecone_env"] == "" else data["pinecone_env"] | |
| ) | |
| self.firebase_key = self.decode_firebase_key(data["firebase_key"]) | |
| self.token = data["token"] | |
| self.uuid = data["uuid"] | |
| self.settings = self.Settings() | |
| self.settings.temperature = data["settings"]["temperature"] | |
| def to_json(self): | |
| return { | |
| "openai_key": self.openai_key, | |
| "pinecone_key": self.pinecone_key, | |
| "pinecone_env": self.pinecone_env, | |
| "firebase_key": self.firebase_key, | |
| "settings": {"temperature": self.settings.temperature}, | |
| "token": self.token, | |
| "uuid": self.uuid, | |
| } | |
| def decode_firebase_key(self, data: str) -> Any: | |
| decoded_data = base64.b64decode(data) | |
| # Printing the decoded string | |
| return decoded_data.decode("utf-8") | |