Added OPENAI_API_BASE.

Dockerfile

@@ -13,6 +13,7 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Set the environment variables
 ENV API_KEY=""
 ENV PROXY=""
+ENV OPENAI_API_BASE=""
 ENV OTHER_ARGS="--advanced_mode"
 ENV ADVANCED_MODE="true"
 ENV SERVER_NAME="0.0.0.0"
@@ -21,4 +22,4 @@ ENV SERVER_NAME="0.0.0.0"
 EXPOSE 7860
 
 # Run the script when the container launches
-CMD /usr/local/bin/python3 meta_prompt.py --api_key=${API_KEY} --proxy=${PROXY} --server_name=${SERVER_NAME} ${OTHER_ARGS}
+CMD /usr/local/bin/python3 meta_prompt.py --api_key=${API_KEY} --proxy=${PROXY} --openai_api_base=${OPENAI_API_BASE} --server_name=${SERVER_NAME} ${OTHER_ARGS}

README.md

@@ -72,6 +72,30 @@ If you have access to ChatGPT and want to save costs on GPT-4 API usage, you can
 = 30
 ```
 
+#### Prompt After 4 Interations
+
+```
+ROLE
+
+You are a math tutor.
+
+TASK
+
+Your task is to solve the mathematical expression provided by the user and provide a concise, step-by-step solution. Each step should only include the calculation and the result, without any additional explanations or step labels.
+
+REQUIREMENTS AND RESTRICTIONS
+
+* The solution should be provided in standard mathematical notation.
+* The format of the mathematical expressions should be consistent with the user's input.
+* The symbols used in the mathematical expressions should be consistent with the user's input.
+* No spaces should be included around the mathematical operators.
+* Avoid unnecessary explanations or verbosity.
+* Do not include any additional information or explanations beyond the direct calculation steps.
+* Do not include a final solution statement.
+
+{user_message}
+```
+
 ### GDP
 
 #### Testing User Prompt

meta_prompt.py

@@ -24,9 +24,14 @@ from prompt_ui import PromptUI
 
 class ChatbotApp:
     def __init__(self, args):
-        os.environ["OPENAI_API_KEY"] = args.api_key
+        if args.api_key:
+            os.environ["OPENAI_API_KEY"] = args.api_key
+
+        if args.openai_api_base:
+            os.environ["OPENAI_API_BASE"] = args.openai_api_base
+
         if args.proxy:
-            openai.proxy = eval(args.proxy)
+            os.environ["OPENAI_PROXY"] = args.proxy
 
         self.prompt_ui = PromptUI(advanced_mode=args.advanced_mode)
 
@@ -46,6 +51,7 @@ def parse_args():
     parser.add_argument("--advanced_mode", action='store_true', default=False,
                         help="Enable advanced mode")
     parser.add_argument("--server_name", type=str, default="127.0.0.1", help="Server name or IP address")
+    parser.add_argument("--openai_api_base", type=str, default=None, help="OpenAI API base URL")
  
     return parser.parse_args()