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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-timbr
-Version: 2.1.6
+Version: 2.1.8
 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/

{langchain_timbr-2.1.6 → langchain_timbr-2.1.8}/src/langchain_timbr/__init__.py

@@ -12,7 +12,7 @@
 from .timbr_llm_connector import TimbrLlmConnector
 from .llm_wrapper.llm_wrapper import LlmWrapper, LlmTypes
 
-from .utils.timbr_llm_utils import (
+from .utils.timbr_utils import (
     generate_key,
     encrypt_prompt,
     decrypt_prompt,

{langchain_timbr-2.1.6 → langchain_timbr-2.1.8}/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.6'
-__version_tuple__ = version_tuple = (2, 1, 6)
+__version__ = version = '2.1.8'
+__version_tuple__ = version_tuple = (2, 1, 8)
 
 __commit_id__ = commit_id = None

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

@@ -30,5 +30,9 @@ 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"
 
 # 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))
+enable_reasoning = to_boolean(os.environ.get('ENABLE_REASONING', 'false'))
+reasoning_steps = to_integer(os.environ.get('REASONING_STEPS', 2))
+
+should_validate_sql = to_boolean(os.environ.get('SHOULD_VALIDATE_SQL', os.environ.get('LLM_SHOULD_VALIDATE_SQL', 'true')))
+retry_if_no_results = to_boolean(os.environ.get('RETRY_IF_NO_RESULTS', os.environ.get('LLM_RETRY_IF_NO_RESULTS', 'true')))
+llm_default_limit = to_integer(os.environ.get('LLM_DEFAULT_LIMIT', 500))  # Default max tokens limit for LLM responses

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

@@ -30,10 +30,10 @@ class ExecuteTimbrQueryChain(Chain):
         include_logic_concepts: Optional[bool] = False,
         include_tags: Optional[Union[list[str], str]] = None,
         exclude_properties: Optional[Union[list[str], str]] = ['entity_id', 'entity_type', 'entity_label'],
-        should_validate_sql: Optional[bool] = False,
+        should_validate_sql: Optional[bool] = config.should_validate_sql,
         retries: Optional[int] = 3,
-        max_limit: Optional[int] = 500,
-        retry_if_no_results: Optional[bool] = False,
+        max_limit: Optional[int] = config.llm_default_limit,
+        retry_if_no_results: Optional[bool] = config.retry_if_no_results,
         no_results_max_retries: Optional[int] = 2,
         note: Optional[str] = '',
         db_is_case_sensitive: Optional[bool] = False,
@@ -42,7 +42,7 @@ 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,
+        enable_reasoning: Optional[bool] = config.enable_reasoning,
         reasoning_steps: Optional[int] = config.reasoning_steps,
         debug: Optional[bool] = False,
         **kwargs,
@@ -71,7 +71,7 @@ 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 enable_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
@@ -141,7 +141,7 @@ 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._enable_reasoning = to_boolean(enable_reasoning)
         self._reasoning_steps = to_integer(reasoning_steps)
 
 
@@ -215,7 +215,7 @@ 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,
+            enable_reasoning=self._enable_reasoning,
             reasoning_steps=self._reasoning_steps,
             debug=self._debug,
         )

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

@@ -29,9 +29,9 @@ class GenerateTimbrSqlChain(Chain):
         include_tags: Optional[Union[list[str], str]] = None,
         include_logic_concepts: Optional[bool] = False,
         exclude_properties: Optional[Union[list[str], str]] = ['entity_id', 'entity_type', 'entity_label'],
-        should_validate_sql: Optional[bool] = False,
+        should_validate_sql: Optional[bool] = config.should_validate_sql,
         retries: Optional[int] = 3,
-        max_limit: Optional[int] = 500,
+        max_limit: Optional[int] = config.llm_default_limit,
         note: Optional[str] = '',
         db_is_case_sensitive: Optional[bool] = False,
         graph_depth: Optional[int] = 1,
@@ -39,7 +39,7 @@ 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,
+        enable_reasoning: Optional[bool] = config.enable_reasoning,
         reasoning_steps: Optional[int] = config.reasoning_steps,
         debug: Optional[bool] = False,
         **kwargs,
@@ -66,7 +66,7 @@ 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 enable_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
@@ -134,7 +134,7 @@ 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._enable_reasoning = to_boolean(enable_reasoning)
         self._reasoning_steps = to_integer(reasoning_steps)
 
 
@@ -191,7 +191,7 @@ 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,
+            enable_reasoning=self._enable_reasoning,
             reasoning_steps=self._reasoning_steps,
             debug=self._debug,
         )

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

@@ -22,10 +22,10 @@ class TimbrSqlAgent(BaseSingleActionAgent):
         include_logic_concepts: Optional[bool] = False,
         include_tags: Optional[Union[list[str], str]] = None,
         exclude_properties: Optional[Union[list[str], str]] = ['entity_id', 'entity_type', 'entity_label'],
-        should_validate_sql: Optional[bool] = True,
+        should_validate_sql: Optional[bool] = config.should_validate_sql,
         retries: Optional[int] = 3,
-        max_limit: Optional[int] = 500,
-        retry_if_no_results: Optional[bool] = False,
+        max_limit: Optional[int] = config.llm_default_limit,
+        retry_if_no_results: Optional[bool] = config.retry_if_no_results,
         no_results_max_retries: Optional[int] = 2,
         generate_answer: Optional[bool] = False,
         note: Optional[str] = '',
@@ -35,7 +35,7 @@ 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,
+        enable_reasoning: Optional[bool] = config.enable_reasoning,
         reasoning_steps: Optional[int] = config.reasoning_steps,
         debug: Optional[bool] = False
     ):
@@ -64,7 +64,7 @@ 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 enable_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
@@ -118,7 +118,7 @@ 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),
+            enable_reasoning=to_boolean(enable_reasoning),
             reasoning_steps=to_integer(reasoning_steps),
             debug=to_boolean(debug),
         )
@@ -332,10 +332,10 @@ def create_timbr_sql_agent(
     include_logic_concepts: Optional[bool] = False,
     include_tags: Optional[Union[list[str], str]] = None,
     exclude_properties: Optional[Union[list[str], str]] = ['entity_id', 'entity_type', 'entity_label'],
-    should_validate_sql: Optional[bool] = True,
+    should_validate_sql: Optional[bool] = config.should_validate_sql,
     retries: Optional[int] = 3,
-    max_limit: Optional[int] = 500,
-    retry_if_no_results: Optional[bool] = False,
+    max_limit: Optional[int] = config.llm_default_limit,
+    retry_if_no_results: Optional[bool] = config.retry_if_no_results,
     no_results_max_retries: Optional[int] = 2,
     generate_answer: Optional[bool] = False,
     note: Optional[str] = '',
@@ -345,7 +345,7 @@ 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,
+    enable_reasoning: Optional[bool] = config.enable_reasoning,
     reasoning_steps: Optional[int] = config.reasoning_steps,
     debug: Optional[bool] = False
 ) -> AgentExecutor:
@@ -376,7 +376,7 @@ 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 enable_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:
@@ -444,7 +444,7 @@ def create_timbr_sql_agent(
         is_jwt=is_jwt,
         jwt_tenant_id=jwt_tenant_id,
         conn_params=conn_params,
-        with_reasoning=with_reasoning,
+        enable_reasoning=enable_reasoning,
         reasoning_steps=reasoning_steps,
         debug=debug,
     )

{langchain_timbr-2.1.6 → langchain_timbr-2.1.8}/src/langchain_timbr/langchain/validate_timbr_sql_chain.py

@@ -32,7 +32,7 @@ class ValidateTimbrSqlChain(Chain):
         include_logic_concepts: Optional[bool] = False,
         include_tags: Optional[Union[list[str], str]] = None,
         exclude_properties: Optional[Union[list[str], str]] = ['entity_id', 'entity_type', 'entity_label'],
-        max_limit: Optional[int] = 500,
+        max_limit: Optional[int] = config.llm_default_limit,
         note: Optional[str] = '',
         db_is_case_sensitive: Optional[bool] = False,
         graph_depth: Optional[int] = 1,
@@ -40,6 +40,8 @@ class ValidateTimbrSqlChain(Chain):
         is_jwt: Optional[bool] = False,
         jwt_tenant_id: Optional[str] = None,
         conn_params: Optional[dict] = None,
+        enable_reasoning: Optional[bool] = config.enable_reasoning,
+        reasoning_steps: Optional[int] = config.reasoning_steps,
         debug: Optional[bool] = False,
         **kwargs,
     ):
@@ -64,6 +66,8 @@ class ValidateTimbrSqlChain(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 enable_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
         
         ## Example
@@ -127,8 +131,10 @@ class ValidateTimbrSqlChain(Chain):
         self._verify_ssl = to_boolean(verify_ssl)
         self._is_jwt = to_boolean(is_jwt)
         self._jwt_tenant_id = jwt_tenant_id
-        self._debug = to_boolean(debug)
         self._conn_params = conn_params or {}
+        self._enable_reasoning = to_boolean(enable_reasoning)
+        self._reasoning_steps = to_integer(reasoning_steps)
+        self._debug = to_boolean(debug)
 
 
     @property
@@ -193,6 +199,8 @@ class ValidateTimbrSqlChain(Chain):
                 note=f"{prompt_extension}The original SQL query (`{sql}`) was invalid with this error from query {error}. Please take this in consideration while generating the query.",
                 db_is_case_sensitive=self._db_is_case_sensitive,
                 graph_depth=self._graph_depth,
+                enable_reasoning=self._enable_reasoning,
+                reasoning_steps=self._reasoning_steps,
                 debug=self._debug,
             )
             sql = generate_res.get("sql", "")
