chore: drop unrelated ci and aws test noise

Author: fede-kamel

.github/workflows/ci.yml

@@ -77,11 +77,7 @@ jobs:
 
           echo "Current cohere installation: $(poetry show cohere)"
           make test
-
-          # Skip test_langchain_tool_calling_agent: uses deprecated models and hits
-          # the live API without temperature=0, causing non-deterministic failures.
-          # See https://github.com/cohere-ai/cohere-python/pull/738
-          poetry run pytest tests/integration_tests/ --deselect tests/integration_tests/test_langgraph_agents.py::test_langchain_tool_calling_agent
+          make integration_test
           echo "tests passed"
 
           # reset poetry changes

tests/test_aws_client_unit.py

@@ -10,13 +10,27 @@
 import inspect
 import json
 import os
+import sys
+import types
 import unittest
 from unittest.mock import MagicMock, patch
 
 import httpx
 
 from cohere.manually_maintained.cohere_aws.mode import Mode
 
+if "tokenizers" not in sys.modules:
+    tokenizers_stub = types.ModuleType("tokenizers")
+    tokenizers_stub.Tokenizer = object
+    sys.modules["tokenizers"] = tokenizers_stub
+
+if "fastavro" not in sys.modules:
+    fastavro_stub = types.ModuleType("fastavro")
+    fastavro_stub.parse_schema = lambda schema: schema
+    fastavro_stub.reader = lambda *args, **kwargs: iter(())
+    fastavro_stub.writer = lambda *args, **kwargs: None
+    sys.modules["fastavro"] = fastavro_stub
+
 
 class TestSigV4HostHeader(unittest.TestCase):
     """Fix 1: The headers dict passed to AWSRequest for SigV4 signing must
@@ -47,8 +61,11 @@ def capture_aws_request(**kwargs):  # type: ignore
         mock_session.get_credentials.return_value = MagicMock()
         mock_boto3.Session.return_value = mock_session
 
-        with patch("cohere.aws_client.lazy_botocore", return_value=mock_botocore), \
-             patch("cohere.aws_client.lazy_boto3", return_value=mock_boto3):
+        import cohere.aws_client as aws_client_module
+
+        with patch.object(aws_client_module, "lazy_botocore", return_value=mock_botocore), patch.object(
+            aws_client_module, "lazy_boto3", return_value=mock_boto3
+        ):
 
             from cohere.aws_client import map_request_to_bedrock
 
@@ -58,7 +75,7 @@ def capture_aws_request(**kwargs):  # type: ignore
                 method="POST",
                 url="https://api.cohere.com/v1/chat",
                 headers={"connection": "keep-alive"},
-                json={"model": "cohere.I gues-v1:0", "message": "hello"},
+                json={"model": "cohere.command-r-plus-v1:0", "message": "hello"},
             )
 
             self.assertEqual(request.url.host, "api.cohere.com")