diff --git a/gpt3discord.py b/gpt3discord.py index 3c9a65d..970fba1 100644 --- a/gpt3discord.py +++ b/gpt3discord.py @@ -53,7 +53,7 @@ except Exception: pinecone_service = None if PINECONE_TOKEN: - pinecone.init(api_key=PINECONE_TOKEN, environment="us-west1-gcp") + pinecone.init(api_key=PINECONE_TOKEN, environment=EnvService.get_pinecone_region()) PINECONE_INDEX = "conversation-embeddings" if PINECONE_INDEX not in pinecone.list_indexes(): print("Creating pinecone index. Please wait...") diff --git a/services/environment_service.py b/services/environment_service.py index e60a315..ee0334a 100644 --- a/services/environment_service.py +++ b/services/environment_service.py @@ -351,3 +351,11 @@ class EnvService: return google_search_engine_id except Exception: return None + + @staticmethod + def get_pinecone_region(): + try: + pinecone_region = os.getenv("PINECONE_REGION") + return pinecone_region + except Exception: + return "us-west1-gcp"