File size: 1,089 Bytes
4dd9ebd 465b890 ce40f85 2fd0e0b 32c0f7f 6882039 62fc389 6882039 62fc389 56f2c57 eb6daca 1bdc7fe 62fc389 bfe8a00 |
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 |
# 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]
|