更新riper-5.mdc配置以启用自动应用,优化DifyApi类,新增文档子分段管理功能,包括获取、添加、更新和删除子分段的方法,同时改进OpenAiLLM类的初始化逻辑以支持环境变量配置。
This commit is contained in:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user