mirror of
https://github.com/hwchase17/langchain.git
synced 2026-06-09 18:50:33 +00:00
infra: add xai to scheduled testing (#29852)
This commit is contained in:
@@ -5,7 +5,6 @@ from typing import Optional, Type
|
||||
import pytest # type: ignore[import-not-found]
|
||||
from langchain_core.language_models import BaseChatModel
|
||||
from langchain_core.rate_limiters import InMemoryRateLimiter
|
||||
from langchain_core.tools import BaseTool
|
||||
from langchain_tests.integration_tests import ( # type: ignore[import-not-found]
|
||||
ChatModelIntegrationTests, # type: ignore[import-not-found]
|
||||
)
|
||||
@@ -27,7 +26,7 @@ class TestXAIStandard(ChatModelIntegrationTests):
|
||||
@property
|
||||
def chat_model_params(self) -> dict:
|
||||
return {
|
||||
"model": "grok-beta",
|
||||
"model": "grok-2",
|
||||
"rate_limiter": rate_limiter,
|
||||
}
|
||||
|
||||
@@ -39,21 +38,3 @@ class TestXAIStandard(ChatModelIntegrationTests):
|
||||
@pytest.mark.xfail(reason="Not yet supported.")
|
||||
def test_usage_metadata_streaming(self, model: BaseChatModel) -> None:
|
||||
super().test_usage_metadata_streaming(model)
|
||||
|
||||
@pytest.mark.xfail(reason="Can't handle AIMessage with empty content.")
|
||||
def test_tool_message_error_status(
|
||||
self, model: BaseChatModel, my_adder_tool: BaseTool
|
||||
) -> None:
|
||||
super().test_tool_message_error_status(model, my_adder_tool)
|
||||
|
||||
@pytest.mark.xfail(reason="Can't handle AIMessage with empty content.")
|
||||
def test_structured_few_shot_examples(
|
||||
self, model: BaseChatModel, my_adder_tool: BaseTool
|
||||
) -> None:
|
||||
super().test_structured_few_shot_examples(model, my_adder_tool)
|
||||
|
||||
@pytest.mark.xfail(reason="Can't handle AIMessage with empty content.")
|
||||
def test_tool_message_histories_string_content(
|
||||
self, model: BaseChatModel, my_adder_tool: BaseTool
|
||||
) -> None:
|
||||
super().test_tool_message_histories_string_content(model, my_adder_tool)
|
||||
|
||||
Reference in New Issue
Block a user