Agent Zero
591 lines
19 KiB
Python
591 lines
19 KiB
Python
"""AI Assistant plugin routes — providers, models, presets, agents,
|
|
sessions, messages, tools, and streaming chat.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import uuid
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException, Query, Request
|
|
from fastapi.responses import StreamingResponse
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.core.db import get_db, set_tenant_context
|
|
from app.core.visibility import apply_visibility_filter
|
|
from app.deps import get_current_user, require_permission
|
|
from app.plugins.builtins.ai_assistant.models import (
|
|
AIAgent,
|
|
AIChatFolder,
|
|
AIModel,
|
|
AIPreset,
|
|
AIProvider,
|
|
)
|
|
from app.plugins.builtins.ai_assistant.schemas import (
|
|
AIAgentCreate,
|
|
AIAgentUpdate,
|
|
AIModelCreate,
|
|
AIModelUpdate,
|
|
AIPresetCreate,
|
|
AIPresetUpdate,
|
|
AIProviderCreate,
|
|
AIProviderUpdate,
|
|
ChatFolderCreate,
|
|
ChatFolderUpdate,
|
|
ChatSendRequest,
|
|
)
|
|
from app.plugins.builtins.ai_assistant.services import (
|
|
agent_to_response,
|
|
folder_to_response,
|
|
get_agent_by_id,
|
|
get_comm_messages,
|
|
get_default_agent,
|
|
get_preset_by_id,
|
|
get_provider_by_id,
|
|
model_to_response,
|
|
preset_to_response,
|
|
provider_to_response,
|
|
stream_chat_comm,
|
|
)
|
|
from app.plugins.builtins.ai_assistant.tool_registry import get_tool_registry
|
|
|
|
router = APIRouter(prefix="/api/v1/ai", tags=["ai-assistant"])
|
|
|
|
|
|
# ─── Providers ───
|
|
|
|
@router.get("/providers", dependencies=[Depends(require_permission("ai:config"))])
|
|
async def list_providers(
|
|
current_user: dict = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
tenant_id = uuid.UUID(current_user["tenant_id"])
|
|
result = await db.execute(
|
|
select(AIProvider).where(AIProvider.tenant_id == tenant_id)
|
|
)
|
|
providers = list(result.scalars().all())
|
|
return [provider_to_response(p) for p in providers]
|
|
|
|
|
|
@router.post("/providers", dependencies=[Depends(require_permission("ai:config"))])
|
|
async def create_provider(
|
|
data: AIProviderCreate,
|
|
current_user: dict = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
tenant_id = uuid.UUID(current_user["tenant_id"])
|
|
await set_tenant_context(db, tenant_id)
|
|
|
|
# If is_default, unset other defaults
|
|
if data.is_default:
|
|
existing = await db.execute(
|
|
select(AIProvider)
|
|
.where(AIProvider.tenant_id == tenant_id)
|
|
.where(AIProvider.is_default.is_(True))
|
|
)
|
|
for p in existing.scalars().all():
|
|
p.is_default = False
|
|
|
|
provider = AIProvider(
|
|
name=data.name,
|
|
provider_type=data.provider_type,
|
|
api_key=data.api_key,
|
|
base_url=data.base_url,
|
|
is_active=data.is_active,
|
|
is_default=data.is_default,
|
|
config=data.config,
|
|
tenant_id=tenant_id,
|
|
)
|
|
db.add(provider)
|
|
await db.flush()
|
|
await db.refresh(provider)
|
|
await db.commit()
|
|
return provider_to_response(provider)
|
|
|
|
|
|
@router.put("/providers/{provider_id}", dependencies=[Depends(require_permission("ai:config"))])
|
|
async def update_provider(
|
|
provider_id: str,
|
|
data: AIProviderUpdate,
|
|
current_user: dict = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
tenant_id = uuid.UUID(current_user["tenant_id"])
|
|
provider = await get_provider_by_id(db, uuid.UUID(provider_id), tenant_id)
|
|
if not provider:
|
|
raise HTTPException(status_code=404, detail="Provider not found")
|
|
|
|
if data.is_default:
|
|
existing = await db.execute(
|
|
select(AIProvider)
|
|
.where(AIProvider.tenant_id == tenant_id)
|
|
.where(AIProvider.is_default.is_(True))
|
|
.where(AIProvider.id != provider.id)
|
|
)
|
|
for p in existing.scalars().all():
|
|
p.is_default = False
|
|
|
|
for field, val in data.model_dump(exclude_unset=True).items():
|
|
setattr(provider, field, val)
|
|
await db.flush()
|
|
await db.refresh(provider)
|
|
await db.commit()
|
|
return provider_to_response(provider)
|
|
|
|
|
|
@router.delete("/providers/{provider_id}", dependencies=[Depends(require_permission("ai:config"))])
|
|
async def delete_provider(
|
|
provider_id: str,
|
|
current_user: dict = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
tenant_id = uuid.UUID(current_user["tenant_id"])
|
|
provider = await get_provider_by_id(db, uuid.UUID(provider_id), tenant_id)
|
|
if not provider:
|
|
raise HTTPException(status_code=404, detail="Provider not found")
|
|
await db.delete(provider)
|
|
await db.commit()
|
|
return {"ok": True}
|
|
|
|
|
|
# ─── Models ───
|
|
|
|
@router.get("/models", dependencies=[Depends(require_permission("ai:config"))])
|
|
async def list_models(
|
|
provider_id: str | None = Query(None),
|
|
current_user: dict = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
tenant_id = uuid.UUID(current_user["tenant_id"])
|
|
stmt = select(AIModel).where(AIModel.tenant_id == tenant_id)
|
|
if provider_id:
|
|
stmt = stmt.where(AIModel.provider_id == uuid.UUID(provider_id))
|
|
result = await db.execute(stmt)
|
|
models = list(result.scalars().all())
|
|
return [model_to_response(m) for m in models]
|
|
|
|
|
|
@router.post("/models", dependencies=[Depends(require_permission("ai:config"))])
|
|
async def create_model(
|
|
data: AIModelCreate,
|
|
current_user: dict = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
tenant_id = uuid.UUID(current_user["tenant_id"])
|
|
await set_tenant_context(db, tenant_id)
|
|
|
|
model = AIModel(
|
|
provider_id=uuid.UUID(data.provider_id),
|
|
model_id=data.model_id,
|
|
display_name=data.display_name,
|
|
context_window=data.context_window,
|
|
supports_tools=data.supports_tools,
|
|
supports_streaming=data.supports_streaming,
|
|
is_active=data.is_active,
|
|
config=data.config,
|
|
tenant_id=tenant_id,
|
|
)
|
|
db.add(model)
|
|
await db.flush()
|
|
await db.refresh(model)
|
|
await db.commit()
|
|
return model_to_response(model)
|
|
|
|
|
|
@router.put("/models/{model_id}", dependencies=[Depends(require_permission("ai:config"))])
|
|
async def update_model(
|
|
model_id: str,
|
|
data: AIModelUpdate,
|
|
current_user: dict = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
tenant_id = uuid.UUID(current_user["tenant_id"])
|
|
result = await db.execute(
|
|
select(AIModel)
|
|
.where(AIModel.id == uuid.UUID(model_id))
|
|
.where(AIModel.tenant_id == tenant_id)
|
|
)
|
|
model = result.scalar_one_or_none()
|
|
if not model:
|
|
raise HTTPException(status_code=404, detail="Model not found")
|
|
|
|
for field, val in data.model_dump(exclude_unset=True).items():
|
|
setattr(model, field, val)
|
|
await db.flush()
|
|
await db.refresh(model)
|
|
await db.commit()
|
|
return model_to_response(model)
|
|
|
|
|
|
@router.delete("/models/{model_id}", dependencies=[Depends(require_permission("ai:config"))])
|
|
async def delete_model(
|
|
model_id: str,
|
|
current_user: dict = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
tenant_id = uuid.UUID(current_user["tenant_id"])
|
|
result = await db.execute(
|
|
select(AIModel)
|
|
.where(AIModel.id == uuid.UUID(model_id))
|
|
.where(AIModel.tenant_id == tenant_id)
|
|
)
|
|
model = result.scalar_one_or_none()
|
|
if not model:
|
|
raise HTTPException(status_code=404, detail="Model not found")
|
|
await db.delete(model)
|
|
await db.commit()
|
|
return {"ok": True}
|
|
|
|
|
|
# ─── Presets ───
|
|
|
|
@router.get("/presets", dependencies=[Depends(require_permission("ai:config"))])
|
|
async def list_presets(
|
|
current_user: dict = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
tenant_id = uuid.UUID(current_user["tenant_id"])
|
|
result = await db.execute(
|
|
select(AIPreset).where(AIPreset.tenant_id == tenant_id)
|
|
)
|
|
presets = list(result.scalars().all())
|
|
return [preset_to_response(p) for p in presets]
|
|
|
|
|
|
@router.post("/presets", dependencies=[Depends(require_permission("ai:config"))])
|
|
async def create_preset(
|
|
data: AIPresetCreate,
|
|
current_user: dict = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
tenant_id = uuid.UUID(current_user["tenant_id"])
|
|
await set_tenant_context(db, tenant_id)
|
|
|
|
preset = AIPreset(
|
|
name=data.name,
|
|
model_id=data.model_id,
|
|
provider_id=uuid.UUID(data.provider_id) if data.provider_id else None,
|
|
temperature=data.temperature,
|
|
max_tokens=data.max_tokens,
|
|
top_p=data.top_p,
|
|
system_prompt=data.system_prompt,
|
|
config=data.config,
|
|
is_active=data.is_active,
|
|
tenant_id=tenant_id,
|
|
)
|
|
db.add(preset)
|
|
await db.flush()
|
|
await db.refresh(preset)
|
|
await db.commit()
|
|
return preset_to_response(preset)
|
|
|
|
|
|
@router.put("/presets/{preset_id}", dependencies=[Depends(require_permission("ai:config"))])
|
|
async def update_preset(
|
|
preset_id: str,
|
|
data: AIPresetUpdate,
|
|
current_user: dict = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
tenant_id = uuid.UUID(current_user["tenant_id"])
|
|
preset = await get_preset_by_id(db, uuid.UUID(preset_id), tenant_id)
|
|
if not preset:
|
|
raise HTTPException(status_code=404, detail="Preset not found")
|
|
|
|
update_data = data.model_dump(exclude_unset=True)
|
|
if "provider_id" in update_data and update_data["provider_id"]:
|
|
update_data["provider_id"] = uuid.UUID(update_data["provider_id"])
|
|
for field, val in update_data.items():
|
|
setattr(preset, field, val)
|
|
await db.flush()
|
|
await db.refresh(preset)
|
|
await db.commit()
|
|
return preset_to_response(preset)
|
|
|
|
|
|
@router.delete("/presets/{preset_id}", dependencies=[Depends(require_permission("ai:config"))])
|
|
async def delete_preset(
|
|
preset_id: str,
|
|
current_user: dict = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
tenant_id = uuid.UUID(current_user["tenant_id"])
|
|
preset = await get_preset_by_id(db, uuid.UUID(preset_id), tenant_id)
|
|
if not preset:
|
|
raise HTTPException(status_code=404, detail="Preset not found")
|
|
await db.delete(preset)
|
|
await db.commit()
|
|
return {"ok": True}
|
|
|
|
|
|
# ─── Agents ───
|
|
|
|
@router.get("/agents", dependencies=[Depends(require_permission("ai:read"))])
|
|
async def list_agents(
|
|
current_user: dict = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
tenant_id = uuid.UUID(current_user["tenant_id"])
|
|
user_id = uuid.UUID(current_user["user_id"])
|
|
is_system_admin = current_user.get("is_system_admin", False)
|
|
query = select(AIAgent).where(AIAgent.tenant_id == tenant_id)
|
|
query = await apply_visibility_filter(
|
|
db, query, "ai_agent", AIAgent, user_id, tenant_id, is_system_admin
|
|
)
|
|
result = await db.execute(query)
|
|
agents = list(result.scalars().all())
|
|
return [agent_to_response(a) for a in agents]
|
|
|
|
|
|
@router.post("/agents", dependencies=[Depends(require_permission("ai:agents"))])
|
|
async def create_agent(
|
|
data: AIAgentCreate,
|
|
current_user: dict = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
tenant_id = uuid.UUID(current_user["tenant_id"])
|
|
user_id = uuid.UUID(current_user["user_id"])
|
|
await set_tenant_context(db, tenant_id)
|
|
|
|
agent = AIAgent(
|
|
name=data.name,
|
|
description=data.description,
|
|
system_prompt=data.system_prompt,
|
|
preset_id=uuid.UUID(data.preset_id) if data.preset_id else None,
|
|
tool_ids=data.tool_ids,
|
|
is_active=data.is_active,
|
|
config=data.config,
|
|
tenant_id=tenant_id,
|
|
owner_id=user_id,
|
|
)
|
|
db.add(agent)
|
|
await db.flush()
|
|
await db.refresh(agent)
|
|
await db.commit()
|
|
return agent_to_response(agent)
|
|
|
|
|
|
@router.put("/agents/{agent_id}", dependencies=[Depends(require_permission("ai:agents"))])
|
|
async def update_agent(
|
|
agent_id: str,
|
|
data: AIAgentUpdate,
|
|
current_user: dict = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
tenant_id = uuid.UUID(current_user["tenant_id"])
|
|
agent = await get_agent_by_id(db, uuid.UUID(agent_id), tenant_id)
|
|
if not agent:
|
|
raise HTTPException(status_code=404, detail="Agent not found")
|
|
|
|
update_data = data.model_dump(exclude_unset=True)
|
|
if "preset_id" in update_data and update_data["preset_id"]:
|
|
update_data["preset_id"] = uuid.UUID(update_data["preset_id"])
|
|
for field, val in update_data.items():
|
|
setattr(agent, field, val)
|
|
await db.flush()
|
|
await db.refresh(agent)
|
|
await db.commit()
|
|
return agent_to_response(agent)
|
|
|
|
|
|
@router.delete("/agents/{agent_id}", dependencies=[Depends(require_permission("ai:agents"))])
|
|
async def delete_agent(
|
|
agent_id: str,
|
|
current_user: dict = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
tenant_id = uuid.UUID(current_user["tenant_id"])
|
|
agent = await get_agent_by_id(db, uuid.UUID(agent_id), tenant_id)
|
|
if not agent:
|
|
raise HTTPException(status_code=404, detail="Agent not found")
|
|
if agent.is_default:
|
|
raise HTTPException(status_code=400, detail="Cannot delete default agent")
|
|
await db.delete(agent)
|
|
await db.commit()
|
|
return {"ok": True}
|
|
|
|
|
|
# ─── Tools ───
|
|
|
|
@router.get("/tools", dependencies=[Depends(require_permission("ai:read"))])
|
|
async def list_tools(
|
|
current_user: dict = Depends(get_current_user),
|
|
):
|
|
registry = get_tool_registry()
|
|
items = registry.list_for_api()
|
|
return {"items": items, "total": len(items)}
|
|
|
|
|
|
# ─── Chat Folders ───
|
|
|
|
@router.get("/folders", dependencies=[Depends(require_permission("ai:read"))])
|
|
async def list_folders(
|
|
current_user: dict = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
tenant_id = uuid.UUID(current_user["tenant_id"])
|
|
user_id = uuid.UUID(current_user["user_id"])
|
|
result = await db.execute(
|
|
select(AIChatFolder)
|
|
.where(AIChatFolder.tenant_id == tenant_id)
|
|
.where(AIChatFolder.user_id == user_id)
|
|
.order_by(AIChatFolder.name.asc())
|
|
)
|
|
folders = list(result.scalars().all())
|
|
return [folder_to_response(f) for f in folders]
|
|
|
|
|
|
@router.post("/folders", dependencies=[Depends(require_permission("ai:write"))])
|
|
async def create_folder(
|
|
data: ChatFolderCreate,
|
|
current_user: dict = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
tenant_id = uuid.UUID(current_user["tenant_id"])
|
|
user_id = uuid.UUID(current_user["user_id"])
|
|
await set_tenant_context(db, tenant_id)
|
|
|
|
folder = AIChatFolder(
|
|
name=data.name,
|
|
parent_id=uuid.UUID(data.parent_id) if data.parent_id else None,
|
|
user_id=user_id,
|
|
tenant_id=tenant_id,
|
|
)
|
|
db.add(folder)
|
|
await db.flush()
|
|
await db.refresh(folder)
|
|
await db.commit()
|
|
return folder_to_response(folder)
|
|
|
|
|
|
@router.put("/folders/{folder_id}", dependencies=[Depends(require_permission("ai:write"))])
|
|
async def update_folder(
|
|
folder_id: str,
|
|
data: ChatFolderUpdate,
|
|
current_user: dict = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
tenant_id = uuid.UUID(current_user["tenant_id"])
|
|
user_id = uuid.UUID(current_user["user_id"])
|
|
result = await db.execute(
|
|
select(AIChatFolder)
|
|
.where(AIChatFolder.id == uuid.UUID(folder_id))
|
|
.where(AIChatFolder.tenant_id == tenant_id)
|
|
.where(AIChatFolder.user_id == user_id)
|
|
)
|
|
folder = result.scalar_one_or_none()
|
|
if not folder:
|
|
raise HTTPException(status_code=404, detail="Folder not found")
|
|
|
|
update_data = data.model_dump(exclude_unset=True)
|
|
if "parent_id" in update_data:
|
|
if update_data["parent_id"]:
|
|
update_data["parent_id"] = uuid.UUID(update_data["parent_id"])
|
|
else:
|
|
update_data["parent_id"] = None
|
|
for field, val in update_data.items():
|
|
setattr(folder, field, val)
|
|
await db.flush()
|
|
await db.refresh(folder)
|
|
await db.commit()
|
|
return folder_to_response(folder)
|
|
|
|
|
|
@router.delete("/folders/{folder_id}", dependencies=[Depends(require_permission("ai:write"))])
|
|
async def delete_folder(
|
|
folder_id: str,
|
|
current_user: dict = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
tenant_id = uuid.UUID(current_user["tenant_id"])
|
|
user_id = uuid.UUID(current_user["user_id"])
|
|
result = await db.execute(
|
|
select(AIChatFolder)
|
|
.where(AIChatFolder.id == uuid.UUID(folder_id))
|
|
.where(AIChatFolder.tenant_id == tenant_id)
|
|
.where(AIChatFolder.user_id == user_id)
|
|
)
|
|
folder = result.scalar_one_or_none()
|
|
if not folder:
|
|
raise HTTPException(status_code=404, detail="Folder not found")
|
|
await db.delete(folder)
|
|
await db.commit()
|
|
return {"ok": True}
|
|
|
|
|
|
# ─── AI Chat Streaming (via comm_conversations) ───
|
|
|
|
@router.post("/conversations/{conversation_id}/stream", dependencies=[Depends(require_permission("ai:write"))])
|
|
async def chat_stream_comm(
|
|
conversation_id: str,
|
|
data: ChatSendRequest,
|
|
request: Request,
|
|
current_user: dict = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""Stream AI chat response for a comm conversation with conversation_type='ai'."""
|
|
tenant_id = uuid.UUID(current_user["tenant_id"])
|
|
user_id = uuid.UUID(current_user["user_id"])
|
|
|
|
from app.core.rate_limit import RateLimitPolicy, check_rate_limit_policy
|
|
await check_rate_limit_policy(
|
|
f"rate:ai:chat:{tenant_id}:{user_id}",
|
|
RateLimitPolicy.AI,
|
|
)
|
|
|
|
await set_tenant_context(db, tenant_id)
|
|
|
|
agent = None
|
|
if data.agent_id:
|
|
agent = await get_agent_by_id(db, uuid.UUID(data.agent_id), tenant_id)
|
|
if not agent:
|
|
agent = await get_default_agent(db, tenant_id)
|
|
if not agent:
|
|
raise HTTPException(status_code=400, detail="No agent available")
|
|
|
|
user_context = {
|
|
"user_id": current_user["user_id"],
|
|
"tenant_id": current_user["tenant_id"],
|
|
"role": current_user.get("role", ""),
|
|
"permissions": current_user.get("permissions", []),
|
|
"denied_permissions": current_user.get("denied_permissions", []),
|
|
"is_system_admin": current_user.get("is_system_admin", False),
|
|
"field_permissions": current_user.get("field_permissions", {}),
|
|
}
|
|
|
|
async def event_stream():
|
|
from app.core.db import get_session_factory
|
|
factory = get_session_factory()
|
|
async with factory() as stream_db:
|
|
await set_tenant_context(stream_db, tenant_id)
|
|
async for chunk in stream_chat_comm(
|
|
stream_db, uuid.UUID(conversation_id), agent, data.content, user_context, tenant_id, user_id
|
|
):
|
|
yield chunk
|
|
yield "data: [DONE]\n\n"
|
|
|
|
return StreamingResponse(
|
|
event_stream(),
|
|
media_type="text/event-stream",
|
|
headers={
|
|
"Cache-Control": "no-cache",
|
|
"Connection": "keep-alive",
|
|
"X-Accel-Buffering": "no",
|
|
},
|
|
)
|
|
|
|
|
|
@router.get("/conversations/{conversation_id}/messages", dependencies=[Depends(require_permission("ai:read"))])
|
|
async def list_comm_messages(
|
|
conversation_id: str,
|
|
current_user: dict = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""List messages for an AI comm conversation."""
|
|
tenant_id = uuid.UUID(current_user["tenant_id"])
|
|
await set_tenant_context(db, tenant_id)
|
|
messages = await get_comm_messages(db, uuid.UUID(conversation_id), tenant_id)
|
|
return [{"role": m["role"], "content": m["content"]} for m in messages]
|
|
|
|
|