Unverified Commit b6d61a81 authored by taokuizu's avatar taokuizu Committed by GitHub

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

parent 8495ffaa
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): ...@@ -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
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment