groq[patch]: add usage_metadata to (a)invoke and (a)stream (#22834)

This commit is contained in:
ccurme
2024-06-13 10:26:27 -04:00
committed by GitHub
parent e01e5d5a91
commit b626c3ca23
5 changed files with 81 additions and 48 deletions

View File

@@ -9,22 +9,11 @@ from langchain_standard_tests.integration_tests import ChatModelIntegrationTests
from langchain_groq import ChatGroq
class TestMistralStandard(ChatModelIntegrationTests):
class TestGroqStandard(ChatModelIntegrationTests):
@pytest.fixture
def chat_model_class(self) -> Type[BaseChatModel]:
return ChatGroq
@pytest.mark.xfail(reason="Not implemented.")
def test_usage_metadata(
self,
chat_model_class: Type[BaseChatModel],
chat_model_params: dict,
) -> None:
super().test_usage_metadata(
chat_model_class,
chat_model_params,
)
@pytest.mark.xfail(reason="Not yet implemented.")
def test_tool_message_histories_list_content(
self,