310491a557
- Updated python_coding_microproject.csv to use NVIDIA NIM Kimi K2 - Updated kali_linux_shell_simulator.csv to use NVIDIA NIM Kimi K2 - Made more general-purpose (flexible targets, expanded tools) - Added nemotron-mini-agent.csv for fast local inference via Ollama - Added nemotron-agent.csv for advanced reasoning via Ollama - Added wiki page: Projects for NVIDIA NIMs and Nemotron
65 lines
2.5 KiB
Python
65 lines
2.5 KiB
Python
"""
|
|
User schemas for GT 2.0 Tenant Backend
|
|
|
|
Pydantic models for user-related API request/response validation.
|
|
Implements user preferences management per GT 2.0 specifications.
|
|
"""
|
|
|
|
from pydantic import BaseModel, Field
|
|
from typing import List, Optional, Dict, Any
|
|
from datetime import datetime
|
|
|
|
|
|
class CustomCategory(BaseModel):
|
|
"""User-defined custom category with metadata"""
|
|
name: str = Field(..., description="Category name (lowercase, unique per user)")
|
|
description: str = Field(..., description="Category description")
|
|
created_at: Optional[str] = Field(None, description="ISO timestamp when category was created")
|
|
|
|
|
|
class UserPreferences(BaseModel):
|
|
"""User preferences stored in JSONB"""
|
|
favorite_agent_ids: Optional[List[str]] = Field(default_factory=list, description="List of favorited agent UUIDs")
|
|
custom_categories: Optional[List[CustomCategory]] = Field(default_factory=list, description="User's custom agent categories")
|
|
# Future preferences can be added here
|
|
|
|
|
|
class UserPreferencesResponse(BaseModel):
|
|
"""Response for getting user preferences"""
|
|
preferences: Dict[str, Any] = Field(..., description="User preferences dictionary")
|
|
|
|
|
|
class UpdateUserPreferencesRequest(BaseModel):
|
|
"""Request to update user preferences (merges with existing)"""
|
|
preferences: Dict[str, Any] = Field(..., description="Preferences to merge with existing")
|
|
|
|
|
|
class FavoriteAgentsResponse(BaseModel):
|
|
"""Response for getting favorite agent IDs"""
|
|
favorite_agent_ids: List[str] = Field(..., description="List of favorited agent UUIDs")
|
|
|
|
|
|
class UpdateFavoriteAgentsRequest(BaseModel):
|
|
"""Request to update favorite agent IDs (replaces existing list)"""
|
|
agent_ids: List[str] = Field(..., description="List of agent UUIDs to set as favorites")
|
|
|
|
|
|
class AddFavoriteAgentRequest(BaseModel):
|
|
"""Request to add a single agent to favorites"""
|
|
agent_id: str = Field(..., description="Agent UUID to add to favorites")
|
|
|
|
|
|
class RemoveFavoriteAgentRequest(BaseModel):
|
|
"""Request to remove a single agent from favorites"""
|
|
agent_id: str = Field(..., description="Agent UUID to remove from favorites")
|
|
|
|
|
|
class CustomCategoriesResponse(BaseModel):
|
|
"""Response for getting custom categories"""
|
|
categories: List[CustomCategory] = Field(..., description="List of user's custom categories")
|
|
|
|
|
|
class UpdateCustomCategoriesRequest(BaseModel):
|
|
"""Request to update custom categories (replaces entire list)"""
|
|
categories: List[CustomCategory] = Field(..., description="Complete list of custom categories")
|