langchain-timbr 2.1.4__tar.gz → 2.1.6__tar.gz

{langchain_timbr-2.1.4 → langchain_timbr-2.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-timbr
-Version: 2.1.4
+Version: 2.1.6
 Summary: LangChain & LangGraph extensions that parse LLM prompts into Timbr semantic SQL and execute them.
 Project-URL: Homepage, https://github.com/WPSemantix/langchain-timbr
 Project-URL: Documentation, https://docs.timbr.ai/doc/docs/integration/langchain-sdk/
@@ -19,44 +19,17 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Python: <3.13,>=3.10
-Requires-Dist: anthropic==0.42.0
-Requires-Dist: azure-identity==1.25.0; python_version >= '3.11'
-Requires-Dist: azure-identity>=1.16.1; python_version == '3.10'
 Requires-Dist: cryptography==45.0.7; python_version >= '3.11'
 Requires-Dist: cryptography>=44.0.3; python_version == '3.10'
-Requires-Dist: databricks-langchain==0.7.1
-Requires-Dist: databricks-sdk==0.64.0
-Requires-Dist: google-generativeai==0.8.4
-Requires-Dist: langchain-anthropic==0.3.5; python_version >= '3.11'
-Requires-Dist: langchain-anthropic>=0.3.1; python_version == '3.10'
 Requires-Dist: langchain-community==0.3.30; python_version >= '3.11'
-Requires-Dist: langchain-community>=0.3.20; python_version == '3.10'
-Requires-Dist: langchain-core==0.3.78; python_version >= '3.11'
-Requires-Dist: langchain-core>=0.3.58; python_version == '3.10'
-Requires-Dist: langchain-google-genai==2.0.10; python_version >= '3.11'
-Requires-Dist: langchain-google-genai>=2.0.9; python_version == '3.10'
-Requires-Dist: langchain-google-vertexai==2.1.2; python_version >= '3.11'
-Requires-Dist: langchain-google-vertexai>=2.0.28; python_version == '3.10'
-Requires-Dist: langchain-openai==0.3.34; python_version >= '3.11'
-Requires-Dist: langchain-openai>=0.3.16; python_version == '3.10'
-Requires-Dist: langchain-tests==0.3.22; python_version >= '3.11'
-Requires-Dist: langchain-tests>=0.3.20; python_version == '3.10'
+Requires-Dist: langchain-community>=0.3.27; python_version == '3.10'
+Requires-Dist: langchain-core>=0.3.80
 Requires-Dist: langchain==0.3.27; python_version >= '3.11'
 Requires-Dist: langchain>=0.3.25; python_version == '3.10'
 Requires-Dist: langgraph==0.6.8; python_version >= '3.11'
 Requires-Dist: langgraph>=0.3.20; python_version == '3.10'
-Requires-Dist: openai==2.1.0; python_version >= '3.11'
-Requires-Dist: openai>=1.77.0; python_version == '3.10'
-Requires-Dist: opentelemetry-api==1.38.0; python_version == '3.10'
-Requires-Dist: opentelemetry-sdk==1.38.0; python_version == '3.10'
 Requires-Dist: pydantic==2.10.4
-Requires-Dist: pytest==8.3.4
-Requires-Dist: pytimbr-api==2.0.0; python_version >= '3.11'
-Requires-Dist: pytimbr-api>=2.0.0; python_version == '3.10'
-Requires-Dist: snowflake-snowpark-python==1.39.1; python_version >= '3.11'
-Requires-Dist: snowflake-snowpark-python>=1.39.1; python_version == '3.10'
-Requires-Dist: snowflake==1.8.0; python_version >= '3.11'
-Requires-Dist: snowflake>=1.8.0; python_version == '3.10'
+Requires-Dist: pytimbr-api>=2.1.0
 Requires-Dist: tiktoken==0.8.0
 Requires-Dist: transformers==4.57.0; python_version >= '3.11'
 Requires-Dist: transformers>=4.53; python_version == '3.10'
@@ -70,6 +43,7 @@ Requires-Dist: databricks-sdk==0.64.0; extra == 'all'
 Requires-Dist: google-generativeai==0.8.4; extra == 'all'
 Requires-Dist: langchain-anthropic==0.3.5; (python_version >= '3.11') and extra == 'all'
 Requires-Dist: langchain-anthropic>=0.3.1; (python_version == '3.10') and extra == 'all'
+Requires-Dist: langchain-aws<1,>=0.2.35; extra == 'all'
 Requires-Dist: langchain-google-genai==2.0.10; (python_version >= '3.11') and extra == 'all'
 Requires-Dist: langchain-google-genai>=2.0.9; (python_version == '3.10') and extra == 'all'
 Requires-Dist: langchain-google-vertexai==2.1.2; (python_version >= '3.11') and extra == 'all'
@@ -97,6 +71,8 @@ Requires-Dist: langchain-openai==0.3.34; (python_version >= '3.11') and extra ==
 Requires-Dist: langchain-openai>=0.3.16; (python_version == '3.10') and extra == 'azure-openai'
 Requires-Dist: openai==2.1.0; (python_version >= '3.11') and extra == 'azure-openai'
 Requires-Dist: openai>=1.77.0; (python_version == '3.10') and extra == 'azure-openai'
+Provides-Extra: bedrock
+Requires-Dist: langchain-aws==0.2.35; extra == 'bedrock'
 Provides-Extra: databricks
 Requires-Dist: databricks-langchain==0.7.1; extra == 'databricks'
 Requires-Dist: databricks-sdk==0.64.0; extra == 'databricks'
@@ -158,7 +134,7 @@ python -m pip install langchain-timbr
 
 ### Install with selected LLM providers
 
-#### One of: openai, anthropic, google, azure_openai, snowflake, databricks, vertex_ai (or 'all')
+#### One of: openai, anthropic, google, azure_openai, snowflake, databricks, vertex_ai, bedrock (or 'all')
 
 ```bash
 python -m pip install 'langchain-timbr[<your selected providers, separated by comma w/o space>]'

{langchain_timbr-2.1.4 → langchain_timbr-2.1.6}/README.md

@@ -29,7 +29,7 @@ python -m pip install langchain-timbr
 
 ### Install with selected LLM providers
 
-#### One of: openai, anthropic, google, azure_openai, snowflake, databricks, vertex_ai (or 'all')
+#### One of: openai, anthropic, google, azure_openai, snowflake, databricks, vertex_ai, bedrock (or 'all')
 
 ```bash
 python -m pip install 'langchain-timbr[<your selected providers, separated by comma w/o space>]'

{langchain_timbr-2.1.4 → langchain_timbr-2.1.6}/pyproject.toml

@@ -24,53 +24,24 @@ classifiers = [
 
 dependencies = [
   # Core dependencies that are the same across all Python versions
-  "anthropic==0.42.0",
-  "databricks-langchain==0.7.1",
-  "databricks-sdk==0.64.0",
-  "google-generativeai==0.8.4",
   "pydantic==2.10.4",
-  "pytest==8.3.4",
   "tiktoken==0.8.0",
   "uvicorn==0.34.0",
+  "pytimbr-api>=2.1.0",
+  "langchain-core>=0.3.80",
 
   # Python 3.11+ versions (same for 3.11 and 3.12+)
-  "azure-identity==1.25.0; python_version>='3.11'",
   "cryptography==45.0.7; python_version>='3.11'",
   "langchain==0.3.27; python_version>='3.11'",
-  "langchain-anthropic==0.3.5; python_version>='3.11'",
   "langchain-community==0.3.30; python_version>='3.11'",
-  "langchain-core==0.3.78; python_version>='3.11'",
-  "langchain-google-genai==2.0.10; python_version>='3.11'",
-  "langchain-google-vertexai==2.1.2; python_version>='3.11'",
-  "langchain-openai==0.3.34; python_version>='3.11'",
-  "langchain-tests==0.3.22; python_version>='3.11'",
   "langgraph==0.6.8; python_version>='3.11'",
-  "openai==2.1.0; python_version>='3.11'",
-  "opentelemetry-api==1.38.0; python_version=='3.10'",
-  "opentelemetry-sdk==1.38.0; python_version=='3.10'",
-  "pytimbr_api==2.0.0; python_version>='3.11'",
-  "snowflake==1.8.0; python_version>='3.11'",
-  "snowflake-snowpark-python==1.39.1; python_version>='3.11'",
   "transformers==4.57.0; python_version>='3.11'",
 
   # Python 3.10 specific versions
-  "azure-identity>=1.16.1; python_version=='3.10'",
   "cryptography>=44.0.3; python_version=='3.10'",
   "langchain>=0.3.25; python_version=='3.10'",
-  "langchain-anthropic>=0.3.1; python_version=='3.10'",
-  "langchain_community>=0.3.20; python_version=='3.10'",
-  "langchain-core>=0.3.58; python_version=='3.10'",
-  "langchain-google-genai>=2.0.9; python_version=='3.10'",
-  "langchain-google-vertexai>=2.0.28; python_version=='3.10'",
-  "langchain-openai>=0.3.16; python_version=='3.10'",
-  "langchain-tests>=0.3.20; python_version=='3.10'",
+  "langchain-community>=0.3.27; python_version=='3.10'",
   "langgraph>=0.3.20; python_version=='3.10'",
-  "openai>=1.77.0; python_version=='3.10'",
-  "opentelemetry-api==1.38.0; python_version=='3.10'",
-  "opentelemetry-sdk==1.38.0; python_version=='3.10'",
-  "pytimbr-api>=2.0.0; python_version=='3.10'",
-  "snowflake>=1.8.0; python_version=='3.10'",
-  "snowflake-snowpark-python>=1.39.1; python_version=='3.10'",
   "transformers>=4.53; python_version=='3.10'",
 ]
 
@@ -117,6 +88,9 @@ databricks = [
   "databricks-langchain==0.7.1",
   "databricks-sdk==0.64.0"
 ]
+bedrock = [
+  "langchain-aws==0.2.35",
+]
 
 # Development and testing
 dev = [
@@ -128,12 +102,13 @@ dev = [
 
 # All optional dependencies
 all = [
-  "anthropic==0.42.0",
-  "google-generativeai==0.8.4",
   "pytest==8.3.4",
   "uvicorn==0.34.0",
+  "anthropic==0.42.0",
+  "google-generativeai==0.8.4",
   "databricks-langchain==0.7.1",
   "databricks-sdk==0.64.0",
+  "langchain-aws>=0.2.35,<1",
   # Python 3.11+ versions
   "azure-identity==1.25.0; python_version>='3.11'",
   "langchain-anthropic==0.3.5; python_version>='3.11'",

{langchain_timbr-2.1.4 → langchain_timbr-2.1.6}/requirements.txt

@@ -6,8 +6,9 @@ databricks-sdk==0.64.0
 google-generativeai==0.8.4
 langchain==0.3.27
 langchain-anthropic==0.3.5
+langchain-aws==0.2.35
 langchain-community==0.3.30
-langchain-core==0.3.78
+langchain-core==0.3.80
 langchain-google-genai==2.0.10
 langchain-google-vertexai==2.1.2
 langchain-openai==0.3.34
@@ -17,7 +18,7 @@ langgraph-checkpoint==3.0.0
 openai==2.1.0
 pydantic==2.10.4
 pytest==8.3.4
-pytimbr_api==2.0.0
+pytimbr-api>=2.1.0
 snowflake==1.8.0
 snowflake-snowpark-python==1.39.1
 tiktoken==0.8.0

{langchain_timbr-2.1.4 → langchain_timbr-2.1.6}/requirements310.txt

@@ -6,8 +6,9 @@ databricks-sdk==0.64.0
 google-generativeai==0.8.4
 langchain>=0.3.25
 langchain-anthropic>=0.3.1
-langchain_community>=0.3.20
-langchain-core>=0.3.58
+langchain-aws==0.2.35
+langchain-community>=0.3.27
+langchain-core>=0.3.80
 langchain-google-genai>=2.0.9
 langchain-google-vertexai>=2.0.28
 langchain-openai>=0.3.16
@@ -18,7 +19,7 @@ opentelemetry-api==1.38.0 # Required when using snowflake with Python under 3.12
 opentelemetry-sdk==1.38.0 # Required when using snowflake with Python under 3.12
 pydantic==2.10.4
 pytest==8.3.4
-pytimbr-api>=2.0.0
+pytimbr-api>=2.1.0
 snowflake>=1.8.0
 snowflake-snowpark-python>=1.23.0
 snowflake-connector-python[pandas]>=3.13.1,<4

{langchain_timbr-2.1.4 → langchain_timbr-2.1.6}/requirements311.txt

@@ -6,8 +6,9 @@ databricks-sdk==0.64.0
 google-generativeai==0.8.4
 langchain==0.3.27
 langchain-anthropic==0.3.5
+langchain-aws==0.2.35
 langchain-community==0.3.30
-langchain-core==0.3.78
+langchain-core==0.3.80
 langchain-google-genai==2.0.10
 langchain-google-vertexai==2.1.2
 langchain-openai==0.3.34
@@ -18,7 +19,7 @@ opentelemetry-api==1.38.0 # Required when using snowflake with Python under 3.12
 opentelemetry-sdk==1.38.0 # Required when using snowflake with Python under 3.12
 pydantic==2.10.4
 pytest==8.3.4
-pytimbr_api==2.0.0
+pytimbr-api>=2.1.0
 snowflake==1.8.0
 snowflake-snowpark-python==1.39.1
 tiktoken==0.8.0

{langchain_timbr-2.1.4 → langchain_timbr-2.1.6}/src/langchain_timbr/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '2.1.4'
-__version_tuple__ = version_tuple = (2, 1, 4)
+__version__ = version = '2.1.6'
+__version_tuple__ = version_tuple = (2, 1, 6)
 
 __commit_id__ = commit_id = None

{langchain_timbr-2.1.4 → langchain_timbr-2.1.6}/src/langchain_timbr/config.py

@@ -27,4 +27,8 @@ llm_client_id = os.environ.get('LLM_CLIENT_ID', None)
 llm_client_secret = os.environ.get('LLM_CLIENT_SECRET', None)
 llm_endpoint = os.environ.get('LLM_ENDPOINT', None)
 llm_api_version = os.environ.get('LLM_API_VERSION', None)
-llm_scope = os.environ.get('LLM_SCOPE', "https://cognitiveservices.azure.com/.default")  # e.g. "api://<your-client-id>/.default"
+llm_scope = os.environ.get('LLM_SCOPE', "https://cognitiveservices.azure.com/.default")  # e.g. "api://<your-client-id>/.default"
+
+# Whether to enable reasoning during SQL generation
+with_reasoning = to_boolean(os.environ.get('WITH_REASONING', 'false'))
+reasoning_steps = to_integer(os.environ.get('REASONING_STEPS', 2))

{langchain_timbr-2.1.4 → langchain_timbr-2.1.6}/src/langchain_timbr/langchain/execute_timbr_query_chain.py

@@ -42,6 +42,8 @@ class ExecuteTimbrQueryChain(Chain):
         is_jwt: Optional[bool] = False,
         jwt_tenant_id: Optional[str] = None,
         conn_params: Optional[dict] = None,
+        with_reasoning: Optional[bool] = config.with_reasoning,
+        reasoning_steps: Optional[int] = config.reasoning_steps,
         debug: Optional[bool] = False,
         **kwargs,
     ):
@@ -69,6 +71,8 @@ class ExecuteTimbrQueryChain(Chain):
         :param is_jwt: Whether to use JWT authentication (default is False).
         :param jwt_tenant_id: JWT tenant ID for multi-tenant environments (required when is_jwt=True).
         :param conn_params: Extra Timbr connection parameters sent with every request (e.g., 'x-api-impersonate-user').
+        :param with_reasoning: Whether to enable reasoning during SQL generation (default is False).
+        :param reasoning_steps: Number of reasoning steps to perform if reasoning is enabled (default is 2).
         :param kwargs: Additional arguments to pass to the base
         :return: A list of rows from the Timbr query
 
@@ -137,6 +141,8 @@ class ExecuteTimbrQueryChain(Chain):
         self._jwt_tenant_id = jwt_tenant_id
         self._debug = to_boolean(debug)
         self._conn_params = conn_params or {}
+        self._with_reasoning = to_boolean(with_reasoning)
+        self._reasoning_steps = to_integer(reasoning_steps)
 
 
     @property
@@ -209,6 +215,8 @@ class ExecuteTimbrQueryChain(Chain):
             note=(self._note or '') + err_txt,
             db_is_case_sensitive=self._db_is_case_sensitive,
             graph_depth=self._graph_depth,
+            with_reasoning=self._with_reasoning,
+            reasoning_steps=self._reasoning_steps,
             debug=self._debug,
         )
 
@@ -239,6 +247,7 @@ class ExecuteTimbrQueryChain(Chain):
             concept_name = inputs.get("concept", self._concept)
             is_sql_valid = True
             error = None
+            reasoning_status = None
             usage_metadata = {}
 
             if sql and self._should_validate_sql:
@@ -255,6 +264,7 @@ class ExecuteTimbrQueryChain(Chain):
                     schema_name = generate_res.get("schema", schema_name)
                     concept_name = generate_res.get("concept", concept_name)
                     is_sql_valid = generate_res.get("is_sql_valid")
+                    reasoning_status = generate_res.get("reasoning_status")
                     if not is_sql_valid and not self._should_validate_sql:
                         is_sql_valid = True
 
@@ -293,6 +303,7 @@ class ExecuteTimbrQueryChain(Chain):
                 "schema": schema_name,
                 "concept": concept_name,
                 "error": error if not is_sql_valid else None,
+                "reasoning_status": reasoning_status,
                 self.usage_metadata_key: usage_metadata,
             }
 

{langchain_timbr-2.1.4 → langchain_timbr-2.1.6}/src/langchain_timbr/langchain/generate_timbr_sql_chain.py

@@ -39,6 +39,8 @@ class GenerateTimbrSqlChain(Chain):
         is_jwt: Optional[bool] = False,
         jwt_tenant_id: Optional[str] = None,
         conn_params: Optional[dict] = None,
+        with_reasoning: Optional[bool] = config.with_reasoning,
+        reasoning_steps: Optional[int] = config.reasoning_steps,
         debug: Optional[bool] = False,
         **kwargs,
     ):
@@ -64,6 +66,9 @@ class GenerateTimbrSqlChain(Chain):
         :param is_jwt: Whether to use JWT authentication (default is False).
         :param jwt_tenant_id: JWT tenant ID for multi-tenant environments (required when is_jwt=True).
         :param conn_params: Extra Timbr connection parameters sent with every request (e.g., 'x-api-impersonate-user').
+        :param with_reasoning: Whether to enable reasoning during SQL generation (default is False).
+        :param reasoning_steps: Number of reasoning steps to perform if reasoning is enabled (default is 2).
+        :param debug: Whether to enable debug mode for detailed logging
         :param kwargs: Additional arguments to pass to the base
         
         ## Example
@@ -129,6 +134,8 @@ class GenerateTimbrSqlChain(Chain):
         self._jwt_tenant_id = jwt_tenant_id
         self._debug = to_boolean(debug)
         self._conn_params = conn_params or {}
+        self._with_reasoning = to_boolean(with_reasoning)
+        self._reasoning_steps = to_integer(reasoning_steps)
 
 
     @property
@@ -184,6 +191,8 @@ class GenerateTimbrSqlChain(Chain):
             note=self._note,
             db_is_case_sensitive=self._db_is_case_sensitive,
             graph_depth=self._graph_depth,
+            with_reasoning=self._with_reasoning,
+            reasoning_steps=self._reasoning_steps,
             debug=self._debug,
         )
         
