test_groq.py — langchain Source File

Source Code

"""Test groq block translator."""

from typing import cast

import pytest

from langchain_core.messages import AIMessage
from langchain_core.messages import content as types
from langchain_core.messages.base import _extract_reasoning_from_additional_kwargs
from langchain_core.messages.block_translators import PROVIDER_TRANSLATORS
from langchain_core.messages.block_translators.groq import (
    _parse_code_json,
    translate_content,
)


def test_groq_translator_registered() -> None:
    """Test that groq translator is properly registered."""
    assert "groq" in PROVIDER_TRANSLATORS
    assert "translate_content" in PROVIDER_TRANSLATORS["groq"]
    assert "translate_content_chunk" in PROVIDER_TRANSLATORS["groq"]


def test_extract_reasoning_from_additional_kwargs_exists() -> None:
    """Test that _extract_reasoning_from_additional_kwargs can be imported."""
    # Verify it's callable
    assert callable(_extract_reasoning_from_additional_kwargs)


def test_groq_translate_content_basic() -> None:
    """Test basic groq content translation."""
    # Test with simple text message
    message = AIMessage(content="Hello world")
    blocks = translate_content(message)

    assert isinstance(blocks, list)
    assert len(blocks) == 1
    assert blocks[0]["type"] == "text"
    assert blocks[0]["text"] == "Hello world"


def test_groq_translate_content_with_reasoning() -> None:
    """Test groq content translation with reasoning content."""
    # Test with reasoning content in additional_kwargs
    message = AIMessage(
        content="Final answer",
        additional_kwargs={"reasoning_content": "Let me think about this..."},
    )
    blocks = translate_content(message)

    assert isinstance(blocks, list)
    assert len(blocks) == 2

    # First block should be reasoning
    assert blocks[0]["type"] == "reasoning"
    assert blocks[0]["reasoning"] == "Let me think about this..."

    # Second block should be text
    assert blocks[1]["type"] == "text"
    assert blocks[1]["text"] == "Final answer"
// ... (81 more lines)