pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 12:32:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: Pass kwargs in python api #10699 (#10808)

Browse Source 
### What problem does this PR solve?

Fix: Pass kwargs in python api #10699

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Billy Bao
2025-10-27 15:18:56 +08:00
committed by GitHub
parent 5312b75362
commit 16ec6ad346
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
sdk/python/ragflow_sdk/modules/session.py
View File

@ -40,7 +40,7 @@ class Session(Base):
If stream=False, returns a single Message object for the final answer. If stream=False, returns a single Message object for the final answer.
""" """
if self.__session_type == "agent": if self.__session_type == "agent":
res = self._ask_agent(question, stream) res = self._ask_agent(question, stream, **kwargs)
elif self.__session_type == "chat": elif self.__session_type == "chat":
res = self._ask_chat(question, stream, **kwargs) res = self._ask_chat(question, stream, **kwargs)
else: else:
@ -97,9 +97,11 @@ class Session(Base):
json_data, stream=stream) json_data, stream=stream)
return res return res
def _ask_agent(self, question: str, stream: bool): def _ask_agent(self, question: str, stream: bool, **kwargs):
json_data = {"question": question, "stream": stream, "session_id": self.id}
json_data.update(kwargs)
res = self.post(f"/agents/{self.agent_id}/completions", res = self.post(f"/agents/{self.agent_id}/completions",
{"question": question, "stream": stream, "session_id": self.id}, stream=stream) json_data, stream=stream)
return res return res
def update(self, update_message): def update(self, update_message):