148 lines
3.2 KiB
Python
148 lines
3.2 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Utils package for qwen-agent.
|
|
"""
|
|
|
|
from .file_utils import (
|
|
download_file,
|
|
get_file_hash,
|
|
remove_file_or_directory,
|
|
extract_zip_file,
|
|
get_document_preview,
|
|
is_file_already_processed,
|
|
load_processed_files_log,
|
|
save_processed_files_log
|
|
)
|
|
|
|
from .dataset_manager import (
|
|
download_dataset_files,
|
|
generate_dataset_structure,
|
|
remove_dataset_directory,
|
|
remove_dataset_directory_by_key
|
|
)
|
|
|
|
from .project_manager import (
|
|
get_content_from_messages,
|
|
generate_project_readme,
|
|
save_project_readme,
|
|
get_project_status,
|
|
remove_project,
|
|
list_projects,
|
|
get_project_stats
|
|
)
|
|
|
|
# Import agent management modules
|
|
from .file_loaded_agent_manager import (
|
|
get_global_agent_manager,
|
|
init_global_agent_manager
|
|
)
|
|
|
|
from .agent_pool import (
|
|
AgentPool,
|
|
get_agent_pool,
|
|
set_agent_pool,
|
|
init_global_agent_pool,
|
|
get_agent_from_pool,
|
|
release_agent_to_pool
|
|
)
|
|
|
|
from .organize_dataset_files import (
|
|
is_file_already_processed,
|
|
organize_single_project_files,
|
|
organize_dataset_files
|
|
)
|
|
|
|
from .api_models import (
|
|
Message,
|
|
DatasetRequest,
|
|
ChatRequest,
|
|
FileProcessRequest,
|
|
DatasetResponse,
|
|
ChatCompletionResponse,
|
|
ChatResponse,
|
|
FileProcessResponse,
|
|
ErrorResponse,
|
|
HealthCheckResponse,
|
|
SystemStatusResponse,
|
|
CacheStatusResponse,
|
|
ProjectStatusResponse,
|
|
ProjectListResponse,
|
|
ProjectStatsResponse,
|
|
ProjectActionResponse,
|
|
QueueTaskRequest,
|
|
QueueTaskResponse,
|
|
QueueStatusResponse,
|
|
TaskStatusResponse,
|
|
create_success_response,
|
|
create_error_response,
|
|
create_chat_response
|
|
)
|
|
|
|
__all__ = [
|
|
# file_utils
|
|
'download_file',
|
|
'get_file_hash',
|
|
'remove_file_or_directory',
|
|
'extract_zip_file',
|
|
'get_document_preview',
|
|
'is_file_already_processed',
|
|
'load_processed_files_log',
|
|
'save_processed_files_log',
|
|
|
|
# dataset_manager
|
|
'download_dataset_files',
|
|
'generate_dataset_structure',
|
|
'remove_dataset_directory',
|
|
'remove_dataset_directory_by_key',
|
|
|
|
# project_manager
|
|
'get_content_from_messages',
|
|
'generate_project_readme',
|
|
'save_project_readme',
|
|
'get_project_status',
|
|
'remove_project',
|
|
'list_projects',
|
|
'get_project_stats',
|
|
|
|
# file_loaded_agent_manager
|
|
'get_global_agent_manager',
|
|
'init_global_agent_manager',
|
|
|
|
# agent_pool
|
|
'AgentPool',
|
|
'get_agent_pool',
|
|
'set_agent_pool',
|
|
'init_global_agent_pool',
|
|
'get_agent_from_pool',
|
|
'release_agent_to_pool',
|
|
|
|
# organize_dataset_files
|
|
'is_file_already_processed',
|
|
'organize_single_project_files',
|
|
'organize_dataset_files',
|
|
|
|
# api_models
|
|
'Message',
|
|
'DatasetRequest',
|
|
'ChatRequest',
|
|
'FileProcessRequest',
|
|
'DatasetResponse',
|
|
'ChatCompletionResponse',
|
|
'ChatResponse',
|
|
'FileProcessResponse',
|
|
'ErrorResponse',
|
|
'HealthCheckResponse',
|
|
'SystemStatusResponse',
|
|
'CacheStatusResponse',
|
|
'ProjectStatusResponse',
|
|
'ProjectListResponse',
|
|
'ProjectStatsResponse',
|
|
'ProjectActionResponse',
|
|
'QueueTaskRequest',
|
|
'QueueTaskResponse',
|
|
'QueueStatusResponse',
|
|
'TaskStatusResponse',
|
|
'create_success_response',
|
|
'create_error_response',
|
|
'create_chat_response'
|
|
] |