Use QuantPlatformKit shared strategy contract

Author: Pigbibi

requirements.txt

@@ -1,6 +1,6 @@
 flask
 gunicorn
-quant-platform-kit @ git+https://github.com/QuantStrategyLab/QuantPlatformKit.git@v0.4.0
+quant-platform-kit @ git+https://github.com/QuantStrategyLab/QuantPlatformKit.git@v0.5.0
 pandas
 requests
 pytz

strategy_registry.py

@@ -1,20 +1,15 @@
 from __future__ import annotations
 
-from dataclasses import dataclass
-
-US_EQUITY_DOMAIN = "us_equity"
-CRYPTO_DOMAIN = "crypto"
+from quant_platform_kit.common.strategies import (
+    US_EQUITY_DOMAIN,
+    StrategyDefinition,
+    get_supported_profiles_for_platform as qpk_get_supported_profiles_for_platform,
+    resolve_strategy_definition as qpk_resolve_strategy_definition,
+)
 
 LONGBRIDGE_PLATFORM = "longbridge"
 
 
-@dataclass(frozen=True)
-class StrategyDefinition:
-    profile: str
-    domain: str
-    supported_platforms: frozenset[str]
-
-
 DEFAULT_STRATEGY_PROFILE = "semiconductor_rotation_income"
 
 STRATEGY_DEFINITIONS: dict[str, StrategyDefinition] = {
@@ -33,11 +28,10 @@ class StrategyDefinition:
 
 
 def get_supported_profiles_for_platform(platform_id: str) -> frozenset[str]:
-    return frozenset(
-        profile
-        for profile, definition in STRATEGY_DEFINITIONS.items()
-        if platform_id in definition.supported_platforms
-        and definition.domain in PLATFORM_SUPPORTED_DOMAINS.get(platform_id, frozenset())
+    return qpk_get_supported_profiles_for_platform(
+        STRATEGY_DEFINITIONS,
+        PLATFORM_SUPPORTED_DOMAINS,
+        platform_id=platform_id,
     )
 
 
@@ -46,23 +40,10 @@ def resolve_strategy_definition(
     *,
     platform_id: str,
 ) -> StrategyDefinition:
-    profile = (raw_value or DEFAULT_STRATEGY_PROFILE).strip().lower()
-    definition = STRATEGY_DEFINITIONS.get(profile)
-    supported = ", ".join(sorted(get_supported_profiles_for_platform(platform_id)))
-
-    if definition is None:
-        raise ValueError(
-            f"Unsupported STRATEGY_PROFILE={raw_value!r}; supported values: {supported}"
-        )
-
-    if platform_id not in definition.supported_platforms:
-        raise ValueError(
-            f"Unsupported STRATEGY_PROFILE={raw_value!r}; supported values: {supported}"
-        )
-
-    if definition.domain not in PLATFORM_SUPPORTED_DOMAINS.get(platform_id, frozenset()):
-        raise ValueError(
-            f"Unsupported strategy domain {definition.domain!r} for platform {platform_id!r}"
-        )
-
-    return definition
+    return qpk_resolve_strategy_definition(
+        raw_value,
+        platform_id=platform_id,
+        strategy_definitions=STRATEGY_DEFINITIONS,
+        platform_supported_domains=PLATFORM_SUPPORTED_DOMAINS,
+        default_profile=DEFAULT_STRATEGY_PROFILE,
+    )

tests/test_runtime_config_support.py

@@ -8,6 +8,9 @@
 ROOT = Path(__file__).resolve().parents[1]
 if str(ROOT) not in sys.path:
     sys.path.insert(0, str(ROOT))
+QPK_SRC = ROOT.parent / "QuantPlatformKit" / "src"
+if str(QPK_SRC) not in sys.path:
+    sys.path.insert(0, str(QPK_SRC))
 
 from runtime_config_support import (
     DEFAULT_ACCOUNT_REGION,