浏览代码

Chroe: some misc cleanup (#23203)

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
tags/1.7.2
Yongtao Huang 3 个月前
父节点
当前提交
a434f6240f
没有帐户链接到提交者的电子邮件

+ 1
- 1
api/core/ops/ops_trace_manager.py 查看文件

:return: :return:
""" """
# auth check # auth check
if enabled == True:
if enabled:
try: try:
provider_config_map[tracing_provider] provider_config_map[tracing_provider]
except KeyError: except KeyError:

+ 1
- 1
api/core/tools/builtin_tool/providers/time/tools/timezone_conversion.py 查看文件

target_time = self.timezone_convert(current_time, current_timezone, target_timezone) # type: ignore target_time = self.timezone_convert(current_time, current_timezone, target_timezone) # type: ignore
if not target_time: if not target_time:
yield self.create_text_message( yield self.create_text_message(
f"Invalid datatime and timezone: {current_time},{current_timezone},{target_timezone}"
f"Invalid datetime and timezone: {current_time},{current_timezone},{target_timezone}"
) )
return return



+ 1
- 1
api/core/workflow/nodes/document_extractor/node.py 查看文件



for i in range(1, len(raw_results)): for i in range(1, len(raw_results)):
spk, txt = raw_results[i] spk, txt = raw_results[i]
if spk == None:
if spk is None:
merged_results.append((None, current_text)) merged_results.append((None, current_text))
continue continue



+ 1
- 18
api/core/workflow/nodes/llm/node.py 查看文件

import json import json
import logging import logging
from collections.abc import Generator, Mapping, Sequence from collections.abc import Generator, Mapping, Sequence
from typing import TYPE_CHECKING, Any, Optional, cast
from typing import TYPE_CHECKING, Any, Optional


from core.app.entities.app_invoke_entities import ModelConfigWithCredentialsEntity from core.app.entities.app_invoke_entities import ModelConfigWithCredentialsEntity
from core.file import FileType, file_manager from core.file import FileType, file_manager
UserPromptMessage, UserPromptMessage,
) )
from core.model_runtime.entities.model_entities import ( from core.model_runtime.entities.model_entities import (
AIModelEntity,
ModelFeature, ModelFeature,
ModelPropertyKey, ModelPropertyKey,
ModelType, ModelType,
) )
from core.model_runtime.model_providers.__base.large_language_model import LargeLanguageModel
from core.model_runtime.utils.encoders import jsonable_encoder from core.model_runtime.utils.encoders import jsonable_encoder
from core.prompt.entities.advanced_prompt_entities import CompletionModelPromptTemplate, MemoryConfig from core.prompt.entities.advanced_prompt_entities import CompletionModelPromptTemplate, MemoryConfig
from core.prompt.utils.prompt_message_util import PromptMessageUtil from core.prompt.utils.prompt_message_util import PromptMessageUtil
) )
return saved_file return saved_file


def _fetch_model_schema(self, provider: str) -> AIModelEntity | None:
"""
Fetch model schema
"""
model_name = self._node_data.model.name
model_manager = ModelManager()
model_instance = model_manager.get_model_instance(
tenant_id=self.tenant_id, model_type=ModelType.LLM, provider=provider, model=model_name
)
model_type_instance = model_instance.model_type_instance
model_type_instance = cast(LargeLanguageModel, model_type_instance)
model_credentials = model_instance.credentials
model_schema = model_type_instance.get_model_schema(model_name, model_credentials)
return model_schema

@staticmethod @staticmethod
def fetch_structured_output_schema( def fetch_structured_output_schema(
*, *,

正在加载...
取消
保存