langchain-timbr 2.0.2__py3-none-any.whl → 2.0.3__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.0.2'
-__version_tuple__ = version_tuple = (2, 0, 2)
+__version__ = version = '2.0.3'
+__version_tuple__ = version_tuple = (2, 0, 3)
 
 __commit_id__ = commit_id = None

langchain_timbr/llm_wrapper/llm_wrapper.py

@@ -14,6 +14,7 @@ class LlmTypes(Enum):
   AzureOpenAI = 'azure-openai-chat'
   Snowflake = 'snowflake-cortex'
   Databricks = 'chat-databricks'
+  VertexAI = 'chat-vertexai'
   Timbr = 'timbr'
 
 
@@ -54,10 +55,7 @@ class LlmWrapper(LLM):
       # Validation: Ensure we have the required parameters
       if not selected_llm_type:
           raise ValueError("llm_type must be provided either as parameter or in config (LLM_TYPE environment variable)")
-      
-      if not selected_api_key:
-          raise ValueError("api_key must be provided either as parameter or in config (LLM_API_KEY environment variable)")
-      
+            
       self.client = self._connect_to_llm(
         selected_llm_type,
         selected_api_key,
@@ -80,8 +78,75 @@ class LlmWrapper(LLM):
         llm_params["temperature"] = config.llm_temperature
     return llm_params
 
-  
-  def _connect_to_llm(self, llm_type, api_key, model, **llm_params):
+  def _try_build_vertexai_credentials(self,params, api_key):
+    from google.oauth2 import service_account
+    from google.auth import default
+
+    # Try multiple authentication methods in order of preference
+    creds = None
+    scope = pop_param_value(params, ['vertex_scope', 'llm_scope', 'scope'], default=config.llm_scope)
+    scopes = [scope] if scope else ["https://www.googleapis.com/auth/cloud-platform"]
+    
+    # Method 1: Service Account File (json_path)
+    json_path = pop_param_value(params, ['azure_json_path', 'llm_json_path', 'json_path'])
+    if json_path:
+      try:
+        creds = service_account.Credentials.from_service_account_file(
+          json_path,
+          scopes=scopes,
+        )
+      except Exception as e:
+        raise ValueError(f"Failed to load service account from file '{json_path}': {e}")
+    
+    # Method 2: Service Account Info (as dictionary)
+    if not creds:
+      service_account_info = pop_param_value(params, ['service_account_info', 'vertex_service_account_info'])
+      if service_account_info:
+        try:
+          creds = service_account.Credentials.from_service_account_info(
+            service_account_info,
+            scopes=scopes,
+          )
+        except Exception as e:
+          raise ValueError(f"Failed to load service account from info: {e}")
+    
+    # Method 3: Service Account Email + Private Key
+    if not creds:
+      service_account_email = pop_param_value(params, ['service_account_email', 'vertex_email', 'service_account'])
+      private_key = pop_param_value(params, ['private_key', 'vertex_private_key']) or api_key
+      
+      if service_account_email and private_key:
+        try:
+          service_account_info = {
+            "type": "service_account",
+            "client_email": service_account_email,
+            "private_key": private_key,
+            "token_uri": "https://oauth2.googleapis.com/token",
+          }
+          creds = service_account.Credentials.from_service_account_info(
+            service_account_info,
+            scopes=scopes,
+          )
+        except Exception as e:
+          raise ValueError(f"Failed to create service account from email and private key: {e}")
+    
+    # Method 4: Default Google Cloud Credentials (fallback)
+    if not creds:
+      try:
+        creds, _ = default(scopes=scopes)
+      except Exception as e:
+        raise ValueError(
+          "VertexAI authentication failed. Please provide one of:\n"
+          "1. 'json_path' - path to service account JSON file\n"
+          "2. 'service_account_info' - service account info as dictionary\n"
+          "3. 'service_account_email' + 'private_key' - service account credentials\n"
+          "4. Set up default Google Cloud credentials (gcloud auth application-default login)\n"
+          f"Error: {e}"
+        )
+
+    return creds  
+
+  def _connect_to_llm(self, llm_type, api_key = None, model = None, **llm_params):
     if is_llm_type(llm_type, LlmTypes.OpenAI):
       from langchain_openai import ChatOpenAI as OpenAI
       llm_model = model or "gpt-4o-2024-11-20"
@@ -171,6 +236,21 @@ class LlmWrapper(LLM):
         endpoint=llm_model,
         workspace_client=w,  # Using authenticated client
         **params,
+      )    
+    elif is_llm_type(llm_type, LlmTypes.VertexAI):
+      from langchain_google_vertexai import ChatVertexAI
+      llm_model = model or "gemini-2.5-flash-lite"
+      params = self._add_temperature(LlmTypes.VertexAI.name, llm_model, **llm_params)
+
+      project = pop_param_value(params, ['vertex_project', 'llm_project', 'project'])
+      if project:
+        params['project'] = project
+
+      creds = self._try_build_vertexai_credentials(params, api_key)
+      return ChatVertexAI(
+        model_name=llm_model,
+        credentials=creds,
+        **params,
       )
     else:
       raise ValueError(f"Unsupported LLM type: {llm_type}")
@@ -179,6 +259,8 @@ class LlmWrapper(LLM):
   def get_model_list(self) -> list[str]:
     """Return the list of available models for the LLM."""
     models = []
+    llm_type_name = None
+
     try:
       if is_llm_type(self._llm_type, LlmTypes.OpenAI):
         from openai import OpenAI
@@ -237,10 +319,14 @@ class LlmWrapper(LLM):
           models = [ep.name for ep in w.serving_endpoints.list()]
 
       # elif self._is_llm_type(self._llm_type, LlmTypes.Timbr):
-        
+      elif is_llm_type(self._llm_type, LlmTypes.VertexAI):
+        from google import genai
+        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()]
+
     except Exception:
       # If model list fetching throws an exception, return default value using get_supported_models
-      llm_type_name = None
       if hasattr(self, '_llm_type'):
         # Try to extract the LLM type name from the _llm_type
         for llm_enum in LlmTypes:

{langchain_timbr-2.0.2.dist-info → langchain_timbr-2.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-timbr
-Version: 2.0.2
+Version: 2.0.3
 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/
@@ -31,17 +31,18 @@ Requires-Dist: tiktoken==0.8.0
 Requires-Dist: transformers>=4.53
 Provides-Extra: all
 Requires-Dist: anthropic==0.42.0; extra == 'all'
-Requires-Dist: azure-identity==1.16.1; extra == 'all'
+Requires-Dist: azure-identity>=1.16.1; extra == 'all'
 Requires-Dist: databricks-langchain==0.3.0; (python_version < '3.10') and extra == 'all'
 Requires-Dist: databricks-langchain==0.7.1; (python_version >= '3.10') and extra == 'all'
 Requires-Dist: databricks-sdk==0.64.0; extra == 'all'
 Requires-Dist: google-generativeai==0.8.4; extra == 'all'
 Requires-Dist: langchain-anthropic>=0.3.1; extra == 'all'
 Requires-Dist: langchain-google-genai>=2.0.9; extra == 'all'
+Requires-Dist: langchain-google-vertexai>=2.0.28; extra == 'all'
 Requires-Dist: langchain-openai>=0.3.16; extra == 'all'
 Requires-Dist: langchain-tests>=0.3.20; extra == 'all'
 Requires-Dist: openai>=1.77.0; extra == 'all'
-Requires-Dist: pyarrow<19.0.0; extra == 'all'
+Requires-Dist: pyarrow<20.0.0,>=19.0.1; extra == 'all'
 Requires-Dist: pytest==8.3.4; extra == 'all'
 Requires-Dist: snowflake-snowpark-python>=1.6.0; extra == 'all'
 Requires-Dist: snowflake>=0.8.0; extra == 'all'
@@ -50,7 +51,7 @@ Provides-Extra: anthropic
 Requires-Dist: anthropic==0.42.0; extra == 'anthropic'
 Requires-Dist: langchain-anthropic>=0.3.1; extra == 'anthropic'
 Provides-Extra: azure-openai
-Requires-Dist: azure-identity==1.16.1; extra == 'azure-openai'
+Requires-Dist: azure-identity>=1.16.1; extra == 'azure-openai'
 Requires-Dist: langchain-openai>=0.3.16; extra == 'azure-openai'
 Requires-Dist: openai>=1.77.0; extra == 'azure-openai'
 Provides-Extra: databricks
@@ -59,7 +60,7 @@ Requires-Dist: databricks-langchain==0.7.1; (python_version >= '3.10') and extra
 Requires-Dist: databricks-sdk==0.64.0; extra == 'databricks'
 Provides-Extra: dev
 Requires-Dist: langchain-tests>=0.3.20; extra == 'dev'
-Requires-Dist: pyarrow<19.0.0; extra == 'dev'
+Requires-Dist: pyarrow<20.0.0,>=19.0.1; extra == 'dev'
 Requires-Dist: pytest==8.3.4; extra == 'dev'
 Requires-Dist: uvicorn==0.34.0; extra == 'dev'
 Provides-Extra: google
@@ -71,6 +72,9 @@ Requires-Dist: openai>=1.77.0; extra == 'openai'
 Provides-Extra: snowflake
 Requires-Dist: snowflake-snowpark-python>=1.6.0; extra == 'snowflake'
 Requires-Dist: snowflake>=0.8.0; extra == 'snowflake'
+Provides-Extra: vertex-ai
+Requires-Dist: google-generativeai==0.8.4; extra == 'vertex-ai'
+Requires-Dist: langchain-google-vertexai>=2.0.28; extra == 'vertex-ai'
 Description-Content-Type: text/markdown
 
 ![Timbr logo description](https://timbr.ai/wp-content/uploads/2025/01/logotimbrai230125.png)
@@ -89,8 +93,8 @@ Timbr LangChain LLM SDK is a Python SDK that extends LangChain and LangGraph wit
 
 ![Timbr LangGraph pipeline](https://docs.timbr.ai/doc/assets/images/timbr-langgraph-fcf8e2eb7e26dc9dfa8b56b62937281e.png)
 
-
 ## Dependencies
+
 - Access to a timbr-server
 - Python 3.9.13 or newer
 
@@ -103,13 +107,15 @@ python -m pip install langchain-timbr
 ```
 
 ### Install with selected LLM providers
-#### One of: openai, anthropic, google, azure_openai, snowflake, databricks (or 'all')
+
+#### One of: openai, anthropic, google, azure_openai, snowflake, databricks, vertex_ai (or 'all')
 
 ```bash
 python -m pip install 'langchain-timbr[<your selected providers, separated by comma w/o space>]'
 ```
 
 ### Using pip from github
+
 ```bash
 pip install git+https://github.com/WPSemantix/langchain-timbr
 ```

{langchain_timbr-2.0.2.dist-info → langchain_timbr-2.0.3.dist-info}/RECORD

@@ -1,5 +1,5 @@
 langchain_timbr/__init__.py,sha256=gxd6Y6QDmYZtPlYVdXtPIy501hMOZXHjWh2qq4qzt_s,828
-langchain_timbr/_version.py,sha256=UFd-y4dR-HVHWzUJlgmJ3T05hh9C4Q-la-D9QTIoZw0,704
+langchain_timbr/_version.py,sha256=piVud-sBtnpf1MQjPI4VkcaUcsjojJEmPPb85g-VsVw,704
 langchain_timbr/config.py,sha256=PEtvNgvnA9UseZJjKgup_O6xdG-VYk3N11nH8p8W1Kg,1410
 langchain_timbr/timbr_llm_connector.py,sha256=1jDicBZkW7CKB-PvQiQ1_AMXYm9JJHaoNaPqy54nhh8,13096
 langchain_timbr/langchain/__init__.py,sha256=ejcsZKP9PK0j4WrrCCcvBXpDpP-TeRiVb21OIUJqix8,580
@@ -15,14 +15,14 @@ langchain_timbr/langgraph/generate_response_node.py,sha256=BLmsDZfbhncRpO7PEfDpy
 langchain_timbr/langgraph/generate_timbr_sql_node.py,sha256=wkau-NajblSVzNIro9IyqawULvz3XaCYSEdYW95vWco,4911
 langchain_timbr/langgraph/identify_concept_node.py,sha256=aiLDFEcz_vM4zZ_ULe1SvJKmI-e4Fb2SibZQaEPz_eY,3649
 langchain_timbr/langgraph/validate_timbr_query_node.py,sha256=-2fuieCz1hv6ua-17zfonme8LQ_OoPnoOBTdGSXkJgs,4793
-langchain_timbr/llm_wrapper/llm_wrapper.py,sha256=QoTKx_VKd2ToglX53Z49e71xoJUEH6-pRBVOzjURxCo,11320
+langchain_timbr/llm_wrapper/llm_wrapper.py,sha256=xLBTy2E6mTDny5tSSxIeeMUffos1r3a2BkBAjn3zxO4,14935
 langchain_timbr/llm_wrapper/timbr_llm_wrapper.py,sha256=sDqDOz0qu8b4WWlagjNceswMVyvEJ8yBWZq2etBh-T0,1362
 langchain_timbr/utils/general.py,sha256=MtY-ZExKJrcBzV3EQNn6G1ESKpiQB2hJCp95BrUayUo,5707
 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=7-nnTa1T9XOcgIb-aJP3Pgon_gOrCMnDPiIPiAT3UCg,23016
 langchain_timbr/utils/timbr_utils.py,sha256=p21DwTGhF4iKTLDQBkeBaJDFcXt-Hpu1ij8xzQt00Ng,16958
-langchain_timbr-2.0.2.dist-info/METADATA,sha256=VtT8woei5BJOA6rvMtpJublCk7nwDiQ4j0gfCww7n90,7771
-langchain_timbr-2.0.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langchain_timbr-2.0.2.dist-info/licenses/LICENSE,sha256=0ITGFk2alkC7-e--bRGtuzDrv62USIiVyV2Crf3_L_0,1065
-langchain_timbr-2.0.2.dist-info/RECORD,,
+langchain_timbr-2.0.3.dist-info/METADATA,sha256=l50ahPxMk0CgSVwdLEpQh-cs--hlyqj7nR2jvG2KOy4,8028
+langchain_timbr-2.0.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langchain_timbr-2.0.3.dist-info/licenses/LICENSE,sha256=0ITGFk2alkC7-e--bRGtuzDrv62USIiVyV2Crf3_L_0,1065
+langchain_timbr-2.0.3.dist-info/RECORD,,