mirror of
https://github.com/awslabs/amazon-bedrock-agentcore-samples.git
synced 2025-09-08 20:50:46 +00:00
292462b94c
- Copy SRE Agent codebase to 02-use-cases/04-SRE-agent - Update LICENSE link to reference main repository LICENSE - Configure .gitignore to exclude wheel files from version control
49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import logging
|
|
from typing import Annotated, Any, Dict, List, Literal, Optional, TypedDict
|
|
|
|
from langchain_core.messages import BaseMessage
|
|
from langgraph.graph.message import add_messages
|
|
|
|
# Configure logging with basicConfig
|
|
logging.basicConfig(
|
|
level=logging.INFO, # Set the log level to INFO
|
|
# Define log message format
|
|
format="%(asctime)s,p%(process)s,{%(filename)s:%(lineno)d},%(levelname)s,%(message)s",
|
|
)
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class AgentState(TypedDict):
|
|
"""State shared across all agents in the multi-agent system.
|
|
|
|
This state is passed between agents and maintains conversation history,
|
|
intermediate results, and routing information.
|
|
"""
|
|
|
|
# Conversation messages using LangGraph's message annotation
|
|
messages: Annotated[List[BaseMessage], add_messages]
|
|
|
|
# Which agent should act next (set by supervisor)
|
|
next: Literal["kubernetes", "logs", "metrics", "runbooks", "FINISH"]
|
|
|
|
# Intermediate results from each agent
|
|
agent_results: Dict[str, Any]
|
|
|
|
# Current query being processed
|
|
current_query: Optional[str]
|
|
|
|
# Metadata about the conversation
|
|
metadata: Dict[str, Any]
|
|
|
|
# Flag to indicate if we need multiple agents
|
|
requires_collaboration: bool
|
|
|
|
# List of agents that have already responded
|
|
agents_invoked: List[str]
|
|
|
|
# Final aggregated response (set by supervisor)
|
|
final_response: Optional[str]
|