diff --git a/pyproject.toml b/pyproject.toml index 27499a0..a11eb79 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -60,6 +60,7 @@ dependencies = [ "typer>=0.15.0", "questionary>=2.0.0", "cryptography>=44.0.0", + "pytest-asyncio>=1.4.0" ] [project.scripts] diff --git a/windows_use/providers/mistral/llm.py b/windows_use/providers/mistral/llm.py index 7e03891..808c6bf 100644 --- a/windows_use/providers/mistral/llm.py +++ b/windows_use/providers/mistral/llm.py @@ -4,7 +4,7 @@ from collections.abc import AsyncIterator, Iterator from typing import Any, overload -from mistralai import Mistral +from mistralai.client import Mistral from pydantic import BaseModel from windows_use.messages import ( diff --git a/windows_use/providers/openai/llm.py b/windows_use/providers/openai/llm.py index 68aa146..85f902f 100644 --- a/windows_use/providers/openai/llm.py +++ b/windows_use/providers/openai/llm.py @@ -72,6 +72,7 @@ def __init__( self.base_url = base_url or os.environ.get("OPENAI_BASE_URL") self.temperature = temperature self.extra_body = extra_body + self.reasoning_effort = reasoning_effort self.client = OpenAI( api_key=self.api_key,