diff options
Diffstat (limited to 'services/strategy-engine/src/strategy_engine/stock_selector.py')
| -rw-r--r-- | services/strategy-engine/src/strategy_engine/stock_selector.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/services/strategy-engine/src/strategy_engine/stock_selector.py b/services/strategy-engine/src/strategy_engine/stock_selector.py index 5acef0f..8657b93 100644 --- a/services/strategy-engine/src/strategy_engine/stock_selector.py +++ b/services/strategy-engine/src/strategy_engine/stock_selector.py @@ -1,5 +1,6 @@ """3-stage stock selector engine: sentiment → technical → LLM.""" +import asyncio import json import logging import re @@ -218,11 +219,13 @@ class StockSelector: self._model = anthropic_model self._max_picks = max_picks self._http_session: aiohttp.ClientSession | None = None + self._session_lock = asyncio.Lock() async def _ensure_session(self) -> aiohttp.ClientSession: - if self._http_session is None or self._http_session.closed: - self._http_session = aiohttp.ClientSession() - return self._http_session + async with self._session_lock: + if self._http_session is None or self._http_session.closed: + self._http_session = aiohttp.ClientSession() + return self._http_session async def close(self) -> None: if self._http_session and not self._http_session.closed: |