youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-04 01:18:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: support agent node

Browse Source
This commit is contained in:
Yeuoly
2024-12-09 23:02:11 +08:00
parent 16b49ac436
commit ae72514cb4
15 changed files with 459 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
api/core/agent/entities.py
View File

@ -3,7 +3,7 @@ from typing import Any, Optional, Union
from pydantic import BaseModel
from core.tools.entities.tool_entities import ToolProviderType
from core.tools.entities.tool_entities import ToolInvokeMessage, ToolProviderType
class AgentToolEntity(BaseModel):
@ -83,3 +83,11 @@ class AgentEntity(BaseModel):
prompt: Optional[AgentPromptEntity] = None
tools: Optional[list[AgentToolEntity]] = None
max_iteration: int = 5
class AgentInvokeMessage(ToolInvokeMessage):
"""
Agent Invoke Message.
"""
pass