alita-sdk 0.3.137__py3-none-any.whl → 0.3.139__py3-none-any.whl

alita_sdk/clients/client.py

@@ -65,14 +65,20 @@ class AlitaClient:
         self.configurations: list = configurations or []
 
     def get_mcp_toolkits(self):
-        url = self.mcp_tools_list
-        data = requests.get(url, headers=self.headers, verify=False).json()
-        return data
+        if user_id := self._get_real_user_id():
+            url = f"{self.mcp_tools_list}/{user_id}"
+            data = requests.get(url, headers=self.headers, verify=False).json()
+            return data
+        else:
+            return []
 
     def mcp_tool_call(self, params: dict[str, Any]):
-        url = self.mcp_tools_call
-        data = requests.post(url, headers=self.headers, json=params, verify=False).json()
-        return data
+        if user_id := self._get_real_user_id():
+            url = f"{self.mcp_tools_call}/{user_id}"
+            data = requests.post(url, headers=self.headers, json=params, verify=False).json()
+            return data
+        else:
+            return f"Error: Could not determine user ID for MCP tool call: {e}"
 
     def prompt(self, prompt_id, prompt_version_id, chat_history=None, return_tool=False):
         url = f"{self.prompt_versions}/{prompt_id}/{prompt_version_id}"
@@ -410,3 +416,12 @@ class AlitaClient:
         content = resp_data["findings"]
         references = resp_data['references']
         return AIMessage(content=content, additional_kwargs={"references": references})
+    
+    def _get_real_user_id(self):
+        try:
+            import tasknode_task
+            monitoring_meta = tasknode_task.meta.get("monitoring", {})
+            return monitoring_meta["user_id"]
+        except Exception as e:
+            logger.warning(f"Error: Could not determine user ID for MCP tool: {e}")
+            return None

alita_sdk/community/analysis/ado_analyse/api_wrapper.py

@@ -76,7 +76,7 @@ class AdoAnalyseWrapper(BaseToolApiWrapper):
 
         return (
             f"You have access to {len(result)} project. "
-            f"Data has been downloaded to the bucket as 'projects_info_AzureDevOps.csv'."
+            f"Data has been downloaded to the bucket as 'projects_info.csv'."
         )
 
     @with_streamlit_logs(tool_name="get_work_items")

alita_sdk/community/analysis/gitlab_analyse/api_wrapper.py

@@ -85,8 +85,7 @@ class GitLabAnalyseWrapper(BaseToolApiWrapper):
             f"Data has been downloaded to the bucket as 'gitlab_projects_that_in_Jira.csv'."
         )
 
-    def get_gitlab_commits(self, project_ids: Optional[str], since_date: str,
-    ) -> str:
+    def get_gitlab_commits(self, since_date: str, project_ids: Optional[str] = None) -> str:
         """
         Get commit data for specified GitLab project.
 
@@ -112,7 +111,7 @@ class GitLabAnalyseWrapper(BaseToolApiWrapper):
             f"Data has been downloaded to the bucket as 'commits_details_{project_ids}.csv'."
         )
 
-    def get_gitlab_merge_requests(self, project_ids: Optional[str], since_date: str) -> str:
+    def get_gitlab_merge_requests(self, since_date: str, project_ids: Optional[str] = None) -> str:
         """
         Get merge requests for specified GitLab project.
 

alita_sdk/toolkits/tools.py

@@ -121,11 +121,11 @@ def get_tools(tools_list: list, alita_client, llm) -> list:
 def _mcp_tools(tools_list, alita):
     try:
         all_available_toolkits = alita.get_mcp_toolkits()
-        toolkit_lookup = {tk["name"].lower(): tk for tk in all_available_toolkits}
+        toolkit_lookup = {tk["name"]: tk for tk in all_available_toolkits}
         tools = []
         #
         for selected_toolkit in tools_list:
