langchain-timbr 2.1.14__py3-none-any.whl → 3.0.0__py3-none-any.whl

langchain_timbr/langchain/validate_timbr_sql_chain.py

@@ -2,6 +2,8 @@ from typing import Optional, Union, Dict, Any
 from langchain.chains.base import Chain
 from langchain.llms.base import LLM
 
+from langchain_timbr.utils.timbr_utils import get_timbr_agent_options
+
 from ..utils.general import parse_list, to_integer, to_boolean, validate_timbr_connection_params
 from ..utils.timbr_llm_utils import generate_sql
 from ..utils.timbr_utils import validate_sql
@@ -17,6 +19,8 @@ class ValidateTimbrSqlChain(Chain):
     compatible with the target Timbr ontology/knowledge graph structure. It uses an LLM
     for validation and connects to Timbr via URL and token.
     """
+
+    _ontology: Optional[str] = None
     
     def __init__(
         self,
@@ -36,6 +40,7 @@ class ValidateTimbrSqlChain(Chain):
         note: Optional[str] = '',
         db_is_case_sensitive: Optional[bool] = False,
         graph_depth: Optional[int] = 1,
+        agent: Optional[str] = None,
         verify_ssl: Optional[bool] = True,
         is_jwt: Optional[bool] = False,
         jwt_tenant_id: Optional[str] = None,
@@ -62,6 +67,7 @@ class ValidateTimbrSqlChain(Chain):
         :param note: Optional additional note to extend our llm prompt
         :param db_is_case_sensitive: Whether the database is case sensitive (default is False).
         :param graph_depth: Maximum number of relationship hops to traverse from the source concept during schema exploration (default is 1).
+        :param agent: Optional Timbr agent name for options setup.
         :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).
@@ -111,30 +117,58 @@ class ValidateTimbrSqlChain(Chain):
         
         self._url = url if url is not None else config.url
         self._token = token if token is not None else config.token
-        self._ontology = ontology if ontology is not None else config.ontology
         
         # Validate required parameters
         validate_timbr_connection_params(self._url, self._token)
         
-        self._schema = schema
-        self._concept = concept
-        self._retries = retries
-        self._concepts_list = parse_list(concepts_list)
-        self._views_list = parse_list(views_list)
-        self._include_logic_concepts = to_boolean(include_logic_concepts)
-        self._include_tags = parse_list(include_tags)
-        self._exclude_properties = parse_list(exclude_properties)
-        self._max_limit = to_integer(max_limit)
-        self._note = note
-        self._db_is_case_sensitive = to_boolean(db_is_case_sensitive)
-        self._graph_depth = to_integer(graph_depth)
         self._verify_ssl = to_boolean(verify_ssl)
         self._is_jwt = to_boolean(is_jwt)
         self._jwt_tenant_id = jwt_tenant_id
-        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)
+        self._conn_params = conn_params or {}
+        self._max_limit = config.llm_default_limit # use default value so the self._get_conn_params() won't fail before agent options are processed
+
+        self._agent = agent
+        if self._agent:
+            agent_options = get_timbr_agent_options(self._agent, conn_params=self._get_conn_params())
+
+            self._ontology = agent_options.get("ontology") if "ontology" in agent_options else None
+            self._schema = agent_options.get("schema") if "schema" in agent_options else schema
+            self._concept = agent_options.get("concept") if "concept" in agent_options else None
+            self._retries = to_integer(agent_options.get("retries") if "retries" in agent_options else retries)
+            self._concepts_list = parse_list(agent_options.get("concepts_list")) if "concepts_list" in agent_options else None
+            self._views_list = parse_list(agent_options.get("views_list")) if "views_list" in agent_options else None
+            self._include_logic_concepts = to_boolean(agent_options.get("include_logic_concepts")) if "include_logic_concepts" in agent_options else False
+            self._include_tags = parse_list(agent_options.get("include_tags")) if "include_tags" in agent_options else None
+            self._exclude_properties = parse_list(agent_options.get("exclude_properties")) if "exclude_properties" in agent_options else ['entity_id', 'entity_type', 'entity_label']
+            self._max_limit = to_integer(agent_options.get("max_limit")) if "max_limit" in agent_options else config.llm_default_limit
+            self._db_is_case_sensitive = to_boolean(agent_options.get("db_is_case_sensitive")) if "db_is_case_sensitive" in agent_options else False
+            self._graph_depth = to_integer(agent_options.get("graph_depth")) if "graph_depth" in agent_options else 1
+            self._note = agent_options.get("note") if "note" in agent_options else ''
+            if note:
+                self._note = ((self._note + '\n') if self._note else '') + note
+            self._enable_reasoning = to_boolean(agent_options.get("enable_reasoning")) if "enable_reasoning" in agent_options else config.enable_reasoning
+            if enable_reasoning is not None and enable_reasoning != self._enable_reasoning:
+                self._enable_reasoning = to_boolean(enable_reasoning)
+            self._reasoning_steps = to_integer(agent_options.get("reasoning_steps")) if "reasoning_steps" in agent_options else config.reasoning_steps
+            if reasoning_steps is not None and reasoning_steps != self._reasoning_steps:
+                self._reasoning_steps = to_integer(reasoning_steps)
+        else:
+            self._ontology = ontology if ontology is not None else config.ontology
+            self._schema = schema
+            self._concept = concept
+            self._retries = to_integer(retries)
+            self._concepts_list = parse_list(concepts_list)
+            self._views_list = parse_list(views_list)
+            self._include_logic_concepts = to_boolean(include_logic_concepts)
+            self._include_tags = parse_list(include_tags)
+            self._exclude_properties = parse_list(exclude_properties)
+            self._max_limit = to_integer(max_limit)
+            self._db_is_case_sensitive = to_boolean(db_is_case_sensitive)
+            self._graph_depth = to_integer(graph_depth)
+            self._note = note
+            self._enable_reasoning = to_boolean(enable_reasoning)
+            self._reasoning_steps = to_integer(reasoning_steps)
 
 
     @property
@@ -163,7 +197,7 @@ class ValidateTimbrSqlChain(Chain):
         return {
             "url": self._url,
             "token": self._token,
-            "ontology": self._ontology,
+            "ontology": self._ontology if self._ontology is not None else config.ontology,
             "verify_ssl": self._verify_ssl,
             "is_jwt": self._is_jwt,
             "jwt_tenant_id": self._jwt_tenant_id,
@@ -179,6 +213,8 @@ class ValidateTimbrSqlChain(Chain):
         schema = self._schema
         concept = self._concept
         reasoning_status = None
+        identify_concept_reason = None
+        generate_sql_reason = None
 
         is_sql_valid, error, sql = validate_sql(sql, self._get_conn_params())
         if not is_sql_valid:
@@ -211,6 +247,8 @@ class ValidateTimbrSqlChain(Chain):
             is_sql_valid = generate_res.get("is_sql_valid")
             reasoning_status = generate_res.get("reasoning_status")
             error = generate_res.get("error")
+            identify_concept_reason = generate_res.get("identify_concept_reason")
+            generate_sql_reason = generate_res.get("generate_sql_reason")
 
         return {
             "sql": sql,
@@ -219,5 +257,7 @@ class ValidateTimbrSqlChain(Chain):
             "is_sql_valid": is_sql_valid,
             "error": error,
             "reasoning_status": reasoning_status,
+            "identify_concept_reason": identify_concept_reason,
+            "generate_sql_reason": generate_sql_reason,
             self.usage_metadata_key: usage_metadata,
         }

langchain_timbr/langgraph/execute_timbr_query_node.py

@@ -32,6 +32,7 @@ class ExecuteSemanticQueryNode:
         note: Optional[str] = '',
         db_is_case_sensitive: Optional[bool] = False,
         graph_depth: Optional[int] = 1,
+        agent: Optional[str] = None,
         verify_ssl: Optional[bool] = True,
         is_jwt: Optional[bool] = False,
         jwt_tenant_id: Optional[str] = None,
@@ -61,6 +62,7 @@ class ExecuteSemanticQueryNode:
         :param note: Optional additional note to extend our llm prompt
         :param db_is_case_sensitive: Whether the database is case sensitive (default is False).
         :param graph_depth: Maximum number of relationship hops to traverse from the source concept during schema exploration (default is 1).
+        :param agent: Optional Timbr agent name for options setup.
         :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).
@@ -89,6 +91,7 @@ class ExecuteSemanticQueryNode:
             note=note,
             db_is_case_sensitive=db_is_case_sensitive,
             graph_depth=graph_depth,
+            agent=agent,
             verify_ssl=verify_ssl,
             is_jwt=is_jwt,
             jwt_tenant_id=jwt_tenant_id,

langchain_timbr/langgraph/generate_response_node.py

@@ -16,11 +16,12 @@ class GenerateResponseNode:
         llm: Optional[LLM] = None,
         url: Optional[str] = None,
         token: Optional[str] = None,
+        note: Optional[str] = '',
+        agent: Optional[str] = None,
         verify_ssl: Optional[bool] = True,
         is_jwt: Optional[bool] = False,
         jwt_tenant_id: Optional[str] = None,
         conn_params: Optional[dict] = None,
-        note: Optional[str] = '',
         debug: Optional[bool] = False,
         **kwargs,
     ):
@@ -28,11 +29,12 @@ class GenerateResponseNode:
         :param llm: An LLM instance or a function that takes a prompt string and returns the LLM's response (optional, will use LlmWrapper with env variables if not provided)
         :param url: Timbr server url (optional, defaults to TIMBR_URL environment variable)
         :param token: Timbr password or token value (optional, defaults to TIMBR_TOKEN environment variable)
+        :param note: Optional additional note to extend our llm prompt
+        :param agent: Optional Timbr agent name for options setup.
         :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 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,
@@ -43,6 +45,7 @@ class GenerateResponseNode:
             jwt_tenant_id=jwt_tenant_id,
             conn_params=conn_params,
             note=note,
+            agent=agent,
             debug=debug,
             **kwargs,
         )

langchain_timbr/langgraph/generate_timbr_sql_node.py

@@ -29,6 +29,7 @@ class GenerateTimbrSqlNode:
         note: Optional[str] = '',
         db_is_case_sensitive: Optional[bool] = False,
         graph_depth: Optional[int] = 1,
+        agent: Optional[str] = None,
         verify_ssl: Optional[bool] = True,
         is_jwt: Optional[bool] = False,
         jwt_tenant_id: Optional[str] = None,
@@ -56,6 +57,7 @@ class GenerateTimbrSqlNode:
         :param note: Optional additional note to extend our llm prompt
         :param db_is_case_sensitive: Whether the database is case sensitive (default is False).
         :param graph_depth: Maximum number of relationship hops to traverse from the source concept during schema exploration (default is 1).
+        :param agent: Optional Timbr agent name for options setup.
         :param verify_ssl: Whether to verify SSL certificates (default is True).
         :param is_jwt: Whether to use JWT authentication (default: False)
         :param jwt_tenant_id: Tenant ID for JWT authentication when using multi-tenant setup
@@ -81,6 +83,7 @@ class GenerateTimbrSqlNode:
             note=note,
             db_is_case_sensitive=db_is_case_sensitive,
             graph_depth=graph_depth,
+            agent=agent,
             verify_ssl=verify_ssl,
             is_jwt=is_jwt,
             jwt_tenant_id=jwt_tenant_id,

langchain_timbr/langgraph/identify_concept_node.py

@@ -19,6 +19,7 @@ class IdentifyConceptNode:
         should_validate: Optional[bool] = False,
         retries: Optional[int] = 3,
         note: Optional[str] = None,
+        agent: Optional[str] = None,
         verify_ssl: Optional[bool] = True,
         is_jwt: Optional[bool] = False,
         jwt_tenant_id: Optional[str] = None,
@@ -38,6 +39,7 @@ class IdentifyConceptNode:
         :param should_validate: Whether to validate the identified concept before returning it
         :param retries: Number of retry attempts if the identified concept is invalid
         :param note: Optional additional note to extend our llm prompt
+        :param agent: Optional Timbr agent name for options setup.
         :param verify_ssl: Whether to verify SSL certificates
         :param is_jwt: Whether to use JWT authentication (default: False)
         :param jwt_tenant_id: Tenant ID for JWT authentication when using multi-tenant setup
@@ -55,6 +57,7 @@ class IdentifyConceptNode:
             should_validate=should_validate,
             retries=retries,
             note=note,
+            agent=agent,
             verify_ssl=verify_ssl,
             is_jwt=is_jwt,
             jwt_tenant_id=jwt_tenant_id,

langchain_timbr/langgraph/validate_timbr_query_node.py

@@ -29,6 +29,7 @@ class ValidateSemanticSqlNode:
         note: Optional[str] = None,
         db_is_case_sensitive: Optional[bool] = False,
         graph_depth: Optional[int] = 1,
+        agent: Optional[str] = None,
         verify_ssl: Optional[bool] = True,
         is_jwt: Optional[bool] = False,
         jwt_tenant_id: Optional[str] = None,
@@ -55,6 +56,7 @@ class ValidateSemanticSqlNode:
         :param note: Optional additional note to extend our llm prompt
         :param db_is_case_sensitive: Whether the database is case sensitive (default is False).
         :param graph_depth: Maximum number of relationship hops to traverse from the source concept during schema exploration (default is 1).
+        :param agent: Optional Timbr agent name for options setup.
         :param verify_ssl: Whether to verify SSL certificates (default is True).
         :param is_jwt: Whether to use JWT authentication (default: False)
         :param jwt_tenant_id: Tenant ID for JWT authentication when using multi-tenant setup
@@ -79,6 +81,7 @@ class ValidateSemanticSqlNode:
             note=note,
             db_is_case_sensitive=db_is_case_sensitive,
             graph_depth=graph_depth,
+            agent=agent,
             verify_ssl=verify_ssl,
             is_jwt=is_jwt,
             jwt_tenant_id=jwt_tenant_id,