mirror of
https://github.com/langgenius/dify.git
synced 2026-05-09 04:36:31 +08:00
[autofix.ci] apply automated fixes
This commit is contained in:
parent
6b87dda00e
commit
ed2a102953
@ -118,7 +118,7 @@ class LLMQuotaLayer(GraphEngineLayer):
|
||||
model_instance = cast("ParameterExtractorNode", node).model_instance
|
||||
case BuiltinNodeTypes.QUESTION_CLASSIFIER:
|
||||
typed_node: QuestionClassifierNode = cast("QuestionClassifierNode", node)
|
||||
model_instance = cast(PreparedLLMProtocol, getattr(typed_node, "_model_instance"))
|
||||
model_instance = cast(PreparedLLMProtocol, typed_node._model_instance)
|
||||
case _:
|
||||
return None
|
||||
except AttributeError:
|
||||
|
||||
@ -254,7 +254,7 @@ class PluginModelRuntime(ModelRuntime):
|
||||
stop=list(stop) if stop else None,
|
||||
stream=stream,
|
||||
)
|
||||
|
||||
|
||||
@overload
|
||||
def invoke_llm_with_structured_output(
|
||||
self,
|
||||
@ -294,10 +294,7 @@ class PluginModelRuntime(ModelRuntime):
|
||||
prompt_messages: Sequence[PromptMessage],
|
||||
stop: Sequence[str] | None,
|
||||
stream: bool,
|
||||
) -> (
|
||||
LLMResultWithStructuredOutput
|
||||
| Generator[LLMResultChunkWithStructuredOutput, None, None]
|
||||
):
|
||||
) -> LLMResultWithStructuredOutput | Generator[LLMResultChunkWithStructuredOutput, None, None]:
|
||||
# TODO: added to pass type check.
|
||||
# it is a new method from upstream that is not invoked at all.
|
||||
raise NotImplementedError
|
||||
|
||||
Loading…
Reference in New Issue
Block a user