Remove xfail on some more tests

libs/cerebras/langchain_cerebras/chat_models.py

@@ -3,13 +3,7 @@
 from typing import Any, AsyncIterator, Dict, Iterator, List, Optional, cast
 
 import openai
-from langchain_core.callbacks import (
-    AsyncCallbackManagerForLLMRun,
-    CallbackManagerForLLMRun,
-)
 from langchain_core.language_models.chat_models import LangSmithParams
-from langchain_core.messages import BaseMessage
-from langchain_core.outputs import ChatGenerationChunk
 from langchain_core.utils import (
     from_env,
     secret_from_env,
@@ -379,38 +373,3 @@ def validate_environment(self) -> Self:
             )
             self.async_client = self.root_async_client.chat.completions
         return self
-
-    # Patch tool calling w/ streaming.
-    def _stream(
-        self,
-        messages: List[BaseMessage],
-        stop: Optional[List[str]] = None,
-        run_manager: Optional[CallbackManagerForLLMRun] = None,
-        **kwargs: Any,
-    ) -> Iterator[ChatGenerationChunk]:
-        if kwargs.get("tools"):
-            yield cast(
-                ChatGenerationChunk,
-                super()._generate(messages, stop, run_manager, **kwargs).generations[0],
-            )
-        else:
-            yield from super()._stream(messages, stop, run_manager, **kwargs)
-
-    async def _astream(
-        self,
-        messages: List[BaseMessage],
-        stop: Optional[List[str]] = None,
-        run_manager: Optional[AsyncCallbackManagerForLLMRun] = None,
-        **kwargs: Any,
-    ) -> AsyncIterator[ChatGenerationChunk]:
-        if kwargs.get("tools"):
-            generation = await super()._agenerate(messages, stop, run_manager, **kwargs)
-            yield (
-                cast(
-                    ChatGenerationChunk,
-                    generation.generations[0],
-                )
-            )
-        else:
-            async for msg in super()._astream(messages, stop, run_manager, **kwargs):
-                yield msg

libs/cerebras/tests/integration_tests/test_base_standard.py

@@ -16,20 +16,8 @@ def chat_model_class(self) -> Type[BaseChatModel]:
 
     @property
     def chat_model_params(self) -> dict:
-        return {"model": "llama3.1-8b"}
+        return {"model": "llama3.1-70b"}
 
-    @pytest.mark.xfail(reason=("Not supported"))
-    def test_structured_output(self, model: BaseChatModel) -> None:
-        super().test_structured_output(model)
-
-    @pytest.mark.xfail(reason=("Not supported"))
-    def test_structured_output_pydantic_2_v1(self, model: BaseChatModel) -> None:
-        super().test_structured_output_pydantic_2_v1(model)
-
-    @pytest.mark.xfail(reason=("Not supported"))
-    def test_tool_calling_with_no_arguments(self, model: BaseChatModel) -> None:
-        super().test_tool_calling_with_no_arguments(model)
-
-    @pytest.mark.xfail(reason=("Not supported"))
+    @pytest.mark.xfail(reason=("Array input not supported"))
     def test_tool_message_histories_list_content(self, model: BaseChatModel) -> None:
         super().test_tool_message_histories_list_content(model)