增加模型可配置功能

ui.py

@@ -1,3 +1,4 @@
+import os
 from typing import List
 
 import streamlit as st
@@ -83,14 +84,16 @@ def show_tabs():
     #     if tabs[index]["name"] == selected_tab:
     #         set_current_page(value["name"])
 
-model_options = {
-        "Qwen2.5-72B": "openai:Qwen/Qwen2.5-72B-Instruct",
-        "o3-mini": "openai:o3-mini",
-        "gpt-4o": "openai:gpt-4o",
-        "gemini-2.0-flash-exp": "google:gemini-2.0-flash-exp",
-        "claude-3-5-sonnet": "anthropic:claude-3-5-sonnet-20241022",
-        "llama-3.3-70b": "groq:llama-3.3-70b-versatile",
-    }
+# model_options = {
+#         "Qwen2.5-72B": "openai:Qwen/Qwen2.5-72B-Instruct",
+#         "o3-mini": "openai:o3-mini",
+#         "gpt-4o": "openai:gpt-4o",
+#         "gemini-2.0-flash-exp": "google:gemini-2.0-flash-exp",
+#         "claude-3-5-sonnet": "anthropic:claude-3-5-sonnet-20241022",
+#         "llama-3.3-70b": "groq:llama-3.3-70b-versatile",
+#     }
+
+model_options = dict(item.split('=') for item in os.getenv("MODEL_LIST", "Qwen2.5-72B=openai:Qwen/Qwen2.5-72B-Instruct").split('&'))
 
 def get_modul_option(id: int = 0) -> str:
     """Return the selected module option"""