From 17807d5a569eaab3be4fe0327f7d436d49d82b72 Mon Sep 17 00:00:00 2001 From: Luke Wulf Date: Thu, 27 Jun 2024 16:32:02 -0700 Subject: [PATCH] Address comments --- backend/danswer/chat/load_yamls.py | 4 +++- backend/requirements/model_server.txt | 6 +++--- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/backend/danswer/chat/load_yamls.py b/backend/danswer/chat/load_yamls.py index 3320a02f2..193d3e6f2 100644 --- a/backend/danswer/chat/load_yamls.py +++ b/backend/danswer/chat/load_yamls.py @@ -82,7 +82,9 @@ def load_personas_from_yaml( persona_id=(-1 * p_id) if p_id is not None else None, name=persona["name"], description=persona["description"], - num_chunks=persona.get("num_chunks", default_chunks), + num_chunks=persona.get("num_chunks") + if persona.get("num_chunks") is not None + else default_chunks, use_recent_documents=persona.get("use_recent_documents", False), num_days=persona.get("num_days"), llm_relevance_filter=persona.get("llm_relevance_filter"), diff --git a/backend/requirements/model_server.txt b/backend/requirements/model_server.txt index bc6388621..4ef8ffa5b 100644 --- a/backend/requirements/model_server.txt +++ b/backend/requirements/model_server.txt @@ -1,9 +1,9 @@ fastapi==0.109.2 -h5py==3.10.0 +h5py==3.9.0 pydantic==1.10.13 safetensors==0.4.2 sentence-transformers==2.6.1 -tensorflow==2.16.1 -torch==2.3.1 +tensorflow==2.15.0 +torch==2.0.1 transformers==4.39.2 uvicorn==0.21.1