Browse Source

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

tags/0.5.8
taokuizu 1 year ago
parent
commit
b6d61a818e
No account linked to committer's email address
1 changed files with 5 additions and 2 deletions
  1. 5
    2
      api/core/model_runtime/model_providers/localai/llm/llm.py

+ 5
- 2
api/core/model_runtime/model_providers/localai/llm/llm.py 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

Loading…
Cancel
Save