Humanlearning commited on
Commit
8c16f30
·
1 Parent(s): 5c1c332

feat: introduce common agent state definitions and update README metadata.

Browse files
README.md CHANGED
@@ -7,5 +7,6 @@ sdk: gradio
7
  sdk_version: 6.0.1
8
  app_file: app.py
9
  pinned: false
 
10
  ---
11
 
 
7
  sdk_version: 6.0.1
8
  app_file: app.py
9
  pinned: false
10
+ python_version: 3.11
11
  ---
12
 
src/credentialwatch_agent/agents/common.py CHANGED
@@ -1,11 +1,14 @@
1
  from typing import TypedDict, List, Optional, Any, Dict, Annotated
2
  import operator
3
 
 
 
 
4
  class AgentState(TypedDict):
5
  """
6
  Common state for agents.
7
  """
8
- messages: List[Dict[str, Any]]
9
  # Add other common fields if needed
10
 
11
  class ExpirySweepState(TypedDict):
 
1
  from typing import TypedDict, List, Optional, Any, Dict, Annotated
2
  import operator
3
 
4
+ from langchain_core.messages import BaseMessage
5
+ from langgraph.graph.message import add_messages
6
+
7
  class AgentState(TypedDict):
8
  """
9
  Common state for agents.
10
  """
11
+ messages: Annotated[List[BaseMessage], add_messages]
12
  # Add other common fields if needed
13
 
14
  class ExpirySweepState(TypedDict):