Initial commit of integrated agent system

shared/schema/portfolio.py

@@ -0,0 +1,23 @@
+# -*- coding: utf-8 -*-
+"""Portfolio-related schemas."""
+
+from pydantic import BaseModel
+
+
+class Position(BaseModel):
+    """Position information - for Portfolio mode"""
+
+    long: int = 0  # Long position quantity (shares)
+    short: int = 0  # Short position quantity (shares)
+    long_cost_basis: float = 0.0  # Long position average cost
+    short_cost_basis: float = 0.0  # Short position average cost
+
+
+class Portfolio(BaseModel):
+    """Portfolio - for Portfolio mode"""
+
+    cash: float = 100000.0  # Available cash
+    positions: dict[str, Position] = {}  # ticker -> Position mapping
+    # Margin requirement (0.0 means shorting disabled, 0.5 means 50% margin)
+    margin_requirement: float = 0.0
+    margin_used: float = 0.0  # Margin used