Spaces:
Sleeping
Sleeping
Commit ·
9a36c19
1
Parent(s): 898f1d0
fix abc func name
Browse files- 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.
|
| 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
|
| 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 |
|