mrfirdauss commited on
Commit
9a36c19
·
1 Parent(s): 898f1d0

fix abc func name

Browse files
Files changed (1) hide show
  1. src/FinancialAgentApp.py +2 -2
src/FinancialAgentApp.py CHANGED
@@ -57,7 +57,7 @@ class FinancialAgentApp (ABC):
57
 
58
  # Step 2: Check if context is needed
59
  if response_state.isNeedContext:
60
- context_prompt = self.handle_context(response_state)
61
  self.generate_final_answer(context_prompt)
62
  else:
63
  self.display_final_answer(response_state.response)
@@ -69,7 +69,7 @@ class FinancialAgentApp (ABC):
69
  return buf
70
 
71
  @abstractmethod
72
- def handle_context(self, response_state: ResponseState) -> str:
73
  """Handle context if need to add context from data/pdf"""
74
  pass
75
 
 
57
 
58
  # Step 2: Check if context is needed
59
  if response_state.isNeedContext:
60
+ context_prompt = self.__handle_context__(response_state)
61
  self.generate_final_answer(context_prompt)
62
  else:
63
  self.display_final_answer(response_state.response)
 
69
  return buf
70
 
71
  @abstractmethod
72
+ def __handle_context__(self, response_state: ResponseState) -> str:
73
  """Handle context if need to add context from data/pdf"""
74
  pass
75