wassname/stampy-chat
1
0
Fork 0
mirror of https://github.com/wassname/stampy-chat.git synced 2026-09-12 13:00:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

Moving Pinecone env variables assignments to settings

Browse Source
This commit is contained in:
henri123lemoine
2023-06-27 10:36:15 -04:00
parent 084b7139a1
commit 2025940db9
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/dataset/pinecone_db_handler.py
+1 -4
View File
@@ -4,7 +4,7 @@ import os
import json
import pinecone
from .settings import PINECONE_INDEX_NAME, PINECONE_VALUES_DIMS, PINECONE_METRIC, PINECONE_METADATA_ENTRIES
from .settings import PINECONE_INDEX_NAME, PINECONE_VALUES_DIMS, PINECONE_METRIC, PINECONE_METADATA_ENTRIES, PINECONE_API_KEY, PINECONE_ENVIRONMENT
import logging
logger = logging.getLogger(__name__)
@@ -17,9 +17,6 @@ class PineconeDB:
):
self.index_name = PINECONE_INDEX_NAME
PINECONE_API_KEY = os.environ["PINECONE_API_KEY"]
PINECONE_ENVIRONMENT = os.environ["PINECONE_ENVIRONMENT"]
pinecone.init(
api_key = PINECONE_API_KEY,
environment = PINECONE_ENVIRONMENT,
src/dataset/settings.py
+3
View File
@@ -1,5 +1,6 @@
# dataset/settings.py
import os
from pathlib import Path
### FILE PATHS ###
@@ -19,6 +20,8 @@ PINECONE_INDEX_NAME = "stampy-chat-embeddings-test"
PINECONE_VALUES_DIMS = EMBEDDINGS_DIMS
PINECONE_METRIC = "cosine"
PINECONE_METADATA_ENTRIES = ["entry_id", "source", "title", "authors", "text"]
PINECONE_API_KEY = os.environ["PINECONE_API_KEY"]
PINECONE_ENVIRONMENT = os.environ["PINECONE_ENVIRONMENT"]
### MISCELLANEOUS ###
MAX_NUM_AUTHORS_IN_SIGNATURE = 3