notte-agent 0.0.dev0__py3-none-any.whl

notte_agent/main.py

@@ -0,0 +1,77 @@
+import asyncio
+from collections.abc import Callable
+
+from notte_browser.window import BrowserWindow
+from notte_core.credentials.base import BaseVault
+from notte_core.llms.engine import LlmModel
+from notte_sdk.types import DEFAULT_MAX_NB_STEPS, AgentCreateRequest
+
+from notte_agent.common.base import BaseAgent
+from notte_agent.common.notifier import BaseNotifier, NotifierAgent
+from notte_agent.common.types import AgentResponse
+from notte_agent.falco.agent import FalcoAgent, FalcoAgentConfig
+from notte_agent.falco.types import StepAgentOutput
+
+
+class Agent:
+    def __init__(
+        self,
+        headless: bool = False,
+        reasoning_model: LlmModel = LlmModel.default(),  # type: ignore[reportCallInDefaultInitializer]
+        max_steps: int = DEFAULT_MAX_NB_STEPS,
+        use_vision: bool = True,
+        # /!\ web security is disabled by default to increase agent performance
+        # turn it off if you need to input confidential information on trajectories
+        web_security: bool = False,
+        chrome_args: list[str] | None = None,
+        vault: BaseVault | None = None,
+        notifier: BaseNotifier | None = None,
+        window: BrowserWindow | None = None,
+    ):
+        # just validate the request to create type dependency
+        _ = AgentCreateRequest(
+            reasoning_model=reasoning_model,
+            use_vision=use_vision,
+            max_steps=max_steps,
+            persona_id=None,
+            vault_id=None,
+        )
+        self.config: FalcoAgentConfig = (
+            FalcoAgentConfig()
+            .use_vision(use_vision)
+            .model(reasoning_model, deep=True)
+            .map_session(
+                lambda session: (
+                    session.agent_mode()
+                    .steps(max_steps)
+                    .headless(headless)
+                    .web_security(web_security)
+                    .set_chrome_args(chrome_args)
+                )
+            )
+        )
+        self.vault: BaseVault | None = vault
+        self.notifier: BaseNotifier | None = notifier
+        self.window: BrowserWindow | None = window
+
+    def create_agent(
+        self,
+        step_callback: Callable[[str, StepAgentOutput], None] | None = None,
+    ) -> BaseAgent:
+        agent = FalcoAgent(
+            config=self.config,
+            vault=self.vault,
+            window=self.window,
+            step_callback=step_callback,
+        )
+        if self.notifier:
+            agent = NotifierAgent(agent, notifier=self.notifier)
+        return agent
+
+    async def arun(self, task: str, url: str | None = None) -> AgentResponse:
+        agent = self.create_agent()
+        return await agent.run(task, url=url)
+
+    def run(self, task: str, url: str | None = None) -> AgentResponse:
+        agent = self.create_agent()
+        return asyncio.run(agent.run(task, url=url))

notte_agent-0.0.dev0.dist-info/METADATA

@@ -0,0 +1,8 @@
+Metadata-Version: 2.4
+Name: notte-agent
+Version: 0.0.dev0
+Summary: Notte Web AI Agents
+Author-email: Notte Team  <hello@notte.cc>
+Requires-Python: >=3.11
+Requires-Dist: notte-browser==0.0.dev
+Requires-Dist: notte-core==0.0.dev

notte_agent-0.0.dev0.dist-info/RECORD

@@ -0,0 +1,34 @@
+notte_agent/README.md,sha256=qFOkfS1l8n2Jkxk-1KyrrHk4Xiz-Hj7qlJHEAmnJ0Ic,2231
+notte_agent/__init__.py,sha256=LfPglyBqY8BxOzM5GlSWRMnQsTsfFoKqQaGuW6mXvgA,150
+notte_agent/main.py,sha256=zReSxGNHl7L9_qQA0YvpDoc-U7Fv0gt88F6_FAzUafA,2832
+notte_agent/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+notte_agent/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+notte_agent/common/base.py,sha256=AJ07GLlM01iMZne-rp4SNm4NMEEireHHUsMdvODcinE,363
+notte_agent/common/captcha_detector.py,sha256=IKtcFsmVlXIogZRzVSEHn0VAjv-mFb35RKqVgr3tpok,2629
+notte_agent/common/config.py,sha256=z7Lt26BkqIC2WljmT1MtBQGo6orIVPQxFxF-_7-yJuk,8780
+notte_agent/common/conversation.py,sha256=lUs_uXT3JRgQSOcO1MbjcsQYc6ouqK5JQ4cCbVrD1hg,9511
+notte_agent/common/notifier.py,sha256=SNkKWcaNQeG3pMblFZRMQWQrKdcXFyAvBEVrN2wBax4,1569
+notte_agent/common/parser.py,sha256=fb5Det1QW65q7wMQLpNmDTjNIn_f781z8lZbxLlxXaA,2310
+notte_agent/common/perception.py,sha256=65JIeDrJDp_rlOqM96P2q3xLPsSqMe2Pzlpa6OH0phA,474
+notte_agent/common/prompt.py,sha256=-eDCNR51Cw2L4KGChkUDzg_AMoF08S4aqUDefHaBLVU,275
+notte_agent/common/safe_executor.py,sha256=OBzf6PXAaQAm3sEL4eZSJjSNglh50RgUrFX7SBY2SNU,3504
+notte_agent/common/trajectory_history.py,sha256=i2ivaJ2xtaM5Q-eUm-sX-yc73kedzENFmtfiiDxCRok,3818
+notte_agent/common/types.py,sha256=_HKlF6tMPHQbealJRAzeEajGbIdsKOER2wdyvOJ55UQ,1389
+notte_agent/common/validator.py,sha256=ZxecbIeIKAVVwroEk7QoyKx2lJCZGBxIxgC2j9GtK80,2739
+notte_agent/falco/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+notte_agent/falco/agent.py,sha256=PE4rXzDyZibwew8s8sQmhmXY_OTgYAXoYRWqBhTgVjs,15852
+notte_agent/falco/perception.py,sha256=td01ZbuwIoYQ8wJVDUzV41cmcsg8Y8PTjHnCf6ze00c,2671
+notte_agent/falco/prompt.py,sha256=IFJOK5eRDAaYm03XcdU6yOuBGpmMsC4yEtTLKnGgdvE,4912
+notte_agent/falco/trajectory_history.py,sha256=BURxgZeeFYnSvvLvBtZR8XmY2eXPcWt6b0N0SmX2Y_c,1453
+notte_agent/falco/types.py,sha256=ffrU0vHBrp5AIg_z8q55mz6rUikzoIAaOH22dSVp1N4,5215
+notte_agent/falco/prompts/system_prompt_multi_actions.md,sha256=Zc73x25eQOkFImKBjLsJKxbPzU6_7xgqjeYU5MrKTSQ,4716
+notte_agent/falco/prompts/system_prompt_single_action.md,sha256=lRLCgWB4UNhDcLH94UM_M3pO11ga9Tl--yKilu85evE,4980
+notte_agent/gufo/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+notte_agent/gufo/agent.py,sha256=LkbL5wO6sOa1FTSHxm9jab3b5QNxNl5yEbPtNURYgZk,8020
+notte_agent/gufo/parser.py,sha256=hhIJCwG7DUXpxZTnCAiwNA_nxG7JpDb322_KF7v-V1A,3054
+notte_agent/gufo/perception.py,sha256=eaAr4Hp8Qz0DH762SH9sKgS1LyQy7pX58sX2Zh3WWb8,1492
+notte_agent/gufo/prompt.py,sha256=ipyPbgRahrlyw7oQc7K6tMKJyDDM-VxyLlvR-FJrhEg,2274
+notte_agent/gufo/system.md,sha256=gFoMRVTfir-_asj8w25up5HvGCWWLc0PPDqMpCmXVJk,427
+notte_agent-0.0.dev0.dist-info/METADATA,sha256=osGZAi0Sq-yUprinFxBPOtw9O30nJ8Yf30vDkaEMEao,227
+notte_agent-0.0.dev0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+notte_agent-0.0.dev0.dist-info/RECORD,,

notte_agent-0.0.dev0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.27.0
+Root-Is-Purelib: true
+Tag: py3-none-any