jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Refactor/react agent (#3355)

Browse source
This commit is contained in:
Yeuoly 2024-04-11 18:34:17 +08:00 • committed by GitHub
commit cea107b165
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
9 changed files with 589 additions and 511 deletions
Download patch file Download diff file Expand all files Collapse all files

17
api/core/agent/entities.py
View file

@ -34,12 +34,29 @@ class AgentScratchpadUnit(BaseModel):
action_name: str
action_input: Union[dict, str]
def to_dict(self) -> dict:
"""
Convert to dictionary.
"""
return {
'action': self.action_name,
'action_input': self.action_input,
}
agent_response: Optional[str] = None
thought: Optional[str] = None
action_str: Optional[str] = None
observation: Optional[str] = None
action: Optional[Action] = None
def is_final(self) -> bool:
"""
Check if the scratchpad unit is final.
"""
return self.action is None or (
'final' in self.action.action_name.lower() and
'answer' in self.action.action_name.lower()
)
class AgentEntity(BaseModel):
"""