mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-28 08:11:59 +00:00
2396b1e13a
* build_container.sh - append `python3.11‑devel gcc make` to UBI 9 dnf line so C‑ext wheels build. * tests/integration - + test_ubi9_toolchain.py (fast; asserts pkgs exist in Containerfile) - + test_ubi9_compiles.py (slow; @integration, compiles hello‑world in built image) - – test_container_build.py (removed heavyweight test) * pytest.ini - register `integration` marker. * CI - tests.yml ⇒ `pytest -m "not integration"` - providers_build.yml ⇒ run integration compiler smoke for container jobs
63 lines
1.7 KiB
Python
63 lines
1.7 KiB
Python
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
||
# All rights reserved.
|
||
#
|
||
# This source code is licensed under the terms described in the LICENSE file in
|
||
# the root directory of this source tree.
|
||
|
||
import os
|
||
import subprocess
|
||
import tempfile
|
||
import textwrap
|
||
from pathlib import Path
|
||
|
||
import pytest
|
||
import yaml
|
||
|
||
pytestmark = pytest.mark.integration # filtered out of the fast suite
|
||
|
||
|
||
def _tmp_yaml() -> Path:
|
||
f = tempfile.NamedTemporaryFile(delete=False, suffix=".yaml")
|
||
yaml.safe_dump(
|
||
{
|
||
"version": "2",
|
||
"distribution_spec": {
|
||
"description": "UBI9 compile smoke",
|
||
"providers": {
|
||
"inference": ["remote::ollama"],
|
||
"vector_io": ["inline::faiss"],
|
||
},
|
||
"container_image": "registry.access.redhat.com/ubi9",
|
||
},
|
||
"image_type": "container",
|
||
"image_name": "ci-test",
|
||
},
|
||
f,
|
||
)
|
||
f.close()
|
||
return Path(f.name)
|
||
|
||
|
||
def test_image_compiles_c():
|
||
cfg = _tmp_yaml()
|
||
tag = "ci-test:dev"
|
||
|
||
# Build image (providers‑build.yml already did `uv venv` etc.)
|
||
subprocess.run(
|
||
["llama", "stack", "build", "--config", cfg, "--image-name", "ci-test"],
|
||
check=True,
|
||
env={**os.environ, "USE_COPY_NOT_MOUNT": "true"},
|
||
)
|
||
|
||
# compile a hello‑world c program inside the container
|
||
hello_c = textwrap.dedent("""
|
||
#include <stdio.h>
|
||
int main(){puts("ok");return 0;}
|
||
""")
|
||
with tempfile.NamedTemporaryFile(delete=False, suffix=".c") as f:
|
||
f.write(hello_c.encode())
|
||
|
||
subprocess.run(
|
||
["docker", "run", "--rm", "-v", f.name + ":/tmp/a.c", tag, "bash", "-c", "gcc /tmp/a.c -o /tmp/a && /tmp/a"],
|
||
check=True,
|
||
)
|