e607ccbb00
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>
18 lines
497 B
Python
18 lines
497 B
Python
from __future__ import annotations
|
|
|
|
import httpx
|
|
import respx
|
|
|
|
from tests import constants as c
|
|
from tests.agent_loop.e2e.providers.base import ProviderAPI
|
|
|
|
|
|
class MistralAPI(ProviderAPI):
|
|
base_url = c.MISTRAL_BASE_URL
|
|
post_path = c.CHAT_COMPLETIONS_PATH
|
|
|
|
def setup_router(self, router: respx.MockRouter) -> None:
|
|
super().setup_router(router)
|
|
router.get(c.CONNECTORS_BOOTSTRAP_PATH).mock(
|
|
return_value=httpx.Response(200, json={"connectors": []})
|
|
)
|