Spaces:
Running
Running
Commit ·
4137d9f
1
Parent(s): 59ed4d3
Expand transaction classification options to include Reward_Points_Credit and sweep_in_fd
Browse files
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]
|