test/dify
1
0
Fork 0
mirror of https://gitee.com/dify_ai/dify.git synced 2024-12-04 12:17:52 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix: Show rerank in system for localai (#4652)

Browse Source
This commit is contained in:
Giovanny Gutiérrez 2024-05-26 23:09:51 -05:00 committed by GitHub
parent 8152bc6fbf
commit 2deb23e00e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
api/core/model_runtime/model_providers/localai/rerank/rerank.py
View File

@ -5,6 +5,8 @@ import httpx
from requests import post from requests import post
from yarl import URL from yarl import URL
from core.model_runtime.entities.common_entities import I18nObject
from core.model_runtime.entities.model_entities import AIModelEntity, FetchFrom, ModelType
from core.model_runtime.entities.rerank_entities import RerankDocument, RerankResult from core.model_runtime.entities.rerank_entities import RerankDocument, RerankResult
from core.model_runtime.errors.invoke import ( from core.model_runtime.errors.invoke import (
InvokeAuthorizationError, InvokeAuthorizationError,
@ -118,3 +120,17 @@ class LocalaiRerankModel(RerankModel):
InvokeAuthorizationError: [httpx.HTTPStatusError], InvokeAuthorizationError: [httpx.HTTPStatusError],
InvokeBadRequestError: [httpx.RequestError] InvokeBadRequestError: [httpx.RequestError]
} }
def get_customizable_model_schema(self, model: str, credentials: dict) -> AIModelEntity:
"""
generate custom model entities from credentials
"""
entity = AIModelEntity(
model=model,
label=I18nObject(en_US=model),
model_type=ModelType.RERANK,
fetch_from=FetchFrom.CUSTOMIZABLE_MODEL,
model_properties={}
)
return entity