@@ -197,5 +206,6 @@ class GenerateTimbrSqlChain(Chain):
             "concept": concept,
             "is_sql_valid": generate_res.get("is_sql_valid"),
             "error": generate_res.get("error"),
+            "reasoning_status": generate_res.get("reasoning_status"),
             self.usage_metadata_key: generate_res.get("usage_metadata"),
         }

{langchain_timbr-2.1.4 → langchain_timbr-2.1.6}/src/langchain_timbr/langchain/timbr_sql_agent.py

@@ -6,6 +6,7 @@ from langchain.schema import AgentAction, AgentFinish
 from ..utils.general import parse_list, to_boolean, to_integer
 from .execute_timbr_query_chain import ExecuteTimbrQueryChain
 from .generate_answer_chain import GenerateAnswerChain
+from .. import config
 
 class TimbrSqlAgent(BaseSingleActionAgent):
     def __init__(
@@ -34,6 +35,8 @@ class TimbrSqlAgent(BaseSingleActionAgent):
         is_jwt: Optional[bool] = False,
         jwt_tenant_id: Optional[str] = None,
         conn_params: Optional[dict] = None,
+        with_reasoning: Optional[bool] = config.with_reasoning,
+        reasoning_steps: Optional[int] = config.reasoning_steps,
         debug: Optional[bool] = False
     ):
         """
@@ -61,6 +64,8 @@ class TimbrSqlAgent(BaseSingleActionAgent):
         :param is_jwt: Whether to use JWT authentication (default is False).
         :param jwt_tenant_id: JWT tenant ID for multi-tenant environments (required when is_jwt=True).
         :param conn_params: Extra Timbr connection parameters sent with every request (e.g., 'x-api-impersonate-user').
+        :param with_reasoning: Whether to enable reasoning during SQL generation (default is False).
+        :param reasoning_steps: Number of reasoning steps to perform if reasoning is enabled (default is 2).
 
         ## Example
         ```
@@ -113,6 +118,8 @@ class TimbrSqlAgent(BaseSingleActionAgent):
             is_jwt=to_boolean(is_jwt),
             jwt_tenant_id=jwt_tenant_id,
             conn_params=conn_params,
+            with_reasoning=to_boolean(with_reasoning),
+            reasoning_steps=to_integer(reasoning_steps),
             debug=to_boolean(debug),
         )
         self._generate_answer = to_boolean(generate_answer)
@@ -173,6 +180,7 @@ class TimbrSqlAgent(BaseSingleActionAgent):
                     "sql": None,
                     "schema": None,
                     "concept": None,
+                    "reasoning_status": None,
                     "usage_metadata": {},
                 },
                 log="Empty input received"
@@ -200,6 +208,7 @@ class TimbrSqlAgent(BaseSingleActionAgent):
                     "schema": result.get("schema", ""),
                     "concept": result.get("concept", ""),
                     "error": result.get("error", None),
+                    "reasoning_status": result.get("reasoning_status", None),
                     "usage_metadata": usage_metadata,
                 },
                 log=f"Successfully executed query on concept: {result.get('concept', '')}"
@@ -214,6 +223,7 @@ class TimbrSqlAgent(BaseSingleActionAgent):
                     "sql": None,
                     "schema": None,
                     "concept": None,
+                    "reasoning_status": None,
                     "usage_metadata": {},
                 },
                 log=error_context
@@ -234,6 +244,7 @@ class TimbrSqlAgent(BaseSingleActionAgent):
                     "sql": None,
                     "schema": None,
                     "concept": None,
+                    "reasoning_status": None,
                     "usage_metadata": {},
                 },
                 log="Empty or whitespace-only input received"
@@ -274,6 +285,7 @@ class TimbrSqlAgent(BaseSingleActionAgent):
                     "schema": result.get("schema", ""),
                     "concept": result.get("concept", ""),
                     "error": result.get("error", None),
+                    "reasoning_status": result.get("reasoning_status", None),
                     "usage_metadata": usage_metadata,
                 },
                 log=f"Successfully executed query on concept: {result.get('concept', '')}"
@@ -288,6 +300,7 @@ class TimbrSqlAgent(BaseSingleActionAgent):
                     "sql": None,
                     "schema": None,
                     "concept": None,
+                    "reasoning_status": None,
                     "usage_metadata": {},
                 },
                 log=error_context
@@ -332,6 +345,8 @@ def create_timbr_sql_agent(
     is_jwt: Optional[bool] = False,
     jwt_tenant_id: Optional[str] = None,
     conn_params: Optional[dict] = None,
+    with_reasoning: Optional[bool] = config.with_reasoning,
+    reasoning_steps: Optional[int] = config.reasoning_steps,
     debug: Optional[bool] = False
 ) -> AgentExecutor:
     """
@@ -361,6 +376,8 @@ def create_timbr_sql_agent(
     :param is_jwt: Whether to use JWT authentication (default is False).
     :param jwt_tenant_id: JWT tenant ID for multi-tenant environments (required when is_jwt=True).
     :param conn_params: Extra Timbr connection parameters sent with every request (e.g., 'x-api-impersonate-user').
+    :param with_reasoning: Whether to enable reasoning during SQL generation (default is False).
+    :param reasoning_steps: Number of reasoning steps to perform if reasoning is enabled (default is 2).
 
     Returns:
         AgentExecutor: Configured agent executor ready to use
@@ -427,6 +444,8 @@ def create_timbr_sql_agent(
         is_jwt=is_jwt,
         jwt_tenant_id=jwt_tenant_id,
         conn_params=conn_params,
+        with_reasoning=with_reasoning,
+        reasoning_steps=reasoning_steps,
         debug=debug,
     )
     

{langchain_timbr-2.1.4 → langchain_timbr-2.1.6}/src/langchain_timbr/langgraph/execute_timbr_query_node.py

@@ -3,7 +3,7 @@ from langchain.llms.base import LLM
 from langgraph.graph import StateGraph
 
 from ..langchain.execute_timbr_query_chain import ExecuteTimbrQueryChain
-
+from .. import config
 
 class ExecuteSemanticQueryNode:
     """
@@ -36,6 +36,8 @@ class ExecuteSemanticQueryNode:
         is_jwt: Optional[bool] = False,
         jwt_tenant_id: Optional[str] = None,
         conn_params: Optional[dict] = None,
+        with_reasoning: Optional[bool] = config.with_reasoning,
+        reasoning_steps: Optional[int] = config.reasoning_steps,
         debug: Optional[bool] = False,
         **kwargs,
     ):
@@ -63,6 +65,8 @@ class ExecuteSemanticQueryNode:
         :param is_jwt: Whether to use JWT authentication (default is False).
         :param jwt_tenant_id: JWT tenant ID for multi-tenant environments (required when is_jwt=True).
         :param conn_params: Extra Timbr connection parameters sent with every request (e.g., 'x-api-impersonate-user').
+        :param with_reasoning: Whether to enable reasoning during SQL generation (default is False).
+        :param reasoning_steps: Number of reasoning steps to perform if reasoning is enabled (default is 2).
         :return: A list of rows from the Timbr query
         """
         self.chain = ExecuteTimbrQueryChain(
@@ -89,6 +93,8 @@ class ExecuteSemanticQueryNode:
             is_jwt=is_jwt,
             jwt_tenant_id=jwt_tenant_id,
             conn_params=conn_params,
+            with_reasoning=with_reasoning,
+            reasoning_steps=reasoning_steps,
             debug=debug,
             **kwargs,
         )

{langchain_timbr-2.1.4 → langchain_timbr-2.1.6}/src/langchain_timbr/langgraph/generate_response_node.py

@@ -20,6 +20,7 @@ class GenerateResponseNode:
         is_jwt: Optional[bool] = False,
         jwt_tenant_id: Optional[str] = None,
         conn_params: Optional[dict] = None,
+        note: Optional[str] = '',
         debug: Optional[bool] = False,
         **kwargs,
     ):
