fix: strip / from base_url

ej52 · Nov 4, 2023 · eeb85e6 · eeb85e6
1 parent 848a69b
commit eeb85e6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/custom_components/ollama_conversation/api.py b/custom_components/ollama_conversation/api.py
@@ -33,7 +33,7 @@ def __init__(
         session: aiohttp.ClientSession,
     ) -> None:
         """Sample API Client."""
-        self._base_url = base_url
+        self._base_url = base_url.rstrip("/")
         self._session = session
 
     async def async_get_heartbeat(self) -> any:

diff --git a/custom_components/ollama_conversation/config_flow.py b/custom_components/ollama_conversation/config_flow.py
@@ -85,7 +85,7 @@ async def async_step_user(
         errors = {}
         try:
             client = OllamaApiClient(
-                base_url=cv.url_no_path(user_input[CONF_BASE_URL]).rstrip("/"),
+                base_url=cv.url_no_path(user_input[CONF_BASE_URL]),
                 session=async_create_clientsession(self.hass),
             )
             await client.async_get_heartbeat()
@@ -129,7 +129,7 @@ async def async_step_init(
 
         try:
             client = OllamaApiClient(
-                base_url=cv.url_no_path(self.config_entry.data[CONF_BASE_URL]).rstrip("/"),
+                base_url=cv.url_no_path(self.config_entry.data[CONF_BASE_URL]),
                 session=async_create_clientsession(self.hass),
             )
             response = await client.async_get_models()