From b6d61a818e82f47e8f9c52e90b3d9b6f54623104 Mon Sep 17 00:00:00 2001 From: taokuizu Date: Fri, 1 Mar 2024 05:07:15 +0000 Subject: [PATCH] fix: Replace path.join with urljoin. (#2631) --- api/core/model_runtime/model_providers/localai/llm/llm.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/api/core/model_runtime/model_providers/localai/llm/llm.py b/api/core/model_runtime/model_providers/localai/llm/llm.py index 694f5891f9..3c805682f3 100644 --- a/api/core/model_runtime/model_providers/localai/llm/llm.py +++ b/api/core/model_runtime/model_providers/localai/llm/llm.py @@ -1,6 +1,6 @@ from collections.abc import Generator -from os.path import join from typing import cast +from urllib.parse import urljoin from httpx import Timeout from openai import ( @@ -313,10 +313,13 @@ class LocalAILarguageModel(LargeLanguageModel): :param credentials: credentials dict :return: client kwargs """ + if not credentials['server_url'].endswith('/'): + credentials['server_url'] += '/' + client_kwargs = { "timeout": Timeout(315.0, read=300.0, write=10.0, connect=5.0), "api_key": "1", - "base_url": join(credentials['server_url'], 'v1'), + "base_url": urljoin(credentials['server_url'], 'v1'), } return client_kwargs