mirror of
https://github.com/browser-use/browser-use
synced 2026-04-22 17:45:09 +02:00
60 lines
1.5 KiB
Python
60 lines
1.5 KiB
Python
"""
|
|
We have switched all of our code from langchain to openai.types.chat.chat_completion_message_param.
|
|
|
|
For easier transition we have
|
|
"""
|
|
|
|
from typing import Any, Protocol, TypeVar, overload, runtime_checkable
|
|
|
|
from pydantic import BaseModel
|
|
|
|
from browser_use.llm.messages import BaseMessage
|
|
from browser_use.llm.views import ChatInvokeCompletion
|
|
|
|
T = TypeVar('T', bound=BaseModel)
|
|
|
|
|
|
@runtime_checkable
|
|
class BaseChatModel(Protocol):
|
|
_verified_api_keys: bool = False
|
|
|
|
model: str
|
|
|
|
@property
|
|
def provider(self) -> str: ...
|
|
|
|
@property
|
|
def name(self) -> str: ...
|
|
|
|
@property
|
|
def model_name(self) -> str:
|
|
# for legacy support
|
|
return self.model
|
|
|
|
@overload
|
|
async def ainvoke(
|
|
self, messages: list[BaseMessage], output_format: None = None, **kwargs: Any
|
|
) -> ChatInvokeCompletion[str]: ...
|
|
|
|
@overload
|
|
async def ainvoke(self, messages: list[BaseMessage], output_format: type[T], **kwargs: Any) -> ChatInvokeCompletion[T]: ...
|
|
|
|
async def ainvoke(
|
|
self, messages: list[BaseMessage], output_format: type[T] | None = None, **kwargs: Any
|
|
) -> ChatInvokeCompletion[T] | ChatInvokeCompletion[str]: ...
|
|
|
|
@classmethod
|
|
def __get_pydantic_core_schema__(
|
|
cls,
|
|
source_type: type,
|
|
handler: Any,
|
|
) -> Any:
|
|
"""
|
|
Allow this Protocol to be used in Pydantic models -> very useful to typesafe the agent settings for example.
|
|
Returns a schema that allows any object (since this is a Protocol).
|
|
"""
|
|
from pydantic_core import core_schema
|
|
|
|
# Return a schema that accepts any object for Protocol types
|
|
return core_schema.any_schema()
|