arbor-ai 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.whl

arbor/server/utils/logging.py

@@ -0,0 +1,298 @@
+"""
+Logging configuration for Arbor.
+Provides centralized logging setup with file and console handlers.
+"""
+
+import logging
+import logging.config
+import sys
+from pathlib import Path
+from typing import Any, Dict, Optional
+
+
+class ColoredFormatter(logging.Formatter):
+    """Custom formatter with colors for console output."""
+
+    # ANSI color codes
+    COLORS = {
+        "DEBUG": "\033[36m",  # Cyan
+        "INFO": "\033[32m",  # Green
+        "WARNING": "\033[33m",  # Yellow
+        "ERROR": "\033[31m",  # Red
+        "CRITICAL": "\033[35m",  # Magenta
+        "RESET": "\033[0m",  # Reset
+    }
+
+    def format(self, record):
+        # Use the name mapping to get short names, fallback to original name
+        name_mappings = {
+            "arbor.server.services.inference_manager": "infer",
+            "arbor.server.services.grpo_manager": "grpo",
+            "arbor.server.services.file_manager": "files",
+            "arbor.server.services.health_manager": "health",
+            "arbor.server.services.job_manager": "jobs",
+            "arbor.server.services.training_manager": "train",
+            "arbor.server.services.comms.comms": "comms",
+            "arbor.server.services.scripts.sft_training": "sft",
+            "arbor.server.services.scripts.grpo_training": "grpo",
+            "arbor.config": "config",
+            "arbor.cli": "cli",
+            "__main__": "main",
+            # Add uvicorn loggers
+            "uvicorn": "api",
+            "uvicorn.access": "api",
+            "uvicorn.error": "api",
+        }
+
+        # Get short name if available, otherwise use the provided name
+        short_name = name_mappings.get(record.name, record.name)
+        name = short_name.upper()
+        record.name = f"[{name}]".rjust(
+            8
+        )  # Right-align the whole bracketed name (8 chars max)
+
+        # Store original level name for color lookup
+        original_level = record.levelname
+
+        # Convert level names to 4-character abbreviations
+        level_abbreviations = {
+            "DEBUG": "DEBG",
+            "INFO": "INFO",
+            "WARNING": "WARN",
+            "ERROR": "ERRO",
+            "CRITICAL": "CRIT",
+        }
+
+        # Get abbreviated level name and add brackets with padding
+        abbreviated_level = level_abbreviations.get(original_level, original_level[:4])
+        record.levelname = f"[{abbreviated_level}]"
+
+        # Add color to the bracketed level name
+        if original_level in self.COLORS:
+            record.levelname = (
+                f"{self.COLORS[original_level]}{record.levelname}{self.COLORS['RESET']}"
+            )
+
+        return super().format(record)
+
+
+def setup_logging(
+    log_level: str = "INFO",
+    log_dir: Optional[Path] = None,
+    enable_file_logging: bool = True,
+    enable_console_logging: bool = True,
+    log_format: Optional[str] = None,
+) -> Dict[str, Any]:
+    """
+    Setup logging configuration for Arbor.
+
+    Args:
+        log_level: Logging level (DEBUG, INFO, WARNING, ERROR, CRITICAL)
+        log_dir: Directory for log files
+        enable_file_logging: Whether to enable file logging
+        enable_console_logging: Whether to enable console logging
+        log_format: Custom log format string
+
+    Returns:
+        Dictionary with logging configuration details
+    """
+
+    # Default log format
+    if log_format is None:
+        log_format = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+
+    # Console format with colors and tight alignment
+    console_format = "%(name)s - %(levelname)s - %(message)s"
+
+    # Create formatters
+    file_formatter = logging.Formatter(log_format)
+    console_formatter = ColoredFormatter(console_format)
+
+    # Get root logger
+    root_logger = logging.getLogger()
+    root_logger.setLevel(getattr(logging, log_level.upper()))
+
+    # Clear existing handlers
+    root_logger.handlers.clear()
+
+    handlers = []
+
+    # Console handler
+    if enable_console_logging:
+        console_handler = logging.StreamHandler(sys.stdout)
+        console_handler.setLevel(getattr(logging, log_level.upper()))
+        console_handler.setFormatter(console_formatter)
+        root_logger.addHandler(console_handler)
+        handlers.append("console")
+
+    # File handlers
+    if enable_file_logging and log_dir:
+        log_dir = Path(log_dir)
+        log_dir.mkdir(parents=True, exist_ok=True)
+
+        # Main log file (all levels)
+        main_log_file = log_dir / "arbor.log"
+        file_handler = logging.FileHandler(main_log_file)
+        file_handler.setLevel(logging.DEBUG)  # Capture all levels in file
+        file_handler.setFormatter(file_formatter)
+        root_logger.addHandler(file_handler)
+        handlers.append("file")
+
+        # Error log file (errors and critical only)
+        error_log_file = log_dir / "arbor_error.log"
+        error_handler = logging.FileHandler(error_log_file)
+        error_handler.setLevel(logging.ERROR)
+        error_handler.setFormatter(file_formatter)
+        root_logger.addHandler(error_handler)
+        handlers.append("error_file")
+
+    # Configure specific loggers for different components
+    configure_component_loggers(log_level)
+
+    return {
+        "level": log_level,
+        "handlers": handlers,
+        "log_dir": str(log_dir) if log_dir else None,
+        "main_log_file": str(log_dir / "arbor.log") if log_dir else None,
+        "error_log_file": str(log_dir / "arbor_error.log") if log_dir else None,
+    }
+
+
+def configure_component_loggers(log_level: str):
+    """Configure loggers for specific Arbor components."""
+
+    # Arbor component loggers with shorter names (8 chars max for proper centering)
+    component_loggers = {
+        "arbor.server.services.inference_manager": "infer",
+        "arbor.server.services.grpo_manager": "grpo",
+        "arbor.server.services.file_manager": "files",
+        "arbor.server.services.health_manager": "health",
+        "arbor.server.services.job_manager": "jobs",
+        "arbor.server.services.training_manager": "train",
+        "arbor.server.services.comms": "comms",
+        "arbor.server.services.scripts.sft_training": "sft",
+        "arbor.server.services.scripts.grpo_training": "grpo",
+        "arbor.config": "config",
+        "arbor.cli": "cli",
+        "__main__": "main",
+    }
+
+    # Create shorter logger aliases
+    for full_name, short_name in component_loggers.items():
+        # Set up the short name logger to use the same handlers as root
+        short_logger = logging.getLogger(short_name)
+        short_logger.setLevel(getattr(logging, log_level.upper()))
+
+        # Also configure the full name for backward compatibility
+        full_logger = logging.getLogger(full_name)
+        full_logger.setLevel(getattr(logging, log_level.upper()))
+
+    # Third-party library loggers (usually more verbose)
+    third_party_loggers = {
+        "uvicorn": "WARNING",
+        "uvicorn.access": "WARNING",
+        "uvicorn.error": "WARNING",
+        "fastapi": "WARNING",
+        "httpx": "WARNING",
+        "urllib3": "WARNING",
+        "vllm": "INFO",
+        "torch": "WARNING",
+        "transformers": "WARNING",
+    }
+
+    for logger_name, level in third_party_loggers.items():
+        logger = logging.getLogger(logger_name)
+        logger.setLevel(getattr(logging, level))
+
+
+def get_logger(name: str) -> logging.Logger:
+    """
+    Get a logger with the specified name, automatically mapping long module names to short ones.
+
+    Args:
+        name: Logger name, typically __name__
+
+    Returns:
+        Configured logger instance with short name if available
+    """
+    # Mapping of full module names to short names (8 chars max for proper centering)
+    name_mappings = {
+        "arbor.server.services.inference_manager": "infer",
+        "arbor.server.services.grpo_manager": "grpo",
+        "arbor.server.services.file_manager": "files",
+        "arbor.server.services.health_manager": "health",
+        "arbor.server.services.job_manager": "jobs",
+        "arbor.server.services.training_manager": "train",
+        "arbor.server.services.comms.comms": "comms",
+        "arbor.server.services.scripts.sft_training": "sft",
+        "arbor.server.services.scripts.grpo_training": "grpo",
+        "arbor.config": "config",
+        "arbor.cli": "cli",
+        "__main__": "main",
+        # Add uvicorn loggers
+        "uvicorn": "api",
+        "uvicorn.access": "api",
+        "uvicorn.error": "api",
+    }
+
+    # Use short name if available, otherwise use the provided name
+    short_name = name_mappings.get(name, name)
+    return logging.getLogger(short_name)
+
+
+def log_system_info():
+    """Log system information at startup."""
+    logger = get_logger("arbor.startup")
+
+    logger.info("=" * 60)
+    logger.info("ARBOR SYSTEM STARTUP")
+    logger.info("=" * 60)
+
+    # This will be populated by the health manager
+    logger.info("System information logged via health manager")
+
+
+def log_configuration(config: Dict[str, Any]):
+    """Log configuration information."""
+    logger = get_logger("arbor.config")
+
+    logger.info("Logging configuration:")
+    for key, value in config.items():
+        logger.info(f"  {key}: {value}")
+
+
+# Context manager for temporary log level changes
+class LogLevel:
+    """Context manager to temporarily change log level."""
+
+    def __init__(self, logger_name: str, level: str):
+        self.logger = logging.getLogger(logger_name)
+        self.level = getattr(logging, level.upper())
+        self.original_level = None
+
+    def __enter__(self):
+        self.original_level = self.logger.level
+        self.logger.setLevel(self.level)
+        return self.logger
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.logger.setLevel(self.original_level)
+
+
+def apply_uvicorn_formatting():
+    """
+    Apply our custom formatting to uvicorn loggers after they're set up.
+    This runs after the FastAPI app is created to avoid interfering with startup.
+    """
+    console_formatter = ColoredFormatter("%(name)s - %(levelname)s - %(message)s")
+
+    # List of uvicorn loggers to modify
+    uvicorn_loggers = ["uvicorn", "uvicorn.error", "uvicorn.access"]
+
+    for logger_name in uvicorn_loggers:
+        logger = logging.getLogger(logger_name)
+
+        # Update formatters for existing handlers
+        for handler in logger.handlers:
+            if hasattr(handler, "setFormatter"):
+                handler.setFormatter(console_formatter)

{arbor_ai-0.2.1.dist-info → arbor_ai-0.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arbor-ai
-Version: 0.2.1
+Version: 0.2.2
 Summary: A framework for fine-tuning and managing language models
 Author-email: Noah Ziems <nziems2@nd.edu>
 Project-URL: Homepage, https://github.com/Ziems/arbor
@@ -14,7 +14,7 @@ Requires-Dist: uvicorn
 Requires-Dist: click
 Requires-Dist: python-multipart
 Requires-Dist: pydantic-settings
-Requires-Dist: vllm>=0.8.5.post1
+Requires-Dist: vllm==0.8.5.post1
 Requires-Dist: transformers
 Requires-Dist: trl>=0.17.0
 Requires-Dist: peft
@@ -43,7 +43,8 @@ Install Arbor via pip:
 pip install -U arbor-ai
 ```
 
-Optionally, you can also install:
+Optionally, you can also install flash attention to speed up inference. <br/>
+This can take 15+ minutes to install on some setups:
 ```bash
 pip install flash-attn --no-build-isolation
 ```
@@ -52,27 +53,16 @@ pip install flash-attn --no-build-isolation
 
 ## ⚡ Quick Start
 
-### 1️⃣ Make an `arbor.yaml` File
-
-This is all dependent on your setup. Here is an example of one:
-```yaml
-inference:
-  gpu_ids: '0'
-
-training:
-  gpu_ids: '1, 2'
-```
-Which will use the `GPU:0` for inference with `GPU:1` and `GPU:2` reserved for training. We generally recommend splitting the GPUs roughly evenly between inference and training.
-
-### 2️⃣ Start the Server
+### 1️⃣ Start the Server
 
 **CLI:**
 
 ```bash
-python -m arbor.cli serve --arbor-config arbor.yaml
+python -m arbor.cli serve
 ```
+On the first run you'll be asked which GPUs will be used for training and which for inference. For more that one GPU, separate the ids by comma: `1, 2`. Your config file will be saved in `~/.arbor/config.yaml` should you want to edit these configs in the future.
 
-### 3️⃣ Optimize a DSPy Program
+### 2️⃣ Optimize a DSPy Program
 
 Follow the DSPy tutorials here to see usage examples:
 [DSPy RL Optimization Examples](https://dspy.ai/tutorials/rl_papillon/)

arbor_ai-0.2.2.dist-info/RECORD

@@ -0,0 +1,51 @@
+arbor/__init__.py,sha256=DKqMupRY_1oOt7WfwU_6g4FVbTBJjU83p7yEGC-JAhs,458
+arbor/cli.py,sha256=YK-MIFxWb4ZfNb-UtK6t-EgTk1XQRq0dVOzdGf7_AY8,4867
+arbor/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arbor/client/api.py,sha256=86bgHuGM_AvI1Uhic_QaCnpF4VFqXie9ZzxmbTXUPpQ,19
+arbor/client/arbor_client.py,sha256=laGmeTMyGzwKOBGjpOWEZdDXTwsp_MUhp6umiLufncs,8320
+arbor/server/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+arbor/server/main.py,sha256=Ww4HfxltENIIe4TCq4x_tis69tJ7IhQ9o70BEfRT7MI,1177
+arbor/server/api/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+arbor/server/api/models/schemas.py,sha256=yQzZEdYpUDi1BOQi2Ws7pn09Bq5Y_V8ae-oOEKMeD4M,6296
+arbor/server/api/routes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arbor/server/api/routes/files.py,sha256=DQC_ogH5zlzhHZSAA4Cj5wzK07XBIBVs2Po91W9rcDY,1835
+arbor/server/api/routes/grpo.py,sha256=VHL68SO0lUp4zAZ1RfPCdgdRfO_ZRE7NA7LXvOZwo-c,1914
+arbor/server/api/routes/inference.py,sha256=Vyh6atkEOhz8YLWgJ34dXUzkdLwqOfZC9LIhez-IfEI,2282
+arbor/server/api/routes/jobs.py,sha256=BNdaSYUBJX6xSd6Pj6qx1DQJiZ5EKVxxbXDbEkfkCpw,3634
+arbor/server/core/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+arbor/server/core/config.py,sha256=8l1BVuV09MZd7tPh1iK-bW2scQaQd0rmgN1K8WjIZWc,11116
+arbor/server/core/config_manager.py,sha256=LTT8T6CQ063MiDM6udfc4IptHTLRW8OewyI1tpGcd_Q,3227
+arbor/server/core/logging.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arbor/server/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arbor/server/services/dependencies.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arbor/server/services/file_manager.py,sha256=kyRz5rJ2VHjdGWCqzqQ-HnNVaL42oLlJZQAuRwiDJLE,12199
+arbor/server/services/grpo_manager.py,sha256=IC3rY2cqC_qRo8hnBWju44CreY7soz_qyw2paURmCsc,20571
+arbor/server/services/health_manager.py,sha256=J5RkxHvXTzGITCaiQ9a_G_h8YYQLAEt69DdL7-Xj1VM,6475
+arbor/server/services/inference_manager.py,sha256=63GpZeHPIlj2QgNH4LXA4K2DSjofhWQRvd_bKlpaJyE,9925
+arbor/server/services/job_manager.py,sha256=wrtB0cO48DZh7D7hq6POt6xgxlauDKVjK8foNTVZVTA,2180
+arbor/server/services/training_manager.py,sha256=SM1TaKNF7Plm-nY6UQ2ytVSh4d5hgfVkLGx9vPXbx0E,21053
+arbor/server/services/comms/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arbor/server/services/comms/comms.py,sha256=zTV1Wo4FGnOcl5tIjBOfcuvDPMu-VTSKBlDBXV4h6LY,7833
+arbor/server/services/inference/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arbor/server/services/inference/vllm_client.py,sha256=EwVuv4bUSK2rOGEL6YdGSfaYtA4NsZfTwWw_KKfMAiQ,18367
+arbor/server/services/inference/vllm_serve.py,sha256=UZAGo7CyshR3-9fhXCTKhXeidqNqbY6LyU9DDNiX_Sw,109543
+arbor/server/services/scripts/dpo_training.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arbor/server/services/scripts/grpo_training.py,sha256=XJZ3PBJdXRP__XfiQBtqEqjRE4YKjjYrJ4ze5W9SRnY,22160
+arbor/server/services/scripts/mmgrpo_training.py,sha256=aDaTGkyfceA-jhAdzfpehAzuSSGIT8gdfD_5z9IoBEk,19373
+arbor/server/services/scripts/sft_training.py,sha256=BvCsOA2kg2SEsyd7P9mkerIICPORoASB-0r8EO3usrQ,3538
+arbor/server/services/scripts/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arbor/server/services/scripts/utils/arg_parser.py,sha256=ur_iyhc_Ie00tjq63vK4Sdeu2PGKwe6Dh6Iax2vw9jc,1022
+arbor/server/services/scripts/utils/callbacks.py,sha256=iPE_bjNHSQyBLO0v7UjxsuWDjDvLvhQAGerBFX8UbnA,1347
+arbor/server/services/scripts/utils/comms_monitors.py,sha256=RE9RxHR1yqXgc8VgRyhVwtwes7ofXeVgFcwIN5AVW_Y,7983
+arbor/server/services/scripts/utils/dataset.py,sha256=O-g-BUwLdVxUiQEsn2G4VobOYJjspA_iXc7MMNXZLQ0,6178
+arbor/server/services/scripts/utils/ingestion_monitor.py,sha256=r5t5vG_TO62PYiFQROfm1mzBESB10WlxFRgGN-IKtaI,1305
+arbor/server/services/scripts/utils/mock_server.py,sha256=NX38oBtxEVS9-kreayTCSzg5Hhuum-3vK0uNcOBkdtw,4237
+arbor/server/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arbor/server/utils/helpers.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arbor/server/utils/logging.py,sha256=UsG3cYgNk04pXadf5-NcYxvI8qYjB6o91N_P08FuoIg,10368
+arbor_ai-0.2.2.dist-info/licenses/LICENSE,sha256=5vFGrbOFeXXM83JV9o16w7ohH4WLeu3-57GocJSz8ow,1067
+arbor_ai-0.2.2.dist-info/METADATA,sha256=1Jw-RJostZFj8mR9mIz_p8LLLpyh7qKZSPmrRn0Otzw,2776
+arbor_ai-0.2.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+arbor_ai-0.2.2.dist-info/entry_points.txt,sha256=PGBX-MfNwfIl8UPFgsX3gjtXLqSogRhOktKMpZUysD0,40
+arbor_ai-0.2.2.dist-info/top_level.txt,sha256=jzWdp3BRYqvZDMFsPajrcftvvlluzVDErkD8IMRfhYs,6
+arbor_ai-0.2.2.dist-info/RECORD,,

arbor_ai-0.2.1.dist-info/RECORD

@@ -1,42 +0,0 @@
-arbor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arbor/cli.py,sha256=6S_nT93Zof6nB01n-xA7hSzssREzY13Oyh_jrElyTTY,3490
-arbor/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arbor/client/api.py,sha256=86bgHuGM_AvI1Uhic_QaCnpF4VFqXie9ZzxmbTXUPpQ,19
-arbor/server/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-arbor/server/main.py,sha256=tY4Vlaaj4oq1FTGYOkbFMGF0quLEeR-VBaKaXhQ5mEE,382
-arbor/server/api/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-arbor/server/api/models/schemas.py,sha256=394FHmIxAWVwED3z5tjnJCsyrgSWXg2SFWvMM1oKqOI,6177
-arbor/server/api/routes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arbor/server/api/routes/files.py,sha256=DQC_ogH5zlzhHZSAA4Cj5wzK07XBIBVs2Po91W9rcDY,1835
-arbor/server/api/routes/grpo.py,sha256=Yc4FxieuUbJ7Dbd-93uN4syQu9h2eQU4R9ZvnE_axRU,1982
-arbor/server/api/routes/inference.py,sha256=txLF4ANa0ZSaROrbvSaPZVFOSzn4so9e7mjNKnt2bcM,2182
-arbor/server/api/routes/jobs.py,sha256=BNdaSYUBJX6xSd6Pj6qx1DQJiZ5EKVxxbXDbEkfkCpw,3634
-arbor/server/core/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-arbor/server/core/config.py,sha256=Mx77S3ByIMvHmPDikQLcczhzA5so3Vrw_U4QefOiHOU,1257
-arbor/server/core/logging.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arbor/server/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arbor/server/services/dependencies.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arbor/server/services/file_manager.py,sha256=Z9z4A4EzvPauid_DBfpim401DDtuJy_TbX4twTWDJWI,12119
-arbor/server/services/grpo_manager.py,sha256=jY4kc7wlKKoi7RigjJiH1VaxX6qJCOxyEc0oYCkqPlQ,18549
-arbor/server/services/inference_manager.py,sha256=a1c5zYbjk6fPM3egX2McKv7ZWPN7c-QH_Qogu9iay90,9597
-arbor/server/services/job_manager.py,sha256=m_d4UPwN_82f7t7K443DaFpFoyv7JZSZKml8tawt1Bk,2186
-arbor/server/services/training_manager.py,sha256=oQdhpfxdgp_lCTb_lxhvjupdLrcg6HL3TEbct_q9F6I,21065
-arbor/server/services/comms/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arbor/server/services/comms/comms.py,sha256=3KN3mzwPvfW2_L5hq02JdAk6yOMyhY0_pBz-DDr5A3o,7694
-arbor/server/services/inference/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arbor/server/services/inference/vllm_client.py,sha256=06-VfdcwKqq8_ZRWaER3OnSVLtvL87bLdljSrkXfm-A,18269
-arbor/server/services/inference/vllm_serve.py,sha256=UZAGo7CyshR3-9fhXCTKhXeidqNqbY6LyU9DDNiX_Sw,109543
-arbor/server/services/scripts/dpo_training.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arbor/server/services/scripts/grpo_training.py,sha256=6kXzMwn3rZXHdEn0xe_Kd9d7tbdYb76zE0zbi02xCm4,31314
-arbor/server/services/scripts/sft_training.py,sha256=jgDMxZn9RFH9ys_7OF9Is8pQ9V97O2KzWg22Gveh3yE,3410
-arbor/server/services/scripts/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arbor/server/services/scripts/utils/arg_parser.py,sha256=ur_iyhc_Ie00tjq63vK4Sdeu2PGKwe6Dh6Iax2vw9jc,1022
-arbor/server/services/scripts/utils/dataset.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arbor/server/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arbor/server/utils/helpers.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arbor_ai-0.2.1.dist-info/licenses/LICENSE,sha256=5vFGrbOFeXXM83JV9o16w7ohH4WLeu3-57GocJSz8ow,1067
-arbor_ai-0.2.1.dist-info/METADATA,sha256=34XAZBm8OLlsSBicLmRn_hhbltn0pDNlAj5WOjn9LtE,2791
-arbor_ai-0.2.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-arbor_ai-0.2.1.dist-info/entry_points.txt,sha256=PGBX-MfNwfIl8UPFgsX3gjtXLqSogRhOktKMpZUysD0,40
-arbor_ai-0.2.1.dist-info/top_level.txt,sha256=jzWdp3BRYqvZDMFsPajrcftvvlluzVDErkD8IMRfhYs,6
-arbor_ai-0.2.1.dist-info/RECORD,,