Address comments

This commit is contained in:
Luke Wulf 2024-06-27 16:32:02 -07:00
parent 7e7c984d3c
commit 17807d5a56
2 changed files with 6 additions and 4 deletions

View File

@ -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"),

View File

@ -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