mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-07-17 05:07:23 +08:00
fix: adding GPUStack chat model requires v1 suffix (#13237)
### What problem does this PR solve? Refer to issue: #13236 The base url for GPUStack chat model requires `/v1` suffix. For the other model type like `Embedding` or `Rerank`, the `/v1` suffix is not required and will be appended in code. So keep the same logic for chat model as other model type. ### Type of change - [X] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
@@ -1634,7 +1634,7 @@ class LiteLLMBase(ABC):
|
||||
elif self.provider == SupportedLiteLLMProvider.GPUStack:
|
||||
completion_args.update(
|
||||
{
|
||||
"api_base": self.base_url,
|
||||
"api_base": urljoin(self.base_url, "v1"),
|
||||
}
|
||||
)
|
||||
elif self.provider == SupportedLiteLLMProvider.Azure_OpenAI:
|
||||
|
||||
Reference in New Issue
Block a user