Chroe: some misc cleanup (#23203)
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
@@ -322,7 +322,7 @@ class OpsTraceManager:
|
|||||||
: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:
|
||||||
|
@@ -27,7 +27,7 @@ class TimezoneConversionTool(BuiltinTool):
|
|||||||
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
|
||||||
|
|
||||||
|
@@ -597,7 +597,7 @@ def _extract_text_from_vtt(vtt_bytes: bytes) -> str:
|
|||||||
|
|
||||||
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
|
||||||
|
|
||||||
|
@@ -3,7 +3,7 @@ import io
|
|||||||
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
|
||||||
@@ -33,12 +33,10 @@ from core.model_runtime.entities.message_entities import (
|
|||||||
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
|
||||||
@@ -1006,21 +1004,6 @@ class LLMNode(BaseNode):
|
|||||||
)
|
)
|
||||||
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(
|
||||||
*,
|
*,
|
||||||
|
Reference in New Issue
Block a user