alita-sdk 0.3.379__py3-none-any.whl → 0.3.627__py3-none-any.whl

alita_sdk/tools/zephyr/__init__.py

@@ -8,7 +8,8 @@ from pydantic import create_model, BaseModel, Field, SecretStr
 from ..base.tool import BaseAction
 from .api_wrapper import ZephyrV1ApiWrapper
 from ..elitea_base import filter_missconfigured_index_tools
-from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
+from ..utils import clean_string, get_max_toolkit_length
+from ...runtime.utils.constants import TOOLKIT_NAME_META, TOOL_NAME_META, TOOLKIT_TYPE_META
 
 name = "zephyr"
 
@@ -23,15 +24,13 @@ def get_tools(tool):
 
 class ZephyrToolkit(BaseToolkit):
     tools: List[BaseTool] = []
-    toolkit_max_length: int = 0
 
     @staticmethod
     def toolkit_config_schema() -> BaseModel:
         selected_tools = {x['name']: x['args_schema'].schema() for x in ZephyrV1ApiWrapper.model_construct().get_available_tools()}
-        ZephyrToolkit.toolkit_max_length = get_max_toolkit_length(selected_tools)
         return create_model(
             name,
-            base_url=(str, Field(description="Base URL", json_schema_extra={'toolkit_name': True, 'max_toolkit_length': ZephyrToolkit.toolkit_max_length})),
+            base_url=(str, Field(description="Base URL")),
             username=(str, Field(description="Username")),
             password=(SecretStr, Field(description="Password", json_schema_extra={'secret': True})),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
@@ -49,18 +48,22 @@ class ZephyrToolkit(BaseToolkit):
     @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         zephyr_api_wrapper = ZephyrV1ApiWrapper(**kwargs)
-        prefix = clean_string(toolkit_name, cls.toolkit_max_length) + TOOLKIT_SPLITTER if toolkit_name else ''
         available_tools = zephyr_api_wrapper.get_available_tools()
         tools = []
         for tool in available_tools:
             if selected_tools:
                 if tool["name"] not in selected_tools:
                     continue
+            description = tool["description"]
+            if toolkit_name:
+                description = f"Toolkit: {toolkit_name}\n{description}"
+            description = description[:1000]
             tools.append(BaseAction(
                 api_wrapper=zephyr_api_wrapper,
                 name=tool["name"],
-                description=tool["description"],
-                args_schema=tool["args_schema"]
+                description=description,
+                args_schema=tool["args_schema"],
+                metadata={TOOLKIT_NAME_META: toolkit_name, TOOLKIT_TYPE_META: name, TOOL_NAME_META: tool["name"]} if toolkit_name else {TOOL_NAME_META: tool["name"]}
             ))
         return cls(tools=tools)
 

alita_sdk/tools/zephyr_enterprise/__init__.py

@@ -5,9 +5,10 @@ from typing import List, Literal, Optional
 from .api_wrapper import ZephyrApiWrapper
 from ..base.tool import BaseAction
 from ..elitea_base import filter_missconfigured_index_tools
-from ..utils import clean_string, get_max_toolkit_length, TOOLKIT_SPLITTER
+from ..utils import clean_string, get_max_toolkit_length
 from ...configurations.pgvector import PgVectorConfiguration
 from ...configurations.zephyr_enterprise import ZephyrEnterpriseConfiguration
+from ...runtime.utils.constants import TOOLKIT_NAME_META, TOOL_NAME_META, TOOLKIT_TYPE_META
 
 name = "zephyr_enterprise"
 
@@ -28,13 +29,11 @@ def get_tools(tool):
 
 class ZephyrEnterpriseToolkit(BaseToolkit):
     tools: List[BaseTool] = []
-    toolkit_max_length: int = 0
 
     @staticmethod
     def toolkit_config_schema() -> BaseModel:
         selected_tools = {x['name']: x['args_schema'].schema() for x in
                           ZephyrApiWrapper.model_construct().get_available_tools()}
-        ZephyrEnterpriseToolkit.toolkit_max_length = get_max_toolkit_length(selected_tools)
         return create_model(
             name,
             zephyr_configuration=(ZephyrEnterpriseConfiguration, Field(description="Zephyr Configuration", json_schema_extra={'configuration_types': ['zephyr_enterprise']})),
@@ -67,17 +66,22 @@ class ZephyrEnterpriseToolkit(BaseToolkit):
             **(kwargs.get('embedding_configuration') or {}),
         }
         zephyr_api_wrapper = ZephyrApiWrapper(**wrapper_payload)
-        prefix = clean_string(toolkit_name, cls.toolkit_max_length) + TOOLKIT_SPLITTER if toolkit_name else ''
         available_tools = zephyr_api_wrapper.get_available_tools()
         tools = []
         for tool in available_tools:
             if selected_tools and tool["name"] not in selected_tools:
                 continue
+            description = tool["description"]
+            if toolkit_name:
+                description = f"Toolkit: {toolkit_name}\n{description}"
+            description = description + "\nZephyr Enterprise instance: " + zephyr_api_wrapper.base_url
+            description = description[:1000]
             tools.append(BaseAction(
                 api_wrapper=zephyr_api_wrapper,
-                name=prefix + tool["name"],
-                description=tool["description"] + "\nZephyr Enterprise instance: " + zephyr_api_wrapper.base_url,
-                args_schema=tool["args_schema"]
+                name=tool["name"],
+                description=description,
+                args_schema=tool["args_schema"],
+                metadata={TOOLKIT_NAME_META: toolkit_name, TOOLKIT_TYPE_META: name, TOOL_NAME_META: tool["name"]} if toolkit_name else {TOOL_NAME_META: tool["name"]}
             ))
         return cls(tools=tools)
 

alita_sdk/tools/zephyr_essential/__init__.py

@@ -6,9 +6,10 @@ from pydantic import create_model, BaseModel, Field
 from .api_wrapper import ZephyrEssentialApiWrapper
 from ..base.tool import BaseAction
 from ..elitea_base import filter_missconfigured_index_tools
-from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
+from ..utils import clean_string, get_max_toolkit_length
 from ...configurations.pgvector import PgVectorConfiguration
 from ...configurations.zephyr_essential import ZephyrEssentialConfiguration
+from ...runtime.utils.constants import TOOLKIT_NAME_META, TOOL_NAME_META, TOOLKIT_TYPE_META
 
 name = "zephyr_essential"
 
@@ -29,12 +30,10 @@ def get_tools(tool):
 
 class ZephyrEssentialToolkit(BaseToolkit):
     tools: List[BaseTool] = []
-    toolkit_max_length: int = 0
 
     @staticmethod
     def toolkit_config_schema() -> BaseModel:
         selected_tools = {x['name']: x['args_schema'].schema() for x in ZephyrEssentialApiWrapper.model_construct().get_available_tools()}
-        ZephyrEssentialToolkit.toolkit_max_length = get_max_toolkit_length(selected_tools)
         return create_model(
             name,
             zephyr_essential_configuration=(ZephyrEssentialConfiguration, Field(description="Zephyr Essential Configuration", json_schema_extra={'configuration_types': ['zephyr_essential']})),
@@ -62,18 +61,22 @@ class ZephyrEssentialToolkit(BaseToolkit):
             **(kwargs.get('pgvector_configuration') or {}),
         }
         zephyr_api_wrapper = ZephyrEssentialApiWrapper(**wrapper_payload)
-        prefix = clean_string(toolkit_name, cls.toolkit_max_length) + TOOLKIT_SPLITTER if toolkit_name else ''
         available_tools = zephyr_api_wrapper.get_available_tools()
         tools = []
         for tool in available_tools:
             if selected_tools:
                 if tool["name"] not in selected_tools:
                     continue
+            description = tool["description"]
+            if toolkit_name:
+                description = f"Toolkit: {toolkit_name}\n{description}"
+            description = description[:1000]
             tools.append(BaseAction(
                 api_wrapper=zephyr_api_wrapper,
-                name=prefix + tool["name"],
-                description=tool["description"],
-                args_schema=tool["args_schema"]
+                name=tool["name"],
+                description=description,
+                args_schema=tool["args_schema"],
+                metadata={TOOLKIT_NAME_META: toolkit_name, TOOLKIT_TYPE_META: name, TOOL_NAME_META: tool["name"]} if toolkit_name else {TOOL_NAME_META: tool["name"]}
             ))
         return cls(tools=tools)
 

alita_sdk/tools/zephyr_essential/api_wrapper.py

@@ -1,3 +1,4 @@
+import hashlib
 import json
 import logging
 from typing import Optional, Generator, Literal
@@ -55,6 +56,7 @@ class ZephyrEssentialApiWrapper(NonCodeIndexerToolkit):
     def create_test_case_issue_link(self, test_case_key: str, json: str):
         """Create an issue link for a test case."""
         issue_link_data = self._parse_json(json)
+        self._validate_issue_link_data(issue_link_data, "test case")
         return self._client.create_test_case_issue_link(test_case_key, issue_link_data)
 
     def create_test_case_web_link(self, test_case_key: str, json: str):
@@ -119,6 +121,7 @@ class ZephyrEssentialApiWrapper(NonCodeIndexerToolkit):
     def create_test_cycle_issue_link(self, test_cycle_id_or_key: str, json: str):
         """Create an issue link for a test cycle."""
         issue_link_data = self._parse_json(json)
+        self._validate_issue_link_data(issue_link_data, "test cycle")
         return self._client.create_test_cycle_issue_link(test_cycle_id_or_key, issue_link_data)
 
     def create_test_cycle_web_link(self, test_cycle_id_or_key: str, json: str):
@@ -164,6 +167,7 @@ class ZephyrEssentialApiWrapper(NonCodeIndexerToolkit):
     def create_test_execution_issue_link(self, test_execution_id_or_key: str, json: str):
         """Create an issue link for a test execution."""
         issue_link_data = self._parse_json(json)
+        self._validate_issue_link_data(issue_link_data, "test execution")
         return self._client.create_test_execution_issue_link(test_execution_id_or_key, issue_link_data)
 
     def list_projects(self, max_results: int = None, start_at: int = None):
@@ -266,6 +270,21 @@ class ZephyrEssentialApiWrapper(NonCodeIndexerToolkit):
         except json.JSONDecodeError as e:
             raise ValueError(f"Invalid JSON string: {str(e)}")
 
+    def _validate_issue_link_data(self, issue_link_data: dict, entity_type: str):
+        """Validate issue link payload has issueId, not issueKey."""
+        if 'issueId' not in issue_link_data:
+            if 'issueKey' in issue_link_data:
+                raise ToolException(
+                    f"Zephyr Essential API requires 'issueId' (numeric Jira issue ID), "
+                    f"not 'issueKey'. You provided issueKey='{issue_link_data['issueKey']}'. "
+                    f"To find the issueId, Jira toolkit can be used or raw Jira API: GET /rest/api/2/issue/{issue_link_data['issueKey']} "
+                    f"and look for the 'id' field in the response."
+                )
+            raise ToolException(
+                f"Missing required field 'issueId' in JSON payload for {entity_type} issue link. "
+                f"Example: {{\"issueId\": 10100}}"
+            )
+
     def _index_tool_params(self):
         return {
             'chunking_tool':(Literal['json', ''], Field(description="Name of chunking tool", default='json'))
@@ -284,22 +303,20 @@ class ZephyrEssentialApiWrapper(NonCodeIndexerToolkit):
                 if isinstance(v, (str, int, float, bool, list, dict))
             }
             metadata['type'] = "TEST_CASE"
-
-            yield Document(page_content="", metadata=metadata)
-
-    def _extend_data(self, documents: Generator[Document, None, None]) -> Generator[Document, None, None]:
-        for document in documents:
+            #
             try:
-                if 'type' in document.metadata and document.metadata['type'] == "TEST_CASE":
-                    additional_content = self._process_test_case(document.metadata['key'])
-                    for steps_type, content in additional_content.items():
-                        if content:
-                            page_content = json.dumps(content)
-                            document.metadata[IndexerKeywords.CONTENT_IN_BYTES.value] = page_content.encode('utf-8')
-                            document.metadata["steps_type"] = steps_type
+                additional_content = self._process_test_case(metadata['key'])
+                for steps_type, content in additional_content.items():
+                    if content:
+                        page_content = json.dumps(content)
+                        content_hash = hashlib.sha256(page_content.encode('utf-8')).hexdigest()
+                        metadata[IndexerKeywords.UPDATED_ON.value] = content_hash
+                        metadata[IndexerKeywords.CONTENT_IN_BYTES.value] = page_content.encode('utf-8')
+                        metadata["steps_type"] = steps_type
             except Exception as e:
                 logging.error(f"Failed to process document: {e}")
-            yield document
+            #
+            yield Document(page_content="", metadata=metadata)
 
     def _process_test_case(self, key) -> dict:
         steps = self.get_test_case_test_steps(key)

alita_sdk/tools/zephyr_essential/client.py

@@ -19,9 +19,9 @@ class ZephyrEssentialAPI:
                 if resp.headers.get("Content-Type", "").startswith("application/json"):
                     return resp.json()
                 return resp.text
-            return ToolException(f"Error performing request {method} {api_path}: {resp.content}")
+            return ToolException(f"Unexpected status code {resp.status_code} for {method} {api_path}: {resp.content}")
         except requests.RequestException as e:
-            raise Exception(f"Error performing request {method} {api_path}: {str(e)}")
+            raise Exception(f"Unexpected error while performing request {method} {api_path}: {str(e)}")
 
     # Test Cases
     def list_test_cases(self, project_key=None, folder_id=None, max_results=10, start_at=0):

alita_sdk/tools/zephyr_scale/__init__.py

@@ -7,9 +7,10 @@ from pydantic import create_model, BaseModel, Field
 from .api_wrapper import ZephyrScaleApiWrapper
 from ..base.tool import BaseAction
 from ..elitea_base import filter_missconfigured_index_tools
-from ..utils import clean_string, get_max_toolkit_length, TOOLKIT_SPLITTER
+from ..utils import clean_string, get_max_toolkit_length
 from ...configurations.pgvector import PgVectorConfiguration
 from ...configurations.zephyr import ZephyrConfiguration
+from ...runtime.utils.constants import TOOLKIT_NAME_META, TOOL_NAME_META, TOOLKIT_TYPE_META
 
 name = "zephyr_scale"
 
@@ -32,15 +33,13 @@ def get_tools(tool):
 
 class ZephyrScaleToolkit(BaseToolkit):
     tools: List[BaseTool] = []
-    toolkit_max_length: int = 0
 
     @staticmethod
     def toolkit_config_schema() -> BaseModel:
         selected_tools = {x['name']: x['args_schema'].schema() for x in ZephyrScaleApiWrapper.model_construct().get_available_tools()}
-        ZephyrScaleToolkit.toolkit_max_length = get_max_toolkit_length(selected_tools)
         return create_model(
             name,
-            max_results=(int, Field(default=100, description="Results count to show")),
+            max_results=(int, Field(default=100, description="Results count to show", gt=0)),
             zephyr_configuration=(ZephyrConfiguration, Field(description="Zephyr Configuration",
                                                                        json_schema_extra={'configuration_types': ['zephyr']})),
             pgvector_configuration=(Optional[PgVectorConfiguration], Field(default=None, description="PgVector Configuration",
@@ -75,18 +74,22 @@ class ZephyrScaleToolkit(BaseToolkit):
             **(kwargs.get('pgvector_configuration') or {}),
         }
         zephyr_wrapper = ZephyrScaleApiWrapper(**wrapper_payload)
-        prefix = clean_string(toolkit_name, cls.toolkit_max_length) + TOOLKIT_SPLITTER if toolkit_name else ''
         available_tools = zephyr_wrapper.get_available_tools()
         tools = []
         for tool in available_tools:
             if selected_tools:
                 if tool["name"] not in selected_tools:
                     continue
+            description = tool["description"]
+            if toolkit_name:
+                description = f"Toolkit: {toolkit_name}\n{description}"
+            description = description[:1000]
             tools.append(BaseAction(
                 api_wrapper=zephyr_wrapper,
-                name=prefix + tool["name"],
-                description=tool["description"],
-                args_schema=tool["args_schema"]
+                name=tool["name"],
+                description=description,
+                args_schema=tool["args_schema"],
+                metadata={TOOLKIT_NAME_META: toolkit_name, TOOLKIT_TYPE_META: name, TOOL_NAME_META: tool["name"]} if toolkit_name else {TOOL_NAME_META: tool["name"]}
             ))
         return cls(tools=tools)
 

alita_sdk/tools/zephyr_scale/api_wrapper.py

@@ -177,13 +177,13 @@ ZephyrSearchTestCases = create_model(
     "ZephyrSearchTestCases",
     project_key=(str, Field(description="Jira project key filter")),
     search_term=(Optional[str], Field(description="Optional search term to filter test cases", default=None)),
-    max_results=(Optional[int], Field(description="Maximum number of results to query from the API", default=1000)),
+    max_results=(Optional[int], Field(description="Maximum number of results to query from the API", default=1000, gt=0)),
     start_at=(Optional[int], Field(description="Zero-indexed starting position", default=0)),
     order_by=(Optional[str], Field(description="Field to order results by", default="name")),
     order_direction=(Optional[str], Field(description="Order direction", default="ASC")),
     archived=(Optional[bool], Field(description="Include archived test cases", default=False)),
     fields=(Optional[List[str]], Field(description="Fields to include in the response (default: key, name). Regular fields include key, name, id, labels, folder, etc. Custom fields can be included in the following ways:Individual custom fields via customFields.field_name format, All custom fields via customFields in the fields list", default=["key", "name"])),
-    limit_results=(Optional[int], Field(description="Maximum number of filtered results to return", default=10)),
+    limit_results=(Optional[int], Field(description="Maximum number of filtered results to return", default=10, gt=0)),
     folder_id=(Optional[str], Field(description="Filter test cases by folder ID", default=None)),
     folder_name=(Optional[str], Field(description="Filter test cases by folder name (full or partial)", default=None)),
     exact_folder_match=(Optional[bool], Field(description="Whether to match the folder name exactly or allow partial matches", default=False)),

alita_sdk/tools/zephyr_squad/__init__.py

@@ -6,7 +6,8 @@ from pydantic import create_model, BaseModel, Field, SecretStr
 from .api_wrapper import ZephyrSquadApiWrapper
 from ..base.tool import BaseAction
 from ..elitea_base import filter_missconfigured_index_tools
-from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
+from ..utils import clean_string, get_max_toolkit_length
+from ...runtime.utils.constants import TOOLKIT_NAME_META, TOOL_NAME_META, TOOLKIT_TYPE_META
 
 name = "zephyr_squad"
 
@@ -21,12 +22,10 @@ def get_tools(tool):
 
 class ZephyrSquadToolkit(BaseToolkit):
     tools: List[BaseTool] = []
-    toolkit_max_length: int = 0
 
     @staticmethod
     def toolkit_config_schema() -> BaseModel:
         selected_tools = {x['name']: x['args_schema'].schema() for x in ZephyrSquadApiWrapper.model_construct().get_available_tools()}
-        ZephyrSquadToolkit.toolkit_max_length = get_max_toolkit_length(selected_tools)
         return create_model(
             name,
             account_id=(str, Field(description="AccountID for the user that is going to be authenticating")),
@@ -43,18 +42,22 @@ class ZephyrSquadToolkit(BaseToolkit):
     @filter_missconfigured_index_tools
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         zephyr_api_wrapper = ZephyrSquadApiWrapper(**kwargs)
-        prefix = clean_string(toolkit_name, cls.toolkit_max_length) + TOOLKIT_SPLITTER if toolkit_name else ''
         available_tools = zephyr_api_wrapper.get_available_tools()
         tools = []
         for tool in available_tools:
             if selected_tools:
                 if tool["name"] not in selected_tools:
                     continue
+            description = tool["description"]
+            if toolkit_name:
+                description = f"Toolkit: {toolkit_name}\n{description}"
+            description = description[:1000]
             tools.append(BaseAction(
                 api_wrapper=zephyr_api_wrapper,
-                name=prefix + tool["name"],
-                description=tool["description"],
-                args_schema=tool["args_schema"]
+                name=tool["name"],
+                description=description,
+                args_schema=tool["args_schema"],
+                metadata={TOOLKIT_NAME_META: toolkit_name, TOOLKIT_TYPE_META: name, TOOL_NAME_META: tool["name"]} if toolkit_name else {TOOL_NAME_META: tool["name"]}
             ))
         return cls(tools=tools)
 

{alita_sdk-0.3.379.dist-info → alita_sdk-0.3.627.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.379
+Version: 0.3.627
 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 <lifedj27@gmail.com>, Artem Dubrovskiy <ad13box@gmail.com>
 License-Expression: Apache-2.0
@@ -18,19 +18,20 @@ Requires-Dist: python-dotenv~=1.0.1
 Requires-Dist: jinja2~=3.1.3
 Requires-Dist: pillow~=11.1.0
 Requires-Dist: requests~=2.3
-Requires-Dist: pydantic~=2.10.0
+Requires-Dist: pydantic~=2.12.0
 Requires-Dist: chardet==5.2.0
 Requires-Dist: fastapi==0.115.9
 Requires-Dist: httpcore==1.0.7
 Requires-Dist: urllib3>=2
 Requires-Dist: certifi==2024.8.30
+Requires-Dist: aiohttp>=3.9.0
 Provides-Extra: runtime
 Requires-Dist: streamlit>=1.28.0; extra == "runtime"
 Requires-Dist: langchain_core<0.4.0,>=0.3.76; extra == "runtime"
 Requires-Dist: langchain<0.4.0,>=0.3.22; extra == "runtime"
 Requires-Dist: langchain_community<0.4.0,>=0.3.7; extra == "runtime"
 Requires-Dist: langchain-openai<0.4.0,>=0.3.0; extra == "runtime"
-Requires-Dist: langchain-anthropic<0.4.0,>=0.3.10; extra == "runtime"
+Requires-Dist: langchain-anthropic<0.4.0,>=0.3.20; extra == "runtime"
 Requires-Dist: anthropic>=0.57.0; extra == "runtime"
 Requires-Dist: langgraph<0.5,>=0.4.8; extra == "runtime"
 Requires-Dist: langgraph-prebuilt==0.5.2; extra == "runtime"
@@ -67,12 +68,13 @@ Requires-Dist: pytesseract==0.3.13; extra == "runtime"
 Requires-Dist: markdown==3.5.1; extra == "runtime"
 Requires-Dist: beautifulsoup4==4.12.2; extra == "runtime"
 Requires-Dist: charset_normalizer==3.3.2; extra == "runtime"
-Requires-Dist: opentelemetry-exporter-otlp-proto-grpc==1.25.0; extra == "runtime"
-Requires-Dist: opentelemetry_api==1.25.0; extra == "runtime"
-Requires-Dist: opentelemetry_instrumentation==0.46b0; extra == "runtime"
-Requires-Dist: grpcio_status==1.63.0rc1; extra == "runtime"
-Requires-Dist: protobuf==4.25.7; extra == "runtime"
+Requires-Dist: opentelemetry-exporter-otlp-proto-grpc>=1.25.0; extra == "runtime"
+Requires-Dist: opentelemetry_api>=1.25.0; extra == "runtime"
+Requires-Dist: opentelemetry_instrumentation>=0.46b0; extra == "runtime"
+Requires-Dist: grpcio_status>=1.63.0rc1; extra == "runtime"
+Requires-Dist: protobuf>=4.25.7; extra == "runtime"
 Requires-Dist: langchain-sandbox>=0.0.6; extra == "runtime"
+Requires-Dist: langchain-mcp-adapters<0.2.0,>=0.1.14; extra == "runtime"
 Provides-Extra: tools
 Requires-Dist: dulwich==0.21.6; extra == "tools"
 Requires-Dist: paramiko==3.3.1; extra == "tools"
@@ -133,6 +135,7 @@ Provides-Extra: community
 Requires-Dist: retry-extended==0.2.3; extra == "community"
 Requires-Dist: pyobjtojson==0.3; extra == "community"
 Requires-Dist: elitea-analyse==0.1.2; extra == "community"
+Requires-Dist: networkx>=3.0; extra == "community"
 Provides-Extra: all
 Requires-Dist: alita-sdk[runtime]; extra == "all"
 Requires-Dist: alita-sdk[tools]; extra == "all"
@@ -143,6 +146,11 @@ Requires-Dist: pytest-cov; extra == "dev"
 Requires-Dist: black; extra == "dev"
 Requires-Dist: flake8; extra == "dev"
 Requires-Dist: mypy; extra == "dev"
+Provides-Extra: cli
+Requires-Dist: click>=8.1.0; extra == "cli"
+Requires-Dist: rich>=13.0.0; extra == "cli"
+Requires-Dist: pyyaml>=6.0; extra == "cli"
+Requires-Dist: langchain-mcp-adapters; extra == "cli"
 Dynamic: license-file
 
 Alita SDK
@@ -200,6 +208,144 @@ PROJECT_ID=<your_project_id>
 NOTE: these variables can be grabbed from your Elitea platform configuration page.
 ![Platform configuration](docs/readme_imgs/platform_config.png "Platform configuration")
 
+### Custom .env File Location
+
+By default, the CLI looks for `.env` files in the following order:
+1. `.alita/.env` (recommended)
+2. `.env` in the current directory
+
+You can override this by setting the `ALITA_ENV_FILE` environment variable:
+
+```bash
+export ALITA_ENV_FILE=/path/to/your/.env
+alita-cli agent chat
+```
+
+Using the CLI for Interactive Chat
+----------------------------------
+
+The Alita SDK includes a powerful CLI for interactive agent chat sessions.
+
+### Starting a Chat Session
+
+```bash
+# Interactive selection (shows all available agents + direct chat option)
+alita-cli agent chat
+
+# Chat with a specific local agent
+alita-cli agent chat .alita/agents/my-agent.agent.md
+
+# Chat with a platform agent
+alita-cli agent chat my-agent-name
+```
+
+### Direct Chat Mode (No Agent)
+
+You can start a chat session directly with the LLM without any agent configuration:
+
+```bash
+alita-cli agent chat
+# Select option 1: "Direct chat with model (no agent)"
+```
+
+This is useful for quick interactions or testing without setting up an agent.
+
+### Chat Commands
+
+During a chat session, you can use the following commands:
+
+| Command | Description |
+|---------|-------------|
+| `/help` | Show all available commands |
+| `/model` | Switch to a different model (preserves chat history) |
+| `/add_mcp` | Add an MCP server from your local mcp.json (preserves chat history) |
+| `/add_toolkit` | Add a toolkit from $ALITA_DIR/tools (preserves chat history) |
+| `/clear` | Clear conversation history |
+| `/history` | Show conversation history |
+| `/save` | Save conversation to file |
+| `exit` | End conversation |
+
+### Enhanced Input Features
+
+The chat interface includes readline-based input enhancements:
+
+| Feature | Key/Action |
+|---------|------------|
+| **Tab completion** | Press `Tab` to autocomplete commands (e.g., `/mo` → `/model`) |
+| **Command history** | `↑` / `↓` arrows to navigate through previous messages |
+| **Cursor movement** | `←` / `→` arrows to move within the current line |
+| **Start of line** | `Ctrl+A` jumps to the beginning of the line |
+| **End of line** | `Ctrl+E` jumps to the end of the line |
+| **Delete word** | `Ctrl+W` deletes the word before cursor |
+| **Clear line** | `Ctrl+U` clears from cursor to beginning of line |
+
+### Dynamic Model Switching
+
+Use `/model` to switch models on the fly:
+
+```
+> /model
+
+🔧 Select a model:
+
+#    Model              Type
+1    gpt-4o             openai
+2    gpt-4o-mini        openai
+3    claude-3-sonnet    anthropic
+
+Select model number: 1
+
+✓ Selected: gpt-4o
+╭──────────────────────────────────────────────────────────────╮
+│ ℹ Model switched to gpt-4o. Agent state reset, chat history  │
+│ preserved.                                                    │
+╰──────────────────────────────────────────────────────────────╯
+```
+
+### Adding MCP Servers Dynamically
+
+Use `/add_mcp` to add MCP servers during a chat session. Servers are loaded from your local `mcp.json` file (typically at `.alita/mcp.json`):
+
+```
+> /add_mcp
+
+🔌 Select an MCP server to add:
+
+#    Server       Type     Command/URL
+1    playwright   stdio    npx @playwright/mcp@latest
+2    filesystem   stdio    npx @anthropic/mcp-fs
+
+Select MCP server number: 1
+
+✓ Selected: playwright
+╭──────────────────────────────────────────────────────────────╮
+│ ℹ Added MCP: playwright. Agent state reset, chat history     │
+│ preserved.                                                    │
+╰──────────────────────────────────────────────────────────────╯
+```
+
+### Adding Toolkits Dynamically
+
+Use `/add_toolkit` to add toolkits from your `$ALITA_DIR/tools` directory (default: `.alita/tools`):
+
+```
+> /add_toolkit
+
+🧰 Select a toolkit to add:
+
+#    Toolkit    Type    File
+1    jira       jira    jira-config.json
+2    github     github  github-config.json
+
+Select toolkit number: 1
+
+✓ Selected: jira
+╭──────────────────────────────────────────────────────────────╮
+│ ℹ Added toolkit: jira. Agent state reset, chat history       │
+│ preserved.                                                    │
+╰──────────────────────────────────────────────────────────────╯
+```
+
 
 
 Using SDK with Streamlit for Local Development