fix(llm_api_default.py, config.py): fix a bug that prevented the use of certain models and modified the naming of config parameters.

Author: Robbings

.gitignore

@@ -3,3 +3,7 @@ logs
 app/mrmax.py
 db
 /venv/
+.idea
+.venv
+*.iml
+*.pyc

config/config.py

@@ -8,13 +8,14 @@
 # api 接口封装类
 llm_api_impl = "llm_api.llm_api_default.LLMApiDefault"
 
-# api 配置默使用认UnionLLM，参考：https://github.com/EvalsOne/UnionLLM/tree/main/docs
+# api 配置方式参考 docs/llm_api.md
+# 默认使用认UnionLLM，参考：https://github.com/EvalsOne/UnionLLM/tree/main/docs
 # UnionLLM兼容LiteLLM，参考LiteLLM文档：https://docs.litellm.ai/docs
 api_config = {
-    "OPENAI_API_KEY": "your openai key",
-    "OPENAI_API_BASE": "https://api.openai.com/v1",
-    "MODEL_NAME": model_gpt_4o,
-    "PROVIDER": "openai",
+    "api_key": "your openai key",
+    "api_base": "https://api.openai.com/v1",
+    "model": model_gpt_4o,
+    "provider": "openai",
 }
 
 # 2. 提示词

llm_api/llm_api_default.py

@@ -11,24 +11,22 @@
 class LLMApiDefault(LLMApiInterface):
 
     def __init__(self):
-        self.model_name = None
+        self.params = {}
         self.response = None
-        self.provider = None
 
     def set_config(self, api_config: dict) -> bool:
         if api_config is None:
             raise ValueError("api_config is None")
         for key in api_config:
-            if key == "MODEL_NAME":
-                self.model_name = api_config[key]
-            if key == "PROVIDER":
-                self.provider = api_config[key]
-            os.environ[key] = api_config[key]
+            self.params[key] = api_config[key]
+            # 如果为大写，则写入环境变量
+            if key.isupper():
+                os.environ[key] = api_config[key]
         return True
 
     def generate_text(self, messages: list) -> bool:
         try:
-            self.response = unionchat(provider=self.provider, model=self.model_name, messages=messages)
+            self.response = unionchat(messages=messages, **self.params)
         except Exception as e:
             raise e
         return True