[update]本番と開発環境の切り分けの仕組みを導入。

.env.example

@@ -1,3 +1,4 @@
+ENV=development
 OPENAI_API_KEY=XXX
 PINECONE_API_KEY=YYY
 AWS_ACCESS_KEY_ID=ZZZ

src/aws/s3.py

@@ -10,7 +10,9 @@ from src.utils import setup_logger
 logger = setup_logger(__name__)
 
 trial_bucket_name = "pharma-doc-trials"
-change_request_history_file_path_in_trial_bucket = "bankyo/change_request_history.csv"
+base_change_request_history_file_path_in_trial_bucket = (
+    "bankyo/change_request_history.csv"
+)
 csv_extension = ".csv"
 err_msg_file_is_not_csv = "指定したファイルはcsvではありません。"
 

src/forms/change_request.py

@@ -10,7 +10,7 @@ from dotenv import load_dotenv
 from openai import OpenAI
 from pinecone import Pinecone
 
-from src.aws.s3 import change_request_history_file_path_in_trial_bucket
+from src.aws.s3 import base_change_request_history_file_path_in_trial_bucket
 from src.aws.s3 import get_client as get_s3_client
 from src.aws.s3 import (
     get_csv_as_pd_dataframe_from_s3,
@@ -503,6 +503,19 @@ def change_request_form() -> None:
         bucket_name = "pharma-doc-trials"
 
         s3_client = get_s3_client()
+        env = os.environ["ENV"]
+        if env != "production":
+            file_name, file_extension = os.path.splitext(
+                base_change_request_history_file_path_in_trial_bucket
+            )
+            change_request_history_file_path_in_trial_bucket = (
+                f"{file_name}_{env}{file_extension}"
+            )
+        else:
+            change_request_history_file_path_in_trial_bucket = (
+                base_change_request_history_file_path_in_trial_bucket
+            )
+
         data = get_csv_as_pd_dataframe_from_s3(
             s3_client, bucket_name, change_request_history_file_path_in_trial_bucket
         )
@@ -595,6 +608,18 @@ def save_change_request_history(
 ) -> None:
     s3_client = get_s3_client()
     try:
+        env = os.environ["ENV"]
+        if env != "production":
+            file_name, file_extension = os.path.splitext(
+                base_change_request_history_file_path_in_trial_bucket
+            )
+            change_request_history_file_path_in_trial_bucket = (
+                f"{file_name}_{env}{file_extension}"
+            )
+        else:
+            change_request_history_file_path_in_trial_bucket = (
+                base_change_request_history_file_path_in_trial_bucket
+            )
         df = get_csv_as_pd_dataframe_from_s3(
             s3_client,
             trial_bucket_name,

src/forms/change_request_history.py

@@ -3,7 +3,7 @@ import os
 import streamlit as st
 
 from src.aws.s3 import (
-    change_request_history_file_path_in_trial_bucket,
+    base_change_request_history_file_path_in_trial_bucket,
     get_client,
     get_csv_as_pd_dataframe_from_s3,
     trial_bucket_name,
@@ -14,6 +14,18 @@ from src.forms.change_request import change_request_history_file_path
 def change_request_history() -> None:
     s3_client = get_client()
     try:
+        env = os.environ["ENV"]
+        if env != "production":
+            file_name, file_extension = os.path.splitext(
+                base_change_request_history_file_path_in_trial_bucket
+            )
+            change_request_history_file_path_in_trial_bucket = (
+                f"{file_name}_{env}{file_extension}"
+            )
+        else:
+            change_request_history_file_path_in_trial_bucket = (
+                base_change_request_history_file_path_in_trial_bucket
+            )
         data = get_csv_as_pd_dataframe_from_s3(
             s3_client,
             trial_bucket_name,