Quellcode durchsuchen

feat: 修改配置

huangzhichao vor 2 Tagen
Ursprung
Commit
fb5d8f52ac

+ 3 - 7
app/core/config.py

@@ -14,16 +14,12 @@ except Exception:
 
 @dataclass
 class Settings:
-    llm_provider: str = os.getenv("LLM_PROVIDER", "mock").lower()
-    default_model: str = os.getenv("LLM_MODEL", "mock-echo-001")
-    temperature: float = float(os.getenv("LLM_TEMPERATURE", "1.0"))
-    openai_api_key: Optional[str] = os.getenv("OPENAI_API_KEY")
-    dashscope_api_key: Optional[str] = (
-        os.getenv("DASHSCOPE_API_KEY") or os.getenv("ALIYUN_APIKEY")
-    )
+    airouter_api_key: Optional[str] = os.getenv("AIROUTER_APIKEY")
+    aliyun_api_key: Optional[str] = os.getenv("ALIYUN_APIKEY")
     upload_path: Optional[str] = os.getenv(
         "UPLOAD_PATH", "https://api.piaoquantv.com/ad/file/upload"
     )
+    base_url: Optional[str] = os.getenv("BASE_URL")
 
 
 def get_settings() -> Settings:

+ 2 - 2
app/providers/evaluation_provider.py

@@ -122,8 +122,8 @@ class EvaluationProvider:
 
   def copywriting_evaluation(self, image_url: str, text: str, model: str) -> DataResponse:
         client = OpenAI(
-            api_key = settings.dashscope_api_key or "",
-            base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
+            api_key = settings.aliyun_api_key or '',
+            base_url= settings.base_url  or '',
         )
 
         if not client:

+ 2 - 2
app/providers/speech_provider.py

@@ -17,7 +17,7 @@ from ..core.logger import get_logger
 
 settings = get_settings()
 # Configure DashScope API key from env/.env
-dashscope.api_key = settings.dashscope_api_key or ""
+dashscope.api_key = settings.aliyun_api_key or ""
 UPLOAD_PATH = settings.upload_path or ""
 
 # module logger
@@ -50,7 +50,7 @@ class SpeechProvider:
             logger.error(msg)
             return DataResponse(code=1, data=None, msg=msg)
         if not dashscope.api_key:
-            msg = "DASHSCOPE_API_KEY is missing"
+            msg = "ALIYUN_APIKEY is missing"
             logger.error(msg)
             return DataResponse(code=1, data=None, msg=msg)
 

+ 2 - 2
app/providers/understand_image_provider.py

@@ -245,8 +245,8 @@ class UnderstandImageProvider:
 
     def _create_client(self) -> OpenAI:
         return OpenAI(
-            api_key = settings.dashscope_api_key or "",
-            base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
+            api_key = settings.aliyun_api_key or "",
+            base_url= settings.base_url or "",
         )
 
     def understand_image(self, image_url: str, *, model: str) -> DataResponse:

+ 0 - 0
script/audit.py