src/frontends/streamlit/frontend/side_bar.py [23:112]: - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - HELP_GCS_CHECKBOX, HELP_MESSAGE_MULTIMODALITY, upload_files_to_gcs, ) EMPTY_CHAT_NAME = "Empty chat" NUM_CHAT_IN_RECENT = 3 DEFAULT_BASE_URL = "http://localhost:8000/" DEFAULT_REMOTE_AGENT_ENGINE_ID = "N/A" if os.path.exists("deployment_metadata.json"): with open("deployment_metadata.json") as f: DEFAULT_REMOTE_AGENT_ENGINE_ID = json.load(f)["remote_agent_engine_id"] DEFAULT_AGENT_CALLABLE_PATH = "app.agent_engine_app.AgentEngineApp" class SideBar: """Manages the sidebar components of the Streamlit application.""" def __init__(self, st: Any) -> None: """ Initialize the SideBar. Args: st (Any): The Streamlit object for rendering UI components. """ self.st = st def init_side_bar(self) -> None: """Initialize and render the sidebar components.""" with self.st.sidebar: default_agent_type = ( "Remote URL" if os.path.exists("Dockerfile") else "Local Agent" ) use_agent_path = self.st.selectbox( "Select Agent Type", ["Local Agent", "Remote Agent Engine ID", "Remote URL"], index=["Local Agent", "Remote Agent Engine ID", "Remote URL"].index( default_agent_type ), help="'Local Agent' uses a local implementation, 'Remote Agent Engine ID' connects to a deployed Vertex AI agent, and 'Remote URL' connects to a custom endpoint.", ) if use_agent_path == "Local Agent": self.agent_callable_path = self.st.text_input( label="Agent Callable Path", value=os.environ.get( "AGENT_CALLABLE_PATH", DEFAULT_AGENT_CALLABLE_PATH ), ) self.remote_agent_engine_id = None self.url_input_field = None self.should_authenticate_request = False elif use_agent_path == "Remote Agent Engine ID": self.remote_agent_engine_id = self.st.text_input( label="Remote Agent Engine ID", value=os.environ.get( "REMOTE_AGENT_ENGINE_ID", DEFAULT_REMOTE_AGENT_ENGINE_ID ), ) self.agent_callable_path = None self.url_input_field = None self.should_authenticate_request = False else: self.url_input_field = self.st.text_input( label="Service URL", value=os.environ.get("SERVICE_URL", DEFAULT_BASE_URL), ) self.should_authenticate_request = self.st.checkbox( label="Authenticate request", value=False, help="If checked, any request to the server will contain an" "Identity token to allow authentication. " "See the Cloud Run documentation to know more about authentication:" "https://cloud.google.com/run/docs/authenticating/service-to-service", ) self.agent_callable_path = None self.remote_agent_engine_id = None col1, col2, col3 = self.st.columns(3) with col1: if self.st.button("+ New chat"): if ( len( self.st.session_state.user_chats[ self.st.session_state["session_id"] ]["messages"] ) > 0 ): - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - src/frontends/streamlit_adk/frontend/side_bar.py [23:112]: - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - HELP_GCS_CHECKBOX, HELP_MESSAGE_MULTIMODALITY, upload_files_to_gcs, ) EMPTY_CHAT_NAME = "Empty chat" NUM_CHAT_IN_RECENT = 3 DEFAULT_BASE_URL = "http://localhost:8000/" DEFAULT_REMOTE_AGENT_ENGINE_ID = "N/A" if os.path.exists("deployment_metadata.json"): with open("deployment_metadata.json") as f: DEFAULT_REMOTE_AGENT_ENGINE_ID = json.load(f)["remote_agent_engine_id"] DEFAULT_AGENT_CALLABLE_PATH = "app.agent_engine_app.AgentEngineApp" class SideBar: """Manages the sidebar components of the Streamlit application.""" def __init__(self, st: Any) -> None: """ Initialize the SideBar. Args: st (Any): The Streamlit object for rendering UI components. """ self.st = st def init_side_bar(self) -> None: """Initialize and render the sidebar components.""" with self.st.sidebar: default_agent_type = ( "Remote URL" if os.path.exists("Dockerfile") else "Local Agent" ) use_agent_path = self.st.selectbox( "Select Agent Type", ["Local Agent", "Remote Agent Engine ID", "Remote URL"], index=["Local Agent", "Remote Agent Engine ID", "Remote URL"].index( default_agent_type ), help="'Local Agent' uses a local implementation, 'Remote Agent Engine ID' connects to a deployed Vertex AI agent, and 'Remote URL' connects to a custom endpoint.", ) if use_agent_path == "Local Agent": self.agent_callable_path = self.st.text_input( label="Agent Callable Path", value=os.environ.get( "AGENT_CALLABLE_PATH", DEFAULT_AGENT_CALLABLE_PATH ), ) self.remote_agent_engine_id = None self.url_input_field = None self.should_authenticate_request = False elif use_agent_path == "Remote Agent Engine ID": self.remote_agent_engine_id = self.st.text_input( label="Remote Agent Engine ID", value=os.environ.get( "REMOTE_AGENT_ENGINE_ID", DEFAULT_REMOTE_AGENT_ENGINE_ID ), ) self.agent_callable_path = None self.url_input_field = None self.should_authenticate_request = False else: self.url_input_field = self.st.text_input( label="Service URL", value=os.environ.get("SERVICE_URL", DEFAULT_BASE_URL), ) self.should_authenticate_request = self.st.checkbox( label="Authenticate request", value=False, help="If checked, any request to the server will contain an" "Identity token to allow authentication. " "See the Cloud Run documentation to know more about authentication:" "https://cloud.google.com/run/docs/authenticating/service-to-service", ) self.agent_callable_path = None self.remote_agent_engine_id = None col1, col2, col3 = self.st.columns(3) with col1: if self.st.button("+ New chat"): if ( len( self.st.session_state.user_chats[ self.st.session_state["session_id"] ]["messages"] ) > 0 ): - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -