anmolsahai commited on
Commit
e0da68f
1 Parent(s): 3b5a8b1
Files changed (1) hide show
  1. langchain_pipeline.py +13 -12
langchain_pipeline.py CHANGED
@@ -38,6 +38,7 @@ def pipeline(file, model_name, balance_type, apsn_transactions, max_fees_per_day
38
 
39
  prompt = PromptTemplate.from_template(
40
  """
 
41
 
42
  law context:
43
 
@@ -1076,19 +1077,19 @@ What is the minimum amount overdrawn to incur a fee?: {min_overdrawn_fee}
1076
  What is the minimum transaction amount to trigger an overdraft?: {min_transaction_overdraft}
1077
 
1078
  Please output in the following format:
1079
- {updated_disclosure_text}
1080
  ------
1081
- {reasons_for_changes}
1082
- """
1083
- val = prompt.format(
1084
- context=related_docs,
1085
- disclosure=disclosure_text,
1086
- balance_type=balance_type,
1087
- apsn_transactions=apsn_transactions,
1088
- max_fees_per_day=max_fees_per_day,
1089
- min_overdrawn_fee=min_overdrawn_fee,
1090
- min_transaction_overdraft=min_transaction_overdraft,
1091
- )
1092
  )
1093
 
1094
  chat_response = models[model_name].invoke(input=val)
 
38
 
39
  prompt = PromptTemplate.from_template(
40
  """
41
+ law context:
42
 
43
  law context:
44
 
 
1077
  What is the minimum transaction amount to trigger an overdraft?: {min_transaction_overdraft}
1078
 
1079
  Please output in the following format:
1080
+ {{updated_disclosure_text}}
1081
  ------
1082
+ {{reasons_for_changes}}
1083
+ """
1084
+ )
1085
+ val = prompt.format(
1086
+ context=related_docs,
1087
+ disclosure=disclosure_text,
1088
+ balance_type=balance_type,
1089
+ apsn_transactions=apsn_transactions,
1090
+ max_fees_per_day=max_fees_per_day,
1091
+ min_overdrawn_fee=min_overdrawn_fee,
1092
+ min_transaction_overdraft=min_transaction_overdraft,
1093
  )
1094
 
1095
  chat_response = models[model_name].invoke(input=val)