alita-sdk 0.3.201__py3-none-any.whl → 0.3.202__py3-none-any.whl

alita_sdk/tools/postman/api_wrapper.py

@@ -1283,7 +1283,7 @@ class PostmanApiWrapper(BaseToolApiWrapper):
             self.analyzer.remove_item_ids(collection_data.get("item", []))
 
             response = self._make_request(
-                'POST', '/collections', json={"collection": collection_data})
+                'POST', f'/collections?workspace={self.workspace_id}', json={"collection": collection_data})
             return json.dumps(response, indent=2)
         except Exception as e:
             stacktrace = format_exc()

alita_sdk/tools/slack/__init__.py

@@ -25,6 +25,7 @@ def get_tools(tool):
 
 class SlackToolkit(BaseToolkit):
     tools: List[BaseTool] = []
+    toolkit_max_length: int = 0
 
     @staticmethod
     def toolkit_config_schema() -> BaseModel:
@@ -52,7 +53,7 @@ class SlackToolkit(BaseToolkit):
                                                                              'configuration_title': True})),
              slack_token=(SecretStr, Field(description="Slack Token like XOXB-*****-*****-*****-*****",
                                            json_schema_extra={'secret': True, 'configuration': True})),
-             channel_id=(str, Field(description="Channel ID", json_schema_extra={'configuration': True})),
+             channel_id=(Optional[str], Field(default=None, description="Channel ID", json_schema_extra={'configuration': True})),
              selected_tools=(List[Literal[tuple(selected_tools)]],
                              Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
              __config__={'json_schema_extra': {'metadata': {"label": "Slack", "icon_url": "slack-icon.svg"}}}
@@ -74,7 +75,7 @@ class SlackToolkit(BaseToolkit):
             tools.append(BaseAction(                
                 api_wrapper=slack_api_wrapper,
                 name=prefix + tool["name"],
-                description=tool["description"],
+                description=f"Slack Tool: {tool['description']}",
                 args_schema=tool["args_schema"],
             ))
         return cls(tools=tools)

alita_sdk/tools/slack/api_wrapper.py

@@ -10,12 +10,14 @@ from alita_sdk.tools.elitea_base import BaseToolApiWrapper
 logger = logging.getLogger(__name__)
 
 SendMessageModel = create_model(
-                    "SendMessageModel",                    
+                    "SendMessageModel",
+                    channel_id=(Optional[str], Field(default=None,description="Channel ID, user ID, or conversation ID to send the message to. (like C12345678 for public channels, D12345678 for DMs)")),                   
                     message=(str, Field(description="The message text to send."))
                     )
 
 ReadMessagesModel = create_model(
-                    "ReadMessagesModel",                    
+                    "ReadMessagesModel",
+                    channel_id=(Optional[str], Field(default=None,description="Channel ID, user ID, or conversation ID to read messages from. (like C12345678 for public channels, D12345678 for DMs)")),                    
                     limit=(int, Field(default=10, description="The number of messages to fetch (default is 10)."))
                     )
 
@@ -25,11 +27,27 @@ CreateChannelModel = create_model(
                     is_private=(bool, Field(default=False, description="Whether to make the channel private (default: False)."))
                     )
 
-ListUsersModel = create_model(
-                    "ListUsersModel"                 
+ListChannelUsersModel = create_model(
+                    "ListChannelUsersModel",
+                    channel_id=(Optional[str], Field(default=None,description="Channel ID, user ID, or conversation ID to read messages from. (like C12345678 for public channels, D12345678 for DMs)"))                 
+                    )
+
+ListWorkspaceUsersModel = create_model(
+                    "ListWorkspaceUsersModel"                 
+                    )
+
+ListWorkspaceConversationsModel = create_model(
+                    "ListWorkspaceConversationsModel"                 
+                    )
+
+InviteToConversationModel = create_model(
+                    "InviteToConversationModel" ,
+                    channel_id=(Optional[str], Field(default=None,description="Conversation ID of the channel.")),
+                    user_ids=(list[str], Field(description="List of user IDs to invite."))                  
                     )
 
 
+
 class SlackApiWrapper(BaseToolApiWrapper):
 
     """
@@ -46,28 +64,34 @@ class SlackApiWrapper(BaseToolApiWrapper):
             logger.error("Slack token is required for authentication.")
             raise ValueError("Slack token is required for authentication.")
         return values
+
+    
     def _get_client(self):
         if not hasattr(self, "_client") or self._client is None:
             self._client = WebClient(token=self.slack_token.get_secret_value())
         return self._client
     
-    def send_message(self, message: str):
+    def send_message(self, message: str, channel_id: Optional[str] = None):
         """
         Sends a message to a specified Slack channel, user, or conversation.
+        Uses the provided channel_id if given, otherwise falls back to the instance's channel_id.
         """
         
         try:
 
             client = self._get_client()
-            response = client.chat_postMessage(channel=self.channel_id, text=message)
-            logger.info(f"Message sent to {self.channel_id}: {message}")
-            return f"Message sent successfully to {self.channel_id}."
+            
+            # Use the passed channel_id if provided, else use self.channel_id
+            channel = channel_id if channel_id else self.channel_id
+            response = client.chat_postMessage(channel=channel, text=message)
+            logger.info(f"Message sent to {channel}: {message}")
+            return f"Message sent successfully to {channel}."
         
         except SlackApiError as e:
-            logger.error(f"Failed to send message to {self.channel_id}: {e.response['error']}")
+            logger.error(f"Failed to send message to {channel}: {e.response['error']}")
             return f"Received the error :  {e.response['error']}"
 
-    def read_messages(self, limit=10):
+    def read_messages(self, limit=10, channel_id: Optional[str] = None):
         """
         Reads the latest messages from a Slack channel or conversation.
        
@@ -78,9 +102,11 @@ class SlackApiWrapper(BaseToolApiWrapper):
 
             client = self._get_client()
             logger.info(f"auth test: {client.auth_test()}")
+            # Use the passed channel_id if provided, else use self.channel_id
+            channel = channel_id if channel_id else self.channel_id
             # Fetch conversation history
             response = client.conversations_history(
-                channel=self.channel_id,
+                channel=channel,
                 limit=limit )
             
             # Extract messages from the response
@@ -90,7 +116,7 @@ class SlackApiWrapper(BaseToolApiWrapper):
             
         except SlackApiError as e:
             # Handle errors from the Slack API
-            logger.error(f"Failed to read message from {self.channel_id}: {e.response['error']}")
+            logger.error(f"Failed to read message from {channel}: {e.response['error']}")
             return f"Received the error :  {e.response['error']}"
         
     def create_slack_channel(self,  channel_name: str, is_private=False):
@@ -116,25 +142,91 @@ class SlackApiWrapper(BaseToolApiWrapper):
             print(f"Failed to create channel '{channel_name}': {error_message}")
             return {"success": False, "error": error_message}
     
-    def list_users(self):
+
+
+    def list_channel_users(self, channel_id: Optional[str] = None):
         """
-        Lists all users in the Slack workspace.
+        Lists all users in the specified Slack channel.
 
-        :return: list: List of users with their IDs and names.
+        :param channel_id: Optional[str]: The channel ID to list users from. If not provided, uses self.channel_id.
+        :return: list: List of user dictionaries with their IDs and names.
         """
-        
-        
         try:
             client = self._get_client()
-            print(client.auth_test())
-            response = client.users_list()
-            users = response["members"]
-            return [{"id": user["id"], "name": user["name"]} for user in users if not user["is_bot"]]
-        
+            # Use the passed channel_id if provided, else use self.channel_id
+            channel = channel_id if channel_id else self.channel_id
+            logger.info
+            if not channel:
+                logger.error("No channel_id provided to list_channel_users.")
+                return "Error: channel_id must be provided either as an argument or set on the instance."
+
+            # Get user IDs in the channel
+            members_response = client.conversations_members(channel=channel)
+            user_ids = members_response.get("members", [])
+
+            # Fetch user info for each user ID
+            users = []
+            for user_id in user_ids:
+                user_info = client.users_info(user=user_id)
+                user = user_info.get("user", {})
+                users.append({"id": user.get("id"), "name": user.get("name")})
+
+            return users
+
         except SlackApiError as e:
-            logger.error(f"Failed to list users: {e.response['error']}")
+            logger.error(f"Failed to list users in channel {channel}: {e.response['error']}")
             return f"Received the error :  {e.response['error']}"
+
+    def list_workspace_users(self):
+        """
+        Fetches and returns a list of all users in the Slack workspace with selected user details.
+
+        :return: list: List of user dictionaries containing id, name, is_bot, email, and team.
+        """
+        try:
+            client = self._get_client()
+            response = client.users_list()  # Fetch users
+            members = self.extract_required_user_fields(response.get('members', []))
+            print(f"Found {len(members)} users.")
+            return members  # Return the list of users
+        except SlackApiError as e:
+            print(f"Error fetching users: {e.response['error']}")
+            return []
     
+    def list_workspace_conversations(self):
+        """
+        Retrieves and returns a list of all conversations (channels, groups, and direct messages) in the Slack workspace.
+
+        :return: list: List of conversation/channel dictionaries as returned by the Slack API.
+        """
+        try:
+            client = self._get_client()
+            response = client.conversations_list()  # Fetch conversations
+            channels = response.get("channels", [])
+            print(f"Found {len(channels)} channels.")
+            return channels  # Return the list of channels
+        except SlackApiError as e:
+            print(f"Error fetching conversations: {e.response['error']}")
+            return [] 
+
+    def invite_to_conversation(self, user_ids: list, channel_id: Optional[str] = None ):
+        """
+        Invite one or more users to a Slack channel.
+        :param client: Slack WebClient instance.
+        :param channel_id: Conversation ID of the channel.
+        :param user_ids: List of user IDs to invite.
+        """
+        try:
+            client = self._get_client()
+            # Use the passed channel_id if provided, else use self.channel_id
+            channel = channel_id if channel_id else self.channel_id
+            response = client.conversations_invite(channel=channel, users=",".join(user_ids))
+            print(f"Successfully invited users to {channel}.")
+            return response
+        except SlackApiError as e:
+            print(f"Error inviting users: {e.response['error']}")
+            return None
+                
     def extract_slack_messages(self, data):
         extracted_info = []
 
@@ -150,7 +242,20 @@ class SlackApiWrapper(BaseToolApiWrapper):
             extracted_info.append({"user": user, "message": message, "app_name": app_name})
         
         return extracted_info
-    
+
+    # Function to extract required user details
+    def extract_required_user_fields(self, user_details):
+        extracted_user_data = []
+        for entry in user_details:
+            extracted_entry = {
+                "id": entry.get("id", None),
+                "name": entry.get("name", None),
+                "is_bot": entry.get("is_bot", None),
+                "email": entry.get("profile", {}).get("email", None),
+                "team": entry.get("profile", {}).get("team", None)
+            }
+            extracted_user_data.append(extracted_entry)
+        return extracted_user_data   
     
     def get_available_tools(self):
         return [
@@ -163,21 +268,34 @@ class SlackApiWrapper(BaseToolApiWrapper):
             },
             {
                 "name": "read_messages",
-                "description": self.read_messages.__doc__ or "Send a message to a Slack channel, user, or conversation.",
+                "description": self.read_messages.__doc__ or "Read a message from a Slack channel, user, or conversation.",
                 "args_schema": ReadMessagesModel,
                 "ref": self.read_messages
             },
             {
-                "name": "create_channel",
-                "description": self.create_slack_channel.__doc__ or "Send a message to a Slack channel, user, or conversation.",
+                "name": "create_slack_channel",
+                "description": self.create_slack_channel.__doc__ or "Create a  Slack channel",
                 "args_schema": CreateChannelModel,
                 "ref": self.create_slack_channel             
             },
             {
-                "name": "list_users",
-                "description": self.list_users.__doc__ or "List all users in the Slack workspace.",
-                "args_schema": ListUsersModel,
-                "ref": self.list_users
+                "name": "list_channel_users",
+                "description": self.list_channel_users.__doc__ or "List all users in the Slack channel.",
+                "args_schema": ListChannelUsersModel,
+                "ref": self.list_channel_users
+            },
+            {
+                "name": "list_workspace_users",
+                "description": self.list_workspace_users.__doc__ or "List all users in the Slack workspace.",
+                "args_schema": ListWorkspaceUsersModel,
+                "ref": self.list_workspace_users
+            }
+            ,
+            {
+                "name": "invite_to_conversation",
+                "description": self.invite_to_conversation.__doc__ or "Invite to a conversation in the Slack workspace.",
+                "args_schema": InviteToConversationModel,
+                "ref": self.invite_to_conversation
             }
             
         ]

{alita_sdk-0.3.201.dist-info → alita_sdk-0.3.202.dist-info}/METADATA

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

@@ -248,7 +248,7 @@ alita_sdk/tools/pandas/statsmodels/descriptive.py,sha256=APdofBnEiRhMrn6tLKwH076
 alita_sdk/tools/pandas/statsmodels/hypothesis_testing.py,sha256=fdNAayMB3W7avMfKJCcbf2_P54vUXbq8KVebOB48348,10508
 alita_sdk/tools/pandas/statsmodels/regression.py,sha256=Y1pWK4u_qzrfA740K-FX0nZ5FREGGPk8mfvykPIYoiI,9164
 alita_sdk/tools/postman/__init__.py,sha256=FzVZvqbTrA08mdoHVs0GZH1HcXDTtiMMWmSK07Bdvlc,4766
-alita_sdk/tools/postman/api_wrapper.py,sha256=K7Vq06DX7uMeuhpnc3EQfYvc4szY0_ElrRf9P4FLlfk,93501
+alita_sdk/tools/postman/api_wrapper.py,sha256=7IgbaKfk8Wtvo0-3Wl0DpbiQOU_h00lCEmNo4J3uG2g,93532
 alita_sdk/tools/postman/postman_analysis.py,sha256=2d-Oi2UORosIePIUyncSONw9hY7dw8Zc7BQvCd4aqpg,45115
 alita_sdk/tools/pptx/__init__.py,sha256=vVUrWnj7KWJgEk9oxGSsCAQ2SMSXrp_SFOdUHYQKcAo,3444
 alita_sdk/tools/pptx/pptx_wrapper.py,sha256=yyCYcTlIY976kJ4VfPo4dyxj4yeii9j9TWP6W8ZIpN8,29195
@@ -270,8 +270,8 @@ alita_sdk/tools/sharepoint/__init__.py,sha256=HqKQDFboab1AYh20uJvHxs9HFLJSqVfVTj
 alita_sdk/tools/sharepoint/api_wrapper.py,sha256=qCHCIH4FRDtgdpIK22ewFhZJeOaTv9hT9BVivslxxlE,7119
 alita_sdk/tools/sharepoint/authorization_helper.py,sha256=n-nL5dlBoLMK70nHu7P2RYCb8C6c9HMA_gEaw8LxuhE,2007
 alita_sdk/tools/sharepoint/utils.py,sha256=fZ1YzAu5CTjKSZeslowpOPH974902S8vCp1Wu7L44LM,446
-alita_sdk/tools/slack/__init__.py,sha256=bMuSoJYCkMzaZuMMktRtI7wPwrO5r07FN5PyR0nTys4,3821
-alita_sdk/tools/slack/api_wrapper.py,sha256=dkMcS3xozEhoJ-A1ycEn6OqbIftxmVHjyYttTy3D2JI,7343
+alita_sdk/tools/slack/__init__.py,sha256=mbP2JiHybGSAH0ay8pxvPCqeU2eb9CK_NaCKG1uhPE4,3894
+alita_sdk/tools/slack/api_wrapper.py,sha256=EsfxhQHFhfji7hanxbDzTMbEFQnBnipyCDuu_AZgGJ8,13291
 alita_sdk/tools/sql/__init__.py,sha256=9Lh8YHKO8zD5eeolpR4O9swTUsjpXj9LVDn8fM-T5IM,3506
 alita_sdk/tools/sql/api_wrapper.py,sha256=Rky0_CX9HWDQ2mClHGAgP3LHjYVX4iymPuilZMtaDlQ,3687
 alita_sdk/tools/sql/models.py,sha256=AKJgSl_kEEz4fZfw3kbvdGHXaRZ-yiaqfJOB6YOj3i0,641
@@ -297,8 +297,8 @@ alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=UHVQUVqcBc3SZvDfO78HSuBzwAsRw
 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.201.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.201.dist-info/METADATA,sha256=RFOFTi8B3D4RaQKwv-N029Joo0gLd406t0hXhe03dkI,18804
-alita_sdk-0.3.201.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.201.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.201.dist-info/RECORD,,
+alita_sdk-0.3.202.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.202.dist-info/METADATA,sha256=0c34jEtau84xdrHseyu4ZiBrj2u71UZDTV6mBSdms3Q,18804
+alita_sdk-0.3.202.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.202.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.202.dist-info/RECORD,,