File size: 1,585 Bytes
7042c3c 4a6af9d 21cb336 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
from pydantic import Field
from typing_extensions import (
Annotated
)
import operator
from typing import (
Optional,
Annotated,
Sequence,
List
)
from dataclasses import dataclass
from typing import Optional
from langchain_core.messages import (
BaseMessage,
)
from ea4all.src.ea4all_gra.data import (
ListRequirement,
ListObjective,
UserJourney,
StakeholderList,
BusinessCapability,
)
from ea4all.src.ea4all_gra.state import InputState
# Task-1 Graph State
@dataclass(kw_only=True)
class Task1State(InputState):
"""
Represents the BusinessOutput state of our graph.
Attributes:
message: a message is added after each team member finishes
team_members: the team members are tracked so they are aware of the others' skill-sets
next: used to route work. The supervisor calls a function that will update this every time it makes a decision
requirement: list of business requirements
intent: business objective, goal
userjourney: list of user journeys
stakeholder: list of stakeholder and their concerns
capability: list of business capabilities to deliver intent and requirements
"""
#messages: Optional[Annotated[Sequence[BaseMessage], operator.add]]
business_query: str
team_members: Optional[List[str]]
requirement: Optional[ListRequirement]
intent: Optional[ListObjective]
userjourney: Optional[UserJourney]
stakeholder: Optional[StakeholderList]
biz_capability: Optional[BusinessCapability]
next: Optional[str]
|