How to use create_sql_agent function from langchain with aws lambda function just to invoke bedrock model.

[yadav-shivani]

Checked other resources

• I added a very descriptive title to this question.
• I searched the LangChain documentation with the integrated search.
• I used the GitHub search to find a similar question and didn't find it.

Commit to Help

• I commit to help with one of those options 👆

Example Code

===

Description

How to use 'create_sql_agent' function from LangChain with AWS lambda function just to invoke bedrock model. I cannot fetch database to lambda. I just want lambda function to invoke bedrock model. I already tried CustomLLM class from LangChain to call bedrock function and pass that to create_sql_agent function llm arguement but it does not work. I already 'create_sql_agent' function from langchain function which seems to be working well for my use case but having lambda constraint making it challenging for me to deploy.
Any idea how to do this will be very helpful.

System Info

System Information

OS: Linux
OS Version: #1 SMP Wed Aug 7 16:53:32 UTC 2024
Python Version: 3.10.16 (main, Jan 14 2025, 02:25:17) [GCC 12.2.0]

Package Information

langchain_core: 0.3.29
langchain: 0.3.14
langchain_community: 0.3.7
langsmith: 0.1.147
langchain_aws: 0.2.7
langchain_test: Installed. No version info available.
langchain_text_splitters: 0.3.5

Optional packages not installed

langserve

Other Dependencies

aiohttp: 3.11.11
async-timeout: 4.0.3
boto3: 1.34.131
dataclasses-json: 0.6.7
httpx: 0.28.1
httpx-sse: 0.4.0
jsonpatch: 1.33
langsmith-pyo3: Installed. No version info available.
numpy: 1.26.4
orjson: 3.10.14
packaging: 24.2
pydantic: 2.9.2
pydantic-settings: 2.7.1
PyYAML: 6.0.2
requests: 2.32.3
requests-toolbelt: 1.0.0
SQLAlchemy: 2.0.35
tenacity: 9.0.0
typing-extensions: 4.12.2

[yadav-shivani]

I was able to solve the error by prompting.