skyvern-langchain 0.1.0

No project description provided

Skyvern Langchain

This is a langchain integration for Skyvern.

Installation

pip install skyvern-langchain

Usage

Run a task(sync) with notebook style (code block)

:warning: :warning: if you want to run this code block, you need to run skvyern init command in your terminal to set up skyvern first.

import asyncio
from dotenv import load_dotenv
from langchain_openai import ChatOpenAI
from langchain.agents import initialize_agent, AgentType
from skyvern_langchain.agent import run_observer_task_v_2

load_dotenv()

llm = ChatOpenAI(model="gpt-4o", temperature=0)

agent = initialize_agent(
    llm=llm,
    tools=[run_observer_task_v_2],
    verbose=True,
    agent=AgentType.STRUCTURED_CHAT_ZERO_SHOT_REACT_DESCRIPTION,
)


async def main():
    # to run skyvern agent locally, must run `skvyern init` first
    print(await agent.ainvoke("Create a task by Skyvern. The task is about 'Navigate to the Hacker News homepage and get the top 3 posts.'"))


if __name__ == "__main__":
    asyncio.run(main())

Run a task(async) with notebook style (code block)

:warning: :warning: if you want to run this code block, you need to run skvyern init command in your terminal to set up skyvern first.

import asyncio
from dotenv import load_dotenv
from langchain_openai import ChatOpenAI
from langchain.agents import initialize_agent, AgentType
from skyvern_langchain.agent import create_observer_task_v_2, get_observer_task_v_2

from langchain_community.tools.sleep.tool import SleepTool

load_dotenv()

llm = ChatOpenAI(model="gpt-4o", temperature=0)

agent = initialize_agent(
    llm=llm,
    tools=[
        create_observer_task_v_2,
        get_observer_task_v_2,
        SleepTool(),
    ],
    verbose=True,
    agent=AgentType.STRUCTURED_CHAT_ZERO_SHOT_REACT_DESCRIPTION,
)


async def main():
    # use sleep tool to set up the polling logic until the task is completed, if you only want to run the create task, you can remove the sleep tool
    print(await agent.ainvoke("Create a task by Skyvern. The task is about 'Navigate to the Hacker News homepage and get the top 3 posts.' Then, get this task information until it's completed. The task information re-get interval should be 60s."))


if __name__ == "__main__":
    asyncio.run(main())

Run a task(sync) with client style (callling skyvern cloud api)

no need to run skvyern init command in your terminal to set up skyvern before using this integration.

import asyncio
from dotenv import load_dotenv
from langchain_openai import ChatOpenAI
from langchain.agents import initialize_agent, AgentType
from skyvern_langchain.client import RunSkyvernClientObserverTaskTool

load_dotenv()

llm = ChatOpenAI(model="gpt-4o", temperature=0)

run_observer_task = RunSkyvernClientObserverTaskTool(
    credential="<your_organization_api_key>",
)

agent = initialize_agent(
    llm=llm,
    tools=[run_observer_task],
    verbose=True,
    agent=AgentType.STRUCTURED_CHAT_ZERO_SHOT_REACT_DESCRIPTION,
)

print(await agent.ainvoke("Create a task by Skyvern. The task is about 'Navigate to the Hacker News homepage and get the top 3 posts.'"))

Run a task(async) with client style (callling skyvern cloud api)

no need to run skvyern init command in your terminal to set up skyvern before using this integration.

import asyncio
from dotenv import load_dotenv
from langchain_openai import ChatOpenAI
from langchain.agents import initialize_agent, AgentType
from skyvern_langchain.client import (
    RunSkyvernClientTaskTool,
    RunSkyvernClientObserverTaskTool,
    CreateSkyvernClientTaskTool,
    GetSkyvernClientTaskTool,
    CreateSkyvernClientObserverTaskV2Tool,
    GetSkyvernClientObserverTaskV2Tool,
)

from langchain_community.tools.sleep.tool import SleepTool

load_dotenv()

llm = ChatOpenAI(model="gpt-4o", temperature=0)

create_observer_task_v_2 = CreateSkyvernClientObserverTaskV2Tool(
    credential="<your_organization_api_key>",
)

get_observer_task_v_2 = GetSkyvernClientObserverTaskV2Tool(
    credential="<your_organization_api_key>",
)

agent = initialize_agent(
    llm=llm,
    tools=[
        create_observer_task_v_2,
        get_observer_task_v_2,
        SleepTool(),
    ],
    verbose=True,
    agent=AgentType.STRUCTURED_CHAT_ZERO_SHOT_REACT_DESCRIPTION,
)


async def main():
    # use sleep tool to set up the polling logic until the task is completed, if you only want to run the create task, you can remove the sleep tool
    print(await agent.ainvoke("Create a task by Skyvern. The task is about 'Navigate to the Hacker News homepage and get the top 3 posts.' Then, get this task information until it's completed. The task information re-get interval should be 60s."))


if __name__ == "__main__":
    asyncio.run(main())