langchain-timbr 2.0.4__tar.gz → 2.1.1__tar.gz

{langchain_timbr-2.0.4 → langchain_timbr-2.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-timbr
-Version: 2.0.4
+Version: 2.1.1
 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/
@@ -45,12 +45,10 @@ 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: numpy==1.26.4; python_version >= '3.11'
-Requires-Dist: numpy~=1.26.2; 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: pyarrow<20.0.0,>=19.0.1; python_version == '3.10'
-Requires-Dist: pyarrow==19.0.1; python_version >= '3.11'
+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'
@@ -82,8 +80,6 @@ Requires-Dist: langchain-tests==0.3.22; (python_version >= '3.11') and extra ==
 Requires-Dist: langchain-tests>=0.3.20; (python_version == '3.10') and extra == 'all'
 Requires-Dist: openai==2.1.0; (python_version >= '3.11') and extra == 'all'
 Requires-Dist: openai>=1.77.0; (python_version == '3.10') and extra == 'all'
-Requires-Dist: pyarrow<20.0.0,>=19.0.1; (python_version == '3.10') and extra == 'all'
-Requires-Dist: pyarrow==19.0.1; (python_version >= '3.11') and extra == 'all'
 Requires-Dist: pytest==8.3.4; extra == 'all'
 Requires-Dist: snowflake-snowpark-python==1.39.1; (python_version >= '3.11') and extra == 'all'
 Requires-Dist: snowflake-snowpark-python>=1.39.1; (python_version == '3.10') and extra == 'all'
@@ -107,8 +103,6 @@ Requires-Dist: databricks-sdk==0.64.0; extra == 'databricks'
 Provides-Extra: dev
 Requires-Dist: langchain-tests==0.3.22; (python_version >= '3.11') and extra == 'dev'
 Requires-Dist: langchain-tests>=0.3.20; (python_version == '3.10') and extra == 'dev'
-Requires-Dist: pyarrow<20.0.0,>=19.0.1; (python_version == '3.10') and extra == 'dev'
-Requires-Dist: pyarrow==19.0.1; (python_version >= '3.11') and extra == 'dev'
 Requires-Dist: pytest==8.3.4; extra == 'dev'
 Requires-Dist: uvicorn==0.34.0; extra == 'dev'
 Provides-Extra: google
@@ -121,6 +115,8 @@ Requires-Dist: langchain-openai>=0.3.16; (python_version == '3.10') and extra ==
 Requires-Dist: openai==2.1.0; (python_version >= '3.11') and extra == 'openai'
 Requires-Dist: openai>=1.77.0; (python_version == '3.10') and extra == 'openai'
 Provides-Extra: snowflake
+Requires-Dist: opentelemetry-api==1.38.0; (python_version < '3.12') and extra == 'snowflake'
+Requires-Dist: opentelemetry-sdk==1.38.0; (python_version < '3.12') and extra == 'snowflake'
 Requires-Dist: snowflake-snowpark-python==1.39.1; (python_version >= '3.11') and extra == 'snowflake'
 Requires-Dist: snowflake-snowpark-python>=1.39.1; (python_version == '3.10') and extra == 'snowflake'
 Requires-Dist: snowflake==1.8.0; (python_version >= '3.11') and extra == 'snowflake'

{langchain_timbr-2.0.4 → langchain_timbr-2.1.1}/pyproject.toml

@@ -45,9 +45,9 @@ dependencies = [
   "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'",
-  "numpy==1.26.4; python_version>='3.11'",
   "openai==2.1.0; python_version>='3.11'",
-  "pyarrow==19.0.1; 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'",
@@ -65,9 +65,9 @@ dependencies = [
   "langchain-openai>=0.3.16; python_version=='3.10'",
   "langchain-tests>=0.3.20; python_version=='3.10'",
   "langgraph>=0.3.20; python_version=='3.10'",
-  "numpy~=1.26.2; python_version=='3.10'",
   "openai>=1.77.0; python_version=='3.10'",
-  "pyarrow>=19.0.1,<20.0.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'",
@@ -106,6 +106,8 @@ vertex_ai = [
   "langchain-google-vertexai>=2.0.28; python_version=='3.10'"
 ]
 snowflake = [
+  "opentelemetry-api==1.38.0; python_version<'3.12'",
+  "opentelemetry-sdk==1.38.0; python_version<'3.12'",
   "snowflake==1.8.0; python_version>='3.11'", 
   "snowflake-snowpark-python==1.39.1; python_version>='3.11'",
   "snowflake>=1.8.0; python_version=='3.10'", 
@@ -120,10 +122,8 @@ databricks = [
 dev = [
   "pytest==8.3.4",
   "langchain-tests==0.3.22; python_version>='3.11'",
-  "pyarrow==19.0.1; python_version>='3.11'",
   "uvicorn==0.34.0",
   "langchain-tests>=0.3.20; python_version=='3.10'",
-  "pyarrow>=19.0.1,<20.0.0; python_version=='3.10'"
 ]
 
 # All optional dependencies
@@ -144,7 +144,6 @@ all = [
   "snowflake==1.8.0; python_version>='3.11'",
   "snowflake-snowpark-python==1.39.1; python_version>='3.11'",
   "langchain-tests==0.3.22; python_version>='3.11'",
-  "pyarrow==19.0.1; python_version>='3.11'",
   # Python 3.10 versions
   "azure-identity>=1.16.1; python_version=='3.10'",
   "langchain-anthropic>=0.3.1; python_version=='3.10'",
@@ -155,7 +154,6 @@ all = [
   "snowflake>=1.8.0; python_version=='3.10'",
   "snowflake-snowpark-python>=1.39.1; python_version=='3.10'",
   "langchain-tests>=0.3.20; python_version=='3.10'",
-  "pyarrow>=19.0.1,<20.0.0; python_version=='3.10'"
 ]
 
 [project.urls]

{langchain_timbr-2.0.4 → langchain_timbr-2.1.1}/requirements.txt

@@ -13,9 +13,7 @@ langchain-google-vertexai==2.1.2
 langchain-openai==0.3.34
 langchain-tests==0.3.22
 langgraph==0.6.8
-numpy==1.26.4
 openai==2.1.0
-pyarrow==19.0.1
 pydantic==2.10.4
 pytest==8.3.4
 pytimbr_api==2.0.0

{langchain_timbr-2.0.4 → langchain_timbr-2.1.1}/requirements310.txt

@@ -13,9 +13,9 @@ langchain-google-vertexai>=2.0.28
 langchain-openai>=0.3.16
 langchain-tests>=0.3.20
 langgraph>=0.3.20
-numpy~=1.26.2
 openai>=1.77.0
-pyarrow>=19.0.1,<20.0.0
+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

{langchain_timbr-2.0.4 → langchain_timbr-2.1.1}/requirements311.txt

@@ -13,9 +13,9 @@ langchain-google-vertexai==2.1.2
 langchain-openai==0.3.34
 langchain-tests==0.3.22
 langgraph==0.6.8
-numpy==1.26.4
 openai==2.1.0
-pyarrow==19.0.1
+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

{langchain_timbr-2.0.4 → langchain_timbr-2.1.1}/src/langchain_timbr/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '2.0.4'
-__version_tuple__ = version_tuple = (2, 0, 4)
+__version__ = version = '2.1.1'
+__version_tuple__ = version_tuple = (2, 1, 1)
 
 __commit_id__ = commit_id = None

{langchain_timbr-2.0.4 → langchain_timbr-2.1.1}/src/langchain_timbr/utils/timbr_llm_utils.py

@@ -180,6 +180,10 @@ def _get_response_text(response: Any) -> str:
     else:
         raise ValueError("Unexpected response format from LLM.")
 
+    if "QUESTION VALIDATION ERROR:" in response_text:
+        err = response_text.split("QUESTION VALIDATION ERROR:", 1)[1].strip()
+        raise ValueError(err)
+
     return response_text
 
 def _extract_usage_metadata(response: Any) -> dict:
@@ -282,7 +286,7 @@ def determine_concept(
                 response = _call_llm_with_timeout(llm, prompt, timeout=timeout)
             except TimeoutError as e:
                 error = f"LLM call timed out: {str(e)}"
-                continue
+                raise Exception(error)
             except Exception as e:
                 error = f"LLM call failed: {str(e)}"
                 continue
@@ -505,10 +509,7 @@ def generate_sql(
             response = _call_llm_with_timeout(llm, prompt, timeout=timeout)
         except TimeoutError as e:
             error = f"LLM call timed out: {str(e)}"
-            if should_validate_sql:
-                continue
-            else:
-                raise Exception(error)
+            raise Exception(error)
         except Exception as e:
             error = f"LLM call failed: {str(e)}"
             if should_validate_sql:

{langchain_timbr-2.0.4 → langchain_timbr-2.1.1}/tests/standard/test_chain_documentation.py

@@ -220,7 +220,7 @@ class TestChainExamples:
                 chain = ExecuteTimbrQueryChain(**config_dict)
                 
                 # Test basic functionality
-                result = chain.invoke({"prompt": f"test configuration {i}"})
+                result = chain.invoke({"prompt": "Count all orders"})
                 assert isinstance(result, dict)
                 
             except Exception as e: