硬编码改为配置文件

black-box-data-create/agent/api_client.py

@@ -4,20 +4,22 @@ from typing import Dict, Any, Optional
 
 import requests
 
+from agent.config import load_config
+
 
 class APIClient:
     """DeepSeek API 客户端，含重试逻辑。"""
 
-    def __init__(self, api_key: str, model: str = 'deepseek-v4-flash',
-                 base_url: str = 'https://api.deepseek.com/chat/completions',
-                 max_retries: int = 3, timeout: int = 120,
-                 max_tokens: int = 32768):
-        self.api_key = api_key
-        self.model = model
-        self.base_url = base_url
-        self.max_retries = max_retries
-        self.timeout = timeout
-        self.max_tokens = max_tokens
+    def __init__(self, api_key: str = None, model: str = None,
+                 base_url: str = None, max_retries: int = None,
+                 timeout: int = None, max_tokens: int = None):
+        cfg = load_config()['api']
+        self.api_key = api_key if api_key else cfg['api_key']
+        self.model = model if model else cfg['model']
+        self.base_url = base_url if base_url else cfg['base_url']
+        self.max_retries = max_retries if max_retries is not None else cfg['max_retries']
+        self.timeout = timeout if timeout is not None else cfg['timeout']
+        self.max_tokens = max_tokens if max_tokens is not None else cfg['max_tokens']
 
     def generate(self, prompt: str) -> Dict[str, Any]:
         """发送 prompt 并返回 AI 生成的结果。"""