alita-sdk 0.3.137__tar.gz → 0.3.138__tar.gz

{alita_sdk-0.3.137 → alita_sdk-0.3.138}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.137
+Version: 0.3.138
 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 → alita_sdk-0.3.138}/pyproject.toml

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

{alita_sdk-0.3.137 → alita_sdk-0.3.138}/src/alita_sdk/clients/client.py

@@ -65,14 +65,22 @@ 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 +418,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-0.3.137 → alita_sdk-0.3.138}/src/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-0.3.137 → alita_sdk-0.3.138}/src/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-0.3.137 → alita_sdk-0.3.138}/src/alita_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.137
+Version: 0.3.138
 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