alita-sdk 0.3.266__py3-none-any.whl → 0.3.268__py3-none-any.whl

alita_sdk/runtime/langchain/langraph_agent.py

@@ -422,7 +422,7 @@ def create_graph(
                             if not pipeline_name:
                                 raise ValueError("Subgraph must have a 'tool' node: add required tool to the subgraph node")
                             node_fn = SubgraphRunnable(
-                                inner=tool,
+                                inner=tool.graph,
                                 name=pipeline_name,
                                 input_mapping=node.get('input_mapping', {}),
                                 output_mapping=node.get('output_mapping', {}),
@@ -666,7 +666,9 @@ class LangGraphAgentRunnable(CompiledStateGraph):
                config: Optional[RunnableConfig] = None,
                *args, **kwargs):
         logger.info(f"Incomming Input: {input}")
-        if not config.get("configurable", {}).get("thread_id"):
+        if config is None:
+            config = RunnableConfig()
+        if not config.get("configurable", {}).get("thread_id", ""):
             config["configurable"] = {"thread_id": str(uuid4())}
         thread_id = config.get("configurable", {}).get("thread_id")
         # Handle chat history and current input properly

alita_sdk/runtime/toolkits/subgraph.py

@@ -1,8 +1,11 @@
 from typing import List, Any
 
+from langchain_core.tools import BaseTool
+from langgraph.checkpoint.memory import MemorySaver
 from langgraph.graph.state import CompiledStateGraph
 
 from ..langchain.langraph_agent import create_graph, SUBGRAPH_REGISTRY
+from ..tools.graph import GraphTool
 from ..utils.utils import clean_string
 
 
@@ -16,7 +19,7 @@ class SubgraphToolkit:
         llm,
         app_api_key: str,
         selected_tools: list[str] = []
-    ) -> List[CompiledStateGraph]:
+    ) -> List[BaseTool]:
         from .tools import get_tools
         # from langgraph.checkpoint.memory import MemorySaver
 
@@ -36,18 +39,20 @@ class SubgraphToolkit:
         
         # For backward compatibility, still create a compiled graph stub
         # This is mainly used for identification in the parent graph's tools list
+        # For now the graph toolkit will have its own ephemeral in memory checkpoint memory.
         graph = create_graph(
             client=llm,
             tools=tools,
             yaml_schema=version_details['instructions'],
             debug=False,
             store=None,
-            memory=None,
-            for_subgraph=True,  # compile as raw subgraph
+            memory=MemorySaver(),
+            # for_subgraph=True,  # compile as raw subgraph
         )
-        
+
+        cleaned_subgraph_name = clean_string(subgraph_name)
         # Tag the graph stub for parent lookup
-        graph.name = clean_string(subgraph_name)
+        graph.name = cleaned_subgraph_name
         
         # Return the compiled graph stub for backward compatibility
-        return [graph]
+        return [GraphTool(description=app_details['description'], name=subgraph_name, graph=graph)]

alita_sdk/runtime/tools/graph.py

@@ -0,0 +1,75 @@
+import json
+
+from langgraph.graph.state import CompiledStateGraph
+
+from ..utils.utils import clean_string
+from langchain_core.tools import BaseTool
+from langchain_core.messages import BaseMessage, AIMessage, ToolCall
+from typing import Any, Type, Optional, Union
+from pydantic import create_model, field_validator, BaseModel
+from pydantic.fields import FieldInfo
+from ..langchain.mixedAgentRenderes import convert_message_to_json
+from logging import getLogger
+
+logger = getLogger(__name__)
+
+graphToolSchema = create_model(
+    "graphToolSchema",
+    input=(str, FieldInfo(description="User Input for Graph")),
+    chat_history=(Optional[list[BaseMessage]],
+                  FieldInfo(description="Chat History relevant for Graph in format [{'role': '<user| assistant | etc>', 'content': '<content of the respected message>'}]", default=[]))
+)
+
+
+def formulate_query(kwargs):
+    chat_history = []
+    if kwargs.get('chat_history'):
+        if isinstance(kwargs.get('chat_history')[-1], BaseMessage):
+            chat_history = convert_message_to_json(kwargs.get('chat_history')[:])
+        elif isinstance(kwargs.get('chat_history')[-1], dict):
+            if all([True if message.get('role') and message.get('content') else False for message in
+                    kwargs.get('chat_history')]):
+                chat_history = kwargs.get('chat_history')[:]
+            else:
+                for each in kwargs.get('chat_history')[:]:
+                    chat_history.append(AIMessage(json.dumps(each)))
+        elif isinstance(kwargs.get('chat_history')[-1], str):
+            chat_history = []
+            for each in kwargs.get('chat_history')[:]:
+                chat_history.append(AIMessage(each))
+    elif kwargs.get('messages'):
+        chat_history = convert_message_to_json(kwargs.get('messages')[:])
+    result = {"input": kwargs.get('input'), "chat_history": chat_history}
+    for key, value in kwargs.items():
+        if key not in ("input", "chat_history"):
+            result[key] = value
+    return result
+
+
+class GraphTool(BaseTool):
+    name: str
+    description: str
+    graph: CompiledStateGraph
+    args_schema: Type[BaseModel] = graphToolSchema
+    return_type: str = "str"
+
+    @field_validator('name', mode='before')
+    @classmethod
+    def remove_spaces(cls, v):
+        return clean_string(v)
+
+    def invoke(self, input: Any, config: Optional[dict] = None, **kwargs: Any) -> Any:
+        """Override default invoke to preserve all fields, not just args_schema"""
+        schema_values = self.args_schema(**input).model_dump() if self.args_schema else {}
+        extras = {k: v for k, v in input.items() if k not in schema_values}
+        all_kwargs = {**kwargs, **extras, **schema_values}
+        if config is None:
+            config = {}
+        return self._run(*config, **all_kwargs)
+
+    def _run(self, *args, **kwargs):
+        response = self.graph.invoke(formulate_query(kwargs))
+        if self.return_type == "str":
+            return response["output"]
+        else:
+            return {"messages": [{"role": "assistant", "content": response["output"]}]}

alita_sdk/tools/base_indexer_toolkit.py

@@ -110,8 +110,8 @@ class BaseIndexerToolkit(VectorStoreWrapperBase):
         connection_string = conn.get_secret_value() if isinstance(conn, SecretStr) else conn
         collection_name = kwargs.get('collection_name')
         
-        if 'embedding_model' not in kwargs:
-            kwargs['embedding_model'] = 'HuggingFaceEmbeddings'
+        if not kwargs.get('embedding_model'):
+            kwargs['embedding_model'] = 'text-embedding-ada-002'
         if 'vectorstore_type' not in kwargs:
             kwargs['vectorstore_type'] = 'PGVector'
         vectorstore_type = kwargs.get('vectorstore_type')

alita_sdk/tools/elitea_base.py

@@ -58,8 +58,8 @@ BaseSearchParams = create_model(
         default={},
         examples=["{\"key\": \"value\"}", "{\"status\": \"active\"}"]
     )),
-    cut_off=(Optional[float], Field(description="Cut-off score for search results", default=0.5)),
-    search_top=(Optional[int], Field(description="Number of top results to return", default=10)),
+    cut_off=(Optional[float], Field(description="Cut-off score for search results", default=0.5, ge=0, le=1)),
+    search_top=(Optional[int], Field(description="Number of top results to return", default=10, ge=0)),
     full_text_search=(Optional[Dict[str, Any]], Field(
         description="Full text search parameters. Can be a dictionary with search options.",
         default=None
@@ -88,8 +88,8 @@ BaseStepbackSearchParams = create_model(
         default={},
         examples=["{\"key\": \"value\"}", "{\"status\": \"active\"}"]
     )),
-    cut_off=(Optional[float], Field(description="Cut-off score for search results", default=0.5)),
-    search_top=(Optional[int], Field(description="Number of top results to return", default=10)),
+    cut_off=(Optional[float], Field(description="Cut-off score for search results", default=0.5, ge=0, le=1)),
+    search_top=(Optional[int], Field(description="Number of top results to return", default=10, ge=0)),
     reranker=(Optional[dict], Field(
         description="Reranker configuration. Can be a dictionary with reranking parameters.",
         default={}

alita_sdk/tools/gitlab/api_wrapper.py

@@ -106,18 +106,6 @@ class GitLabAPIWrapper(BaseCodeToolApiWrapper):
     _repo_instance: Any = PrivateAttr()
     _active_branch: Any = PrivateAttr()
 
-    llm: Optional[Any] = None
-    # Alita instance
-    alita: Optional[Any] = None
-
-    # Vector store configuration
-    connection_string: Optional[SecretStr] = None
-    collection_name: Optional[str] = None
-    doctype: Optional[str] = 'code'
-    embedding_model: Optional[str] = "HuggingFaceEmbeddings"
-    embedding_model_params: Optional[Dict[str, Any]] = {"model_name": "sentence-transformers/all-MiniLM-L6-v2"}
-    vectorstore_type: Optional[str] = "PGVector"
-
     @model_validator(mode='before')
     @classmethod
     def validate_toolkit(cls, values: Dict) -> Dict:

{alita_sdk-0.3.266.dist-info → alita_sdk-0.3.268.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.266
+Version: 0.3.268
 Summary: SDK for building langchain agents using resources from Alita
 Author-email: Artem Rozumenko <artyom.rozumenko@gmail.com>, Mikalai Biazruchka <mikalai_biazruchka@epam.com>, Roman Mitusov <roman_mitusov@epam.com>, Ivan Krakhmaliuk <lifedjik@gmail.com>, Artem Dubrovskiy <ad13box@gmail.com>
 License-Expression: Apache-2.0

{alita_sdk-0.3.266.dist-info → alita_sdk-0.3.268.dist-info}/RECORD

@@ -34,7 +34,7 @@ alita_sdk/runtime/langchain/assistant.py,sha256=suBFEt24t9bLyBHMzkR3Mkgd9HIrGBq_
 alita_sdk/runtime/langchain/chat_message_template.py,sha256=kPz8W2BG6IMyITFDA5oeb5BxVRkHEVZhuiGl4MBZKdc,2176
 alita_sdk/runtime/langchain/constants.py,sha256=eHVJ_beJNTf1WJo4yq7KMK64fxsRvs3lKc34QCXSbpk,3319
 alita_sdk/runtime/langchain/indexer.py,sha256=0ENHy5EOhThnAiYFc7QAsaTNp9rr8hDV_hTK8ahbatk,37592
-alita_sdk/runtime/langchain/langraph_agent.py,sha256=m96QQR1zoWQ8cDfwD4mcrlBl41JIrLwxDGPOg6yQzf8,43850
+alita_sdk/runtime/langchain/langraph_agent.py,sha256=3s1QLEPu0U0pz-PVaLBIEZLp7dhqAxy2wNUIjvIt49I,43925
 alita_sdk/runtime/langchain/mixedAgentParser.py,sha256=M256lvtsL3YtYflBCEp-rWKrKtcY1dJIyRGVv7KW9ME,2611
 alita_sdk/runtime/langchain/mixedAgentRenderes.py,sha256=asBtKqm88QhZRILditjYICwFVKF5KfO38hu2O-WrSWE,5964
 alita_sdk/runtime/langchain/store_manager.py,sha256=i8Fl11IXJhrBXq1F1ukEVln57B1IBe-tqSUvfUmBV4A,2218
@@ -87,7 +87,7 @@ alita_sdk/runtime/toolkits/artifact.py,sha256=3AjdKxrsbb0kzQ6NxJla9XqJcgyDJicMAq
 alita_sdk/runtime/toolkits/configurations.py,sha256=kIDAlnryPQfbZyFxV-9SzN2-Vefzx06TX1BBdIIpN90,141
 alita_sdk/runtime/toolkits/datasource.py,sha256=qk78OdPoReYPCWwahfkKLbKc4pfsu-061oXRryFLP6I,2498
 alita_sdk/runtime/toolkits/prompt.py,sha256=WIpTkkVYWqIqOWR_LlSWz3ug8uO9tm5jJ7aZYdiGRn0,1192
-alita_sdk/runtime/toolkits/subgraph.py,sha256=ZYqI4yVLbEPAjCR8dpXbjbL2ipX598Hk3fL6AgaqFD4,1758
+alita_sdk/runtime/toolkits/subgraph.py,sha256=wwUK8JjPXkGzyVZ3tAukmvST6eGbqx_U11rpnmbrvtg,2105
 alita_sdk/runtime/toolkits/tools.py,sha256=jNgWazbCb2t-H0OOErg7oPDxKXM-ezUAc6y9wqlI6GY,7875
 alita_sdk/runtime/toolkits/vectorstore.py,sha256=BGppQADa1ZiLO17fC0uCACTTEvPHlodEDYEzUcBRbAA,2901
 alita_sdk/runtime/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -97,6 +97,7 @@ alita_sdk/runtime/tools/artifact.py,sha256=yIn-kfI9OWoaxbBeqdqF0M1HPeMtNnvZ_pCPo
 alita_sdk/runtime/tools/datasource.py,sha256=pvbaSfI-ThQQnjHG-QhYNSTYRnZB0rYtZFpjCfpzxYI,2443
 alita_sdk/runtime/tools/echo.py,sha256=spw9eCweXzixJqHnZofHE1yWiSUa04L4VKycf3KCEaM,486
 alita_sdk/runtime/tools/function.py,sha256=ZFpd7TGwIawze2e7BHlKwP0NHwNw42wwrmmnXyJQJhk,2600
+alita_sdk/runtime/tools/graph.py,sha256=MbnZYqdmvZY7SGDp43lOVVIjUt5ARHSgj43mdtBjSjQ,3092
 alita_sdk/runtime/tools/indexer_tool.py,sha256=whSLPevB4WD6dhh2JDXEivDmTvbjiMV1MrPl9cz5eLA,4375
 alita_sdk/runtime/tools/llm.py,sha256=NsrsP-SblyxDdzgMCn9_OBUL0sUGDVS5yqer49V7ciE,15069
 alita_sdk/runtime/tools/loop.py,sha256=uds0WhZvwMxDVFI6MZHrcmMle637cQfBNg682iLxoJA,8335
@@ -119,8 +120,8 @@ alita_sdk/runtime/utils/toolkit_runtime.py,sha256=MU63Fpxj0b5_r1IUUc0Q3-PN9VwL7r
 alita_sdk/runtime/utils/toolkit_utils.py,sha256=I9QFqnaqfVgN26LUr6s3XlBlG6y0CoHURnCzG7XcwVs,5311
 alita_sdk/runtime/utils/utils.py,sha256=VXNLsdeTmf6snn9EtUyobv4yL-xzLhUcH8P_ORMifYc,675
 alita_sdk/tools/__init__.py,sha256=ko5TToGYZFmBrho26DRAVvrkHWxQ2sfs8gVAASinYp8,10611
-alita_sdk/tools/base_indexer_toolkit.py,sha256=17v9AIMGLO0OBUWS6A_-1xvHMArqTwK84mfkpTrwPwM,18434
-alita_sdk/tools/elitea_base.py,sha256=PfelIUb5YFTjDN_1jNYT9tJbjfYr11PAUrPQHyW2d5I,32830
+alita_sdk/tools/base_indexer_toolkit.py,sha256=JcjsN-xdZfwNJ7qy_TUgv6F1dIZhsQnguxsWsnsmj14,18437
+alita_sdk/tools/elitea_base.py,sha256=zx0ai3aN0re-Uc7_AKJjbTPeJ_K4OpFfK0jsaGvdMMU,32866
 alita_sdk/tools/non_code_indexer_toolkit.py,sha256=v9uq1POE1fQKCd152mbqDtF-HSe0qoDj83k4E5LAkMI,1080
 alita_sdk/tools/ado/__init__.py,sha256=u2tdDgufGuDb-7lIgKKQlqgStL9Wd1gzNmRNYems2c0,1267
 alita_sdk/tools/ado/utils.py,sha256=PTCludvaQmPLakF2EbCGy66Mro4-rjDtavVP-xcB2Wc,1252
@@ -230,7 +231,7 @@ alita_sdk/tools/github/schemas.py,sha256=yFsqivfjCPRk9GxFJrL8sTz6nnjFCZ0j5DIfPtG
 alita_sdk/tools/github/tool.py,sha256=Jnnv5lenV5ds8AAdyo2m8hSzyJ117HZBjzHC6T1ck-M,1037
 alita_sdk/tools/github/tool_prompts.py,sha256=y6ZW_FpUCE87Uop3WuQAZVRnzxO5t7xjBOI5bCqiluw,30194
 alita_sdk/tools/gitlab/__init__.py,sha256=wLTgBEeX44A3YkPkyUoPe4GVkDwHgLUpNEIBwJvaGUY,4488
-alita_sdk/tools/gitlab/api_wrapper.py,sha256=KYCRO2pF8EPTLhWuEj64XsHPCYSucsf8S3R_ofJttrA,22301
+alita_sdk/tools/gitlab/api_wrapper.py,sha256=x2AdR4CkqnMPxVOWwIAgxXuQuzel58drz8m2jyf2R0M,21833
 alita_sdk/tools/gitlab/tools.py,sha256=vOGTlSaGaFmWn6LS6YFP-FuTqUPun9vnv1VrUcUHAZQ,16500
 alita_sdk/tools/gitlab/utils.py,sha256=Z2XiqIg54ouqqt1to-geFybmkCb1I6bpE91wfnINH1I,2320
 alita_sdk/tools/gitlab_org/__init__.py,sha256=k8ECZzeGCmgKJ9UaVmPPlFxf-h1E6taC2SnPVrxPpHY,3775
@@ -335,8 +336,8 @@ alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=HOt9ShtJI_1tVPcwd3Rwk-VS0SMLq
 alita_sdk/tools/zephyr_squad/__init__.py,sha256=0AI_j27xVO5Gk5HQMFrqPTd4uvuVTpiZUicBrdfEpKg,2796
 alita_sdk/tools/zephyr_squad/api_wrapper.py,sha256=kmw_xol8YIYFplBLWTqP_VKPRhL_1ItDD0_vXTe_UuI,14906
 alita_sdk/tools/zephyr_squad/zephyr_squad_cloud_client.py,sha256=R371waHsms4sllHCbijKYs90C-9Yu0sSR3N4SUfQOgU,5066
-alita_sdk-0.3.266.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.266.dist-info/METADATA,sha256=zPwf0nv3u6t4QxbLBnNNSifv802d_j070ob8wr47fyw,18897
-alita_sdk-0.3.266.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.266.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.266.dist-info/RECORD,,
+alita_sdk-0.3.268.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.268.dist-info/METADATA,sha256=PSq9ld7E0116iWPhcL1hXxmgvIDs-p4zrEaF2eoHEx8,18897
+alita_sdk-0.3.268.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.268.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.268.dist-info/RECORD,,