fahmiaziz98 commited on
Commit
87ef7db
·
1 Parent(s): 5e48e1f
Files changed (2) hide show
  1. app.py +2 -8
  2. core/model_manager.py +1 -1
app.py CHANGED
@@ -52,17 +52,12 @@ High-performance API for document reranking using multiple state-of-the-art mode
52
  - Input: Query + documents + optional instruction
53
  - Output: Ranked documents with relevance scores
54
 
55
- **Warning**: Not use production!.
56
  """,
57
  version="1.0.0",
58
  lifespan=lifespan
59
  )
60
 
61
-
62
- # -------------------------
63
- # API Endpoints
64
- # -------------------------
65
-
66
  @app.post("/rerank", response_model=RerankResponse, tags=["Reranking"])
67
  async def rerank_documents(request: RerankRequest):
68
  """
@@ -105,8 +100,7 @@ async def rerank_documents(request: RerankRequest):
105
 
106
  model = model_manager.get_model(request.model_id)
107
  original_indices, documents = zip(*valid_docs)
108
- logger.info(f"Query: {request.query.strip()}")
109
- logger.info(f"Document: {list(documents)}")
110
  scores = model.rerank(
111
  query=request.query.strip(),
112
  documents=list(documents),
 
52
  - Input: Query + documents + optional instruction
53
  - Output: Ranked documents with relevance scores
54
 
55
+ ⚠️ **Warning**: Not for production use!.
56
  """,
57
  version="1.0.0",
58
  lifespan=lifespan
59
  )
60
 
 
 
 
 
 
61
  @app.post("/rerank", response_model=RerankResponse, tags=["Reranking"])
62
  async def rerank_documents(request: RerankRequest):
63
  """
 
100
 
101
  model = model_manager.get_model(request.model_id)
102
  original_indices, documents = zip(*valid_docs)
103
+
 
104
  scores = model.rerank(
105
  query=request.query.strip(),
106
  documents=list(documents),
core/model_manager.py CHANGED
@@ -128,7 +128,7 @@ class ModelManager:
128
  "id": model_id,
129
  "name": config.get("model_name"),
130
  "type": config.get("model_type"),
131
- "language": config.get("language"),
132
  "description": config.get("description"),
133
  "repository": config.get("repository"),
134
  "loaded": model.loaded if model else False
 
128
  "id": model_id,
129
  "name": config.get("model_name"),
130
  "type": config.get("model_type"),
131
+ "language": config.get("languages"),
132
  "description": config.get("description"),
133
  "repository": config.get("repository"),
134
  "loaded": model.loaded if model else False