From 76a1462e370297d3dab0d2b2c4ef4fa8739c9232 Mon Sep 17 00:00:00 2001 From: cs0lar Date: Sat, 8 Apr 2023 16:11:31 +0100 Subject: [PATCH] moved pinecone api config settings into provider class --- scripts/main.py | 2 -- scripts/providers/pinecone.py | 3 +++ 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/scripts/main.py b/scripts/main.py index 51495e20..4740c626 100644 --- a/scripts/main.py +++ b/scripts/main.py @@ -281,8 +281,6 @@ next_action_count = 0 # Make a constant: user_input = "Determine which next command to use, and respond using the format specified above:" -# raise an exception if pinecone_api_key or region is not provided -if not cfg.pinecone_api_key or not cfg.pinecone_region: raise Exception("Please provide pinecone_api_key and pinecone_region") # Initialize memory and make sure it is empty. # this is particularly important for indexing and referencing pinecone memory memory = MemoryFactory.get_memory(cfg.memory_provider) diff --git a/scripts/providers/pinecone.py b/scripts/providers/pinecone.py index 971ef186..e8cf019a 100644 --- a/scripts/providers/pinecone.py +++ b/scripts/providers/pinecone.py @@ -6,6 +6,9 @@ cfg = Config() class PineconeMemory(Memory): def __init__(self): + # raise an exception if pinecone_api_key or region is not provided + if not cfg.pinecone_api_key or not cfg.pinecone_region: raise Exception("Please provide pinecone_api_key and pinecone_region") + pinecone_api_key = cfg.pinecone_api_key pinecone_region = cfg.pinecone_region pinecone.init(api_key=pinecone_api_key, environment=pinecone_region)