| # app_agent_config.py | |
| import streamlit as st | |
| from tool_loader import ToolLoader | |
| from tool_config import tool_names | |
| from logger import log_enabled | |
| from PIL import Image | |
| import numpy as np | |
| class AgentConfig: | |
| def __init__(self): | |
| self.tool_checkboxes = [] | |
| self.url_endpoint = "" | |
| self.image = [] | |
| self.document = "" | |
| self.log_enabled = False | |
| self.context = "" | |
| self.tool_loader = ToolLoader(tool_names) | |
| def configure(self): | |
| st.markdown("Change the agent's configuration here.") | |
| self.url_endpoint = st.selectbox("Select Inference URL", [ | |
| "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" | |
| ]) | |
| tool_loader = ToolLoader(tool_names) | |
| self.log_enabled = st.checkbox("Enable Logging") | |
| self.tool_checkboxes = [st.checkbox(f"{tool.name} --- {tool.description} ") for tool in tool_loader.tools] | |