@@ -31,6 +32,7 @@ class GenerateResponseNode:
         :param is_jwt: Whether to use JWT authentication (default is False).
         :param jwt_tenant_id: JWT tenant ID for multi-tenant environments (required when is_jwt=True).
         :param conn_params: Extra Timbr connection parameters sent with every request (e.g., 'x-api-impersonate-user').
+        :param note: Optional additional note to extend our llm prompt
         """
         self.chain = GenerateAnswerChain(
             llm=llm,
@@ -40,6 +42,7 @@ class GenerateResponseNode:
             is_jwt=is_jwt,
             jwt_tenant_id=jwt_tenant_id,
             conn_params=conn_params,
+            note=note,
             debug=debug,
             **kwargs,
         )

{langchain_timbr-2.1.4 → langchain_timbr-2.1.6}/src/langchain_timbr/langgraph/generate_timbr_sql_node.py

@@ -3,6 +3,7 @@ from langchain.llms.base import LLM
 from langgraph.graph import StateGraph
 
 from ..langchain.generate_timbr_sql_chain import GenerateTimbrSqlChain
+from .. import config
 
 class GenerateTimbrSqlNode:
     """
@@ -32,6 +33,8 @@ class GenerateTimbrSqlNode:
         is_jwt: Optional[bool] = False,
         jwt_tenant_id: Optional[str] = None,
         conn_params: Optional[dict] = None,
