fix: Replace path.join with urljoin. (#2631)

pull/2639/head
taokuizu 2 years ago committed by GitHub
parent 8495ffaa45
commit b6d61a818e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -1,6 +1,6 @@
from collections.abc import Generator from collections.abc import Generator
from os.path import join
from typing import cast from typing import cast
from urllib.parse import urljoin
from httpx import Timeout from httpx import Timeout
from openai import ( from openai import (
@ -313,10 +313,13 @@ class LocalAILarguageModel(LargeLanguageModel):
:param credentials: credentials dict :param credentials: credentials dict
:return: client kwargs :return: client kwargs
""" """
if not credentials['server_url'].endswith('/'):
credentials['server_url'] += '/'
client_kwargs = { client_kwargs = {
"timeout": Timeout(315.0, read=300.0, write=10.0, connect=5.0), "timeout": Timeout(315.0, read=300.0, write=10.0, connect=5.0),
"api_key": "1", "api_key": "1",
"base_url": join(credentials['server_url'], 'v1'), "base_url": urljoin(credentials['server_url'], 'v1'),
} }
return client_kwargs return client_kwargs

Loading…
Cancel
Save