agent-reference-implemenation / app_agent_config.py
Chris4K's picture
updates
273c375
raw
history blame
1.87 kB
"""
Module: app_agent_config
This module defines the AgentConfig class, which holds configuration settings for the agent.
Dependencies:
- utils.tool_loader: Module providing the ToolLoader class for loading tools.
- utils.tool_config: Module providing tool_names for configuration.
Classes:
- AgentConfig: A class for managing configuration settings for the agent.
"""
from utils.tool_loader import ToolLoader # Importing ToolLoader class from utils.tool_loader module
from utils.tool_config import tool_names # Importing tool_names from utils.tool_config module
class AgentConfig:
"""
A class for managing configuration settings for the agent.
"""
def __init__(self):
"""
Initializes an instance of the AgentConfig class.
Attributes:
- url_endpoint (str): The URL endpoint for the agent.
- tool_checkboxes (list): Checkboxes for available tools.
- s_tool_checkboxes (list): Selected checkboxes for tools.
- image (list): Image data.
- document (str): Document data.
- log_enabled (bool): Flag indicating whether logging is enabled.
- context (str): Context data.
- tool_loader (ToolLoader): Instance of ToolLoader class for loading tools.
- agent_urls (list): URLs for different agents.
"""
self.url_endpoint = ""
self.tool_checkboxes = []
self.s_tool_checkboxes = []
self.image = []
self.document = ""
self.log_enabled = False
self.context = ""
self.tool_loader = ToolLoader(tool_names)
self.agent_urls = [
"/static-proxy?url=https%3A%2F%2Fapi-inference.huggingface.co%2Fmodels%2Fbigcode%2Fstarcoder%26quot%3B%3C%2Fspan%3E%2C%3C!-- HTML_TAG_END -->
"/static-proxy?url=https%3A%2F%2Fapi-inference.huggingface.co%2Fmodels%2FOpenAssistant%2Foasst-sft-4-pythia-12b-epoch-3.5%26quot%3B%3C%2Fspan%3E%2C%3C!-- HTML_TAG_END -->
"/static-proxy?url=https%3A%2F%2Fapi-inference.huggingface.co%2Fmodels%2Fgpt2%26quot%3B%3C%2Fspan%3E%3C!-- HTML_TAG_END -->
]