github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-27 06:18:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

community: model config cassandradatabasetool (#26296)

Browse Source
This commit is contained in:
Erick Friis
2024-09-10 19:17:22 -07:00
committed by GitHub
parent a56812f970
commit 19bce9aba1
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
libs/community/langchain_community/tools/cassandra_database/tool.py
View File

@@ -7,7 +7,7 @@ from typing import TYPE_CHECKING, Any, Dict, Optional, Sequence, Type, Union
from langchain_core.callbacks import CallbackManagerForToolRun from langchain_core.callbacks import CallbackManagerForToolRun
from langchain_core.tools import BaseTool 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 from langchain_community.utilities.cassandra_database import CassandraDatabase
@@ -20,6 +20,10 @@ class BaseCassandraDatabaseTool(BaseModel):
db: CassandraDatabase = Field(exclude=True) db: CassandraDatabase = Field(exclude=True)
model_config = ConfigDict(
arbitrary_types_allowed=True,
)
class _QueryCassandraDatabaseToolInput(BaseModel): class _QueryCassandraDatabaseToolInput(BaseModel):
query: str = Field(..., description="A detailed and correct CQL query.") query: str = Field(..., description="A detailed and correct CQL query.")