mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-25 21:36:52 +08:00
25d9cf2890
Co-authored-by: openhands <openhands@all-hands.dev> Co-authored-by: Graham Neubig <neubig@gmail.com> Co-authored-by: Engel Nyst <enyst@users.noreply.github.com>
41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
from typing import TYPE_CHECKING
|
|
|
|
import openhands.agenthub.loc_agent.function_calling as locagent_function_calling
|
|
from openhands.agenthub.codeact_agent import CodeActAgent
|
|
from openhands.core.config import AgentConfig
|
|
from openhands.core.logger import openhands_logger as logger
|
|
from openhands.llm.llm_registry import LLMRegistry
|
|
|
|
if TYPE_CHECKING:
|
|
from openhands.events.action import Action
|
|
from openhands.llm.llm import ModelResponse
|
|
|
|
|
|
class LocAgent(CodeActAgent):
|
|
VERSION = '1.0'
|
|
|
|
def __init__(
|
|
self,
|
|
config: AgentConfig,
|
|
llm_registry: LLMRegistry,
|
|
) -> None:
|
|
"""Initializes a new instance of the LocAgent class.
|
|
|
|
Parameters:
|
|
- llm (LLM): The llm to be used by this agent
|
|
- config (AgentConfig): The configuration for the agent
|
|
"""
|
|
|
|
super().__init__(config, llm_registry)
|
|
|
|
self.tools = locagent_function_calling.get_tools()
|
|
logger.debug(
|
|
f'TOOLS loaded for LocAgent: {", ".join([tool.get("function").get("name") for tool in self.tools])}'
|
|
)
|
|
|
|
def response_to_actions(self, response: 'ModelResponse') -> list['Action']:
|
|
return locagent_function_calling.response_to_actions(
|
|
response,
|
|
mcp_tool_names=list(self.mcp_tools.keys()),
|
|
)
|