morikotikk/browser-ai
1
1import os2from fastapi import FastAPI3from pydantic import BaseModel4from browser_use import Agent, Browser, BrowserConfig5from langchain_openai import ChatOpenAI6 7app = FastAPI()8 9# Используем Nvidia или OpenRouter (через OpenAI адаптер)10llm = ChatOpenAI(11 base_url="https://integrate.api.nvidia.com/v1",12 api_key=os.getenv("NVIDIA_API_KEY"),13 model="meta/llama-3.1-405b-instruct"14)15 16class Task(BaseModel):17 instruction: str18 19@app.post("/run")20async def run_task(task: Task):21 # Настройка браузера для работы с виртуальным дисплеем22 browser = Browser(config=BrowserConfig(headless=False)) 23 agent = Agent(task=task.instruction, llm=llm, browser=browser)24 25 result = await agent.run()26 return {"result": result.final_result()}