Thomas Richardson commited on
Commit
4c99672
·
unverified ·
2 Parent(s): 6cc2ac1aadfb3f

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()