nagarmayank commited on
Commit
4137d9f
·
1 Parent(s): 59ed4d3

Expand transaction classification options to include Reward_Points_Credit and sweep_in_fd

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ class TransactionParser(BaseModel):
46
  class TransactionClassification(BaseModel):
47
  """This Pydantic class is used to classify the transaction message. The message is taken and the output is structured in a specific format based upon below definition."""
48
 
49
- classification: str = Field(description="Classification of the transaction. Strictly choose one of the values - Transaction, OTP, Scheduled, Reminder")
50
 
51
  class AgentState(TypedDict):
52
  messages: Annotated[list[AnyMessage], operator.add]
 
46
  class TransactionClassification(BaseModel):
47
  """This Pydantic class is used to classify the transaction message. The message is taken and the output is structured in a specific format based upon below definition."""
48
 
49
+ classification: str = Field(description="Classification of the transaction. Strictly choose one of the values - Transaction, OTP, Scheduled, Reminder, Reward_Points_Credit, sweep_in_fd")
50
 
51
  class AgentState(TypedDict):
52
  messages: Annotated[list[AnyMessage], operator.add]