liqiang
b119135836
Build latest book artifacts / build (push) Canceled after 0s
dependency resolution / resolve (3.11) (push) Canceled after 0s
dependency resolution / resolve (3.13) (push) Canceled after 0s
deploy-pages / build (push) Canceled after 0s
deploy-pages / deploy (push) Canceled after 0s
i18n consistency check / check (push) Canceled after 0s
provider adoption tests / test (chapter2/context-compression) (push) Canceled after 0s
provider adoption tests / test (chapter2/prompt-injection) (push) Canceled after 0s
provider adoption tests / test (chapter2/system-hint) (push) Canceled after 0s
provider adoption tests / test (chapter3/log-sanitization) (push) Canceled after 0s
web-search-agent tests / test (push) Canceled after 0s
web-search-agent tests / agentbook (push) Canceled after 0s
58 lines
1.5 KiB
Python
58 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) -> ChatInvokeCompletion[str]: ...
|
|
|
|
@overload
|
|
async def ainvoke(self, messages: list[BaseMessage], output_format: type[T]) -> ChatInvokeCompletion[T]: ...
|
|
|
|
async def ainvoke(
|
|
self, messages: list[BaseMessage], output_format: type[T] | None = None
|
|
) -> 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()
|