File size: 1,866 Bytes
273c375
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6882039
62fc389
273c375
 
 
62fc389
273c375
 
 
 
 
 
 
 
 
 
 
 
 
 
62fc389
91bf3c7
d920a9f
6882039
 
62fc389
56f2c57
eb6daca
032b798
273c375
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
"""
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 = [
            "https://api-inference.huggingface.co/models/bigcode/starcoder",
            "https://api-inference.huggingface.co/models/OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5",
            "https://api-inference.huggingface.co/models/gpt2"
        ]