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

This commit is contained in:
taokuizu 2024-03-01 05:07:15 +00:00 committed by GitHub
parent 8495ffaa45
commit b6d61a818e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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