dojozero 0.2.1__tar.gz → 0.2.2__tar.gz

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

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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dojozero
-Version: 0.2.1
+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

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

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

@@ -831,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.
 
@@ -842,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)
@@ -937,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
@@ -949,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).
 

{dojozero-0.2.1 → 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')"
     )

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

@@ -34,6 +34,7 @@ from dojozero.data.polymarket import PolymarketAPI
 from dojozero.utils import utc_iso_to_local
 from dojozero.core import TrialBuilderNotFoundError as _TrialBuilderNotFoundError
 from dojozero.dashboard_server import InitialTrialSourceDict
+from dojozero.dashboard_server._scheduler import SchedulerStore
 
 try:  # Optional Ray dependency
     from dojozero.ray_runtime import RayActorRuntimeProvider
@@ -389,6 +390,26 @@ def _build_parser() -> argparse.ArgumentParser:
         action="store_true",
         help="Disable HTTP gateway for external agents (gateway is enabled by default).",
     )
+    # Cluster arguments
+    serve_parser.add_argument(
+        "--server-id",
+        dest="server_id",
+        default=None,
+        help="Unique server identifier for cluster mode (default: hostname).",
+    )
+    serve_parser.add_argument(
+        "--server-url",
+        dest="server_url",
+        default=None,
+        help="Externally reachable URL for this server (default: http://{host}:{port}).",
+    )
+    serve_parser.add_argument(
+        "--cluster-redis-url",
+        dest="cluster_redis_url",
+        default=os.environ.get("DOJOZERO_CLUSTER_REDIS_URL"),
+        help="Redis URL for cluster leader election and peer discovery (enables redis mode). "
+        "Falls back to DOJOZERO_CLUSTER_REDIS_URL env var.",
+    )
 
     # Arena Server command
     arena_parser = subparsers.add_parser(
@@ -2020,10 +2041,17 @@ async def _serve_command(args: argparse.Namespace) -> int:
         args.store_directory if args.store_directory else Path(DEFAULT_STORE_DIRECTORY)
     )
 
-    # Create scheduler store (uses store root directory for persistence)
-    from dojozero.dashboard_server._scheduler import FileSchedulerStore
+    # Create scheduler store — Redis in cluster mode, file otherwise
+    cluster_redis_url: str | None = getattr(args, "cluster_redis_url", None)
 
-    scheduler_store = FileSchedulerStore(store_path)
+    if cluster_redis_url:
+        from dojozero.dashboard_server._scheduler import RedisSchedulerStore
+
+        scheduler_store: SchedulerStore = RedisSchedulerStore(cluster_redis_url)
+    else:
+        from dojozero.dashboard_server._scheduler import FileSchedulerStore
+
+        scheduler_store = FileSchedulerStore(store_path)
 
     # Expand glob patterns and load trial source configurations
     import glob as glob_module
@@ -2109,6 +2137,43 @@ async def _serve_command(args: argparse.Namespace) -> int:
         "enabled" if local_auth else "disabled",
     )
 
+    # Build cluster config if Redis cluster URL is provided
+    cluster_config = None
+    server_id_arg: str | None = getattr(args, "server_id", None)
+    server_url_arg: str | None = getattr(args, "server_url", None)
+
+    if cluster_redis_url:
+        from dojozero.dashboard_server._cluster import ClusterConfig
+
+        if server_url_arg:
+            server_url = server_url_arg
+        elif host in ("0.0.0.0", "::"):
+            # Resolve to a reachable IP so peers can forward trials to us.
+            # In Kubernetes, expose status.podIP as the POD_IP env var via
+            # the Downward API.  The socket fallback works on plain VMs but
+            # may return 127.0.0.1 inside containers — warn if that happens.
+            import socket
+
+            pod_ip = os.environ.get("POD_IP", "").strip()
+            if not pod_ip:
+                pod_ip = socket.gethostbyname(socket.gethostname())
+                if pod_ip.startswith("127."):
+                    LOGGER.warning(
+                        "Resolved server IP is %s which is not routable by "
+                        "peers. Set --server-url or the POD_IP env var to "
+                        "this server's reachable address.",
+                        pod_ip,
+                    )
+            server_url = f"http://{pod_ip}:{port}"
+        else:
+            server_url = f"http://{host}:{port}"
+        cluster_config = ClusterConfig(
+            server_id=server_id_arg or "",
+            server_url=server_url,
+            redis_url=cluster_redis_url,
+        )
+        LOGGER.info("Cluster mode enabled (redis_url=%s)", cluster_redis_url)
+
     await run_dashboard_server(
         orchestrator=orchestrator,
         scheduler_store=scheduler_store,
@@ -2124,6 +2189,7 @@ async def _serve_command(args: argparse.Namespace) -> int:
         enable_gateway=enable_gateway,
         authenticator=authenticator,
         no_scheduler=no_scheduler,
+        cluster_config=cluster_config,
     )
     return 0
 

