alita-sdk 0.3.240__tar.gz → 0.3.242__tar.gz

{alita_sdk-0.3.240 → alita_sdk-0.3.242}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.240
+Version: 0.3.242
 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.240 → alita_sdk-0.3.242}/alita_sdk/configurations/ado.py

@@ -14,8 +14,8 @@ class AdoConfiguration(BaseModel):
             }
         }
     )
-    organization_url: Optional[str] = Field(description="Base API URL")
-    project: Optional[str] = Field(description="ADO project", default=None)
+    organization_url: str = Field(description="Base API URL")
+    project: str = Field(description="ADO project")
     token: Optional[SecretStr] = Field(description="ADO Token")
 
 

{alita_sdk-0.3.240 → alita_sdk-0.3.242}/alita_sdk/configurations/azure_search.py

@@ -9,6 +9,7 @@ class AzureSearchConfiguration(BaseModel):
             "metadata": {
                 "label": "Azure Search",
                 "icon_url": "azure-search.svg",
+                "hidden": True,
                 "section": "credentials",
                 "type": "azure_search",
                 "categories": ["search"],

{alita_sdk-0.3.240 → alita_sdk-0.3.242}/alita_sdk/configurations/bigquery.py

@@ -9,6 +9,7 @@ class BigQueryConfiguration(BaseModel):
             "metadata": {
                 "label": "Google BigQuery",
                 "icon_url": "google.svg",
+                "hidden": True,
                 "section": "credentials",
                 "type": "bigquery",
                 "categories": ["database"],

{alita_sdk-0.3.240 → alita_sdk-0.3.242}/alita_sdk/configurations/delta_lake.py

@@ -9,6 +9,7 @@ class DeltaLakeConfiguration(BaseModel):
             "metadata": {
                 "label": "AWS Delta Lake",
                 "icon_url": "delta-lake.svg",
+                "hidden": True,
                 "section": "credentials",
                 "type": "delta_lake",
                 "categories": ["database"],

{alita_sdk-0.3.240 → alita_sdk-0.3.242}/alita_sdk/configurations/embedding.py

@@ -13,7 +13,4 @@ class EmbeddingConfiguration(BaseModel):
         }
     )
     embedding_model: str = Field(description="Embedding model: i.e. 'HuggingFaceEmbeddings', etc.",
-                                 default="HuggingFaceEmbeddings")
-    embedding_model_params: dict = Field(
-        description="Embedding model parameters: i.e. `{'model_name': 'sentence-transformers/all-MiniLM-L6-v2'}",
-        default={"model_name": "sentence-transformers/all-MiniLM-L6-v2"})
+                                 default="HuggingFaceEmbeddings")

{alita_sdk-0.3.240 → alita_sdk-0.3.242}/alita_sdk/configurations/postman.py

@@ -27,6 +27,6 @@ class PostmanConfiguration(BaseModel):
             }
         }
     )
-    base_url: str = Field(description="Postman API base URL", default="https://api.getpostman.com")
+    base_url: str = Field(description="Postman API base URL")
     workspace_id: str = Field(description="Default workspace ID")
     api_key: Optional[SecretStr] = Field(description="Postman API Key", default=None)

{alita_sdk-0.3.240 → alita_sdk-0.3.242}/alita_sdk/configurations/slack.py

@@ -33,4 +33,4 @@ class SlackConfiguration(BaseModel):
     )
     name: Optional[SecretStr] = Field(description="Slack Bot Token")
     slack_token: Optional[SecretStr] = Field(description="Slack Token like XOXB-*****-*****-*****-*****")
-    channel_id:Optional[str] = Field(default=None, description="Channel ID")
+    channel_id:Optional[str] = Field(description="Channel ID")

{alita_sdk-0.3.240 → alita_sdk-0.3.242}/alita_sdk/configurations/xray.py

@@ -28,5 +28,5 @@ class XrayConfiguration(BaseModel):
         }
     )
     base_url: str = Field(description="Xray URL")
-    client_id: Optional[str] = Field(description="Client ID", default=None)
-    client_secret: Optional[SecretStr] = Field(description="Client secret", default=None)
+    client_id: Optional[str] = Field(description="Client ID")
+    client_secret: Optional[SecretStr] = Field(description="Client secret")

{alita_sdk-0.3.240 → alita_sdk-0.3.242}/alita_sdk/runtime/langchain/document_loaders/AlitaCSVLoader.py

@@ -13,14 +13,15 @@
 # limitations under the License.
 
 from typing import List, Optional, Iterator
-from charset_normalizer import from_path
+from charset_normalizer import from_path, from_bytes
 from csv import DictReader
 from .AlitaTableLoader import AlitaTableLoader
 from typing import Any
 
 class AlitaCSVLoader(AlitaTableLoader):
     def __init__(self,
-                 file_path: str,
+                 file_path: str = None,
+                 file_content: bytes = None,
                  encoding: Optional[str] = 'utf-8',
                  autodetect_encoding: bool = True,
                  json_documents: bool = True,
@@ -28,11 +29,14 @@ class AlitaCSVLoader(AlitaTableLoader):
                  columns: Optional[List[str]] = None,
                  cleanse: bool = True,
                  **kwargs):
-        super().__init__(file_path=file_path, json_documents=json_documents, columns=columns, raw_content=raw_content, cleanse=cleanse)
+        super().__init__(file_path=file_path, file_content=file_content, json_documents=json_documents, columns=columns, raw_content=raw_content, cleanse=cleanse)
         self.encoding = encoding
         self.autodetect_encoding = autodetect_encoding
-        if autodetect_encoding:
-            self.encoding = from_path(self.file_path).best().encoding
+        if self.file_path:
+            if autodetect_encoding:
+                self.encoding = from_path(self.file_path).best().encoding
+        else:
+            self.encoding = from_bytes(self.file_content).best().encoding
 
     def read_lazy(self) -> Iterator[dict]:
         with open(self.file_path, 'r', encoding=self.encoding) as fd:
@@ -43,7 +47,11 @@ class AlitaCSVLoader(AlitaTableLoader):
                 yield row
 
     def read(self) -> Any:
-        with open(self.file_path, 'r', encoding=self.encoding) as fd:
-            if self.raw_content:
-                return [fd.read()]
-            return list(DictReader(fd))
+        if self.file_path:
+            with open(self.file_path, 'r', encoding=self.encoding) as fd:
+                if self.raw_content:
+                    return [fd.read()]
+                return list(DictReader(fd))
+        else:
+            super.row_content = True
+            return self.file_content

{alita_sdk-0.3.240 → alita_sdk-0.3.242}/alita_sdk/runtime/langchain/document_loaders/AlitaTableLoader.py

@@ -22,7 +22,8 @@ from ..tools.log import print_log
 
 class AlitaTableLoader(BaseLoader):
     def __init__(self,
-                 file_path: str,
+                 file_path: str = None,
+                 file_content: bytes = None,
                  json_documents: bool = True,
                  raw_content: bool = False,
                  columns: Optional[List[str]] = None,
@@ -30,6 +31,7 @@ class AlitaTableLoader(BaseLoader):
 
         self.raw_content = raw_content
         self.file_path = file_path
+        self.file_content = file_content
         self.json_documents = json_documents
         self.columns = columns
         self.cleanse = cleanse
@@ -63,6 +65,11 @@ class AlitaTableLoader(BaseLoader):
                 "source": f'{self.file_path}:{idx+1}',
                 "table_source": self.file_path,
             }
+            if len(docs) == 0:
+                header_metadata = metadata.copy()
+                header_metadata["header"] = "true"
+                header = "\t".join([str(value) for value in row.keys()])
+                docs.append(Document(page_content=header, metadata=header_metadata))
             if self.raw_content:
                 docs.append(Document(page_content=row, metadata=metadata))
                 continue

{alita_sdk-0.3.240 → alita_sdk-0.3.242}/alita_sdk/tools/jira/__init__.py

@@ -20,6 +20,7 @@ def get_tools(tool):
         jira_configuration=tool['settings']['jira_configuration'],
         limit=tool['settings'].get('limit', 5),
         labels=parse_list(tool['settings'].get('labels', [])),
+        custom_headers=tool['settings'].get('custom_headers', {}),
         additional_fields=tool['settings'].get('additional_fields', []),
         verify_ssl=tool['settings'].get('verify_ssl', True),
         # indexer settings
@@ -66,12 +67,14 @@ class JiraToolkit(BaseToolkit):
         model = create_model(
             name,
             cloud=(bool, Field(description="Hosting Option", json_schema_extra={'configuration': True})),
-            limit=(int, Field(description="Limit issues")),
+            limit=(int, Field(description="Limit issues. Default is 5", gt=0, default=5)),
             labels=(Optional[str], Field(
                 description="List of comma separated labels used for labeling of agent's created or updated entities",
                 default=None,
                 examples="alita,elitea;another-label"
             )),
+            # optional field for custom headers as dictionary
+            custom_headers=(Optional[dict], Field(description="Custom headers for API requests", default=None)),
             verify_ssl=(bool, Field(description="Verify SSL", default=True)),
             additional_fields=(Optional[str], Field(description="Additional fields", default="")),
             jira_configuration=(Optional[JiraConfiguration], Field(description="Jira Configuration", json_schema_extra={'configuration_types': ['jira']})),

{alita_sdk-0.3.240 → alita_sdk-0.3.242}/alita_sdk/tools/jira/api_wrapper.py

@@ -436,6 +436,11 @@ class JiraApiWrapper(BaseVectorStoreToolApiWrapper):
             cls._client = Jira(url=url, token=token, cloud=cloud, verify_ssl=values['verify_ssl'], api_version=api_version)
         else:
             cls._client = Jira(url=url, username=username, password=api_key, cloud=cloud, verify_ssl=values['verify_ssl'], api_version=api_version)
+        custom_headers = values.get('custom_headers', {})
+        logger.info(f"Jira tool: custom headers length: {len(custom_headers)}")
+        for header, value in custom_headers.items():
+            cls._client._update_header(header, value)
+
         cls.llm=values.get('llm')
         return values
 

{alita_sdk-0.3.240 → alita_sdk-0.3.242}/alita_sdk/tools/sharepoint/api_wrapper.py

@@ -170,7 +170,7 @@ class SharepointApiWrapper(BaseVectorStoreToolApiWrapper):
         except Exception as e:
             logging.error(f"Failed while parsing the file '{document.metadata['Path']}': {e}")
         if isinstance(doc_content, dict):
-            for page, content in doc_content:
+            for page, content in doc_content.items():
                 new_metadata = document.metadata
                 new_metadata['page'] = page
                 yield Document(page_content=str(content), metadata=new_metadata)

{alita_sdk-0.3.240 → alita_sdk-0.3.242}/alita_sdk/tools/utils/content_parser.py

@@ -90,7 +90,9 @@ def parse_file_content(file_name=None, file_content=None, is_capture_image: bool
         "llm": llm,
         "page_number": page_number,
         "sheet_name": sheet_name,
-        "excel_by_sheets": excel_by_sheets
+        "excel_by_sheets": excel_by_sheets,
+        "row_content": True,
+        "json_documents": False
     })
     loader = loader_object['class'](**loader_kwargs)
 
@@ -135,6 +137,8 @@ def load_content(file_path: str, extension: str = None, loader_extra_config: dic
             loader_kwargs.update(loader_extra_config)
         if loader_config['is_multimodal_processing'] and llm:
             loader_kwargs.update({'llm': llm})
+        if "file_path" in loader_kwargs:
+            del loader_kwargs["file_path"]
 
         loader = loader_cls(file_path, **loader_kwargs)
         documents = loader.load()

{alita_sdk-0.3.240 → alita_sdk-0.3.242}/alita_sdk.egg-info/PKG-INFO

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

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