File size: 1,605 Bytes
b66d193 286137e 28af5bc 286137e 28af5bc 286137e 28af5bc 286137e 28af5bc 286137e b66d193 286137e b66d193 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 | 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")
|