更新riper-5.mdc配置以启用自动应用,优化DifyApi类,新增文档子分段管理功能,包括获取、添加、更新和删除子分段的方法,同时改进OpenAiLLM类的初始化逻辑以支持环境变量配置。

This commit is contained in:
2025-07-03 11:34:31 +08:00
parent b8ccdcf287
commit 68e3677c34
3 changed files with 190 additions and 21 deletions
+14 -8
View File
@@ -127,16 +127,22 @@ class XinferenceReRankerModel:
class OpenAiLLM:
def __init__(self, **kwargs):
if kwargs.get("api_key") == None or kwargs.get("base_url") == None or kwargs.get("model") == None:
raise ValueError("api_key, base_url, model 不能为空")
if "api_key" in kwargs:
self._api_key = kwargs.get("api_key")
kwargs.pop("api_key")
self._api_key = kwargs.get("api_key")
self._url = kwargs.get("base_url")
self._model = kwargs.get("model")
if "base_url" in kwargs:
self._url = kwargs.get("base_url")
kwargs.pop("base_url")
else:
self._url = os.getenv("OPENAI_API_BASE")
if "model" in kwargs:
self._model = kwargs.get("model")
kwargs.pop("model")
else:
self._model = os.getenv("LLM_MODEL_NAME")
kwargs.pop("api_key")
kwargs.pop("base_url")
kwargs.pop("model")
self._kwargs = kwargs
def invoke(self, user_prompt="你是谁?", need_retry=True):