alita-sdk 0.3.175__py3-none-any.whl → 0.3.177__py3-none-any.whl

alita_sdk/tools/confluence/api_wrapper.py

@@ -1507,6 +1507,7 @@ class ConfluenceAPIWrapper(BaseVectorStoreToolApiWrapper):
                         'application/doc',  'application/docx',
                         'application/xls', 'application/xlsx',
                         'application/svg', 'application/html',
+                        'application/octet-stream'
                     ]
                     or file_ext in [
                         'xml', 'json', 'md', 'markdown', 'txt',

{alita_sdk-0.3.175.dist-info → alita_sdk-0.3.177.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.175
+Version: 0.3.177
 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
@@ -123,9 +123,9 @@ Requires-Dist: pysnc==1.1.10; extra == "tools"
 Requires-Dist: shortuuid==1.0.13; extra == "tools"
 Requires-Dist: yarl==1.17.1; extra == "tools"
 Requires-Dist: langmem==0.0.27; extra == "tools"
+Requires-Dist: textract-py3==2.1.1; extra == "tools"
 Provides-Extra: community
 Requires-Dist: retry-extended==0.2.3; extra == "community"
-Requires-Dist: browser-use==0.1.43; extra == "community"
 Requires-Dist: pyobjtojson==0.3; extra == "community"
 Requires-Dist: elitea-analyse==0.1.2; extra == "community"
 Provides-Extra: all

{alita_sdk-0.3.175.dist-info → alita_sdk-0.3.177.dist-info}/RECORD

@@ -1,5 +1,5 @@
 alita_sdk/__init__.py,sha256=fxeNiqiVpIFAJls31Oomifyrtd5gT9iPUTdkWjDOB2Y,656
-alita_sdk/community/__init__.py,sha256=2NBUS_1pmf65CHtWQDrTP185ewFbdrXHm_F0SIuZMbA,2974
+alita_sdk/community/__init__.py,sha256=8N7wWwPhoyOq3p8wlV3-pb3l3nJCR8TUrtV9iIPLU88,2523
 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=ADjjyV2z7XmQhOv873Mrz_-W9jWxhWpiDJ0nujm0eZs,4232
@@ -10,37 +10,6 @@ alita_sdk/community/analysis/gitlab_analyse/__init__.py,sha256=6PgqvD0cX5evpT_vg
 alita_sdk/community/analysis/gitlab_analyse/api_wrapper.py,sha256=xGX3io2rT_DJl1PMiP9d6lsw6xhr9L2h910raGk1s4w,6373
 alita_sdk/community/analysis/jira_analyse/__init__.py,sha256=oyAVkjq4KZmSA1RiAijeo4oh9hNDgFmOCkDilUJ8DWs,6023
 alita_sdk/community/analysis/jira_analyse/api_wrapper.py,sha256=Ui1GBWizIFGFOi98BtkPWxVITQbrv7xFZxxYjAFq66Q,9701
-alita_sdk/community/browseruse/__init__.py,sha256=9GhPp8VTuTPJHdlM06PlauHmCqbXR6zMwg_EV1OCpsg,3609
-alita_sdk/community/browseruse/api_wrapper.py,sha256=cu6Rik6Dugrlb4aomlKIwBHuTHmhWZhIfysXbO03-_c,10842
-alita_sdk/community/deep_researcher/__init__.py,sha256=Gq0VWr5v6Mr0m1vsZmIUCh5rITlZIi7TU5BDjfm70ow,3433
-alita_sdk/community/deep_researcher/api_wrapper.py,sha256=WoMkbGHcwDq_BsTnfehPEGu7QZP-mMQe1aWrqLxbMuk,4806
-alita_sdk/community/deep_researcher/deep_research.py,sha256=qAX5Ghtk130e6UQ9ciy2lmkv0o-rBuVGuzneu4V1Wwk,7818
-alita_sdk/community/deep_researcher/iterative_research.py,sha256=ROnIecAWufGXkExT1aUX7tLR_khifaqf_vRLNQ1UTps,16970
-alita_sdk/community/deep_researcher/llm_config.py,sha256=WPSrTy8K9URHvHxauO0gKVkua_8RcO1TFvo4Vfsdsmk,3098
-alita_sdk/community/deep_researcher/main.py,sha256=AAw6WOgx6zcL6bbQKlhIhZuDYgGQxdK7XEAn9f2N5Ik,2593
-alita_sdk/community/deep_researcher/agents/__init__.py,sha256=MG9dF6z2AuDniIghfhrfXupO5DP9t1vNzMJO2SksO2o,52
-alita_sdk/community/deep_researcher/agents/baseclass.py,sha256=vIw1ezRkGM5zCBFWrIRNlKEWhl5frdlfO-ywVavgrFM,6271
-alita_sdk/community/deep_researcher/agents/knowledge_gap_agent.py,sha256=XC648-7LQ3UQglDaMqvf-jWzegEt1ykiigUydM_ZuOI,3467
-alita_sdk/community/deep_researcher/agents/long_writer_agent.py,sha256=WDu_9qT5frX2rkPGmsAeUiD8Ram5UPllUg7-rKq4gUQ,10250
-alita_sdk/community/deep_researcher/agents/planner_agent.py,sha256=aGvpVn0Wrtmzrati87sSQrTkxNdX0P8XjNQwwENmx1U,5736
-alita_sdk/community/deep_researcher/agents/proofreader_agent.py,sha256=MfBoZMlDNKIGQp9c87bpIWBIdb0BBFkJ9ufqfFNRK2I,3618
-alita_sdk/community/deep_researcher/agents/thinking_agent.py,sha256=VYyHVuYVDBP7ZLrgrGKYYD9e6v11krb0XJ4fzcrakGU,2948
-alita_sdk/community/deep_researcher/agents/tool_selector_agent.py,sha256=2PK3CWKM1DVQnOjQNnj_5AY_OTMSH2tFkF9FzaW0Nvg,4026
-alita_sdk/community/deep_researcher/agents/writer_agent.py,sha256=-qyoJjEcXTyPbEvxNVP_ni_xPmq3VNZdkBJwKKpvZbQ,2699
-alita_sdk/community/deep_researcher/agents/tool_agents/__init__.py,sha256=zrbzyCDrUWhQ7ZRX-AK5fqYd-kcq8IHFZeFkmy77V3k,608
-alita_sdk/community/deep_researcher/agents/tool_agents/crawl_agent.py,sha256=VBSsPmiBpz9L8QT49RyQ43nkMYO6rJirYcbmzXWxieE,3552
-alita_sdk/community/deep_researcher/agents/tool_agents/search_agent.py,sha256=sQzUdt44Xm2sRB6lQ3nTTRtS41rVT7Zm4tACH3I5prA,3986
-alita_sdk/community/deep_researcher/agents/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-alita_sdk/community/deep_researcher/agents/utils/parse_output.py,sha256=_9Jq3bvChaRaKuZbayc92DmqXZaOoj7nLSOZzyUGXWw,4963
-alita_sdk/community/deep_researcher/examples/deep_example.py,sha256=xWDojQ8dwPLaZCPU2zkKfyeesFk2T1bwRlCO3hYqvEo,699
-alita_sdk/community/deep_researcher/examples/iterative_example.py,sha256=k291pvU3BnMkNXFdw6GdekvpXSztouv5CYRXXaBT_-4,863
-alita_sdk/community/deep_researcher/examples/report_plan_example.py,sha256=9xrLVksmnCPwhdgHqieqnmaW2b-DcGwB3vl6xXKiHs8,952
-alita_sdk/community/deep_researcher/tools/__init__.py,sha256=-gLuDJIkSeLW4BRsS-maAIIz6pcyr_VrSUs34G63dTg,88
-alita_sdk/community/deep_researcher/tools/crawl_website.py,sha256=oQ12KgpXB3P-6mbUVUONMmEqgsnZ_m5JTvrkjt4idik,4446
-alita_sdk/community/deep_researcher/tools/web_search.py,sha256=PucwnPCwJwMyaEfHGCy2OxNyCUhF7Uvv61M-yzNO_Yc,11036
-alita_sdk/community/deep_researcher/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-alita_sdk/community/deep_researcher/utils/md_to_pdf.py,sha256=EgCaUGLsP5-5F301aB60ZbwhhaxKK-EsSx0Ol_35fvw,246
-alita_sdk/community/deep_researcher/utils/os.py,sha256=Q1xX7c7_p7EmuzzXIAY9TDmraDNvU0GGcpfgIfWKQ2A,793
 alita_sdk/runtime/__init__.py,sha256=4W0UF-nl3QF2bvET5lnah4o24CoTwSoKXhuN0YnwvEE,828
 alita_sdk/runtime/clients/__init__.py,sha256=BdehU5GBztN1Qi1Wul0cqlU46FxUfMnI6Vq2Zd_oq1M,296
 alita_sdk/runtime/clients/artifact.py,sha256=4N2t5x3GibyXLq3Fvrv2o_VA7Z000yNfc-UN4eGsHZg,2679
@@ -160,18 +129,18 @@ alita_sdk/tools/browser/google_search_rag.py,sha256=QVHFbVwymiJGuno_HLSJOK1c_Mpg
 alita_sdk/tools/browser/utils.py,sha256=4k3YM_f1Kqlhjz9vt2pNsGkvCjhy-EmY3nvcwdFCsLA,2501
 alita_sdk/tools/browser/wiki.py,sha256=Qh3HBFd4dkS2VavXbFJOm4b8SjVSIe5xSD7CY1vEkKE,1126
 alita_sdk/tools/carrier/__init__.py,sha256=pP-nk-dpqOkrvwcRY_szgwqoowyVNl_GobD4Inp-Qus,4435
-alita_sdk/tools/carrier/api_wrapper.py,sha256=Y_Qznko3ReDZipDn64_IivehdN92hUpX0HLCfiLpYTw,8696
+alita_sdk/tools/carrier/api_wrapper.py,sha256=R8fyGesykTDJ1D2t7z1h_W9TO9B5GaqKaWI7OU9tdRE,8877
 alita_sdk/tools/carrier/backend_reports_tool.py,sha256=2Z1DCt6f0XTrQcEoynxMFUnenhtYMuIt_epq4YBSLDE,12234
-alita_sdk/tools/carrier/backend_tests_tool.py,sha256=Y1Va-VxDtnbRvrEdlwe63t-oDvqzQv5jxZW1eH0XeTY,6246
+alita_sdk/tools/carrier/backend_tests_tool.py,sha256=xfcp0-CHLW_yi5eNeoepd9w8He_1ls_knnhuWt_Kjv4,21723
 alita_sdk/tools/carrier/cancel_ui_test_tool.py,sha256=pD1sKEcZGBWJqFpgjeohMk93uuUPWruVJRPVVg90rpo,6438
-alita_sdk/tools/carrier/carrier_sdk.py,sha256=wb3d1W6dvo5NRLMiJcBcKy663J6IkbTwpKFTZX30QFQ,13672
+alita_sdk/tools/carrier/carrier_sdk.py,sha256=JDPgZ0MSUGEQf07ZBwBYpY-9RvYpXyyDj-cKZsx-3m0,14513
 alita_sdk/tools/carrier/create_ui_excel_report_tool.py,sha256=8aSpkyIGXsOBTo8Ye_6p19v8OOl1y7QW47IJxZ6QDgM,20163
-alita_sdk/tools/carrier/create_ui_test_tool.py,sha256=sHi7-D1uqIUHEyoywI92h6MdUVybKfBXs_XttTu-Ck4,8624
+alita_sdk/tools/carrier/create_ui_test_tool.py,sha256=knKvPOo9usI2XHqZtcbBEBzKwB9tS7GEl9KIX78vJiA,8184
 alita_sdk/tools/carrier/excel_reporter.py,sha256=fXptz7iaBDBcFSc8Ah8nZ9CSgugTONc5JMC1XcQEnfM,21487
 alita_sdk/tools/carrier/lighthouse_excel_reporter.py,sha256=mVuU63tl2n-Gntx9RuedjEU0U5AP1APKsSx1DvJs7wk,6684
-alita_sdk/tools/carrier/run_ui_test_tool.py,sha256=vmEtP-cpnOCyLLZfbw9Nq2ItGUFlcnsf1LmI_XyLrM8,21053
+alita_sdk/tools/carrier/run_ui_test_tool.py,sha256=Wqfxi_jyOU6XxYGsTje2ftgm8O7PJRXRDHUwWcw8opM,26277
 alita_sdk/tools/carrier/tickets_tool.py,sha256=d-wFyFWWTvV01o-hyssb2S-oLnr51b6tlNTUqA_CohY,8099
-alita_sdk/tools/carrier/tools.py,sha256=cCLYcNzC_z0-AaqB46fyt7iOeP20LStVQKI5Dg1zWA4,1588
+alita_sdk/tools/carrier/tools.py,sha256=6j0-_DJqOOBwMtpdSFHlOBcS44HWLBINh4RttpVN_cU,1679
 alita_sdk/tools/carrier/ui_reports_tool.py,sha256=Y6EstTRCa9d11ipFUFGOYlpiEhFx7aOQcgZ_M5Gd1lQ,13708
 alita_sdk/tools/carrier/update_ui_test_schedule_tool.py,sha256=jh9Q86cMCEqpsFopJPNIP0wlr7sYVa_3lhlq6lRmkGg,11850
 alita_sdk/tools/carrier/utils.py,sha256=rl7aq-F6ed_PapDM15w8EtS0BkgsjpDrNdKYuDCMOaI,4376
@@ -217,7 +186,7 @@ alita_sdk/tools/code/loaders/codesearcher.py,sha256=XoXXZtIQZhvjIwZlnl_4wVGHC-3s
 alita_sdk/tools/code/sonar/__init__.py,sha256=u8wpgXJ_shToLl3G9-XEtGDor5dhmsnurIImh1-e-U0,3165
 alita_sdk/tools/code/sonar/api_wrapper.py,sha256=nNqxcWN_6W8c0ckj-Er9HkNuAdgQLoWBXh5UyzNutis,2653
 alita_sdk/tools/confluence/__init__.py,sha256=1wGSP4CjPbfpdZLsjC1SVftTV7XSvW3QCAMHuh9tIxA,6885
-alita_sdk/tools/confluence/api_wrapper.py,sha256=nCU4wAh3f6EN3NVXya3BVZQX47mi7JYAzylxJJOvMMc,88725
+alita_sdk/tools/confluence/api_wrapper.py,sha256=zP8S85oVulhqrom1wlgelN-TSuKtPuYxwwGAlOSn_I0,88776
 alita_sdk/tools/confluence/loader.py,sha256=aHqgdIQMqkyRry8feHAhyd-a_ASEyW3JrV6epTRG6-c,9162
 alita_sdk/tools/confluence/utils.py,sha256=Lxo6dBD0OlvM4o0JuK6qeB_4LV9BptiwJA9e1vqNcDw,435
 alita_sdk/tools/custom_open_api/__init__.py,sha256=9aT5SPNPWcJC6jMZEM-3rUCXVULj_3-qJLQKmnreKNo,2537
@@ -326,8 +295,8 @@ alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=UHVQUVqcBc3SZvDfO78HSuBzwAsRw
 alita_sdk/tools/zephyr_squad/__init__.py,sha256=rq4jOb3lRW2GXvAguk4H1KinO5f-zpygzhBJf-E1Ucw,2773
 alita_sdk/tools/zephyr_squad/api_wrapper.py,sha256=iOMxyE7vOc_LwFB_nBMiSFXkNtvbptA4i-BrTlo7M0A,5854
 alita_sdk/tools/zephyr_squad/zephyr_squad_cloud_client.py,sha256=IYUJoMFOMA70knLhLtAnuGoy3OK80RuqeQZ710oyIxE,3631
-alita_sdk-0.3.175.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.175.dist-info/METADATA,sha256=3ecYvmsESpDa4HnPLjEW0y_G05ofONK9nPClKYPNah4,18757
-alita_sdk-0.3.175.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.175.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.175.dist-info/RECORD,,
+alita_sdk-0.3.177.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.177.dist-info/METADATA,sha256=avoSyHmQjQ--BnjrSryL8zpwaJdw0q8FJpy9Mc3R7fA,18753
+alita_sdk-0.3.177.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.177.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.177.dist-info/RECORD,,

alita_sdk/community/browseruse/__init__.py

@@ -1,73 +0,0 @@
-from typing import List, Literal, Optional, Dict, Any
-from langchain_community.agent_toolkits.base import BaseToolkit
-from .api_wrapper import BrowserUseAPIWrapper
-from langchain_core.tools import BaseTool
-from alita_sdk.tools.base.tool import BaseAction
-from pydantic import create_model, BaseModel, ConfigDict, Field
-from alita_sdk.tools.utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
-
-name = "browser_use"
-
-def get_tools(tool):
-    return BrowserUseToolkit().get_toolkit(
-        selected_tools=tool['settings'].get('selected_tools', []),
-        headless=tool['settings'].get('headless', True),
-        width=tool['settings'].get('width', 1280),
-        height=tool['settings'].get('height', 800),
-        cookies=tool['settings'].get('cookies', None),
-        disable_security=tool['settings'].get('disable_security', True),
-        proxy=tool['settings'].get('proxy', None),
-        extra_chromium_args=tool['settings'].get('extra_chromium_args', []),
-        bucket=tool['settings'].get('bucket', None),
-        alita=tool['settings'].get('alita'),
-        toolkit_name=tool.get('toolkit_name')
-    ).get_tools()
-
-toolkit_max_length = 25
-
-class BrowserUseToolkit(BaseToolkit):
-    tools: List[BaseTool] = []
-
-    @staticmethod
-    def toolkit_config_schema() -> BaseModel:
-        selected_tools = {x['name']: x['args_schema'].schema() for x in
-                          BrowserUseAPIWrapper.model_construct().get_available_tools()}
-        return create_model(
-            name,
-            headless=(bool, Field(description="Run browser in headless mode", default=True)),
-            width=(int, Field(description="Browser window width", default=1280)),
-            height=(int, Field(description="Browser window height", default=800)),
-            bucket=(Optional[str], Field(description="Bucket to store test results", default=None)),
-            cookies=(Optional[Dict[str, Any]], Field(description="Browser cookies as JSON", default=None)),
-            disable_security=(bool, Field(description="Disable browser security features", default=True)),
-            proxy=(Optional[Dict[str, str]], Field(description="Proxy settings", default=None)),
-            extra_chromium_args=(List[str], Field(description="Extra arguments to pass to the browser", default=[])),
-            selected_tools=(List[Literal[tuple(selected_tools)]],
-                           Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
-            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "Browser Use", "icon_url": None, "hidden": True}})
-        )
-    
-    @classmethod
-    def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
-        if selected_tools is None:
-            selected_tools = []
-        _api_wrapper = BrowserUseAPIWrapper(**kwargs)
-        prefix = clean_string(toolkit_name, toolkit_max_length) + TOOLKIT_SPLITTER if toolkit_name else ''
-        available_tools = _api_wrapper.get_available_tools()
-        tools = []
-        for tool in available_tools:
-            if selected_tools:
-                if tool["name"] not in selected_tools:
-                    continue
-            tools.append(
-                BaseAction(
-                    api_wrapper=_api_wrapper,
-                    name=prefix + tool["name"],
-                    description=f"Browser automation tool: {tool['name']}" + tool["description"],
-                    args_schema=tool["args_schema"],
-                )
-            )
-        return cls(tools=tools)
-
-    def get_tools(self):
-        return self.tools

alita_sdk/community/browseruse/api_wrapper.py

@@ -1,288 +0,0 @@
-from datetime import datetime
-from typing import Dict, List, Any, Optional, Type
-from pydantic import BaseModel, Field
-from browser_use import Agent, ActionResult, Browser, BrowserConfig, BrowserContextConfig
-from browser_use.agent.views import AgentHistoryList
-from playwright._impl._api_structures import ProxySettings
-from alita_sdk.tools.elitea_base import BaseToolApiWrapper
-from pydantic import create_model, Field, model_validator
-from tempfile import TemporaryDirectory, NamedTemporaryFile
-from browser_use.controller.service import Controller
-from langchain_core.callbacks import dispatch_custom_event
-from pyobjtojson import obj_to_json
-import os
-import asyncio
-
-import socket
-from browser_use.browser.utils.screen_resolution import get_screen_resolution, get_window_adjustments
-from playwright.async_api import Playwright, Browser as PlaywrightBrowser
-from browser_use.browser.chrome import (
-	CHROME_ARGS,
-	CHROME_DEBUG_PORT,
-	CHROME_DETERMINISTIC_RENDERING_ARGS,
-	CHROME_DISABLE_SECURITY_ARGS,
-	CHROME_DOCKER_ARGS,
-	CHROME_HEADLESS_ARGS,
-)
-IN_DOCKER = os.environ.get('IN_DOCKER', 'false').lower()[0] in 'ty1'
-
-class BrowserEx(Browser):
-    def __init__(self, config: BrowserConfig):
-        super().__init__(config)
-        self.config = config
-
-    async def _setup_builtin_browser(self, playwright: Playwright) -> PlaywrightBrowser:
-        """Sets up and returns a Playwright Browser instance with anti-detection measures."""
-        assert self.config.browser_binary_path is None, 'browser_binary_path should be None if trying to use the builtin browsers'
-
-		# Use the configured window size from new_context_config if available
-        if (
-			not self.config.headless
-			and hasattr(self.config, 'new_context_config')
-			and hasattr(self.config.new_context_config, 'browser_window_size')
-		):
-            screen_size = self.config.new_context_config.browser_window_size.model_dump()
-            offset_x, offset_y = get_window_adjustments()
-        elif self.config.headless:
-            screen_size = {'width': 1920, 'height': 1080}
-            offset_x, offset_y = 0, 0
-        else:
-            screen_size = get_screen_resolution()
-            offset_x, offset_y = get_window_adjustments()
-
-        chrome_args = {
-			f'--remote-debugging-port={self.config.chrome_remote_debugging_port}',
-			*CHROME_ARGS,
-			*(CHROME_DOCKER_ARGS if IN_DOCKER else []),
-			*(CHROME_HEADLESS_ARGS if self.config.headless else []),
-			*(CHROME_DISABLE_SECURITY_ARGS if self.config.disable_security else []),
-			*(CHROME_DETERMINISTIC_RENDERING_ARGS if self.config.deterministic_rendering else []),
-			f'--window-position={offset_x},{offset_y}',
-			f'--window-size={screen_size["width"]},{screen_size["height"]}',
-			*self.config.extra_browser_args,
-		}
-
-		# check if chrome remote debugging port is already taken,
-		# if so remove the remote-debugging-port arg to prevent conflicts
-        with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
-            if s.connect_ex(('localhost', self.config.chrome_remote_debugging_port)) == 0:
-                chrome_args.remove(f'--remote-debugging-port={self.config.chrome_remote_debugging_port}')
-
-        browser_class = getattr(playwright, self.config.browser_class)
-        args = {
-			'chromium': list(chrome_args),
-			'firefox': [
-				*{
-					'-no-remote',
-					*self.config.extra_browser_args,
-				}
-			],
-			'webkit': [
-				*{
-					'--no-startup-window',
-					*self.config.extra_browser_args,
-				}
-			],
-		}
-
-        browser = await browser_class.launch(
-			headless=self.config.headless,
-			channel='chromium',
-			args=args[self.config.browser_class],
-			proxy=self.config.proxy.model_dump() if self.config.proxy else None,
-			handle_sigterm=False,
-			handle_sigint=False,
-		)
-        return browser
-
-
-BrowserTask = create_model(
-    "BrowserTask",
-    task=(str, Field(description="Task to perform")),
-    max_steps=(Optional[int], Field(description="Maximum number of steps to perform")),
-    debug=(Optional[bool], Field(description="Whether debug mode is enabled")),
-    __config__=Field(description="Browser Use API Wrapper")
-)
-
-BrowserTasks = create_model(
-    "BrowserTasks",
-    tasks=(List[str], Field(description="List of tasks to perform")),
-    max_steps=(Optional[int], Field(description="Maximum number of steps to perform")),
-    debug=(Optional[bool], Field(description="Whether debug mode is enabled")),
-    __config__=Field(description="Browser Use API Wrapper")
-)
-
-async def thinking_processor(agent):
-    """Hook to be called after each step."""
-    if hasattr(agent, "state"):
-        history = agent.state.history
-    else:
-        history = None
-        return
-
-    # Process model thoughts
-    model_thoughts = obj_to_json(
-        obj=history.model_thoughts(),
-        check_circular=False
-    )
-    if len(model_thoughts) > 0:
-        model_thoughts_last_elem = model_thoughts[-1]
-        evalualtion = model_thoughts_last_elem.get('evaluation_previous_goal')
-        memory = model_thoughts_last_elem.get('memory')
-        next_goal = model_thoughts_last_elem.get('next_goal')
-        dispatch_custom_event(
-            name="thinking_step",
-            data={
-                "message": f"**Memory** : \n\n{memory}\n\n**Evaluation goal**:\n\n{evalualtion}\n\n**Next goal**:\n\n{next_goal}",
-                "tool_name": "task",
-                "toolkit": "browser_use"
-            }
-        )
-
-
-class DoneResult(BaseModel):
-	title: str
-	comments: str
-	hours_since_start: int
-
-gif_default_location = './agent_history.gif'
-default_bucket = 'browseruse'
-
-class BrowserUseAPIWrapper(BaseToolApiWrapper):
-    """Wrapper for Browser Use API."""
-    headless: bool = True
-    width: int = 1280
-    height: int = 800
-    use_vision: bool = False
-    trace_actions: bool = False
-    trace_actions_path: Optional[str] = None
-    cookies: Optional[Dict[str, Any]] = None
-    disable_security: bool = True
-    proxy: Any = None
-    extra_chromium_args: List[str] = []
-    client: Any = None # AlitaClient
-    artifact: Any = None # Artifact
-    llm: Any = None # LLMLikeObject
-    bucket: str = None
-    proxy_settings: Any = None
-    validate_output: bool = False
-    planner_llm: Any = None
-    browser_window_size: Dict[str, int] = None
-
-    
-    @model_validator(mode='before')
-    @classmethod
-    def validate_toolkit(cls, values):
-        """Validate toolkit parameters."""
-        values['proxy'] = ProxySettings(**values['proxy']) if values.get('proxy') else None
-        values['extra_chromium_args'] = values.get('extra_chromium_args') or []
-        values['browser_window_size'] = {"width": values.get('width', 1280), "height": values.get('height', 800)}
-        values['artifact'] = values.get('client').artifact(values.get('bucket', default_bucket))        
-        return values
-        
-        
-    def _create_browser(self):
-        cookies_file = None
-        if self.cookies:
-            cookies_file = NamedTemporaryFile(delete=False)
-            cookies_file.write(self.cookies)
-            cookies_file.close()
-            
-        context_config = BrowserContextConfig(
-                cookies_file=cookies_file,
-                wait_for_network_idle_page_load_time=10.0, # TODO: Make this configurable
-                highlight_elements=True,
-                browser_window_size=self.browser_window_size
-            )
-        browser_config = BrowserConfig(
-            headless=self.headless,
-            browser_class='chromium', # TODO: Make this configurable
-            disable_security=self.disable_security,
-            extra_chromium_args=self.extra_chromium_args,
-            proxy=self.proxy,
-            new_context_config=context_config
-        )
-        return BrowserEx(config=browser_config)
-    
-    def task(self, task: str, max_steps: Optional[int] = 20, debug: Optional[bool] = False):
-        """Perform a task using the browser."""
-        return asyncio.run(self._tasks([task], max_steps, debug))
-    
-    async def _tasks(self, tasks: List[str], max_steps: Optional[int] = 20, debug: Optional[bool] = False):
-        browser = self._create_browser()
-        context_config = BrowserContextConfig(
-                wait_for_network_idle_page_load_time=10.0, # TODO: Make this configurable
-                highlight_elements=True,
-                browser_window_size=self.browser_window_size
-            )
-        async with await browser.new_context(context_config) as browser:
-            start = tasks[0]
-            if len(tasks) == 1:
-                tasks = []
-            agent = Agent(
-                task=start,
-                llm=self.llm,
-                browser_context=browser,
-                max_actions_per_step=20,
-                use_vision=self.use_vision,
-                save_conversation_path=None,
-                generate_gif=True,
-                planner_llm=self.planner_llm,
-                controller=Controller(),
-                message_context = "Carefully check every step, and make sure to provide detailed feedback on the results.",
-                validate_output=self.validate_output
-            )
-            for task in tasks:
-                agent.add_new_task(task) 
-            history: AgentHistoryList = await agent.run(
-                max_steps=max_steps, 
-                on_step_end=thinking_processor
-                )
-        await browser.close()
-        files = self._save_execution(history.model_dump_json())
-
-        return {
-            "run_data": str(history.extracted_content()), 
-            "files": files
-        }
-
-    def _save_execution(self, data_content: Any):
-        """Saves tasks execution gif"""
-
-        try:
-            with open(gif_default_location, 'rb') as file:
-                artifact_data = file.read()
-        except FileNotFoundError:
-            artifact_data = None
-
-        filename = f"tasks_{datetime.now().strftime('%Y%m%d_%H%M%S')}"
-        files = []
-        if data_content:
-            self.artifact.create(f'{filename}.json', data_content)
-            files.append(f'{filename}.json')
-        
-        if artifact_data:
-            self.artifact.create(f'{filename}.gif', artifact_data)
-            files.append(f'{filename}.gif')
-        return files
-
-
-    def tasks(self, tasks: List[str], max_steps: Optional[int] = 20, debug: Optional[bool] = False):
-        """Perform a list of tasks using the browser."""
-        return asyncio.run(self._tasks(tasks, max_steps, debug))
-    
-    def get_available_tools(self):
-        return [
-            {
-                "name": "task",
-                "description": self.task.__doc__,
-                "args_schema": BrowserTask,
-                "ref": self.task
-            },
-            {
-                "name": "tasks",
-                "description": self.tasks.__doc__,
-                "args_schema": BrowserTasks,
-                "ref": self.tasks
-            }
-        ]

