CoolFace
Apppublic

morikotikk/browser-ai

sourceHugging Faceupdated 8mo agoView on Hugging Face
1likes
app.py26 linesDownload Raw Back to root
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()}