+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 typing import Optional
|
||||||
from fastapi import FastAPI, HTTPException
|
from fastapi import FastAPI, HTTPException
|
||||||
from pydantic import BaseModel
|
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 browser_use.agent.views import AgentHistoryList
|
||||||
from langchain_openai import ChatOpenAI
|
|
||||||
import uvicorn
|
import uvicorn
|
||||||
|
|
||||||
# Обход проверки API ключа browser-use
|
# Обход проверки 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:
|
if not api_key:
|
||||||
raise ValueError("OPENROUTER_API_KEY not set")
|
raise ValueError("OPENROUTER_API_KEY not set")
|
||||||
|
|
||||||
|
# Используем ChatOpenAI с base_url для OpenRouter
|
||||||
llm = ChatOpenAI(
|
llm = ChatOpenAI(
|
||||||
model=model,
|
model=model,
|
||||||
openai_api_base="https://openrouter.ai/api/v1",
|
base_url="https://openrouter.ai/api/v1",
|
||||||
openai_api_key=api_key,
|
api_key=api_key
|
||||||
)
|
)
|
||||||
|
|
||||||
# Инициализация браузера
|
# Инициализация браузера
|
||||||
browser = Browser()
|
browser = Browser()
|
||||||
controller = Controller()
|
|
||||||
|
|
||||||
# Создание агента
|
# Создание агента
|
||||||
agent = Agent(
|
agent = Agent(
|
||||||
task=task,
|
task=task,
|
||||||
llm=llm,
|
llm=llm,
|
||||||
browser=browser,
|
browser=browser,
|
||||||
controller=controller,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
# Выполнение задачи
|
# Выполнение задачи
|
||||||
|
|||||||
Reference in new issue
Block a user