v2.18.0 (#843)
Co-authored-by: Clément Drouin <clement.drouin@mistral.ai> Co-authored-by: Clément Sirieix <clement.sirieix@mistral.ai> Co-authored-by: Cyprien <courtot.c@gmail.com> Co-authored-by: Guillaume LE GOFF <guillaume.lgf@gmail.com> Co-authored-by: Jean Burellier <sheplu@users.noreply.github.com> Co-authored-by: Kim-Adeline Miguel <51720070+kimadeline@users.noreply.github.com> Co-authored-by: Mathias Gesbert <mathias.gesbert@mistral.ai> Co-authored-by: Nelson PROIA <144663685+Nelson-PROIA@users.noreply.github.com> Co-authored-by: Paul VEZIA <166131032+le-codeur-rapide@users.noreply.github.com> Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai> Co-authored-by: Quentin <quentin.torroba@mistral.ai> Co-authored-by: Vincent G <10739306+VinceOPS@users.noreply.github.com> Co-authored-by: josephine-delas <57808586+josephine-delas@users.noreply.github.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
This commit is contained in:
parent
725d3a56ce
commit
e607ccbb00
242 changed files with 7372 additions and 1974 deletions
|
|
@ -35,6 +35,7 @@ class FakeBackend:
|
|||
self._requests_messages: list[list[LLMMessage]] = []
|
||||
self._requests_extra_headers: list[dict[str, str] | None] = []
|
||||
self._requests_metadata: list[dict[str, str] | None] = []
|
||||
self._requests_max_tokens: list[int | None] = []
|
||||
self._exception_to_raise = exception_to_raise
|
||||
|
||||
self._streams: list[list[LLMChunk]]
|
||||
|
|
@ -67,6 +68,10 @@ class FakeBackend:
|
|||
def requests_metadata(self) -> list[dict[str, str] | None]:
|
||||
return self._requests_metadata
|
||||
|
||||
@property
|
||||
def requests_max_tokens(self) -> list[int | None]:
|
||||
return self._requests_max_tokens
|
||||
|
||||
async def __aenter__(self):
|
||||
return self
|
||||
|
||||
|
|
@ -91,6 +96,7 @@ class FakeBackend:
|
|||
self._requests_messages.append(list(messages))
|
||||
self._requests_extra_headers.append(extra_headers)
|
||||
self._requests_metadata.append(metadata)
|
||||
self._requests_max_tokens.append(max_tokens)
|
||||
|
||||
if self._streams:
|
||||
stream = self._streams.pop(0)
|
||||
|
|
@ -119,6 +125,7 @@ class FakeBackend:
|
|||
self._requests_messages.append(list(messages))
|
||||
self._requests_extra_headers.append(extra_headers)
|
||||
self._requests_metadata.append(metadata)
|
||||
self._requests_max_tokens.append(max_tokens)
|
||||
|
||||
if self._streams:
|
||||
stream = list(self._streams.pop(0))
|
||||
|
|
|
|||
|
|
@ -50,7 +50,7 @@ class FakeConnectorRegistry(ConnectorRegistry):
|
|||
connector_name, ConnectorAuthAction.NONE
|
||||
)
|
||||
|
||||
def get_tools(self) -> dict[str, type[BaseTool]]:
|
||||
def get_tools(self, *, force_refresh: bool = False) -> dict[str, type[BaseTool]]:
|
||||
if self._cache is None:
|
||||
self._build_cache()
|
||||
|
||||
|
|
@ -60,8 +60,10 @@ class FakeConnectorRegistry(ConnectorRegistry):
|
|||
result.update(tools)
|
||||
return result
|
||||
|
||||
async def get_tools_async(self) -> dict[str, type[BaseTool]]:
|
||||
return self.get_tools()
|
||||
async def get_tools_async(
|
||||
self, *, force_refresh: bool = False
|
||||
) -> dict[str, type[BaseTool]]:
|
||||
return self.get_tools(force_refresh=force_refresh)
|
||||
|
||||
async def get_auth_url(self, alias: str) -> str | None:
|
||||
"""Return a fake auth URL for connectors that have no tools (not connected)."""
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue