Merge pull request #5 from ittia-research/dev

update get stack
ittia-research · Aug 5, 2024 · e235fb8 · e235fb8
2 parents 5d0e761 + cb8bc1c
commit e235fb8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/utils.py b/src/utils.py
@@ -138,7 +138,8 @@ async def get_stack():
         "Embedding model": settings.EMBEDDING_MODEL_NAME,
         "Rerank model": settings.RERANK_MODEL_NAME,
         "RAG chunk sizes": settings.RAG_CHUNK_SIZES,
-        "RAG deploy mode": settings.RAG_MODEL_DEPLOY,
+        "Embedding deploy mode": settings.EMBEDDING_MODEL_DEPLOY,
+        "Rerank deploy mode": settings.RERANK_MODEL_DEPLOY,
     }
     return stack