diff --git a/sdk/python/ragflow_sdk/modules/chat.py b/sdk/python/ragflow_sdk/modules/chat.py index 39e681536..01083b37d 100644 --- a/sdk/python/ragflow_sdk/modules/chat.py +++ b/sdk/python/ragflow_sdk/modules/chat.py @@ -31,7 +31,7 @@ class Chat(Base): class LLM(Base): def __init__(self, rag, res_dict): - self.model_name = "deepseek-chat" + self.model_name = None self.temperature = 0.1 self.top_p = 0.3 self.presence_penalty = 0.4 @@ -59,8 +59,7 @@ class Chat(Base): super().__init__(rag, res_dict) def update(self, update_message: dict): - res = self.put(f'/chats/{self.id}', - update_message) + res = self.put(f"/chats/{self.id}", update_message) res = res.json() if res.get("code") != 0: raise Exception(res["message"]) @@ -69,13 +68,11 @@ class Chat(Base): res = self.post(f"/chats/{self.id}/sessions", {"name": name}) res = res.json() if res.get("code") == 0: - return Session(self.rag, res['data']) + return Session(self.rag, res["data"]) raise Exception(res["message"]) - def list_sessions(self, page: int = 1, page_size: int = 30, orderby: str = "create_time", desc: bool = True, - id: str = None, name: str = None) -> list[Session]: - res = self.get(f'/chats/{self.id}/sessions', - {"page": page, "page_size": page_size, "orderby": orderby, "desc": desc, "id": id, "name": name}) + def list_sessions(self, page: int = 1, page_size: int = 30, orderby: str = "create_time", desc: bool = True, id: str = None, name: str = None) -> list[Session]: + res = self.get(f"/chats/{self.id}/sessions", {"page": page, "page_size": page_size, "orderby": orderby, "desc": desc, "id": id, "name": name}) res = res.json() if res.get("code") == 0: result_list = []