{dojozero-0.2.1 → dojozero-0.2.2}/src/dojozero/core/_filesystem_orchestrator_store.py

@@ -74,12 +74,22 @@ class FileSystemOrchestratorStore(OrchestratorStore):
         spec = self._read_spec(spec_path)
         status_path = trial_dir / self.STATUS_FILE
         status = self._read_status(status_path) if status_path.exists() else None
-        return TrialRecord(spec=spec, last_status=status)
+        # Read owner_server_id from spec file (stored alongside spec)
+        spec_payload = self._read_json(spec_path)
+        owner_server_id = spec_payload.get("owner_server_id")
+        return TrialRecord(
+            spec=spec,
+            last_status=status,
+            owner_server_id=owner_server_id,
+        )
 
     def upsert_trial_record(self, record: TrialRecord) -> None:
         trial_dir = self._trial_dir(record.trial_id)
         trial_dir.mkdir(parents=True, exist_ok=True)
-        self._write_json(trial_dir / self.SPEC_FILE, self._serialize_spec(record.spec))
+        spec_data = self._serialize_spec(record.spec)
+        if record.owner_server_id is not None:
+            spec_data["owner_server_id"] = record.owner_server_id
+        self._write_json(trial_dir / self.SPEC_FILE, spec_data)
         if record.last_status is not None:
             self._write_json(
                 trial_dir / self.STATUS_FILE, self._serialize_status(record.last_status)

{dojozero-0.2.1 → dojozero-0.2.2}/src/dojozero/core/_trial_orchestrator.py

@@ -297,6 +297,7 @@ class TrialRecord:
 
     spec: TrialSpec
     last_status: TrialStatus | None = None
+    owner_server_id: str | None = None
 
     @property
     def trial_id(self) -> str:
@@ -448,15 +449,24 @@ class TrialOrchestrator:
         """Access the underlying OrchestratorStore."""
         return self._store
 
-    async def launch_trial(self, spec: TrialSpec) -> TrialStatus:
-        """Instantiate and start every actor defined in *spec*."""
+    async def launch_trial(
+        self,
+        spec: TrialSpec,
+        owner_server_id: str | None = None,
+    ) -> TrialStatus:
+        """Instantiate and start every actor defined in *spec*.
+
+        Args:
+            spec: Trial specification.
+            owner_server_id: Server that owns this trial (cluster mode).
+        """
 
         LOGGER.info("launching trial '%s'", spec.trial_id)
         spec = self._apply_resume_from_spec(spec)
         normalized_spec = self._normalize_spec(spec)
         record = self._catalog.get(spec.trial_id)
         if record is None:
-            record = TrialRecord(spec=normalized_spec)
+            record = TrialRecord(spec=normalized_spec, owner_server_id=owner_server_id)
         elif record.spec != normalized_spec:
             raise OrchestratorError(
                 f"trial '{spec.trial_id}' already registered with a different configuration"

{dojozero-0.2.1 → dojozero-0.2.2}/src/dojozero/dashboard_server/__init__.py

@@ -8,6 +8,10 @@ This package provides the Dashboard Server functionality including:
 Moved from core module to keep server-related code separate from core abstractions.
 """
 
+from ._cluster import (
+    ClusterConfig,
+    PeerInfo,
+)
 from ._game_discovery import (
     GameInfo,
     NBAGameFetcher,
@@ -17,6 +21,7 @@ from ._game_discovery import (
 )
 from ._scheduler import (
     FileSchedulerStore,
+    RedisSchedulerStore,
     ScheduledTrial,
     ScheduledTrialPhase,
     ScheduleManager,
@@ -60,6 +65,9 @@ from ._types import (
 )
 
 __all__ = [
+    # Cluster
+    "ClusterConfig",
+    "PeerInfo",
     # Server
     "create_dashboard_app",
     "run_dashboard_server",
@@ -78,6 +86,7 @@ __all__ = [
     "ScheduledTrialPhase",
     "SchedulerStore",
     "FileSchedulerStore",
+    "RedisSchedulerStore",
     # Trial Sources
     "TrialSource",
     "TrialSourceConfig",