mirror of
https://github.com/FuzzingLabs/fuzzforge_ai.git
synced 2026-02-12 19:12:49 +00:00
baace0eac4
- Disable FuzzForge MCP connection (no Prefect backend) - Add a2a_wrapper module for programmatic A2A agent tasks - Add task_agent (LiteLLM A2A agent) on port 10900 - Create volumes/env/ for centralized Docker config - Update docker-compose.yml with task-agent service - Remove workflow_automation_skill from agent card
30 lines
777 B
Python
30 lines
777 B
Python
"""Root agent definition for the LiteLLM hot-swap shell."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from google.adk.agents import Agent
|
|
|
|
from .callbacks import (
|
|
before_agent_callback,
|
|
before_model_callback,
|
|
provide_instruction,
|
|
)
|
|
from .config import AGENT_DESCRIPTION, AGENT_NAME, DEFAULT_MODEL, DEFAULT_PROVIDER
|
|
from .state import HotSwapState
|
|
from .tools import HOTSWAP_TOOLS
|
|
|
|
_initial_state = HotSwapState(model=DEFAULT_MODEL, provider=DEFAULT_PROVIDER)
|
|
|
|
root_agent = Agent(
|
|
name=AGENT_NAME,
|
|
model=_initial_state.instantiate_llm(),
|
|
description=AGENT_DESCRIPTION,
|
|
instruction=provide_instruction,
|
|
tools=HOTSWAP_TOOLS,
|
|
before_agent_callback=before_agent_callback,
|
|
before_model_callback=before_model_callback,
|
|
)
|
|
|
|
|
|
__all__ = ["root_agent"]
|