langchain-timbr 2.1.3__py3-none-any.whl → 2.1.4__py3-none-any.whl

langchain_timbr/_version.py

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

langchain_timbr/langchain/execute_timbr_query_chain.py

@@ -169,6 +169,7 @@ class ExecuteTimbrQueryChain(Chain):
             "verify_ssl": self._verify_ssl,
             "is_jwt": self._is_jwt,
             "jwt_tenant_id": self._jwt_tenant_id,
+            "additional_headers": {"results-limit": str(self._max_limit)},
             **self._conn_params,
         }
 
@@ -265,7 +266,8 @@ class ExecuteTimbrQueryChain(Chain):
                 rows = run_query(
                     sql,
                     self._get_conn_params(),
-                    llm_prompt=prompt
+                    llm_prompt=prompt,
+                    use_query_limit=True,
                 ) if is_sql_valid and is_sql_not_tried else []
                 
                 if iteration < self._no_results_max_retries:

langchain_timbr/langchain/generate_answer_chain.py

@@ -23,6 +23,7 @@ class GenerateAnswerChain(Chain):
         is_jwt: Optional[bool] = False,
         jwt_tenant_id: Optional[str] = None,
         conn_params: Optional[dict] = None,
+        note: Optional[str] = '',
         debug: Optional[bool] = False,
         **kwargs,
     ):
@@ -33,6 +34,7 @@ class GenerateAnswerChain(Chain):
         :param verify_ssl: Whether to verify SSL certificates (default is True).
         :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 note: Optional additional note to extend our llm prompt
         :param conn_params: Extra Timbr connection parameters sent with every request (e.g., 'x-api-impersonate-user').
         
         ## Example
@@ -77,6 +79,7 @@ class GenerateAnswerChain(Chain):
         self._jwt_tenant_id = jwt_tenant_id
         self._debug = to_boolean(debug)
         self._conn_params = conn_params or {}
+        self._note = note
 
 
     @property
@@ -116,6 +119,7 @@ class GenerateAnswerChain(Chain):
             conn_params=self._get_conn_params(),
             results=rows,
             sql=sql,
+            note=self._note,
             debug=self._debug,
         )
 

langchain_timbr/langchain/generate_timbr_sql_chain.py

@@ -160,6 +160,7 @@ class GenerateTimbrSqlChain(Chain):
             "verify_ssl": self._verify_ssl,
             "is_jwt": self._is_jwt,
             "jwt_tenant_id": self._jwt_tenant_id,
+            "additional_headers": {"results-limit": str(self._max_limit)},
             **self._conn_params,
         }
 

langchain_timbr/langchain/timbr_sql_agent.py

@@ -126,6 +126,7 @@ class TimbrSqlAgent(BaseSingleActionAgent):
             is_jwt=to_boolean(is_jwt),
             jwt_tenant_id=jwt_tenant_id,
             conn_params=conn_params,
+            note=note,
             debug=to_boolean(debug),
         ) if self._generate_answer else None
 

langchain_timbr/langchain/validate_timbr_sql_chain.py

@@ -161,6 +161,7 @@ class ValidateTimbrSqlChain(Chain):
             "verify_ssl": self._verify_ssl,
             "is_jwt": self._is_jwt,
             "jwt_tenant_id": self._jwt_tenant_id,
+            "additional_headers": {"results-limit": str(self._max_limit)},
             **self._conn_params,
         }
 

langchain_timbr/timbr_llm_connector.py

@@ -115,6 +115,7 @@ class TimbrLlmConnector:
             "verify_ssl": self.verify_ssl,
             "is_jwt": self.is_jwt,
             "jwt_tenant_id": self.jwt_tenant_id,
+            "additional_headers": {"results-limit": str(self.max_limit)},
             **self.conn_params,
         }
 

langchain_timbr/utils/timbr_llm_utils.py

@@ -545,8 +545,9 @@ def answer_question(
     conn_params: dict,
     results: str,
     sql: Optional[str] = None,
-    debug: Optional[bool] = False,
     timeout: Optional[int] = None,
+    note: Optional[str] = '',
+    debug: Optional[bool] = False,
 ) -> dict[str, Any]:
     # Use config default timeout if none provided
     if timeout is None:
