dojozero 0.2.0__tar.gz → 0.2.2__tar.gz

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

@@ -158,7 +158,8 @@ uv.lock
 Thumbs.db
 
 # Sample scenario artifacts
-dojozero-store
+dojozero-store/
+store/
 
 # Data
 /outputs/
@@ -171,5 +172,11 @@ dojozero-store
 package-lock.json
 
 # logs
-/logs/node_modules/
+logs/
+
+# node modules
 node_modules/
+
+
+agents/llms/*.yaml
+!agents/llms/default.yaml

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dojozero
-Version: 0.2.0
+Version: 0.2.2
 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.2/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.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dojozero"
-version = "0.2.0"
+version = "0.2.2"
 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.2}/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.2}/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.2}/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.2}/src/dojozero/arena_server/_endpoints.py

@@ -357,11 +357,11 @@ def register_rest_endpoints(app: FastAPI) -> None:
             default=None,
             description="Filter by league: 'NBA', 'NFL', etc.",
         ),
-        limit: int = Query(
-            default=20,
-            description="Maximum number of agents to return.",
+        limit: int | None = Query(
+            default=None,
+            description="Maximum number of agents to return. Returns all if not specified.",
             ge=1,
-            le=100,
+            le=1000,
         ),
     ) -> JSONResponse:
         """Get agent leaderboard ranked by winnings.
@@ -379,8 +379,9 @@ def register_rest_endpoints(app: FastAPI) -> None:
         if leaderboard is None:
             leaderboard = await refresher.refresh_leaderboard_on_demand(league=league)
 
-        # Apply limit
-        leaderboard = leaderboard[:limit]
+        # Apply limit only if specified
+        if limit is not None:
+            leaderboard = leaderboard[:limit]
 
         response = LeaderboardResponse(leaderboard=leaderboard)
         return JSONResponse(

{dojozero-0.2.0 → dojozero-0.2.2}/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.2}/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)
@@ -804,7 +831,7 @@ def _compute_leaderboard_from_spans(
     spans_by_trial: dict[str, list[SpanData]],
     agent_info_cache: dict[str, AgentInfo],
     trial_ids: list[str] | None = None,
-    limit: int = 20,
+    limit: int | None = None,
 ) -> list[LeaderboardEntry]:
     """Compute agent leaderboard from pre-fetched spans.
 
@@ -815,7 +842,7 @@ def _compute_leaderboard_from_spans(
         spans_by_trial: Pre-fetched spans grouped by trial_id
         agent_info_cache: Pre-populated agent info cache (agent_id -> AgentInfo)
         trial_ids: Optional list to filter which trials to process (None = all)
-        limit: Maximum entries to return
+        limit: Maximum entries to return (None = all)
 
     Returns:
         List of agents sorted by winnings (highest first)
@@ -910,9 +937,9 @@ def _compute_leaderboard_from_spans(
 
     # Sort by winnings (descending) and add rank
     leaderboard.sort(key=lambda x: x.winnings, reverse=True)
+    entries = leaderboard[:limit] if limit is not None else leaderboard
     ranked = [
-        entry.model_copy(update={"rank": i + 1})
-        for i, entry in enumerate(leaderboard[:limit])
+        entry.model_copy(update={"rank": i + 1}) for i, entry in enumerate(entries)
     ]
 
     return ranked
@@ -922,7 +949,7 @@ async def _compute_leaderboard(
     trace_reader: TraceReader,
     trial_ids: list[str],
     cache: "LandingPageCache | None" = None,
-    limit: int = 20,
+    limit: int | None = None,
 ) -> list[LeaderboardEntry]:
     """Compute agent leaderboard (on-demand version).
 
@@ -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.2}/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.2}/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.2}/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.2}/src/dojozero/betting/_models.py

@@ -435,6 +435,10 @@ class AgentInfo(BaseModel):
     """Agent registration payload for tracing (agent.agent_initialize span)."""
 
     agent_id: str = Field(default="", description="Unique ID for the agent")
+    display_name: str = Field(
+        default="",
+        description="Human-readable display name (e.g., GitHub username or custom name)",
+    )
     persona: str = Field(
         default="", description="Agent persona tag (e.g., 'degen', 'whale', 'shark')"
     )