+        with_reasoning: Optional[bool] = config.with_reasoning,
+        reasoning_steps: Optional[int] = config.reasoning_steps,
         debug: Optional[bool] = False,
         **kwargs,
     ):
@@ -57,6 +60,8 @@ class GenerateTimbrSqlNode:
         :param is_jwt: Whether to use JWT authentication (default: False)
         :param jwt_tenant_id: Tenant ID for JWT authentication when using multi-tenant setup
         :param conn_params: Extra Timbr connection parameters sent with every request (e.g., 'x-api-impersonate-user').
+        :param with_reasoning: Whether to enable reasoning during SQL generation (default is False).
+        :param reasoning_steps: Number of reasoning steps to perform if reasoning is enabled (default is 2).
         """
         self.chain = GenerateTimbrSqlChain(
             llm=llm,
@@ -80,6 +85,8 @@ class GenerateTimbrSqlNode:
             is_jwt=is_jwt,
             jwt_tenant_id=jwt_tenant_id,
             conn_params=conn_params,
+            with_reasoning=with_reasoning,
+            reasoning_steps=reasoning_steps,
             debug=debug,
             **kwargs,
         )

{langchain_timbr-2.1.4 → langchain_timbr-2.1.6}/src/langchain_timbr/llm_wrapper/llm_wrapper.py

@@ -15,6 +15,7 @@ class LlmTypes(Enum):
   Snowflake = 'snowflake-cortex'
   Databricks = 'chat-databricks'
   VertexAI = 'chat-vertexai'
+  Bedrock = 'amazon_bedrock_converse_chat'
   Timbr = 'timbr'
 
 
@@ -252,6 +253,28 @@ class LlmWrapper(LLM):
         credentials=creds,
         **params,
       )
+    elif is_llm_type(llm_type, LlmTypes.Bedrock):
+      from langchain_aws import ChatBedrockConverse
+      llm_model = model or "openai.gpt-oss-20b-1:0"
+      params = self._add_temperature(LlmTypes.Bedrock.name, llm_model, **llm_params)
+
+      aws_region = pop_param_value(params, ['aws_region', 'llm_region', 'region'])
+      if aws_region:
+        params['region_name'] = aws_region
+      aws_access_key_id = pop_param_value(params, ['aws_access_key_id', 'llm_access_key_id', 'access_key_id'])
+      if aws_access_key_id:
+        params['aws_access_key_id'] = aws_access_key_id
+      aws_secret_access_key = pop_param_value(params, ['aws_secret_access_key', 'llm_secret_access_key', 'secret_access_key'], default=api_key)
+      if aws_secret_access_key:
+        params['aws_secret_access_key'] = aws_secret_access_key
+      aws_session_token = pop_param_value(params, ['aws_session_token', 'llm_session_token', 'session_token'])
+      if aws_session_token:
+        params['aws_session_token'] = aws_session_token
+
+      return ChatBedrockConverse(
+        model=llm_model,
+        **params,
+      )
     else:
       raise ValueError(f"Unsupported LLM type: {llm_type}")
 
@@ -324,6 +347,31 @@ class LlmWrapper(LLM):
         if self.client.credentials:
           client = genai.Client(credentials=self.client.credentials, vertexai=True, project=self.client.project, location=self.client.location)
           models = [m.name.split('/')[-1] for m in client.models.list()]
+      elif is_llm_type(self._llm_type, LlmTypes.Bedrock):
+        import boto3
+        
+        # Extract SecretStr values properly
+        aws_access_key_id = getattr(self.client, 'aws_access_key_id', None)
+        if aws_access_key_id and hasattr(aws_access_key_id, '_secret_value'):
+          aws_access_key_id = aws_access_key_id._secret_value
+        
+        aws_secret_access_key = getattr(self.client, 'aws_secret_access_key', None)
+        if aws_secret_access_key and hasattr(aws_secret_access_key, '_secret_value'):
+          aws_secret_access_key = aws_secret_access_key._secret_value
+        
+        aws_session_token = getattr(self.client, 'aws_session_token', None)
+        if aws_session_token and hasattr(aws_session_token, '_secret_value'):
+          aws_session_token = aws_session_token._secret_value
+        
+        bedrock_client = boto3.client(
+          service_name='bedrock',
+          region_name=getattr(self.client, 'region_name', None),
+          aws_access_key_id=aws_access_key_id,
+          aws_secret_access_key=aws_secret_access_key,
+          aws_session_token=aws_session_token,
+        )
+        response = bedrock_client.list_foundation_models()
+        models = [model['modelId'] for model in response.get('modelSummaries', [])]
 
     except Exception:
       # If model list fetching throws an exception, return default value using get_supported_models
@@ -341,6 +389,9 @@ class LlmWrapper(LLM):
 
 
   def _call(self, prompt, **kwargs):
+    # TODO: Remove this condition on next langchain-timbr major release
+    if is_llm_type(self._llm_type, LlmTypes.Bedrock):
+      return self.client.invoke(prompt, **kwargs)
     return self.client(prompt, **kwargs)