Configure model for rag recipe

Signed-off-by: Maysun J Faisal <[email protected]>
containers · Aug 9, 2024 · 3a25158 · 3a25158
1 parent 2249889
commit 3a25158
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/recipes/natural_language_processing/rag/app/rag_app.py b/recipes/natural_language_processing/rag/app/rag_app.py
@@ -12,6 +12,7 @@
 
 model_service = os.getenv("MODEL_ENDPOINT","http://0.0.0.0:8001")
 model_service = f"{model_service}/v1"
+model_name = os.getenv("MODEL_NAME", "") 
 chunk_size = os.getenv("CHUNK_SIZE", 150)
 embedding_model = os.getenv("EMBEDDING_MODEL","BAAI/bge-base-en-v1.5")
 vdb_vendor = os.getenv("VECTORDB_VENDOR", "chromadb")
@@ -75,6 +76,7 @@ def read_file(file):
 
 llm = ChatOpenAI(base_url=model_service, 
                  api_key="EMPTY",
+                 model=model_name,
                  streaming=True,
                  callbacks=[StreamlitCallbackHandler(st.container(),
                                                      collapse_completed_thoughts=True)])