Spaces:
Sleeping
Sleeping
bugfix
Browse files
agent.py
CHANGED
@@ -16,14 +16,14 @@ def create_assistant_tools(cfg):
|
|
16 |
|
17 |
class QueryCFPBComplaints(BaseModel):
|
18 |
query: str = Field(description="The user query.")
|
19 |
-
|
20 |
default=None,
|
21 |
description="The company that the complaint is about.",
|
22 |
examples=['CAPITAL ONE FINANCIAL CORPORATION', 'BANK OF AMERICA, NATIONAL ASSOCIATION', 'CITIBANK, N.A.', 'WELLS FARGO & COMPANY', 'JPMORGAN CHASE & CO.']
|
23 |
)
|
24 |
-
|
25 |
default=None,
|
26 |
-
|
27 |
examples=['CA', 'FL', 'NY', 'TX', 'GA']
|
28 |
)
|
29 |
|
|
|
16 |
|
17 |
class QueryCFPBComplaints(BaseModel):
|
18 |
query: str = Field(description="The user query.")
|
19 |
+
company: Optional[str] = Field(
|
20 |
default=None,
|
21 |
description="The company that the complaint is about.",
|
22 |
examples=['CAPITAL ONE FINANCIAL CORPORATION', 'BANK OF AMERICA, NATIONAL ASSOCIATION', 'CITIBANK, N.A.', 'WELLS FARGO & COMPANY', 'JPMORGAN CHASE & CO.']
|
23 |
)
|
24 |
+
state: Optional[str] = Field(
|
25 |
default=None,
|
26 |
+
description="The two-character state code where the consumer lives.",
|
27 |
examples=['CA', 'FL', 'NY', 'TX', 'GA']
|
28 |
)
|
29 |
|