File size: 678 Bytes
4dd9ebd b8b0b89 6882039 62fc389 91bf3c7 d920a9f 6882039 62fc389 56f2c57 eb6daca 032b798 1bdc7fe |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
# app_agent_config.py
from utils.tool_loader import ToolLoader
from utils.tool_config import tool_names
class AgentConfig:
def __init__(self):
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"
]
|