59 lines
2.1 KiB
Python
59 lines
2.1 KiB
Python
from ahserver.serverenv import ServerEnv
|
|
from appPublic.worker import awaitify
|
|
from .core import (
|
|
harnessed_execute_tool,
|
|
harnessed_manage_memory,
|
|
harnessed_get_intelligent_memory_context,
|
|
harnessed_search_sessions,
|
|
harnessed_manage_skills,
|
|
harnessed_manage_remote_skills,
|
|
harnessed_get_config,
|
|
harnessed_get_current_user,
|
|
# Orchestrator functions
|
|
harnessed_create_workflow,
|
|
harnessed_add_task_to_workflow,
|
|
harnessed_execute_workflow,
|
|
harnessed_list_workflows,
|
|
harnessed_list_executions
|
|
)
|
|
from .config_functions import (
|
|
harnessed_get_agent_config,
|
|
harnessed_save_agent_config
|
|
)
|
|
from .llm_client import (
|
|
llm_chat,
|
|
llm_chat_stream,
|
|
llm_list_models,
|
|
llm_simple,
|
|
llm_get_config,
|
|
)
|
|
|
|
def load_harnessed_agent():
|
|
env = ServerEnv()
|
|
# Existing functions
|
|
env.harnessed_execute_tool = harnessed_execute_tool
|
|
env.harnessed_manage_memory = harnessed_manage_memory
|
|
env.harnessed_get_intelligent_memory_context = harnessed_get_intelligent_memory_context
|
|
env.harnessed_search_sessions = harnessed_search_sessions
|
|
env.harnessed_manage_skills = harnessed_manage_skills
|
|
env.harnessed_manage_remote_skills = harnessed_manage_remote_skills
|
|
env.harnessed_get_config = harnessed_get_config
|
|
env.harnessed_get_current_user = harnessed_get_current_user
|
|
|
|
# Orchestrator functions
|
|
env.harnessed_create_workflow = harnessed_create_workflow
|
|
env.harnessed_add_task_to_workflow = harnessed_add_task_to_workflow
|
|
env.harnessed_execute_workflow = harnessed_execute_workflow
|
|
env.harnessed_list_workflows = harnessed_list_workflows
|
|
env.harnessed_list_executions = harnessed_list_executions
|
|
|
|
# Configuration management functions
|
|
env.harnessed_get_agent_config = harnessed_get_agent_config
|
|
env.harnessed_save_agent_config = harnessed_save_agent_config
|
|
|
|
# LLM client -- calls supplier LLM APIs (OpenAI-compatible)
|
|
env.llm_chat = llm_chat
|
|
env.llm_chat_stream = llm_chat_stream
|
|
env.llm_list_models = llm_list_models
|
|
env.llm_simple = llm_simple
|
|
env.llm_get_config = llm_get_config |