fix: Azure AI Foundry model cannot be used in the workflow (#13323)
Signed-off-by: -LAN- <laipz8200@outlook.com>
This commit is contained in:
parent
3f42fabff8
commit
f9515901cc
2 changed files with 4 additions and 3 deletions
|
|
@ -4,6 +4,7 @@ from typing import Any, Optional
|
|||
from pydantic import BaseModel, Field, field_validator
|
||||
|
||||
from core.model_runtime.entities import ImagePromptMessageContent
|
||||
from core.model_runtime.entities.llm_entities import LLMMode
|
||||
from core.prompt.entities.advanced_prompt_entities import ChatModelMessage, CompletionModelPromptTemplate, MemoryConfig
|
||||
from core.workflow.entities.variable_entities import VariableSelector
|
||||
from core.workflow.nodes.base import BaseNodeData
|
||||
|
|
@ -12,7 +13,7 @@ from core.workflow.nodes.base import BaseNodeData
|
|||
class ModelConfig(BaseModel):
|
||||
provider: str
|
||||
name: str
|
||||
mode: str
|
||||
mode: LLMMode = LLMMode.COMPLETION
|
||||
completion_params: dict[str, Any] = {}
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue