Llm_builders

This module contains functionality related to the the llm_builders module for common.builders.

Llm_builders

OpenAIBuilder

Builder for creating OpenAI language model instances.

Provides factory method to create configured OpenAI objects.

Source code in src/common/builders/llm_builders.py 
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
class OpenAIBuilder:
    """Builder for creating OpenAI language model instances.

    Provides factory method to create configured OpenAI objects.
    """

    @staticmethod
    @inject
    def build(configuration: "OpenAILLMConfiguration") -> OpenAI:
        """Creates a configured OpenAI language model.

        Args:
            configuration: Model settings including API key and parameters.

        Returns:
            OpenAI: Configured language model instance.
        """
        return OpenAI(
            api_key=configuration.secrets.api_key.get_secret_value(),
            model=configuration.name,
            max_tokens=configuration.max_tokens,
            max_retries=configuration.max_retries,
        )

build(configuration) staticmethod

Creates a configured OpenAI language model.

Parameters:
  • configuration (OpenAILLMConfiguration) –

    Model settings including API key and parameters.
Returns:
  • OpenAI( OpenAI ) –

    Configured language model instance.
Source code in src/common/builders/llm_builders.py 
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
@staticmethod
@inject
def build(configuration: "OpenAILLMConfiguration") -> OpenAI:
    """Creates a configured OpenAI language model.

    Args:
        configuration: Model settings including API key and parameters.

    Returns:
        OpenAI: Configured language model instance.
    """
    return OpenAI(
        api_key=configuration.secrets.api_key.get_secret_value(),
        model=configuration.name,
        max_tokens=configuration.max_tokens,
        max_retries=configuration.max_retries,
    )

OpenAILikeBuilder

Builder for creating OpenAI-compatible language model instances.

Provides factory method to create configured OpenAILike objects.

Source code in src/common/builders/llm_builders.py 
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
class OpenAILikeBuilder:
    """Builder for creating OpenAI-compatible language model instances.

    Provides factory method to create configured OpenAILike objects.
    """

    @staticmethod
    @inject
    def build(configuration: "OpenAILikeLLMConfiguration") -> OpenAILike:
        """Creates a configured OpenAI-compatible language model.

        Args:
            configuration: Model settings including API endpoints and parameters.

        Returns:
            OpenAILike: Configured language model instance.
        """
        return OpenAILike(
            api_base=configuration.secrets.api_base.get_secret_value(),
            api_key=configuration.secrets.api_key.get_secret_value(),
            model=configuration.name.value,
            max_tokens=configuration.max_tokens,
            max_retries=configuration.max_retries,
            context_window=configuration.context_window,
            logprobs=None,
            api_version="",
        )

build(configuration) staticmethod

Creates a configured OpenAI-compatible language model.

Parameters:
  • configuration (OpenAILikeLLMConfiguration) –

    Model settings including API endpoints and parameters.
Returns:
  • OpenAILike( OpenAILike ) –

    Configured language model instance.
Source code in src/common/builders/llm_builders.py 
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
@staticmethod
@inject
def build(configuration: "OpenAILikeLLMConfiguration") -> OpenAILike:
    """Creates a configured OpenAI-compatible language model.

    Args:
        configuration: Model settings including API endpoints and parameters.

    Returns:
        OpenAILike: Configured language model instance.
    """
    return OpenAILike(
        api_base=configuration.secrets.api_base.get_secret_value(),
        api_key=configuration.secrets.api_key.get_secret_value(),
        model=configuration.name.value,
        max_tokens=configuration.max_tokens,
        max_retries=configuration.max_retries,
        context_window=configuration.context_window,
        logprobs=None,
        api_version="",
    )