opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-13 08:57:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: dynamically set created_by_role and created_by in BaseAgentRunner

Browse Source 
Signed-off-by: NeatGuyCoding <15627489+NeatGuyCoding@users.noreply.github.com>
This commit is contained in:
NeatGuyCoding 2026-01-12 14:40:14 +04:00
parent b63dfbf654
commit 391d33aad9
No known key found for this signature in database
GPG Key ID: 05140F80A94768BD
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
api/core/agent/base_agent_runner.py
View File

@ -282,12 +282,25 @@ class BaseAgentRunner(AppRunner):
return prompt_tool
def _get_creator_info(self) -> tuple[CreatorUserRole, str]:
"""
Determine created_by_role and created_by based on message source.
Returns tuple of (created_by_role, created_by).
"""
if self.message.from_end_user_id:
return CreatorUserRole.END_USER, self.message.from_end_user_id
elif self.message.from_account_id:
return CreatorUserRole.ACCOUNT, self.message.from_account_id
else:
return CreatorUserRole.ACCOUNT, self.user_id
def create_agent_thought(
self, message_id: str, message: str, tool_name: str, tool_input: str, messages_ids: list[str]
) -> str:
"""
Create agent thought
"""
created_by_role, created_by = self._get_creator_info()
thought = MessageAgentThought(
message_id=message_id,
message_chain_id=None,
@ -312,8 +325,8 @@ class BaseAgentRunner(AppRunner):
position=self.agent_thought_count + 1,
currency="USD",
latency=0,
created_by_role=CreatorUserRole.ACCOUNT,
created_by=self.user_id,
created_by_role=created_by_role,
created_by=created_by,
)
db.session.add(thought)