Skip to content

Commit 0a98783

Browse files
committed
update chainserver with comments
1 parent d333986 commit 0a98783

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

RetrievalAugmentedGeneration/common/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -126,8 +126,8 @@ def get_vector_index() -> VectorStoreIndex:
126126

127127
logger.info(f"Using {config.vector_store.name} as vector store")
128128
if config.vector_store.name == "pgvector":
129-
connection_string = f"postgresql://{os.getenv('POSTGRES_USER', '')}:{os.getenv('POSTGRES_PASSWORD', '')}@{config.vector_store.url}/{os.getenv('POSTGRES_DB', '')}"
130-
db_name = os.getenv('POSTGRES_DB', '')
129+
db_name = os.getenv('POSTGRES_DB', 'vector_db')
130+
connection_string = f"postgresql://{os.getenv('POSTGRES_USER', '')}:{os.getenv('POSTGRES_PASSWORD', '')}@{config.vector_store.url}/{db_name}"
131131

132132
conn = psycopg2.connect(connection_string)
133133
conn.autocommit = True

deploy/compose/compose.env

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,6 @@ export MODEL_DIRECTORY="/home/nvidia/llama2_13b_chat_hf_v1/"
66
# Fill this out if you dont have a GPU. Leave this empty if you have a local GPU
77
export NVIDIA_API_KEY="nvapi-*"
88

9-
export AI_PLAYGROUND_API_KEY="nvapi-*"
10-
119
# flag to enable activation aware quantization for the LLM
1210
# export QUANTIZATION="int4_awq"
1311

0 commit comments

Comments
 (0)