Add System Information Retrieval

[CuriousDolphin]

TITLE

Add System Information Retrieval

CONTEXT

This PR was created to provide users with detailed system diagnostics and robust HTTP request capabilities, improving resource management and integration.

KEY CHANGES

• Introduced SystemInfo and GPUInfo classes for comprehensive system diagnostics.
• Enhanced HttpClient with detailed docstrings and additional HTTP methods.
• Updated Makefile to include a test target.
• Added new dependencies: gputil, psutil, and setuptools.
• Implemented tests for system information retrieval and HTTP client functionality.
• Updated the notebook to include system information retrieval.

IMPACT

These changes provide users with enhanced capabilities to gather system information and perform HTTP requests, facilitating better integration and resource management.

TECHNICAL DETAILS

System Information

Utilizes platform, psutil, and GPUtil to gather system metrics such as CPU, memory, disk, and GPU details. The get_system_info function returns a SystemInfo object with this data.

from focoos.utils.system import get_system_info

system_info = get_system_info()
system_info.pretty_print()

HTTP Client

The HttpClient class now includes detailed docstrings for its methods, providing clarity on usage. It supports GET, POST, and DELETE requests with customizable headers and parameters.

client = HttpClient(api_key="test_key", host_url="http://example.com")
response = client.get("test/path")

Available infos

class SystemInfo(FocoosBaseModel):
    focoos_host: Optional[str] = None
    system: Optional[str] = None
    system_name: Optional[str] = None
    cpu_type: Optional[str] = None
    cpu_cores: Optional[int] = None
    memory_gb: Optional[float] = None
    memory_used_percentage: Optional[float] = None
    available_providers: Optional[list[str]] = None
    disk_space_total_gb: Optional[float] = None
    disk_space_used_percentage: Optional[float] = None
    gpu_count: Optional[int] = None
    gpu_driver: Optional[str] = None
    gpu_cuda_version: Optional[str] = None
    gpus_info: Optional[list[GPUInfo]] = None
    packages_versions: Optional[dict[str, str]] = None

Testing

Added unit tests for the new system information functions and the enhanced HTTP client to ensure reliability and correctness.

[github-actions]

Coverage Report •

File	Stmts	Miss	Cover	Missing
__init__.py	5	0	100%
local_model.py	63	44	30%	69–78, 95–96, 109–111, 116, 120–121, 125, 128, 132, 135, 157–169, 172–175, 177–179, 184, 197
ports.py	201	28	86%	23–24, 102, 104–105, 108, 160–162, 270–280, 282–284, 287–289, 291–292
runtime.py	166	144	13%	56–57, 60, 76–77, 90–93, 111–114, 116, 138–146, 183–185, 188–191, 196–199, 201–209, 211–215, 226–228, 239, 242–247, 259, 261, 263, 265–271, 274–275, 277–287, 295–300, 302, 304, 317–322, 324, 333–336, 338, 340, 343, 356–357, 359–364, 366, 375–376, 378, 380–383, 385, 388–390, 392–394, 396–398, 400–401, 411–412, 434–435, 438–439, 442–443, 446–449, 452
utils
system.py	80	10	87%	88, 113, 130, 158, 160, 222–226
TOTAL	886	468	47%

Tests	Skipped	Failures	Errors	Time
11	0 💤	0 ❌	0 🔥	6.744s ⏱️