alita-sdk 0.3.321b3__tar.gz → 0.3.323__tar.gz

{alita_sdk-0.3.321b3 → alita_sdk-0.3.323}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.321b3
+Version: 0.3.323
 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 <lifedj27@gmail.com>, Artem Dubrovskiy <ad13box@gmail.com>
 License-Expression: Apache-2.0

{alita_sdk-0.3.321b3 → alita_sdk-0.3.323}/alita_sdk/runtime/langchain/langraph_agent.py

@@ -702,7 +702,7 @@ class LangGraphAgentRunnable(CompiledStateGraph):
         
         # Append current input to existing messages instead of overwriting
         if input.get('input'):
-            current_message = HumanMessage(content=input.get('input'))
+            current_message = input.get('input')[-1]
             if input.get('messages'):
                 # Ensure existing messages are LangChain objects
                 input['messages'] = [convert_dict_to_message(msg) for msg in input['messages']]
@@ -718,7 +718,7 @@ class LangGraphAgentRunnable(CompiledStateGraph):
         else:
             result = super().invoke(input, config=config, *args, **kwargs)
         try:
-            if self.output_variables and self.output_variables[0] in result and self.output_variables[0] != "messages":
+            if self.output_variables and self.output_variables[0] != "messages":
                 # If output_variables are specified, use the value of first one or use the last messages as default
                 output = result.get(self.output_variables[0], result['messages'][-1].content)
             else:

{alita_sdk-0.3.321b3 → alita_sdk-0.3.323}/alita_sdk/runtime/tools/application.py

@@ -31,7 +31,8 @@ def formulate_query(kwargs):
             chat_history = []
             for each in kwargs.get('chat_history')[:]:
                 chat_history.append(AIMessage(each))
-    result = {"input": kwargs.get('task'), "chat_history": chat_history}
+    input_message = AIMessage(content=kwargs.get('task'))
+    result = {"input": [input_message], "chat_history": chat_history}
     for key, value in kwargs.items():
         if key not in ("task", "chat_history"):
             result[key] = value

{alita_sdk-0.3.321b3 → alita_sdk-0.3.323}/alita_sdk/runtime/tools/llm.py

@@ -14,7 +14,7 @@ logger = logging.getLogger(__name__)
 
 
 def create_llm_input_with_messages(
-    prompt: Dict[str, str], 
+    prompt: Dict[str, str],
     params: Dict[str, Any]
 ) -> List[BaseMessage]:
     """

{alita_sdk-0.3.321b3 → alita_sdk-0.3.323}/alita_sdk/tools/figma/api_wrapper.py

@@ -4,6 +4,7 @@ import logging
 import re
 from enum import Enum
 from typing import Dict, List, Generator, Optional, Union
+from urllib.parse import urlparse, parse_qs
 
 import requests
 from FigmaPy import FigmaPy
@@ -238,6 +239,7 @@ class FigmaApiWrapper(NonCodeIndexerToolkit):
 
     def _base_loader(
             self,
+            file_or_page_url: Optional[str] = None,
             project_id: Optional[str] = None,
             file_keys_include: Optional[List[str]] = None,
             file_keys_exclude: Optional[List[str]] = None,
@@ -247,6 +249,24 @@ class FigmaApiWrapper(NonCodeIndexerToolkit):
             node_types_exclude: Optional[List[str]] = None,
             **kwargs
     ) -> Generator[Document, None, None]:
+        if file_or_page_url:
+            # If URL is provided and valid, extract and override file_keys_include and node_ids_include
+            try:
+                parsed = urlparse(file_or_page_url)
+                path_parts = parsed.path.strip('/').split('/')
+    
+                # Check if the path matches the expected format
+                if len(path_parts) >= 2 and path_parts[0] == 'design':
+                    file_keys_include = [path_parts[1]]
+                    if len(path_parts) == 3:
+                        # To ensure url structure matches Figma's format with 3 path segments
+                        query_params = parse_qs(parsed.query)
+                        if "node-id" in query_params:
+                            node_ids_include = query_params.get('node-id', [])
+            except Exception as e:
+                raise ToolException(
+                    f"Unexpected error while processing Figma url {file_or_page_url}: {e}")
+        
         # If both include and exclude are provided, use only include
         if file_keys_include:
             self._log_tool_event(f"Loading files: {file_keys_include}")
@@ -364,8 +384,11 @@ class FigmaApiWrapper(NonCodeIndexerToolkit):
     def _index_tool_params(self):
         """Return the parameters for indexing data."""
         return {
+            "file_or_page_url": (Optional[str], Field(
+                description="Url to file or page to index: i.e. https://www.figma.com/design/[YOUR_FILE_KEY]/Login-page-designs?node-id=[YOUR_PAGE_ID]",
+                default=None)),
             "project_id": (Optional[str], Field(
-                description="ID of the project to list files from: i.e. 55391681'",
+                description="ID of the project to list files from: i.e. 55391681",
                 default=None)),
             'file_keys_include': (Optional[List[str]], Field(
                 description="List of file keys to include in index if project_id is not provided: i.e. ['Fp24FuzPwH0L74ODSrCnQo', 'jmhAr6q78dJoMRqt48zisY']",

{alita_sdk-0.3.321b3 → alita_sdk-0.3.323}/alita_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.321b3
+Version: 0.3.323
 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 <lifedj27@gmail.com>, Artem Dubrovskiy <ad13box@gmail.com>
 License-Expression: Apache-2.0

{alita_sdk-0.3.321b3 → alita_sdk-0.3.323}/alita_sdk.egg-info/top_level.txt

@@ -1,3 +1,2 @@
 alita_sdk
 docs
-src

{alita_sdk-0.3.321b3 → alita_sdk-0.3.323}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "alita_sdk"
-version = "0.3.321-beta3"
+version = "0.3.323"
 description = "SDK for building langchain agents using resources from Alita"
 readme = "README.md"
 requires-python = ">=3.10"