v2.9.1 (#644)
Co-authored-by: Brice Carpentier <brice.carpentier@mistral.ai> Co-authored-by: Clément Drouin <clement.drouin@mistral.ai> Co-authored-by: Clément Sirieix <clement.sirieix@mistral.ai> Co-authored-by: Kim-Adeline Miguel <51720070+kimadeline@users.noreply.github.com> Co-authored-by: Lucas Marandat <31749711+lucasmrdt@users.noreply.github.com> Co-authored-by: Michel Thomazo <51709227+michelTho@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: Mistral Vibe <vibe@mistral.ai>
This commit is contained in:
parent
632ea8c032
commit
1fd7eea289
63 changed files with 3482 additions and 301 deletions
|
|
@ -21,9 +21,11 @@ class FakeConnectorRegistry(ConnectorRegistry):
|
|||
self._cache = {}
|
||||
self._connector_names = []
|
||||
self._connector_connected = {}
|
||||
self._alias_to_id = {}
|
||||
for connector_name, tools in self._fake_connectors.items():
|
||||
alias = _normalize_name(connector_name)
|
||||
connector_id = f"fake-id-{connector_name}"
|
||||
self._alias_to_id[alias] = connector_id
|
||||
tool_map: dict[str, type[BaseTool]] = {}
|
||||
for remote in tools:
|
||||
proxy_cls = create_connector_proxy_tool_class(
|
||||
|
|
@ -50,3 +52,13 @@ class FakeConnectorRegistry(ConnectorRegistry):
|
|||
|
||||
async def get_tools_async(self) -> dict[str, type[BaseTool]]:
|
||||
return self.get_tools()
|
||||
|
||||
async def get_auth_url(self, alias: str) -> str | None:
|
||||
"""Return a fake auth URL for connectors that have no tools (not connected)."""
|
||||
if not self.is_connected(alias):
|
||||
return f"https://fake-auth.example.com/{alias}"
|
||||
return None
|
||||
|
||||
async def refresh_connector_async(self, alias: str) -> dict[str, type[BaseTool]]:
|
||||
"""No-op refresh for tests."""
|
||||
return {}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue