gemini_api.py

Define Gemini adapter class.

GeminiAdapter

Bases: LlmAdapter

Define Gemini adapter class.

Source code in taglyatelle/llm_providers/gemini_api.py

class GeminiAdapter(LlmAdapter):
    """Define Gemini adapter class."""

    def __init__(self, model: str, temperature: float | int = 0):
        """
        Initialize Google's llms.

        Parameters
        ----------
        model
            LLM model name

        temperature
            LLM temperature
        """
        self.client = genai.Client()
        self.model = model
        self.temperature = temperature

    def invoke_llm(self, prompt: str) -> str | None:
        """
        Send a request to a LLM.

        Parameters
        ----------
        prompt
            The prompt to send to the LLM

        Returns
        -------
        Answer of the LLM or None
        """
        response = self.client.models.generate_content(
            model=self.model,
            contents=prompt,
            config=types.GenerateContentConfig(
                temperature=self.temperature,
            ),
        )
        return response.text

__init__(model, temperature=0)

Initialize Google's llms.

Parameters

model LLM model name

temperature LLM temperature

Source code in taglyatelle/llm_providers/gemini_api.py

def __init__(self, model: str, temperature: float | int = 0):
    """
    Initialize Google's llms.

    Parameters
    ----------
    model
        LLM model name

    temperature
        LLM temperature
    """
    self.client = genai.Client()
    self.model = model
    self.temperature = temperature

invoke_llm(prompt)

Send a request to a LLM.

Parameters

prompt The prompt to send to the LLM

Returns

Answer of the LLM or None

Source code in taglyatelle/llm_providers/gemini_api.py

def invoke_llm(self, prompt: str) -> str | None:
    """
    Send a request to a LLM.

    Parameters
    ----------
    prompt
        The prompt to send to the LLM

    Returns
    -------
    Answer of the LLM or None
    """
    response = self.client.models.generate_content(
        model=self.model,
        contents=prompt,
        config=types.GenerateContentConfig(
            temperature=self.temperature,
        ),
    )
    return response.text