Kotta commited on
Commit ·
b66d193
1
Parent(s): c8a9733
feature: implemented the base64 decode module for firebase key.
Browse files
Brain/src/model/req_model.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
from typing import Any
|
| 2 |
|
| 3 |
from Brain.src.common.utils import (
|
|
@@ -31,9 +32,7 @@ class ReqModel:
|
|
| 31 |
self.pinecone_env = (
|
| 32 |
PINECONE_ENV if data["pinecone_env"] == "" else data["pinecone_env"]
|
| 33 |
)
|
| 34 |
-
self.firebase_key = (
|
| 35 |
-
FIREBASE_ENV if data["firebase_key"] == "" else data["firebase_key"]
|
| 36 |
-
)
|
| 37 |
self.token = data["token"]
|
| 38 |
self.uuid = data["uuid"]
|
| 39 |
self.settings = self.Settings()
|
|
@@ -49,3 +48,9 @@ class ReqModel:
|
|
| 49 |
"token": self.token,
|
| 50 |
"uuid": self.uuid,
|
| 51 |
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import base64
|
| 2 |
from typing import Any
|
| 3 |
|
| 4 |
from Brain.src.common.utils import (
|
|
|
|
| 32 |
self.pinecone_env = (
|
| 33 |
PINECONE_ENV if data["pinecone_env"] == "" else data["pinecone_env"]
|
| 34 |
)
|
| 35 |
+
self.firebase_key = self.decode_firebase_key(data["firebase_key"])
|
|
|
|
|
|
|
| 36 |
self.token = data["token"]
|
| 37 |
self.uuid = data["uuid"]
|
| 38 |
self.settings = self.Settings()
|
|
|
|
| 48 |
"token": self.token,
|
| 49 |
"uuid": self.uuid,
|
| 50 |
}
|
| 51 |
+
|
| 52 |
+
def decode_firebase_key(self, data: str) -> Any:
|
| 53 |
+
decoded_data = base64.b64decode(data)
|
| 54 |
+
|
| 55 |
+
# Printing the decoded string
|
| 56 |
+
return decoded_data.decode("utf-8")
|