diff --git a/api/apps/sdk/chat.py b/api/apps/sdk/chat.py index 0081077ce..aa717a875 100644 --- a/api/apps/sdk/chat.py +++ b/api/apps/sdk/chat.py @@ -111,7 +111,7 @@ def create(tenant_id): req['prompt_config'] = {} for key in key_list_2: temp = req['prompt_config'].get(key) - if not temp: + if (not temp and key == 'system') or (key not in req["prompt_config"]): req['prompt_config'][key] = default_prompt[key] for p in req['prompt_config']["parameters"]: if p["optional"]: diff --git a/api/apps/sdk/session.py b/api/apps/sdk/session.py index 522c3028f..92ab83c43 100644 --- a/api/apps/sdk/session.py +++ b/api/apps/sdk/session.py @@ -413,7 +413,7 @@ def list_session(chat_id,tenant_id): "document_id": chunk["doc_id"], "document_name": chunk["docnm_kwd"], "dataset_id": chunk["kb_id"], - "image_id": chunk["img_id"], + "image_id": chunk["image_id"], "similarity": chunk["similarity"], "vector_similarity": chunk["vector_similarity"], "term_similarity": chunk["term_similarity"], diff --git a/sdk/python/ragflow_sdk/modules/session.py b/sdk/python/ragflow_sdk/modules/session.py index cefb8d3c3..a619411b5 100644 --- a/sdk/python/ragflow_sdk/modules/session.py +++ b/sdk/python/ragflow_sdk/modules/session.py @@ -17,11 +17,11 @@ class Session(Base): self.__session_type = "agent" super().__init__(rag, res_dict) - def ask(self, question): + def ask(self, question,stream=True): if self.__session_type == "agent": - res=self._ask_agent(question) + res=self._ask_agent(question,stream) elif self.__session_type == "chat": - res=self._ask_chat(question) + res=self._ask_chat(question,stream) for line in res.iter_lines(): line = line.decode("utf-8") if line.startswith("{"): @@ -43,11 +43,11 @@ class Session(Base): yield message - def _ask_chat(self, question: str, stream: bool = False): + def _ask_chat(self, question: str, stream: bool): res = self.post(f"/chats/{self.chat_id}/completions", {"question": question, "stream": True,"session_id":self.id}, stream=stream) return res - def _ask_agent(self,question:str,stream:bool=False): + def _ask_agent(self,question:str,stream:bool): res = self.post(f"/agents/{self.agent_id}/completions", {"question": question, "stream": True,"session_id":self.id}, stream=stream) return res