From 19bce9aba1b5aedb71f0f829383b470c983f9d0e Mon Sep 17 00:00:00 2001 From: Erick Friis Date: Tue, 10 Sep 2024 19:17:22 -0700 Subject: [PATCH] community: model config cassandradatabasetool (#26296) --- .../langchain_community/tools/cassandra_database/tool.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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.")