-            toolkit_name = selected_toolkit['type'].lower()
+            toolkit_name = selected_toolkit['type']
             toolkit_conf = toolkit_lookup.get(toolkit_name)
             #
             if not toolkit_conf:

{alita_sdk-0.3.137.dist-info → alita_sdk-0.3.139.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.137
+Version: 0.3.139
 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>
 Project-URL: Homepage, https://projectalita.ai

{alita_sdk-0.3.137.dist-info → alita_sdk-0.3.139.dist-info}/RECORD

@@ -4,18 +4,18 @@ alita_sdk/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,
 alita_sdk/agents/llamaAgentParser.py,sha256=N_Nw6WJ8xrNX3fr_JFwjUIg_Urai9lLU3poMgZz0Cyk,1701
 alita_sdk/clients/__init__.py,sha256=5O_7WsZmvg5z5uZf_Jkx-f8j5C6yKIuSwQR6AQ-TM84,31
 alita_sdk/clients/artifact.py,sha256=W6oLlthtsUHfUWjhihaDdhLZdRVqj1D2j7oHHELxJfs,2639
-alita_sdk/clients/client.py,sha256=_-2S7wNZYkr6JgOtJJkUQbFWT_XFYdOt1iAi9BivNEE,18793
+alita_sdk/clients/client.py,sha256=iADTqIaVNmMefPBNH37zilPoX42CqcHZTtYBCS-lbtw,19413
 alita_sdk/clients/datasource.py,sha256=HAZovoQN9jBg0_-lIlGBQzb4FJdczPhkHehAiVG3Wx0,1020
 alita_sdk/clients/prompt.py,sha256=li1RG9eBwgNK_Qf0qUaZ8QNTmsncFrAL2pv3kbxZRZg,1447
 alita_sdk/community/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/community/utils.py,sha256=lvuCJaNqVPHOORJV6kIPcXJcdprVW_TJvERtYAEgpjM,249
 alita_sdk/community/analysis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/community/analysis/ado_analyse/__init__.py,sha256=lGlhDwaQB1pqRp9-jR2Y0zLTBMGjao3wJVMi-09XK0g,4108
-alita_sdk/community/analysis/ado_analyse/api_wrapper.py,sha256=NEQxZ6Y0g5wwwFhuoD7TYHwEmCrd6ictf8TakoLfQ08,8687
+alita_sdk/community/analysis/ado_analyse/api_wrapper.py,sha256=PaEoynvxaLiwiQEcrdIQj3nFtkKZvvf5K6fZr_ASBkc,8675
 alita_sdk/community/analysis/github_analyse/__init__.py,sha256=MwHmE6yfunc_h6UmKMqULKxTfMM8dojnpZLbCmdOao0,3602
 alita_sdk/community/analysis/github_analyse/api_wrapper.py,sha256=Otx1kbDqn0O1phz3qNJ5-0soO9_uU4zZ_YnB0tN6l9w,6826
 alita_sdk/community/analysis/gitlab_analyse/__init__.py,sha256=yJQKbr78n61cfvBrt95OjFmI_QPAjw-5SAdQL634ftk,4164
-alita_sdk/community/analysis/gitlab_analyse/api_wrapper.py,sha256=FZ2StzLH0MtO26guL51wV-ar457ygJtuPEvhqg-IXGM,6333
+alita_sdk/community/analysis/gitlab_analyse/api_wrapper.py,sha256=ZHUC9qm2w7hcots4Rxdv7CEzU17R1X1-8pFnTZxQ_dk,6341
 alita_sdk/community/analysis/jira_analyse/__init__.py,sha256=Rm-HKEi_HIxrgHdq9mZ-XzxMKLXm8-81eJwJT2lar-c,5945
 alita_sdk/community/analysis/jira_analyse/api_wrapper.py,sha256=naEgBSdwNonNleUtHCb1UOkWiYdM64ZJ9dfsyszmeX8,9668
 alita_sdk/community/browseruse/__init__.py,sha256=uAxPZEX7ihpt8HtcGDFrzTNv9WcklT1wG1ItTwUO8y4,3601
@@ -75,7 +75,7 @@ alita_sdk/toolkits/artifact.py,sha256=7zb17vhJ3CigeTqvzQ4VNBsU5UOCJqAwz7fOJGMYqX
 alita_sdk/toolkits/datasource.py,sha256=v3FQu8Gmvq7gAGAnFEbA8qofyUhh98rxgIjY6GHBfyI,2494
 alita_sdk/toolkits/prompt.py,sha256=WIpTkkVYWqIqOWR_LlSWz3ug8uO9tm5jJ7aZYdiGRn0,1192
 alita_sdk/toolkits/subgraph.py,sha256=ZYqI4yVLbEPAjCR8dpXbjbL2ipX598Hk3fL6AgaqFD4,1758
-alita_sdk/toolkits/tools.py,sha256=WYR__mieQ-9ouL7ofI8IyRLllnyYbh0N1S6YX9IZNlU,6790
+alita_sdk/toolkits/tools.py,sha256=ltMNYhuIOO8X-424YevdaLUUrE9DarBsOOw_rR92kRU,6774
 alita_sdk/toolkits/vectorstore.py,sha256=di08-CRl0KJ9xSZ8_24VVnPZy58iLqHtXW8vuF29P64,2893
 alita_sdk/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/tools/agent.py,sha256=m98QxOHwnCRTT9j18Olbb5UPS8-ZGeQaGiUyZJSyFck,3162
@@ -101,13 +101,13 @@ alita_sdk/utils/logging.py,sha256=hBE3qAzmcLMdamMp2YRXwOOK9P4lmNaNhM76kntVljs,31
 alita_sdk/utils/save_dataframe.py,sha256=tNwnaVCvN4_B0oi3F4Y3Z13elbIRtwKmdKHsMcj658g,1465
 alita_sdk/utils/streamlit.py,sha256=zp8owZwHI3HZplhcExJf6R3-APtWx-z6s5jznT2hY_k,29124
 alita_sdk/utils/utils.py,sha256=dM8whOJAuFJFe19qJ69-FLzrUp6d2G-G6L7d4ss2XqM,346
-alita_sdk-0.3.137.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.139.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_ado_analysis.py,sha256=wsxB4B2Ycxoiykthh6YbPQ9hqsDbPFie8D9ZK1i_6kg,3311
 tests/test_github_analysis.py,sha256=ulR4CEGmiMRPydJuX7aQcglzvhC7kFOAtZRLLBB9F_M,3148
 tests/test_gitlab_analysis.py,sha256=J7Y2mNi5Sj8-rH2PMRmVbT3uwZ17YeR9pcs0MDIyNW4,3352
 tests/test_jira_analysis.py,sha256=6F3Elikt02L28N6sS_AKDy9lgqgD81_hr979NcdZeg4,3359
-alita_sdk-0.3.137.dist-info/METADATA,sha256=jtSms2_mzUf9ZJ92ctGI_emhFd-Qvwzk_5Uk_XnURrY,7076
-alita_sdk-0.3.137.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.137.dist-info/top_level.txt,sha256=SWRhxB7Et3cOy3RkE5hR7OIRnHoo3K8EXzoiNlkfOmc,25
-alita_sdk-0.3.137.dist-info/RECORD,,
+alita_sdk-0.3.139.dist-info/METADATA,sha256=9ZEhLNU3RXXyKXJCqvTQIstIg22YoQYlRP24-ToFEB8,7076
+alita_sdk-0.3.139.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.139.dist-info/top_level.txt,sha256=SWRhxB7Et3cOy3RkE5hR7OIRnHoo3K8EXzoiNlkfOmc,25
+alita_sdk-0.3.139.dist-info/RECORD,,