@@ -200,6 +208,7 @@ class ValidateTimbrSqlChain(Chain):
             concept = generate_res.get("concept", self._concept)
             usage_metadata.update(generate_res.get("usage_metadata", {}))
             is_sql_valid = generate_res.get("is_sql_valid")
+            reasoning_status = generate_res.get("reasoning_status")
             error = generate_res.get("error")
 
         return {
@@ -208,5 +217,6 @@ class ValidateTimbrSqlChain(Chain):
             "concept": concept,
             "is_sql_valid": is_sql_valid,
             "error": error,
+            "reasoning_status": reasoning_status,
             self.usage_metadata_key: usage_metadata,
         }

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

@@ -24,10 +24,10 @@ class ExecuteSemanticQueryNode:
         include_logic_concepts: Optional[bool] = False,
         include_tags: Optional[Union[list[str], str]] = None,
         exclude_properties: Optional[Union[list[str], str]] = ['entity_id', 'entity_type', 'entity_label'],
-        should_validate_sql: Optional[bool] = False,
+        should_validate_sql: Optional[bool] = config.should_validate_sql,
         retries: Optional[int] = 3,
-        max_limit: Optional[int] = 500,
-        retry_if_no_results: Optional[bool] = False,
+        max_limit: Optional[int] = config.llm_default_limit,
+        retry_if_no_results: Optional[bool] = config.retry_if_no_results,
         no_results_max_retries: Optional[int] = 2,
         note: Optional[str] = '',
         db_is_case_sensitive: Optional[bool] = False,
@@ -36,7 +36,7 @@ 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,
+        enable_reasoning: Optional[bool] = config.enable_reasoning,
         reasoning_steps: Optional[int] = config.reasoning_steps,
         debug: Optional[bool] = False,
         **kwargs,
@@ -65,7 +65,7 @@ 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 enable_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
         """
@@ -93,7 +93,7 @@ class ExecuteSemanticQueryNode:
             is_jwt=is_jwt,
             jwt_tenant_id=jwt_tenant_id,
             conn_params=conn_params,
-            with_reasoning=with_reasoning,
+            enable_reasoning=enable_reasoning,
             reasoning_steps=reasoning_steps,
             debug=debug,
             **kwargs,

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

@@ -23,9 +23,9 @@ class GenerateTimbrSqlNode:
         include_logic_concepts: Optional[bool] = False,
         include_tags: Optional[Union[list[str], str]] = None,
         exclude_properties: Optional[Union[list[str], str]] = ['entity_id', 'entity_type', 'entity_label'],
-        should_validate_sql: Optional[bool] = False,
+        should_validate_sql: Optional[bool] = config.should_validate_sql,
         retries: Optional[int] = 3,
-        max_limit: Optional[int] = 500,
+        max_limit: Optional[int] = config.llm_default_limit,
         note: Optional[str] = '',
         db_is_case_sensitive: Optional[bool] = False,
         graph_depth: Optional[int] = 1,
@@ -33,7 +33,7 @@ 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,
+        enable_reasoning: Optional[bool] = config.enable_reasoning,
         reasoning_steps: Optional[int] = config.reasoning_steps,
         debug: Optional[bool] = False,
         **kwargs,
@@ -60,7 +60,7 @@ 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 enable_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(
@@ -85,7 +85,7 @@ class GenerateTimbrSqlNode:
             is_jwt=is_jwt,
             jwt_tenant_id=jwt_tenant_id,
             conn_params=conn_params,
-            with_reasoning=with_reasoning,
+            enable_reasoning=enable_reasoning,
             reasoning_steps=reasoning_steps,
             debug=debug,
             **kwargs,

{langchain_timbr-2.1.6 → langchain_timbr-2.1.8}/src/langchain_timbr/langgraph/validate_timbr_query_node.py

@@ -3,7 +3,7 @@ from langchain.llms.base import LLM
 from langgraph.graph import StateGraph
 
 from ..langchain.validate_timbr_sql_chain import ValidateTimbrSqlChain
-
+from .. import config
 
 class ValidateSemanticSqlNode:
     """
@@ -25,7 +25,7 @@ class ValidateSemanticSqlNode:
         include_logic_concepts: Optional[bool] = False,
         include_tags: Optional[Union[list[str], str]] = None,
         exclude_properties: Optional[Union[list[str], str]] = ['entity_id', 'entity_type', 'entity_label'],
-        max_limit: Optional[int] = 500,
+        max_limit: Optional[int] = config.llm_default_limit,
         note: Optional[str] = None,
         db_is_case_sensitive: Optional[bool] = False,
         graph_depth: Optional[int] = 1,
