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