Merge pull request #107 from ttt246/bugfix/train_issue
Browse files
Brain/src/router/train_router.py
CHANGED
|
@@ -56,7 +56,7 @@ def construct_blueprint_train_api() -> APIRouter:
|
|
| 56 |
"""@generator.response( status_code=200, schema={"message": "message", "result": {"document_id": "document_id",
|
| 57 |
"page_content":"page_content"}} )"""
|
| 58 |
|
| 59 |
-
@router.post("/{document_id}")
|
| 60 |
def read_one_document(document_id: str, data: BasicReq):
|
| 61 |
# parsing params
|
| 62 |
try:
|
|
|
|
| 56 |
"""@generator.response( status_code=200, schema={"message": "message", "result": {"document_id": "document_id",
|
| 57 |
"page_content":"page_content"}} )"""
|
| 58 |
|
| 59 |
+
@router.post("/read/{document_id}")
|
| 60 |
def read_one_document(document_id: str, data: BasicReq):
|
| 61 |
# parsing params
|
| 62 |
try:
|
Brain/src/service/train_service.py
CHANGED
|
@@ -42,7 +42,7 @@ class TrainService:
|
|
| 42 |
|
| 43 |
"""read all documents from firestore"""
|
| 44 |
|
| 45 |
-
def read_all_documents(self):
|
| 46 |
self.init_firestore()
|
| 47 |
query = self.documents_ref.order_by("timestamp")
|
| 48 |
docs = query.stream()
|
|
|
|
| 42 |
|
| 43 |
"""read all documents from firestore"""
|
| 44 |
|
| 45 |
+
def read_all_documents(self) -> Any:
|
| 46 |
self.init_firestore()
|
| 47 |
query = self.documents_ref.order_by("timestamp")
|
| 48 |
docs = query.stream()
|