|
|
|
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" |
|
] |
|
|
|
|