@@ -558,6 +559,7 @@ def answer_question(
         question=question,
         formatted_rows=results,
         additional_context=f"SQL QUERY:\n{sql}\n\n" if sql else "",
+        note=note,
     )
     
     apx_token_count = _calculate_token_count(llm, prompt)

langchain_timbr/utils/timbr_utils.py

@@ -70,7 +70,7 @@ def cache_with_version_check(func):
     return wrapper
 
 
-def run_query(sql: str, conn_params: dict, llm_prompt: Optional[str] = None) -> list[list]:
+def run_query(sql: str, conn_params: dict, llm_prompt: Optional[str] = None, use_query_limit = False) -> list[list]:
     if not conn_params:
         raise("Please provide connection params.")
 
@@ -79,9 +79,22 @@ def run_query(sql: str, conn_params: dict, llm_prompt: Optional[str] = None) ->
         clean_prompt = llm_prompt.replace('\r\n', ' ').replace('\n', ' ').replace('?', '')
         query = f"-- LLM: {clean_prompt}\n{sql}"
 
+    query_conn_params = conn_params
+    if not use_query_limit:
+        # Remove results-limit
+        if 'additional_headers' in conn_params and 'results-limit' in conn_params['additional_headers']:
+            query_upper = query.strip().upper()
+            if query_upper.startswith('SHOW') or query_upper.startswith('DESC'):
+                query_conn_params = conn_params.copy()
+                query_conn_params['additional_headers'] = conn_params['additional_headers'].copy()
+                del query_conn_params['additional_headers']['results-limit']
+                # If no other additional_headers remain, delete the key entirely
+                if not query_conn_params['additional_headers']:
+                    del query_conn_params['additional_headers']
+
     results = timbr_http_connector.run_query(
         query=query,
-        **conn_params,
+        **query_conn_params,
     )
 
     return results

{langchain_timbr-2.1.3.dist-info → langchain_timbr-2.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-timbr
-Version: 2.1.3
+Version: 2.1.4
 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.3.dist-info → langchain_timbr-2.1.4.dist-info}/RECORD

@@ -1,14 +1,14 @@
 langchain_timbr/__init__.py,sha256=gxd6Y6QDmYZtPlYVdXtPIy501hMOZXHjWh2qq4qzt_s,828
-langchain_timbr/_version.py,sha256=3XX6kpIyOtIVpdYtEgDoTVSW2L9cEQyXaBXhL9YIdM0,704
+langchain_timbr/_version.py,sha256=TYs-mU9m9aJFweRpjovqttRQsYRnpbbHHMifXX9ryi4,704
 langchain_timbr/config.py,sha256=PEtvNgvnA9UseZJjKgup_O6xdG-VYk3N11nH8p8W1Kg,1410
-langchain_timbr/timbr_llm_connector.py,sha256=1jDicBZkW7CKB-PvQiQ1_AMXYm9JJHaoNaPqy54nhh8,13096
+langchain_timbr/timbr_llm_connector.py,sha256=mdkWskpvmXZre5AzVFn6KfPnVH5YN5MIwfEoXWBLMgY,13170
 langchain_timbr/langchain/__init__.py,sha256=ejcsZKP9PK0j4WrrCCcvBXpDpP-TeRiVb21OIUJqix8,580
-langchain_timbr/langchain/execute_timbr_query_chain.py,sha256=g7VzCuywo7xlWIldjtGOV4dYsQB3KsO8VARCfKrZFHM,15461
-langchain_timbr/langchain/generate_answer_chain.py,sha256=euy8GcMOziMKpiJrmpiH-De-qKTaVV48TqTduDa5-PE,4711
-langchain_timbr/langchain/generate_timbr_sql_chain.py,sha256=6NPvidW8pwmaX3w7p7yNFwQTZ8Cnmb40ZznRQJ2dTFM,9023
+langchain_timbr/langchain/execute_timbr_query_chain.py,sha256=pedMajyKDI2ZaoyVp1r64nHX015Wy-r96HoJrRlCh48,15579
+langchain_timbr/langchain/generate_answer_chain.py,sha256=XsaQrgBFwoC9ne3jpnuHueUXL1PzNQ75ECC_HVA61Ks,4871
+langchain_timbr/langchain/generate_timbr_sql_chain.py,sha256=3Z0ut78AFCNHKwLwOYH44hzJDIOA-zNF0x8Tjyrvzp4,9098
 langchain_timbr/langchain/identify_concept_chain.py,sha256=kuzg0jJQpFGIiaxtNhdQ5K4HXveLVwONFNsoipPCteE,7169
-langchain_timbr/langchain/timbr_sql_agent.py,sha256=7S7USe3wnwxdBduBycHYVbXnywnvog-Bg67Tk8-2X_s,19551
-langchain_timbr/langchain/validate_timbr_sql_chain.py,sha256=sxT72asK-Cms4ZTiThWv58PgEBY1p_9hnUQwU35kPIQ,9492
+langchain_timbr/langchain/timbr_sql_agent.py,sha256=HntpalzCZ-PlHd7na5V0syCMqrREFUpppGM4eHstaZQ,19574
+langchain_timbr/langchain/validate_timbr_sql_chain.py,sha256=OcE_7yfb9xpD-I4OS7RG1bY4-yi1UicjvGegOv_vkQU,9567
 langchain_timbr/langgraph/__init__.py,sha256=mKBFd0x01jWpRujUWe-suX3FFhenPoDxrvzs8I0mum0,457
 langchain_timbr/langgraph/execute_timbr_query_node.py,sha256=rPx_V3OOh-JTGOwrEopHmOmFuM-ngBZdswkW9oZ43hU,5536
 langchain_timbr/langgraph/generate_response_node.py,sha256=BLmsDZfbhncRpO7PEfDpy7CnPE7a55j8QV5jfg30heQ,2247
@@ -20,9 +20,9 @@ langchain_timbr/llm_wrapper/timbr_llm_wrapper.py,sha256=sDqDOz0qu8b4WWlagjNceswM
 langchain_timbr/utils/general.py,sha256=KkehHvIj8GoQ_0KVXLcUVeaYaTtkuzgXmYYx2TXJhI4,10253
 langchain_timbr/utils/prompt_service.py,sha256=QT7kiq72rQno77z1-tvGGD7HlH_wdTQAl_1teSoKEv4,11373
 langchain_timbr/utils/temperature_supported_models.json,sha256=d3UmBUpG38zDjjB42IoGpHTUaf0pHMBRSPY99ao1a3g,1832
-langchain_timbr/utils/timbr_llm_utils.py,sha256=c50-dEDh8GnAkDfcbNKdXuOWjbSa-mXpZzDVuNt4jTY,23053
-langchain_timbr/utils/timbr_utils.py,sha256=p21DwTGhF4iKTLDQBkeBaJDFcXt-Hpu1ij8xzQt00Ng,16958
-langchain_timbr-2.1.3.dist-info/METADATA,sha256=3etBzLd5lLYeXINYMxJwu6daB9feBxW1MReZaG_I6fU,12268
-langchain_timbr-2.1.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langchain_timbr-2.1.3.dist-info/licenses/LICENSE,sha256=0ITGFk2alkC7-e--bRGtuzDrv62USIiVyV2Crf3_L_0,1065
-langchain_timbr-2.1.3.dist-info/RECORD,,
+langchain_timbr/utils/timbr_llm_utils.py,sha256=_4Qz5SX5cXW1Rl_fSBcE9P3uPEaI8DBg3GpXA4uQGoI,23102
+langchain_timbr/utils/timbr_utils.py,sha256=SvmQ0wYicODNhmo8c-5_KPDBAfrBVBkUfoO8sPItQhk,17759
+langchain_timbr-2.1.4.dist-info/METADATA,sha256=Wtzy14BfbEujvuYozb4XcaUz6pp87zxjNNBbt9Z5wSQ,12268
+langchain_timbr-2.1.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langchain_timbr-2.1.4.dist-info/licenses/LICENSE,sha256=0ITGFk2alkC7-e--bRGtuzDrv62USIiVyV2Crf3_L_0,1065
+langchain_timbr-2.1.4.dist-info/RECORD,,