ofermend commited on
Commit
d1c53b9
·
1 Parent(s): 8d5563f
Files changed (1) hide show
  1. agent.py +3 -3
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
- 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
- descripition="The two-character state code where the consumer lives.",
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