Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions libs/aws/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,7 @@ test = [
"syrupy>=4.9.1",
"pytest-asyncio>=0.20,<1",
"pytest-watcher>=0.4.3",
"langchain-classic>=1.0.0",
"langchain-tests>=1.0.0",
"langchain>=1.0.0rc2",
]
Expand Down
42 changes: 17 additions & 25 deletions libs/aws/tests/integration_tests/agents/test_bedrock_agents.py
Original file line number Diff line number Diff line change
@@ -1,32 +1,24 @@
# type: ignore

import pytest
import json
import operator
import time
import uuid
from typing import Any, Tuple, Union

pytest.skip(
"Temporarily skipping Bedrock agent tests due to AgentExecutor usage.",
allow_module_level=True,
import boto3
import pytest
from langchain_classic.agents import AgentExecutor
from langchain_core.tools import tool
from typing_extensions import Annotated, TypedDict

import langchain_aws.agents.base
from langchain_aws.agents import (
BedrockAgentAction,
BedrockAgentFinish,
BedrockAgentsRunnable,
BedrockInlineAgentsRunnable,
)
try:
import json
import operator
import time
import uuid
from typing import Any, Tuple, Union

import boto3
from langchain.agents import AgentExecutor
from langchain_core.tools import tool
from typing_extensions import Annotated, TypedDict

import langchain_aws.agents.base
from langchain_aws.agents import (
BedrockAgentAction,
BedrockAgentFinish,
BedrockAgentsRunnable,
BedrockInlineAgentsRunnable,
)
except ImportError:
pass


def _create_iam_client() -> Any:
Expand Down
Loading