main #2
@ -221,6 +221,63 @@ def get_local_agent_list(filters=None, order_by=None, limit_start=None, limit_pa
|
||||
return result
|
||||
|
||||
|
||||
@jingrow.whitelist(allow_guest=True)
|
||||
def get_local_tool_list(filters=None, order_by=None, limit_start=None, limit_page_length=None):
|
||||
"""获取本地工具列表"""
|
||||
|
||||
# 构建查询条件
|
||||
query_filters = {"public": 1}
|
||||
|
||||
if filters:
|
||||
if isinstance(filters, str):
|
||||
import json
|
||||
filters = json.loads(filters)
|
||||
query_filters.update(filters)
|
||||
|
||||
# 默认排序
|
||||
if not order_by:
|
||||
order_by = "tool_name asc"
|
||||
|
||||
# 获取工具列表
|
||||
tools = jingrow.get_all(
|
||||
"Local Tool",
|
||||
filters=query_filters,
|
||||
fields=[
|
||||
"name", "tool_name", "title", "subtitle",
|
||||
"enabled", "public", "team", "status", "repository_url",
|
||||
"file_url", "tool_image", "creation", "modified"
|
||||
],
|
||||
order_by=order_by,
|
||||
limit_start=limit_start,
|
||||
limit_page_length=limit_page_length
|
||||
)
|
||||
|
||||
# 批量获取团队用户名称
|
||||
team_user_names = _get_team_user_names(tools)
|
||||
|
||||
# 格式化返回数据
|
||||
result = []
|
||||
for tool in tools:
|
||||
tool_data = {
|
||||
"name": tool.name,
|
||||
"tool_name": tool.tool_name,
|
||||
"title": tool.title,
|
||||
"subtitle": tool.subtitle,
|
||||
"enabled": tool.enabled,
|
||||
"public": tool.public,
|
||||
"team": team_user_names.get(tool.team),
|
||||
"status": tool.status,
|
||||
"repository_url": tool.repository_url,
|
||||
"file_url": tool.file_url,
|
||||
"tool_image": tool.tool_image,
|
||||
"creation": tool.creation,
|
||||
"modified": tool.modified
|
||||
}
|
||||
result.append(tool_data)
|
||||
|
||||
return result
|
||||
|
||||
|
||||
@jingrow.whitelist(allow_guest=True)
|
||||
def get_local_app(name):
|
||||
"""获取本地应用详情"""
|
||||
@ -638,6 +695,67 @@ def get_my_local_agent_list(filters=None, order_by=None, limit_start=None, limit
|
||||
return result
|
||||
|
||||
|
||||
@dashboard_whitelist()
|
||||
def get_my_local_tool_list(filters=None, order_by=None, limit_start=None, limit_page_length=None):
|
||||
"""获取当前团队的工具列表"""
|
||||
|
||||
team = get_current_team()
|
||||
if not team:
|
||||
return {"success": False, "message": "未找到当前团队信息"}
|
||||
|
||||
# 构建查询条件
|
||||
query_filters = {"team": team}
|
||||
|
||||
if filters:
|
||||
if isinstance(filters, str):
|
||||
import json
|
||||
filters = json.loads(filters)
|
||||
query_filters.update(filters)
|
||||
|
||||
# 默认排序
|
||||
if not order_by:
|
||||
order_by = "tool_name asc"
|
||||
|
||||
# 获取工具列表
|
||||
tools = jingrow.get_all(
|
||||
"Local Tool",
|
||||
filters=query_filters,
|
||||
fields=[
|
||||
"name", "tool_name", "title", "subtitle",
|
||||
"enabled", "public", "team", "status", "repository_url",
|
||||
"file_url", "tool_image", "creation", "modified"
|
||||
],
|
||||
order_by=order_by,
|
||||
limit_start=limit_start,
|
||||
limit_page_length=limit_page_length
|
||||
)
|
||||
|
||||
# 批量获取团队用户名称
|
||||
team_user_names = _get_team_user_names(tools)
|
||||
|
||||
# 格式化返回数据
|
||||
result = []
|
||||
for tool in tools:
|
||||
tool_data = {
|
||||
"name": tool.name,
|
||||
"tool_name": tool.tool_name,
|
||||
"title": tool.title,
|
||||
"subtitle": tool.subtitle,
|
||||
"enabled": tool.enabled,
|
||||
"public": tool.public,
|
||||
"team": team_user_names.get(tool.team),
|
||||
"status": tool.status,
|
||||
"repository_url": tool.repository_url,
|
||||
"file_url": tool.file_url,
|
||||
"tool_image": tool.tool_image,
|
||||
"creation": tool.creation,
|
||||
"modified": tool.modified
|
||||
}
|
||||
result.append(tool_data)
|
||||
|
||||
return result
|
||||
|
||||
|
||||
@jingrow.whitelist(allow_guest=True)
|
||||
def get_local_agent(name):
|
||||
"""获取本地Agent详情"""
|
||||
@ -671,6 +789,38 @@ def get_local_agent(name):
|
||||
return agent_data
|
||||
|
||||
|
||||
@jingrow.whitelist(allow_guest=True)
|
||||
def get_local_tool(name):
|
||||
"""获取本地工具详情"""
|
||||
|
||||
if not jingrow.db.exists("Local Tool", name):
|
||||
return {"success": False, "message": "工具不存在"}
|
||||
|
||||
tool = jingrow.get_pg("Local Tool", name)
|
||||
|
||||
# 获取团队关联用户的名称
|
||||
team_user_names = _get_team_user_names([tool])
|
||||
|
||||
tool_data = {
|
||||
"name": tool.name,
|
||||
"tool_name": tool.tool_name,
|
||||
"title": tool.title,
|
||||
"subtitle": tool.subtitle,
|
||||
"description": tool.description,
|
||||
"enabled": tool.enabled,
|
||||
"public": tool.public,
|
||||
"team": team_user_names.get(tool.team),
|
||||
"status": tool.status,
|
||||
"repository_url": tool.repository_url,
|
||||
"file_url": tool.file_url,
|
||||
"tool_image": tool.tool_image,
|
||||
"creation": tool.creation,
|
||||
"modified": tool.modified
|
||||
}
|
||||
|
||||
return tool_data
|
||||
|
||||
|
||||
@dashboard_whitelist()
|
||||
def create_local_agent(agent_data):
|
||||
"""创建本地Agent"""
|
||||
@ -705,6 +855,37 @@ def create_local_agent(agent_data):
|
||||
return {"success": True, "name": agent.name, "message": "Agent创建成功"}
|
||||
|
||||
|
||||
@dashboard_whitelist()
|
||||
def create_local_tool(tool_data):
|
||||
"""创建本地工具"""
|
||||
team = get_current_team()
|
||||
|
||||
if isinstance(tool_data, str):
|
||||
import json
|
||||
tool_data = json.loads(tool_data)
|
||||
|
||||
# 检查工具名称是否已存在
|
||||
existing_tool = jingrow.db.exists("Local Tool", {"tool_name": tool_data["tool_name"]})
|
||||
if existing_tool:
|
||||
return {"error": "工具名称已存在，请使用其他名称"}
|
||||
|
||||
# 创建工具
|
||||
tool = jingrow.get_pg({
|
||||
"pagetype": "Local Tool",
|
||||
"tool_name": tool_data["tool_name"],
|
||||
"title": tool_data["title"],
|
||||
"subtitle": tool_data.get("subtitle", ""),
|
||||
"description": tool_data.get("description", ""),
|
||||
"team": team,
|
||||
"repository_url": tool_data.get("repository_url"),
|
||||
"file_url": tool_data.get("file_url"),
|
||||
"tool_image": tool_data.get("tool_image")
|
||||
})
|
||||
|
||||
tool.insert()
|
||||
return {"success": True, "name": tool.name, "message": "工具创建成功"}
|
||||
|
||||
|
||||
@dashboard_whitelist()
|
||||
def update_local_agent(name, agent_data):
|
||||
"""更新本地Agent"""
|
||||
@ -746,6 +927,47 @@ def update_local_agent(name, agent_data):
|
||||
return {"success": True, "name": agent.name, "updated_fields": updated_fields, "message": "Agent更新成功"}
|
||||
|
||||
|
||||
@dashboard_whitelist()
|
||||
def update_local_tool(name, tool_data):
|
||||
"""更新本地工具"""
|
||||
|
||||
team = get_current_team()
|
||||
if not team:
|
||||
return {"success": False, "message": "未找到当前团队信息"}
|
||||
|
||||
if not jingrow.db.exists("Local Tool", name):
|
||||
return {"success": False, "message": "工具不存在"}
|
||||
|
||||
tool = jingrow.get_pg("Local Tool", name)
|
||||
|
||||
# 验证权限：只能更新自己团队的工具
|
||||
if tool.team != team:
|
||||
return {"success": False, "message": "您没有权限操作此工具"}
|
||||
|
||||
# 解析 JSON 数据
|
||||
if isinstance(tool_data, str):
|
||||
try:
|
||||
import json
|
||||
tool_data = json.loads(tool_data)
|
||||
except json.JSONDecodeError:
|
||||
return {"success": False, "message": "JSON 数据格式错误"}
|
||||
|
||||
# 更新字段
|
||||
updated_fields = []
|
||||
for field, value in tool_data.items():
|
||||
if hasattr(tool, field):
|
||||
setattr(tool, field, value)
|
||||
updated_fields.append(field)
|
||||
|
||||
# 检查是否有字段被更新
|
||||
if not updated_fields:
|
||||
return {"success": False, "message": "没有有效的字段被更新"}
|
||||
|
||||
# 使用 ignore_permissions=True 因为我们已经手动验证了团队权限
|
||||
tool.save(ignore_permissions=True)
|
||||
return {"success": True, "name": tool.name, "updated_fields": updated_fields, "message": "工具更新成功"}
|
||||
|
||||
|
||||
@dashboard_whitelist()
|
||||
def delete_local_node(name):
|
||||
"""删除本地节点"""
|
||||
@ -790,3 +1012,26 @@ def delete_local_agent(name):
|
||||
agent.delete(ignore_permissions=True)
|
||||
|
||||
return {"success": True, "message": "Agent删除成功"}
|
||||
|
||||
|
||||
@dashboard_whitelist()
|
||||
def delete_local_tool(name):
|
||||
"""删除本地工具"""
|
||||
|
||||
team = get_current_team()
|
||||
if not team:
|
||||
return {"success": False, "message": "未找到当前团队信息"}
|
||||
|
||||
if not jingrow.db.exists("Local Tool", name):
|
||||
return {"success": False, "message": "工具不存在"}
|
||||
|
||||
tool = jingrow.get_pg("Local Tool", name)
|
||||
|
||||
# 验证权限：只能删除自己团队的工具
|
||||
if tool.team != team:
|
||||
return {"success": False, "message": "您没有权限操作此工具"}
|
||||
|
||||
# 使用 ignore_permissions=True 因为我们已经手动验证了团队权限
|
||||
tool.delete(ignore_permissions=True)
|
||||
|
||||
return {"success": True, "message": "工具删除成功"}
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user