feat: Refactors model feature removal logic

Signed-off-by: -LAN- <laipz8200@outlook.com>
pull/20373/head
-LAN- 12 months ago
parent 00199c41bb
commit d954bec8e0
No known key found for this signature in database
GPG Key ID: 6BA0D108DED011FF

@ -356,7 +356,9 @@ class AgentNode(ToolNode):
def _remove_unsupported_model_features_for_old_version(self, model_schema: AIModelEntity) -> AIModelEntity:
if model_schema.features:
for feature in model_schema.features:
if feature.value not in AgentOldVersionModelFeatures:
for feature in model_schema.features[:]: # Create a copy to safely modify during iteration
try:
AgentOldVersionModelFeatures(feature) # Try to create enum member from value
except ValueError:
model_schema.features.remove(feature)
return model_schema

@ -1,4 +1,4 @@
from enum import Enum
from enum import Enum, StrEnum
from typing import Any, Literal, Union
from pydantic import BaseModel
@ -26,7 +26,7 @@ class ParamsAutoGenerated(Enum):
OPEN = 1
class AgentOldVersionModelFeatures(Enum):
class AgentOldVersionModelFeatures(StrEnum):
"""
Enum class for old SDK version llm feature.
"""

Loading…
Cancel
Save