dojozero 0.2.0__tar.gz → 0.2.1__tar.gz

{dojozero-0.2.0 → dojozero-0.2.1}/.gitignore

@@ -171,5 +171,7 @@ dojozero-store
 package-lock.json
 
 # logs
-/logs/node_modules/
-node_modules/
+logs/
+
+# node modules
+node_modules/

{dojozero-0.2.0 → dojozero-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dojozero
-Version: 0.2.0
+Version: 0.2.1
 Summary: A platform for running AI agents on realtime sport data and make predictions about game outcomes.
 Project-URL: Homepage, https://github.com/agentscope-ai/DojoZero
 Project-URL: Repository, https://github.com/agentscope-ai/DojoZero
@@ -44,7 +44,7 @@ Description-Content-Type: text/markdown
 
 # DojoZero
 
-A platform for running AI agents on realtime sport data and make predictions about game outcomes.
+A platform for running AI agents on realtime sport data to reason about game outcomes and make predictions. DojoZero provides an actor-based runtime with live data streams, agent orchestration, trial management, backtesting, and tracing. Currently supports NBA, NFL, and NCAA.
 
 ## Installation
 
@@ -52,83 +52,17 @@ A platform for running AI agents on realtime sport data and make predictions abo
 pip install dojozero
 ```
 
-### Optional extras
+Optional extras:
 
 ```bash
-pip install dojozero[alicloud]   # Alibaba Cloud integration (OSS, credentials, SLS)
+pip install dojozero[alicloud]   # Alibaba Cloud integration (OSS, SLS tracing)
 pip install dojozero[redis]      # Redis-backed data stores
 pip install dojozero[ray]        # Distributed execution via Ray
 ```
 
-## Environment Setup
-
-```bash
-cp .env.example .env
-```
-
-## Quick Start
-
-### Run a single trial locally
-
-```bash
-dojo0 run --params trial_params/nba-moneyline.yaml --trial-id nba-local-001
-```
-
-### Resume an interrupted trial
-
-```bash
-dojo0 run --trial-id nba-local-001 --resume-latest
-```
-
-### Run through dashboard server
-
-```bash
-# Start server with tracing
-dojo0 serve --trace-backend jaeger
-
-# Submit a trial from another terminal
-dojo0 run \
-  --params trial_params/nba-moneyline.yaml \
-  --trial-id nba-server-001 \
-  --server http://localhost:8000
-```
-
-### Automatic scheduling with trial sources
-
-```bash
-dojo0 serve --trace-backend jaeger --trial-source "trial_sources/daily/*.yaml"
-```
-
-### Backtest from captured events
-
-```bash
-dojo0 backtest \
-  --events outputs/2026-01-12/401772976.jsonl \
-  --params outputs/2026-01-12/401772976.yaml \
-  --speed 100 --max-sleep 1
-```
-
-### Arena (live visualization)
-
-```bash
-dojo0 arena --trace-backend jaeger
-```
-
-## Command Index
-
-| Command | Purpose |
-|---------|---------|
-| `dojo0 run` | Start a local or server-submitted trial |
-| `dojo0 serve` | Dashboard / orchestration server |
-| `dojo0 arena` | Trace-backed Arena visualization server |
-| `dojo0 backtest` | Replay persisted event streams |
-| `dojo0 list-sources` | List trial sources |
-| `dojo0 list-trials` | List scheduled trials |
-| `dojo0 clear-schedules` | Clear scheduled runs |
-
 ## Documentation
 
-- [Documentation](https://github.com/agentscope-ai/DojoZero/docs/README.md)
+See the [full documentation](https://github.com/agentscope-ai/DojoZero/tree/main/docs) for setup, configuration, and usage guides.
 
 ## License
 

dojozero-0.2.1/README.md

@@ -0,0 +1,25 @@
+# DojoZero
+
+A platform for running AI agents on realtime sport data to reason about game outcomes and make predictions. DojoZero provides an actor-based runtime with live data streams, agent orchestration, trial management, backtesting, and tracing. Currently supports NBA, NFL, and NCAA.
+
+## Installation
+
+```bash
+pip install dojozero
+```
+
+Optional extras:
+
+```bash
+pip install dojozero[alicloud]   # Alibaba Cloud integration (OSS, SLS tracing)
+pip install dojozero[redis]      # Redis-backed data stores
+pip install dojozero[ray]        # Distributed execution via Ray
+```
+
+## Documentation
+
+See the [full documentation](https://github.com/agentscope-ai/DojoZero/tree/main/docs) for setup, configuration, and usage guides.
+
+## License
+
+MIT

{dojozero-0.2.0 → dojozero-0.2.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dojozero"
-version = "0.2.0"
+version = "0.2.1"
 description = "A platform for running AI agents on realtime sport data and make predictions about game outcomes."
 readme = "README.md"
 requires-python = ">=3.11"

{dojozero-0.2.0 → dojozero-0.2.1}/src/dojozero/agents/_config.py

@@ -278,7 +278,10 @@ def create_model(llm_config: LLMConfig) -> ChatModelBase:
     if model_type in "grok":
         grok_client_kwargs: dict[str, Any] = {"base_url": "https://api.x.ai/v1"}
         return OpenAIChatModel(
-            model_name=model_name, api_key=api_key, client_kwargs=grok_client_kwargs
+            model_name=model_name,
+            api_key=api_key,
+            stream=False,
+            client_kwargs=grok_client_kwargs,
         )
     elif model_type == "openai":
         if base_url:
@@ -286,14 +289,36 @@ def create_model(llm_config: LLMConfig) -> ChatModelBase:
         else:
             client_kwargs = {}
         return OpenAIChatModel(
-            model_name=model_name, api_key=api_key, client_kwargs=client_kwargs
+            model_name=model_name,
+            api_key=api_key,
+            stream=False,
+            client_kwargs=client_kwargs,
         )
     elif model_type == "dashscope":
-        return DashScopeChatModel(model_name=model_name, api_key=api_key)
+        # qwen3.5-* series requires the MultiModalConversation endpoint even
+        # for text-only usage.  The upstream agentscope SDK only auto-detects
+        # "-vl" / "qvq" suffixes/name patterns, so we explicitly opt-in here.
+        multimodality: bool | None = None
+        if model_name.startswith("qwen3.5"):
+            multimodality = True
+        return DashScopeChatModel(
+            model_name=model_name,
+            api_key=api_key,
+            multimodality=multimodality,
+            stream=False,
+        )
     elif model_type == "anthropic":
-        return AnthropicChatModel(model_name=model_name, api_key=api_key)
+        return AnthropicChatModel(
+            model_name=model_name,
+            api_key=api_key,
+            stream=False,
+        )
     elif model_type == "gemini":
-        return GeminiChatModel(model_name=model_name, api_key=api_key)
+        return GeminiChatModel(
+            model_name=model_name,
+            api_key=api_key,
+            stream=False,
+        )
     else:
         raise ValueError(f"Unknown model_type: {model_type}")
 

{dojozero-0.2.0 → dojozero-0.2.1}/src/dojozero/arena_server/_cache.py

@@ -519,6 +519,12 @@ class ReplayMetaInfo:
     odds_update_indices: list[int] = field(
         default_factory=list
     )  # Indices of odds_update events
+    broker_state_update_indices: list[int] = field(
+        default_factory=list
+    )  # Indices of broker.state_update events
+    broker_final_stats_indices: list[int] = field(
+        default_factory=list
+    )  # Indices of broker.final_stats events
 
 
 @dataclass

{dojozero-0.2.0 → dojozero-0.2.1}/src/dojozero/arena_server/_constants.py

@@ -266,18 +266,45 @@ _NFL_TEAMS: dict[str, TeamIdentity] = {
 _DEFAULT_TEAM_COLOR = "#666666"
 
 
-def _get_team_identity(tricode: str, league: str = "NBA") -> TeamIdentity:
+def _get_team_identity(
+    tricode: str, league: str = "NBA", *, team_id: str = ""
+) -> TeamIdentity:
     """Get team identity by tricode.
 
     Returns TeamIdentity from static lookup. Falls back to a minimal identity
-    with the tricode as the name if not found, but still generates a logo URL
-    using the ESPN CDN pattern.
+    with the tricode as the name if not found.
+
+    For NBA/NFL, unknown tricodes use the ESPN CDN pattern
+    ``/teamlogos/{league}/500/{tricode}.png``.
+
+    NCAA mens basketball logos on ESPN use **numeric** team IDs in
+    ``/teamlogos/ncaa/500/{id}.png``; tricode-based URLs usually 404, so pass
+    ``team_id`` from trial metadata when available, otherwise ``logo_url`` is
+    left empty to avoid broken images.
     """
-    teams = _NBA_TEAMS if league == "NBA" else _NFL_TEAMS
+    lu = league.upper()
+    if lu == "NBA":
+        teams = _NBA_TEAMS
+    elif lu == "NFL":
+        teams = _NFL_TEAMS
+    else:
+        teams = {}
     if tricode in teams:
         return teams[tricode]
 
-    # Generate logo URL dynamically for teams not in static lookup
+    if lu == "NCAA":
+        logo_url = ""
+        tid = (team_id or "").strip()
+        if tid.isdigit():
+            logo_url = f"https://a.espncdn.com/i/teamlogos/ncaa/500/{tid}.png"
+        return TeamIdentity(
+            name=tricode,
+            tricode=tricode,
+            team_id=tid,
+            color=_DEFAULT_TEAM_COLOR,
+            logo_url=logo_url,
+        )
+
     league_lower = league.lower()
     logo_url = (
         f"https://a.espncdn.com/i/teamlogos/{league_lower}/500/{tricode.lower()}.png"

{dojozero-0.2.0 → dojozero-0.2.1}/src/dojozero/arena_server/_server.py

@@ -1473,6 +1473,8 @@ class TrialReplayController:
         The snapshot includes:
         - Meta events: agent_initialize, game_initialize, game_start
         - Latest odds_update before the target position
+        - Latest broker snapshot before the target position
+          (prefer final_stats over state_update)
         - Recent play items up to and including the target
 
         Args:
@@ -1497,6 +1499,15 @@ class TrialReplayController:
         # Collect item indices to include in snapshot (use set to avoid duplicates)
         snapshot_indices: set[int] = set()
 
+        def _latest_index_before(indices: list[int]) -> int | None:
+            latest_idx = None
+            for idx in indices:
+                if idx <= target_item_index:
+                    latest_idx = idx
+                else:
+                    break
+            return latest_idx
+
         # 1. Add meta events (agent_initialize, game_initialize, game_start)
         if self.meta.agent_initialize_item_index is not None:
             snapshot_indices.add(self.meta.agent_initialize_item_index)
@@ -1507,17 +1518,21 @@ class TrialReplayController:
 
         # 2. Find and add the latest odds_update before target_item_index
         if self.meta.odds_update_indices:
-            # Find the largest odds_update index that is <= target_item_index
-            latest_odds_idx = None
-            for idx in self.meta.odds_update_indices:
-                if idx <= target_item_index:
-                    latest_odds_idx = idx
-                else:
-                    break  # odds_update_indices are in chronological order
+            latest_odds_idx = _latest_index_before(self.meta.odds_update_indices)
             if latest_odds_idx is not None:
                 snapshot_indices.add(latest_odds_idx)
 
-        # 3. Add recent items up to and including the target
+        # 3. Add the latest broker snapshot before target_item_index.
+        # final_stats is preferred because it contains the ranking/statistics payload.
+        latest_broker_idx = _latest_index_before(self.meta.broker_final_stats_indices)
+        if latest_broker_idx is None:
+            latest_broker_idx = _latest_index_before(
+                self.meta.broker_state_update_indices
+            )
+        if latest_broker_idx is not None:
+            snapshot_indices.add(latest_broker_idx)
+
+        # 4. Add recent items up to and including the target
         start = max(0, target_item_index + 1 - self.snapshot_size)
         for i in range(start, target_item_index + 1):
             snapshot_indices.add(i)

{dojozero-0.2.0 → dojozero-0.2.1}/src/dojozero/arena_server/_utils.py

@@ -46,6 +46,7 @@ TRIAL_INFO_OPERATION_NAMES = [
     "event.game_result",
     "event.nba_game_update",
     "event.nfl_game_update",
+    "event.ncaa_game_update",
 ]
 
 # Tag used by DojoZero to correlate all spans for a trial (see TrialOrchestrator, create_span_from_event).
@@ -258,15 +259,25 @@ def _resolve_team_identity(
     fallback_tricode: str,
     fallback_name: str,
     league: str,
+    *,
+    team_id: str = "",
 ) -> TeamIdentity:
     """Resolve a team to a TeamIdentity, applying fallbacks as needed."""
     if isinstance(team, TeamIdentity) and team:
-        # Ensure tricode is populated
+        resolved = team
         if not team.tricode and fallback_tricode:
-            return team.model_copy(update={"tricode": fallback_tricode})
-        return team
+            resolved = resolved.model_copy(update={"tricode": fallback_tricode})
+        if league.upper() == "NCAA" and not resolved.logo_url:
+            tid = (resolved.team_id or team_id or "").strip()
+            if tid.isdigit():
+                resolved = resolved.model_copy(
+                    update={
+                        "logo_url": f"https://a.espncdn.com/i/teamlogos/ncaa/500/{tid}.png"
+                    }
+                )
+        return resolved
     # Fallback to static lookup, then override name if provided
-    identity = _get_team_identity(fallback_tricode, league)
+    identity = _get_team_identity(fallback_tricode, league, team_id=team_id)
     if fallback_name and fallback_name != identity.name:
         return identity.model_copy(update={"name": fallback_name})
     return identity
@@ -433,17 +444,33 @@ async def _extract_games_from_trials(
         game_init = trial_info.get("game_init")
         if isinstance(game_init, GameInitializeEvent):
             home_team = _resolve_team_identity(
-                game_init.home_team, home_tricode, "", league
+                game_init.home_team,
+                home_tricode,
+                "",
+                league,
+                team_id=str(metadata.get("home_team_id", "") or ""),
             )
             away_team = _resolve_team_identity(
-                game_init.away_team, away_tricode, "", league
+                game_init.away_team,
+                away_tricode,
+                "",
+                league,
+                team_id=str(metadata.get("away_team_id", "") or ""),
             )
         else:
             home_team = _resolve_team_identity(
-                "", home_tricode, metadata.get("home_team_name", ""), league
+                "",
+                home_tricode,
+                metadata.get("home_team_name", ""),
+                league,
+                team_id=str(metadata.get("home_team_id", "") or ""),
             )
             away_team = _resolve_team_identity(
-                "", away_tricode, metadata.get("away_team_name", ""), league
+                "",
+                away_tricode,
+                metadata.get("away_team_name", ""),
+                league,
+                team_id=str(metadata.get("away_team_id", "") or ""),
             )
 
         # Fetch bets for live games only (performance optimization)
@@ -962,6 +989,7 @@ def _compute_replay_meta(
     - play_item_indices: mapping from play_index to item_index
     - periods: list of PeriodInfo with play counts per period
     - meta event indices: agent_initialize, game_initialize, game_start, odds_update
+    - broker snapshot indices: state_update and final_stats
 
     Args:
         items: List of serialized span dicts with "category" and "data" keys
@@ -979,6 +1007,8 @@ def _compute_replay_meta(
     game_initialize_item_index: int | None = None
     game_start_item_index: int | None = None
     odds_update_indices: list[int] = []
+    broker_state_update_indices: list[int] = []
+    broker_final_stats_indices: list[int] = []
 
     current_period: int = 1  # Default period
 
@@ -995,6 +1025,10 @@ def _compute_replay_meta(
             game_start_item_index = item_index
         elif category == "odds_update":
             odds_update_indices.append(item_index)
+        elif category == "state_update":
+            broker_state_update_indices.append(item_index)
+        elif category == "final_stats":
+            broker_final_stats_indices.append(item_index)
 
         # Track core category items (plays)
         if category in core_categories:
@@ -1031,6 +1065,8 @@ def _compute_replay_meta(
         game_initialize_item_index=game_initialize_item_index,
         game_start_item_index=game_start_item_index,
         odds_update_indices=odds_update_indices,
+        broker_state_update_indices=broker_state_update_indices,
+        broker_final_stats_indices=broker_final_stats_indices,
     )
 
 

{dojozero-0.2.0 → dojozero-0.2.1}/src/dojozero/betting/_agent.py

@@ -516,12 +516,7 @@ class BettingAgent(AgentBase, Agent[BettingAgentConfig]):
             summary_request = self._build_summary_request()
             if summary_request:
                 model = self._react_agent.model
-                original_stream = model.stream
-                model.stream = False
-                try:
-                    resp = await model(summary_request)
-                finally:
-                    model.stream = original_stream
+                resp = await model(summary_request)
                 resp_content = getattr(resp, "content", None)
                 events_summary, _ = _parse_response_content(resp_content)
                 if not events_summary:

{dojozero-0.2.0 → dojozero-0.2.1}/src/dojozero/betting/_broker.py

@@ -262,6 +262,11 @@ class BrokerOperator(OperatorBase, Operator[BrokerOperatorConfig]):
             len(self._bets),
         )
 
+    @property
+    def has_unsettled_bets(self) -> bool:
+        """Return True if there are active (unsettled) bets."""
+        return any(bet.status == BetStatus.ACTIVE for bet in self._bets.values())
+
     def ensure_event_initialized(
         self,
         event_id: str,

{dojozero-0.2.0 → dojozero-0.2.1}/src/dojozero/betting/_metadata.py

@@ -67,6 +67,10 @@ class BettingTrialMetadata(BaseTrialMetadata):
     away_team_name: str
     game_date: str
 
+    # ESPN team IDs (for Arena logos, APIs; optional, default from game lookup)
+    home_team_id: str = ""
+    away_team_id: str = ""
+
     # Polymarket (optional)
     market_url: str | None = None
 

{dojozero-0.2.0 → dojozero-0.2.1}/src/dojozero/cli.py

@@ -339,7 +339,9 @@ def _build_parser() -> argparse.ArgumentParser:
         default=[],
         help="Path or glob pattern for trial source YAML files (repeatable). "
         "Enables automatic scheduling for the specified sports/scenarios. "
-        "Requires filesystem store to be configured.",
+        "Requires filesystem store to be configured. "
+        "Optional env DOJOZERO_MAX_DAILY_GAMES overrides config.max_daily_games "
+        "for every loaded source (integer; 0 = unlimited).",
     )
     serve_parser.add_argument(
         "--no-auto-resume",
@@ -347,6 +349,13 @@ def _build_parser() -> argparse.ArgumentParser:
         action="store_true",
         help="Disable automatic resuming of interrupted trials from previous shutdown.",
     )
+    serve_parser.add_argument(
+        "--no-scheduler",
+        dest="no_scheduler",
+        action="store_true",
+        help="Disable the trial scheduler entirely. No trial sources will be loaded "
+        "or auto-resolved, and persisted sources will be ignored.",
+    )
     serve_parser.add_argument(
         "--stale-threshold-hours",
         dest="stale_threshold_hours",
@@ -1881,6 +1890,29 @@ def _expand_compact_trial_source(
         config.setdefault(key, value)
 
 
+def _parse_max_daily_games_env_override() -> int | None:
+    """Parse ``DOJOZERO_MAX_DAILY_GAMES`` for ``serve`` trial source loading.
+
+    Returns:
+        ``None`` if unset or empty (YAML values are used as-is).
+
+    Note:
+        ``0`` means unlimited, consistent with scheduler / trial source YAML.
+    """
+    raw = os.environ.get("DOJOZERO_MAX_DAILY_GAMES", "").strip()
+    if not raw:
+        return None
+    try:
+        value = int(raw, 10)
+    except ValueError as e:
+        raise DojoZeroCLIError(
+            f"DOJOZERO_MAX_DAILY_GAMES must be an integer, got {raw!r}"
+        ) from e
+    if value < 0:
+        raise DojoZeroCLIError(f"DOJOZERO_MAX_DAILY_GAMES must be >= 0, got {value}")
+    return value
+
+
 def _load_trial_source_from_yaml(path: Path) -> InitialTrialSourceDict:
     """Load a trial source configuration from a YAML file.
 
@@ -1956,13 +1988,17 @@ async def _serve_command(args: argparse.Namespace) -> int:
     trace_backend = getattr(args, "trace_backend", None)
     trace_ingest_endpoint = getattr(args, "trace_ingest_endpoint", None)
     service_name = getattr(args, "service_name", "dojozero")
+    no_scheduler = getattr(args, "no_scheduler", False)
     trial_source_files: list[str] = getattr(args, "trial_sources", []) or []
     auto_resume = not getattr(args, "no_auto_resume", False)
     stale_threshold_hours = getattr(args, "stale_threshold_hours", 24.0)
 
-    # Auto-resolve trial sources from DOJOZERO_ENV / SIGMA_APP_STAGE if not
-    # explicitly provided via --trial-source.
-    if not trial_source_files:
+    if no_scheduler:
+        trial_source_files = []
+        LOGGER.info("Scheduler disabled via --no-scheduler")
+    elif not trial_source_files:
+        # Auto-resolve trial sources from DOJOZERO_ENV / SIGMA_APP_STAGE if not
+        # explicitly provided via --trial-source.
         env_tier = _resolve_env_tier()
         trial_sources_dir = Path("trial_sources") / env_tier
         if trial_sources_dir.is_dir():
@@ -2010,6 +2046,21 @@ async def _serve_command(args: argparse.Namespace) -> int:
                 source_data.get("source_id"),
             )
 
+    env_max_daily = _parse_max_daily_games_env_override()
+    if env_max_daily is not None:
+        if initial_trial_sources:
+            for source_data in initial_trial_sources:
+                source_data["config"]["max_daily_games"] = env_max_daily
+            LOGGER.info(
+                "DOJOZERO_MAX_DAILY_GAMES=%s overrides max_daily_games for %d trial source(s)",
+                env_max_daily,
+                len(initial_trial_sources),
+            )
+        else:
+            LOGGER.warning(
+                "DOJOZERO_MAX_DAILY_GAMES is set but no trial sources were loaded; ignoring override"
+            )
+
     LOGGER.info("Starting Dashboard Server at http://%s:%d", host, port)
     LOGGER.info("Trial API: http://%s:%d/api/trials", host, port)
     LOGGER.info("Trial Source API: http://%s:%d/api/trial-sources", host, port)
@@ -2072,6 +2123,7 @@ async def _serve_command(args: argparse.Namespace) -> int:
         stale_threshold_hours=stale_threshold_hours,
         enable_gateway=enable_gateway,
         authenticator=authenticator,
+        no_scheduler=no_scheduler,
     )
     return 0
 

{dojozero-0.2.0 → dojozero-0.2.1}/src/dojozero/core/_models.py

@@ -277,8 +277,10 @@ _OPERATION_NAME_MAP: dict[str, str] = {
 _SPORT_UNIFY_MAP: dict[str, str] = {
     "nba_play": "play",
     "nfl_play": "play",
+    "ncaa_play": "play",
     "nba_game_update": "game_update",
     "nfl_game_update": "game_update",
+    "ncaa_game_update": "game_update",
 }