java-llama.cpp/src/test/java/net/ladenthin/llama/TestConstants.java at 55a6fa03b1da53ea1bc922ad9422913d240bacce · vaiju1981/java-llama.cpp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
// SPDX-FileCopyrightText: 2026 Bernard Ladenthin <bernard.ladenthin@gmail.com>
// SPDX-FileCopyrightText: 2023-2025 Konstantin Herud
//
// SPDX-License-Identifier: MIT

package net.ladenthin.llama;

import net.ladenthin.llama.loader.LlamaSystemProperties;

public class TestConstants {

    /** System property to override GPU layers used in tests. */
    public static final String PROP_TEST_NGL = LlamaSystemProperties.PREFIX + ".test.ngl";

    public static final int DEFAULT_TEST_NGL = 43;

    /** Path to the main text generation model used in tests. */
    public static final String MODEL_PATH = "models/codellama-7b.Q2_K.gguf";

    /** Path to the draft model used for speculative decoding tests. */
    public static final String DRAFT_MODEL_PATH = "models/AMD-Llama-135m-code.Q2_K.gguf";

    /** Path to the Qwen3 thinking model used for reasoning budget tests. */
    public static final String REASONING_MODEL_PATH = "models/Qwen3-0.6B-Q4_K_M.gguf";

    /** Path to the reranking model used in tests (loaded with {@code enableReranking()}). */
    public static final String RERANKING_MODEL_PATH = "models/jina-reranker-v1-tiny-en-Q4_0.gguf";

    /** System property overriding the GGUF used by the real tool-calling integration tests. */
    public static final String PROP_TOOL_MODEL_PATH = LlamaSystemProperties.PREFIX + ".tool.model";

    /** Qwen2.5 tool-capable model used by upstream llama.cpp's blocking and streaming tests. */
    public static final String DEFAULT_TOOL_MODEL_PATH = "models/Qwen2.5-1.5B-Instruct-Q4_K_M.gguf";

    /**
     * System property holding a path to a Nomic embedding model
     * ({@code nomic-embed-text-v1.5.f16.gguf} or a compatible BERT-family encoder).
     * Used by {@link LlamaEmbeddingsTest#testNomicEmbedLoads} to confirm upstream
     * issue #98 (BERT-encoder result_output assertion) stays resolved.
     * When the property is unset the test self-skips.
     */
    public static final String PROP_NOMIC_MODEL_PATH = LlamaSystemProperties.PREFIX + ".nomic.path";

    /** Expected embedding dimension of nomic-embed-text-v1.5 (hidden size = 768). */
    public static final int NOMIC_EMBED_DIM = 768;

    /**
     * System property holding a path to a vision-capable model GGUF. Consumed by
     * {@code MultimodalIntegrationTest} (upstream kherud/java-llama.cpp#103 / #34). The CI default is the
     * SmolVLM-500M Q8_0 GGUF; the test self-skips when the property is unset or
     * the file is missing.
     */
    public static final String PROP_VISION_MODEL_PATH = LlamaSystemProperties.PREFIX + ".vision.model";

    /** System property holding a path to the matching mmproj GGUF for the vision model. */
    public static final String PROP_VISION_MMPROJ_PATH = LlamaSystemProperties.PREFIX + ".vision.mmproj";

    /**
     * System property holding a path to an image used as the visual prompt in
     * {@code MultimodalIntegrationTest}. When unset the test falls back to
     * {@link #DEFAULT_VISION_IMAGE_PATH}, which points at a small image
     * committed under {@code src/test/resources/images/}. Any png/jpeg/webp/gif
     * works; the matching extension drives MIME detection in
     * {@code ContentPart.imageFile(Path)}.
     */
    public static final String PROP_VISION_IMAGE_PATH = LlamaSystemProperties.PREFIX + ".vision.image";

    /**
     * Path used by {@code MultimodalIntegrationTest} when
     * {@link #PROP_VISION_IMAGE_PATH} is unset. Points at the committed test
     * resource so the test needs no network access for the visual prompt.
     */
    public static final String DEFAULT_VISION_IMAGE_PATH = "src/test/resources/images/test-image.jpg";

    /**
     * System property holding a path to an audio-input model GGUF (e.g. Ultravox / Qwen2.5-Omni).
     * Consumed by {@code AudioInputIntegrationTest} (llama.cpp discussion #13759). The test self-skips
     * when this, the mmproj, or the audio clip is unset/missing.
     */
    public static final String PROP_AUDIO_MODEL_PATH = LlamaSystemProperties.PREFIX + ".audio.model";

    /** System property holding a path to the matching audio mmproj (encoder) GGUF. */
    public static final String PROP_AUDIO_MMPROJ_PATH = LlamaSystemProperties.PREFIX + ".audio.mmproj";

    /**
     * System property holding a path to a {@code .wav} or {@code .mp3} clip used as the audio prompt in
     * {@code AudioInputIntegrationTest}. The matching extension drives format detection in
     * {@code ContentPart.audioFile(Path)}.
     */
    public static final String PROP_AUDIO_PATH = LlamaSystemProperties.PREFIX + ".audio.input";
}