Spaces:
Running
Running
File size: 1,051 Bytes
c2ea5ed edf083b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from pydantic import BaseModel, Field
from typing import Dict, List, Optional, Any
# Risk assessment functionality removed
# noqa
from .entity import Entity
from .relation import Relation
from .failure import Failure
from .optimization_recommendation import OptimizationRecommendation
class KnowledgeGraph(BaseModel):
system_name: str = Field("", description="A concise, descriptive name for the agent system")
system_summary: str = Field("", description="A short 2-3 sentence summary of the agent system's purpose and structure")
entities: List[Entity] = Field(default_factory=list, description="List of entities in the knowledge graph")
relations: List[Relation] = Field(default_factory=list, description="List of relations in the knowledge graph")
failures: List[Failure] = Field(default_factory=list, description="List of detected risk or failure events across the trace")
optimizations: List[OptimizationRecommendation] = Field(default_factory=list, description="List of recommendations for optimizing the agent system")
|