Update agents/simple_tools.py

agents/simple_tools.py

@@ -204,8 +204,8 @@ def generate_balance_sheet(file_path: str, user_api_key: str = None, **kwargs) -
             "execution_time": execution_time
         }
 
-@tool  
-def generate_pnl_statement(file_path: str) -> Dict[str, Any]:
+@tool
+def generate_pnl_statement(file_path: str, **kwargs) -> Dict[str, Any]:
     """
     Generate P&L statement from trial balance file using complete pipeline
     Args:
@@ -219,8 +219,7 @@ def generate_pnl_statement(file_path: str) -> Dict[str, Any]:
     try:
         # Use the complete P&L pipeline from existing endpoint
         env = os.environ.copy()
-        if os.getenv("OPENROUTER_API_KEY"):
-            env["OPENROUTER_API_KEY"] = os.getenv("OPENROUTER_API_KEY")
+        
         env["INPUT_FILE"] = "data/clean_financial_data_pnl.json"
         cwd = os.getcwd()
         
@@ -294,7 +293,7 @@ def generate_pnl_statement(file_path: str) -> Dict[str, Any]:
         }
 
 @tool
-def generate_cash_flow_statement(file_path: str) -> Dict[str, Any]:
+def generate_cash_flow_statement(file_path: str, **kwargs) -> Dict[str, Any]:
     """
     Generate cash flow statement from trial balance file using complete pipeline
     Args:
@@ -309,8 +308,7 @@ def generate_cash_flow_statement(file_path: str) -> Dict[str, Any]:
     try:
         # Use the complete CF pipeline from existing endpoint  
         env = os.environ.copy()
-        if os.getenv("OPENROUTER_API_KEY"):
-            env["OPENROUTER_API_KEY"] = os.getenv("OPENROUTER_API_KEY")
+        
         env["INPUT_FILE"] = "data/clean_financial_data_cfs.json"
         env["CFS_JSON_INPUT"] = "data/clean_financial_data_cfs.json"
         env["CFS_JSON_OUTPUT"] = "data/extracted_cfs_data.json"