alita_sdk/community/deep_researcher/__init__.py

@@ -1,70 +0,0 @@
-from .deep_research import DeepResearcher
-from .iterative_research import IterativeResearcher
-from .agents.baseclass import ResearchRunner
-from .llm_config import LLMConfig
-
-__all__ = ["DeepResearcher", "IterativeResearcher", "ResearchRunner", "LLMConfig"]
-
-from typing import Any, List, Literal, Optional
-
-from langchain_core.tools import BaseToolkit, BaseTool
-from pydantic import BaseModel, ConfigDict, create_model, Field
-
-from .api_wrapper import DeepResearcherWrapper
-from ..base.tool import BaseAction
-from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
-
-name = "deep_researcher"
-
-def get_tools(tool):
-    return DeepResearcherToolkit().get_toolkit(
-        selected_tools=tool['settings'].get('selected_tools', []),
-        max_iterations=tool['settings'].get('max_iterations', 5),
-        max_time_minutes=tool['settings'].get('max_time_minutes', 10),
-        verbose=tool['settings'].get('verbose', False),
-        tracing=tool['settings'].get('tracing', False),
-        alita=tool['settings'].get('alita', None),
-        llm=tool['settings'].get('llm', None),
-        toolkit_name=tool.get('toolkit_name')
-    ).get_tools()
-
-
-class DeepResearcherToolkit(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 DeepResearcherWrapper.model_construct().get_available_tools()}
-        DeepResearcherToolkit.toolkit_max_length = get_max_toolkit_length(selected_tools)
-        return create_model(
-            name,
-            max_iterations=(int, Field(default=5, title="Max iterations", description="Maximum number of iterations for research", json_schema_extra={'toolkit_name': True, 'max_toolkit_length': DeepResearcherToolkit.toolkit_max_length})),
-            max_time_minutes=(int, Field(default=10, title="Max time (minutes)", description="Maximum time in minutes for research")),
-            verbose=(bool, Field(default=False, title="Verbose", description="Print status updates to the console")),
-            tracing=(bool, Field(default=False, title="Tracing", description="Enable tracing (only for OpenAI models)")),
-            selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
-            __config__=ConfigDict(json_schema_extra={'metadata': {"label": "Deep Researcher", "icon_url": "research-icon.svg"}})
-        )
-
-    @classmethod
-    def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
-        if selected_tools is None:
-            selected_tools = []
-        deep_researcher_api_wrapper = DeepResearcherWrapper(**kwargs)
-        prefix = clean_string(toolkit_name, cls.toolkit_max_length) + TOOLKIT_SPLITTER if toolkit_name else ''
-        available_tools = deep_researcher_api_wrapper.get_available_tools()
-        tools = []
-        for tool in available_tools:
-            if selected_tools and tool["name"] not in selected_tools:
-                continue
-            tools.append(BaseAction(
-                api_wrapper=deep_researcher_api_wrapper,
-                name=prefix + tool["name"],
-                description=tool["description"],
-                args_schema=tool["args_schema"]
-            ))
-        return cls(tools=tools)
-
-    def get_tools(self):
-        return self.tools

alita_sdk/community/deep_researcher/agents/__init__.py

@@ -1 +0,0 @@
-from .baseclass import ResearchAgent, ResearchRunner