om4r932 commited on
Commit
e8ef860
·
1 Parent(s): 582da9b

Add version response key

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -152,6 +152,7 @@ class DocRequest(BaseModel):
152
  class DocResponse(BaseModel):
153
  doc_id: str
154
  url: str
 
155
  scope: Optional[str] = None
156
  search_time: float
157
 
@@ -480,10 +481,12 @@ def find_document(request: DocRequest):
480
  version = result.split("/")[-1].replace(".zip", "").split("-")[-1]
481
  return DocResponse(
482
  doc_id=request.doc_id,
 
483
  url=result,
484
  search_time=time.time() - start_time
485
  ) if isinstance(finder, TsgDocFinder) else DocResponse(
486
  doc_id=request.doc_id,
 
487
  url=result,
488
  search_time=time.time() - start_time,
489
  scope=finder.indexer_scopes[request.doc_id] if request.doc_id in finder.indexer_scopes else get_scope(request.doc_id, version)
 
152
  class DocResponse(BaseModel):
153
  doc_id: str
154
  url: str
155
+ version: str
156
  scope: Optional[str] = None
157
  search_time: float
158
 
 
481
  version = result.split("/")[-1].replace(".zip", "").split("-")[-1]
482
  return DocResponse(
483
  doc_id=request.doc_id,
484
+ version=version,
485
  url=result,
486
  search_time=time.time() - start_time
487
  ) if isinstance(finder, TsgDocFinder) else DocResponse(
488
  doc_id=request.doc_id,
489
+ version=version,
490
  url=result,
491
  search_time=time.time() - start_time,
492
  scope=finder.indexer_scopes[request.doc_id] if request.doc_id in finder.indexer_scopes else get_scope(request.doc_id, version)