@@ -33,6 +33,8 @@ class ValidateSemanticSqlNode:
         is_jwt: Optional[bool] = False,
         jwt_tenant_id: Optional[str] = None,
         conn_params: Optional[dict] = None,
+        enable_reasoning: Optional[bool] = config.enable_reasoning,
+        reasoning_steps: Optional[int] = config.reasoning_steps,
         debug: Optional[bool] = False,
         **kwargs,
     ):
@@ -57,6 +59,8 @@ class ValidateSemanticSqlNode:
         :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 enable_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 = ValidateTimbrSqlChain(
             llm=llm,
@@ -79,6 +83,8 @@ class ValidateSemanticSqlNode:
             is_jwt=is_jwt,
             jwt_tenant_id=jwt_tenant_id,
             conn_params=conn_params,
+            enable_reasoning=enable_reasoning,
+            reasoning_steps=reasoning_steps,
             debug=debug,
             **kwargs,
         )

{langchain_timbr-2.1.6 → langchain_timbr-2.1.8}/src/langchain_timbr/timbr_llm_connector.py

@@ -30,7 +30,7 @@ class TimbrLlmConnector:
         url: Optional[str] = config.url,
         token: Optional[str] = config.token,
         ontology: Optional[str] = config.ontology,
-        max_limit: Optional[int] = 500,
+        max_limit: Optional[int] = config.llm_default_limit,
         verify_ssl: Optional[bool] = True,
         is_jwt: Optional[bool] = False,
         jwt_tenant_id: Optional[str] = None,
@@ -185,7 +185,7 @@ class TimbrLlmConnector:
         views_list: Optional[list] = None,
         include_logic_concepts: Optional[bool] = False,
         include_tags: Optional[str] = None,
-        should_validate_sql: Optional[bool] = False,
+        should_validate_sql: Optional[bool] = config.should_validate_sql,
         retries: Optional[int] = 3,
         note: Optional[str] = '',
         **chain_kwargs: Any,
@@ -243,7 +243,7 @@ class TimbrLlmConnector:
         views_list: Optional[list] = None,
         include_logic_concepts: Optional[bool] = False,
         include_tags: Optional[str] = None,
-        should_validate_sql: Optional[bool] = True,
+        should_validate_sql: Optional[bool] = config.should_validate_sql,
         retries: Optional[int] = 3,
         note: Optional[str] = '',
         **chain_kwargs: Any,
@@ -272,7 +272,7 @@ class TimbrLlmConnector:
         views_list: Optional[list] = None,
         include_logic_concepts: Optional[bool] = False,
         include_tags: Optional[str] = None,
-        should_validate_sql: Optional[bool] = True,
+        should_validate_sql: Optional[bool] = config.should_validate_sql,
         retries: Optional[int] = 3,
         note: Optional[str] = '',
         **agent_kwargs: Any,
@@ -302,7 +302,7 @@ class TimbrLlmConnector:
         views_list: Optional[list] = None,
         include_logic_concepts: Optional[bool] = False,
         include_tags: Optional[str] = None,
-        should_validate_sql: Optional[bool] = True,
+        should_validate_sql: Optional[bool] = config.should_validate_sql,
         retries: Optional[int] = 3,
         note: Optional[str] = '',
         **nodes_kwargs: Any,

{langchain_timbr-2.1.6 → langchain_timbr-2.1.8}/src/langchain_timbr/utils/prompt_service.py

@@ -184,6 +184,16 @@ class PromptService:
         return self._fetch_template("llm_prompts/generate_sql")
     
     
+    def get_generate_sql_reasoning_template(self) -> ChatPromptTemplate:
+        """
+        Get generate SQL reasoning template from API service (cached)
+        
+        Returns:
+            ChatPromptTemplate object
+        """
+        return self._fetch_template("llm_prompts/generate_sql_reasoning")
+
+    
     def get_generate_answer_template(self) -> ChatPromptTemplate:
         """
         Get generate answer template from API service (cached)
@@ -264,6 +274,22 @@ def get_generate_sql_prompt_template(
     return PromptTemplateWrapper(prompt_service, "get_generate_sql_template")
 
 
+def get_generate_sql_reasoning_prompt_template(
+    conn_params: Optional[dict] = None
+) -> PromptTemplateWrapper:
+    """
+    Get generate SQL reasoning prompt template wrapper
+    
+    Args:
+        conn_params: Connection parameters including url, token, is_jwt, and jwt_tenant_id
+
+    Returns:
+        PromptTemplateWrapper for generate SQL reasoning
+    """
+    prompt_service = PromptService(conn_params=conn_params)
+    return PromptTemplateWrapper(prompt_service, "get_generate_sql_reasoning_template")
+
+
 def get_qa_prompt_template(
     conn_params: Optional[dict] = None
 ) -> PromptTemplateWrapper: