alita-sdk 0.3.177__py3-none-any.whl → 0.3.178__py3-none-any.whl

alita_sdk/tools/__init__.py

@@ -15,19 +15,19 @@ def _safe_import_tool(tool_name, module_path, get_tools_name=None, toolkit_class
     """Safely import a tool module and register available functions/classes."""
     try:
         module = __import__(f'alita_sdk.tools.{module_path}', fromlist=[''])
-        
+
         imported = {}
         if get_tools_name and hasattr(module, get_tools_name):
             imported['get_tools'] = getattr(module, get_tools_name)
-            
+
         if toolkit_class_name and hasattr(module, toolkit_class_name):
             imported['toolkit_class'] = getattr(module, toolkit_class_name)
             AVAILABLE_TOOLKITS[toolkit_class_name] = getattr(module, toolkit_class_name)
-            
+
         if imported:
             AVAILABLE_TOOLS[tool_name] = imported
             logger.debug(f"Successfully imported {tool_name}")
-        
+
     except Exception as e:
         FAILED_IMPORTS[tool_name] = str(e)
         logger.debug(f"Failed to import {tool_name}: {e}")
@@ -92,7 +92,7 @@ def get_tools(tools_list, alita, llm, store: Optional[BaseStore] = None, *args,
         for tool_name in tool.get('settings', {}).get('selected_tools', []):
             if isinstance(tool_name, str) and tool_name.startswith('_'):
                 raise ValueError(f"Tool name '{tool_name}' from toolkit '{tool.get('type', '')}' cannot start with '_'")
-        
+
         tool['settings']['alita'] = alita
         tool['settings']['llm'] = llm
         tool['settings']['store'] = store
@@ -107,10 +107,10 @@ def get_tools(tools_list, alita, llm, store: Optional[BaseStore] = None, *args,
             try:
                 get_tools_func = AVAILABLE_TOOLS[tool_type]['get_tools']
                 tools.extend(get_tools_func(tool))
-                    
+
             except Exception as e:
                 logger.error(f"Error getting tools for {tool_type}: {e}")
-                
+
         # Handle ADO repos special case (it might be requested as azure_devops_repos)
         elif tool_type in ['ado_repos', 'azure_devops_repos'] and 'ado_repos' in AVAILABLE_TOOLS:
             try:
@@ -118,31 +118,35 @@ def get_tools(tools_list, alita, llm, store: Optional[BaseStore] = None, *args,
                 tools.extend(get_tools_func(tool))
             except Exception as e:
                 logger.error(f"Error getting ADO repos tools: {e}")
-                
+
         # Handle custom modules
         elif tool.get("settings", {}).get("module"):
             try:
                 settings = tool.get("settings", {})
                 mod = import_module(settings.pop("module"))
                 tkitclass = getattr(mod, settings.pop("class"))
-                toolkit = tkitclass.get_toolkit(**tool["settings"])
+                #
+                get_toolkit_params = tool["settings"].copy()
+                get_toolkit_params["name"] = tool.get("name")
+                #
+                toolkit = tkitclass.get_toolkit(**get_toolkit_params)
                 tools.extend(toolkit.get_tools())
             except Exception as e:
                 logger.error(f"Error in getting custom toolkit: {e}")
-                
+
         else:
             # Tool not available or not found
             if tool_type in FAILED_IMPORTS:
                 logger.warning(f"Tool '{tool_type}' is not available: {FAILED_IMPORTS[tool_type]}")
             else:
                 logger.warning(f"Unknown tool type: {tool_type}")
-                
+
     return tools
 
 def get_toolkits():
     """Return toolkit configurations for all successfully imported toolkits."""
     toolkit_configs = []
-    
+
     for toolkit_name, toolkit_class in AVAILABLE_TOOLKITS.items():
         try:
             if hasattr(toolkit_class, 'toolkit_config_schema'):
@@ -151,7 +155,7 @@ def get_toolkits():
                 logger.debug(f"Toolkit {toolkit_name} does not have toolkit_config_schema method")
         except Exception as e:
             logger.error(f"Error getting config schema for {toolkit_name}: {e}")
-    
+
     logger.info(f"Successfully loaded {len(toolkit_configs)} toolkit configurations")
     return toolkit_configs
 
@@ -172,13 +176,13 @@ def diagnose_imports():
     available_count = len(AVAILABLE_TOOLS)
     failed_count = len(FAILED_IMPORTS)
     total_count = available_count + failed_count
-    
+
     print(f"=== Tool Import Diagnostic ===")
     print(f"Total tools: {total_count}")
     print(f"Successfully imported: {available_count}")
     print(f"Failed imports: {failed_count}")
     print(f"Success rate: {(available_count/total_count*100):.1f}%")
-    
+
     if AVAILABLE_TOOLS:
         print(f"\n✅ Available tools ({len(AVAILABLE_TOOLS)}):")
         for tool_name in sorted(AVAILABLE_TOOLS.keys()):
@@ -188,12 +192,12 @@ def diagnose_imports():
             if 'toolkit_class' in AVAILABLE_TOOLS[tool_name]:
                 functions.append('toolkit')
             print(f"  - {tool_name}: {', '.join(functions)}")
-    
+
     if FAILED_IMPORTS:
         print(f"\n❌ Failed imports ({len(FAILED_IMPORTS)}):")
         for tool_name, error in FAILED_IMPORTS.items():
             print(f"  - {tool_name}: {error}")
-    
+
     if AVAILABLE_TOOLKITS:
         print(f"\n🔧 Available toolkits ({len(AVAILABLE_TOOLKITS)}):")
         for toolkit_name in sorted(AVAILABLE_TOOLKITS.keys()):
@@ -204,7 +208,7 @@ __all__ = [
     'get_tools',
     'get_toolkits',
     'get_available_tools',
-    'get_failed_imports', 
+    'get_failed_imports',
     'get_available_toolkits',
     'diagnose_imports'
 ]

{alita_sdk-0.3.177.dist-info → alita_sdk-0.3.178.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.177
+Version: 0.3.178
 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.177.dist-info → alita_sdk-0.3.178.dist-info}/RECORD

@@ -98,7 +98,7 @@ alita_sdk/runtime/utils/logging.py,sha256=svPyiW8ztDfhqHFITv5FBCj8UhLxz6hWcqGIY6
 alita_sdk/runtime/utils/save_dataframe.py,sha256=i-E1wp-t4wb17Zq3nA3xYwgSILjoXNizaQAA9opWvxY,1576
 alita_sdk/runtime/utils/streamlit.py,sha256=z4J_bdxkA0zMROkvTB4u379YBRFCkKh-h7PD8RlnZWQ,85644
 alita_sdk/runtime/utils/utils.py,sha256=dM8whOJAuFJFe19qJ69-FLzrUp6d2G-G6L7d4ss2XqM,346
-alita_sdk/tools/__init__.py,sha256=48DhEi14KkaYhNb-KvXuM9XJ4WGC-v9sRcWfN7GFWd4,9963
+alita_sdk/tools/__init__.py,sha256=l3KxV-Qtu-04QQ9YYcovbLtEkZ30fGWDZ7o9nuRF16o,9967
 alita_sdk/tools/elitea_base.py,sha256=NQaIxPX6DVIerHCb18jwUR6maZxxk73NZaTsFHkBQWE,21119
 alita_sdk/tools/ado/__init__.py,sha256=mD6GHcYMTtffPJkJvFPe2rzvye_IRmXmWfI7xYuZhO4,912
 alita_sdk/tools/ado/utils.py,sha256=PTCludvaQmPLakF2EbCGy66Mro4-rjDtavVP-xcB2Wc,1252
@@ -295,8 +295,8 @@ alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=UHVQUVqcBc3SZvDfO78HSuBzwAsRw
 alita_sdk/tools/zephyr_squad/__init__.py,sha256=rq4jOb3lRW2GXvAguk4H1KinO5f-zpygzhBJf-E1Ucw,2773
 alita_sdk/tools/zephyr_squad/api_wrapper.py,sha256=iOMxyE7vOc_LwFB_nBMiSFXkNtvbptA4i-BrTlo7M0A,5854
 alita_sdk/tools/zephyr_squad/zephyr_squad_cloud_client.py,sha256=IYUJoMFOMA70knLhLtAnuGoy3OK80RuqeQZ710oyIxE,3631
-alita_sdk-0.3.177.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.177.dist-info/METADATA,sha256=avoSyHmQjQ--BnjrSryL8zpwaJdw0q8FJpy9Mc3R7fA,18753
-alita_sdk-0.3.177.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.177.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.177.dist-info/RECORD,,
+alita_sdk-0.3.178.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.178.dist-info/METADATA,sha256=eWCrlvpO3Mjv3AQ7AMghklM9k8wTzrQN2k-jp5ymvGg,18753
+alita_sdk-0.3.178.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.178.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.178.dist-info/RECORD,,