+openrouter
This commit is contained in:
1 parent
2c647bb6e5
commit
a8ec5ab98b
1 file changed
+4
-6
@@ -7,9 +7,8 @@ import asyncio
|
||||
from typing import Optional
|
||||
from fastapi import FastAPI, HTTPException
|
||||
from pydantic import BaseModel
|
||||
from browser_use import Agent, Browser, Controller
|
||||
from browser_use import Agent, Browser, ChatOpenAI
|
||||
from browser_use.agent.views import AgentHistoryList
|
||||
from langchain_openai import ChatOpenAI
|
||||
import uvicorn
|
||||
|
||||
# Обход проверки API ключа browser-use
|
||||
@@ -62,22 +61,21 @@ async def run_browser_task(task_id: str, task: str, max_steps: int, model: str):
|
||||
if not api_key:
|
||||
raise ValueError("OPENROUTER_API_KEY not set")
|
||||
|
||||
# Используем ChatOpenAI с base_url для OpenRouter
|
||||
llm = ChatOpenAI(
|
||||
model=model,
|
||||
openai_api_base="https://openrouter.ai/api/v1",
|
||||
openai_api_key=api_key,
|
||||
base_url="https://openrouter.ai/api/v1",
|
||||
api_key=api_key
|
||||
)
|
||||
|
||||
# Инициализация браузера
|
||||
browser = Browser()
|
||||
controller = Controller()
|
||||
|
||||
# Создание агента
|
||||
agent = Agent(
|
||||
task=task,
|
||||
llm=llm,
|
||||
browser=browser,
|
||||
controller=controller,
|
||||
)
|
||||
|
||||
# Выполнение задачи
|
||||
|
||||
Reference in new issue
Block a user