realtime client, timeseries storage

fix_live_trading.py

@@ -0,0 +1,71 @@
+def fix_live_trading():
+    try:
+        # Read the file content as a single string
+        with open('main.py', 'r') as f:
+            content = f.read()
+        
+        print(f"Read {len(content)} characters from main.py")
+        
+        # Fix the live_trading function signature
+        live_trading_pos = content.find('async def live_trading(')
+        if live_trading_pos != -1:
+            print(f"Found live_trading function at position {live_trading_pos}")
+            content = content.replace('async def live_trading(', 'async def live_trading(agent=None, env=None, exchange=None, ')
+            print("Updated live_trading function signature")
+        else:
+            print("WARNING: Could not find live_trading function!")
+        
+        # Fix the TradingEnvironment initialization
+        env_init_pos = content.find('env = TradingEnvironment(')
+        if env_init_pos != -1:
+            print(f"Found env initialization at position {env_init_pos}")
+            
+            # Find the closing parenthesis
+            paren_depth = 0
+            close_pos = env_init_pos
+            
+            for i in range(env_init_pos, len(content)):
+                if content[i] == '(':
+                    paren_depth += 1
+                elif content[i] == ')':
+                    paren_depth -= 1
+                    if paren_depth == 0:
+                        close_pos = i + 1
+                        break
+            
+            # Calculate indentation
+            line_start = content.rfind('\n', 0, env_init_pos) + 1
+            indent = ' ' * (env_init_pos - line_start)
+            
+            # Create the new environment initialization code
+            new_env_init = f'''if env is None:
+{indent}    env = TradingEnvironment(
+{indent}        initial_balance=initial_balance,
+{indent}        leverage=leverage,
+{indent}        window_size=window_size,
+{indent}        commission=commission,
+{indent}        symbol=symbol,
+{indent}        timeframe=timeframe
+{indent}    )'''
+            
+            # Replace the old code with the new code
+            content = content[:env_init_pos] + new_env_init + content[close_pos:]
+            print("Updated TradingEnvironment initialization")
+        else:
+            print("WARNING: Could not find TradingEnvironment initialization!")
+        
+        # Write the updated content back to the file
+        with open('main.py', 'w') as f:
+            f.write(content)
+        
+        print(f"Wrote {len(content)} characters back to main.py")
+        print('Fixed live_trading function and TradingEnvironment initialization')
+        return True
+    except Exception as e:
+        print(f'Error fixing file: {e}')
+        import traceback
+        print(traceback.format_exc())
+        return False
+
+if __name__ == "__main__":
+    fix_live_trading()