diff --git a/libs/community/langchain_community/tools/cassandra_database/tool.py b/libs/community/langchain_community/tools/cassandra_database/tool.py index b96da9973a4..ab6e502fb08 100644 --- a/libs/community/langchain_community/tools/cassandra_database/tool.py +++ b/libs/community/langchain_community/tools/cassandra_database/tool.py @@ -7,7 +7,7 @@ from typing import TYPE_CHECKING, Any, Dict, Optional, Sequence, Type, Union from langchain_core.callbacks import CallbackManagerForToolRun from langchain_core.tools import BaseTool -from pydantic import BaseModel, Field +from pydantic import BaseModel, ConfigDict, Field from langchain_community.utilities.cassandra_database import CassandraDatabase @@ -20,6 +20,10 @@ class BaseCassandraDatabaseTool(BaseModel): db: CassandraDatabase = Field(exclude=True) + model_config = ConfigDict( + arbitrary_types_allowed=True, + ) + class _QueryCassandraDatabaseToolInput(BaseModel): query: str = Field(..., description="A detailed and correct CQL query.")