소스 검색

fix: can not find model bug (#12051)

Signed-off-by: yihong0618 <zouzou0208@gmail.com>
tags/0.15.0
yihong 10 달 전
부모
커밋
7da4fb68da
No account linked to committer's email address
2개의 변경된 파일3개의 추가작업 그리고 3개의 파일을 삭제
  1. 1
    1
      api/core/entities/provider_configuration.py
  2. 2
    2
      api/services/entities/model_provider_entities.py

+ 1
- 1
api/core/entities/provider_configuration.py 파일 보기

@@ -872,7 +872,7 @@ class ProviderConfiguration(BaseModel):
# if llm name not in restricted llm list, remove it
restrict_model_names = [rm.model for rm in restrict_models]
for model in provider_models:
if model.model_type == ModelType.LLM and m.model not in restrict_model_names:
if model.model_type == ModelType.LLM and model.model not in restrict_model_names:
model.status = ModelStatus.NO_PERMISSION
elif not quota_configuration.is_valid:
model.status = ModelStatus.QUOTA_EXCEEDED

+ 2
- 2
api/services/entities/model_provider_entities.py 파일 보기

@@ -7,7 +7,6 @@ from configs import dify_config
from core.entities.model_entities import (
ModelWithProviderEntity,
ProviderModelWithStatusEntity,
SimpleModelProviderEntity,
)
from core.entities.provider_entities import QuotaConfiguration
from core.model_runtime.entities.common_entities import I18nObject
@@ -152,7 +151,8 @@ class ModelWithProviderEntityResponse(ModelWithProviderEntity):
Model with provider entity.
"""

provider: SimpleModelProviderEntity
# FIXME type error ignore here
provider: SimpleProviderEntityResponse # type: ignore

def __init__(self, model: ModelWithProviderEntity) -> None:
super().__init__(**model.model_dump())

Loading…
취소
저장