azettl commited on
Commit
c06c25c
·
1 Parent(s): 903103b
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -112,13 +112,12 @@ def get_or_create_session_state(session_id: str) -> Dict:
112
  "step_continue_event": threading.Event(),
113
  "api_keys": {
114
  "mistral": None,
115
- "sambanova": None,
116
- "huggingface": None
117
  }
118
  }
119
  return user_sessions[session_id]
120
 
121
- def update_session_api_keys(mistral_key, sambanova_key, huggingface_key, session_id_state, request: gr.Request = None):
122
  """Update API keys for THIS SESSION ONLY"""
123
  session_id = get_session_id(request) if not session_id_state else session_id_state
124
  session = get_or_create_session_state(session_id)
@@ -191,8 +190,7 @@ class VisualConsensusEngine:
191
  # Store session keys for API calls
192
  self.session_keys = {
193
  'mistral': mistral_key,
194
- 'sambanova': sambanova_key,
195
- 'huggingface': hf_token
196
  }
197
 
198
  # Role definitions
@@ -989,7 +987,7 @@ with gr.Blocks(title="🎭 Consilium: Visual AI Consensus Platform", theme=gr.th
989
  # Connect the save button
990
  save_keys_btn.click(
991
  update_session_api_keys,
992
- inputs=[mistral_key_input, sambanova_key_input, huggingface_key_input, session_state],
993
  outputs=[keys_status, session_state]
994
  )
995
 
 
112
  "step_continue_event": threading.Event(),
113
  "api_keys": {
114
  "mistral": None,
115
+ "sambanova": None
 
116
  }
117
  }
118
  return user_sessions[session_id]
119
 
120
+ def update_session_api_keys(mistral_key, sambanova_key, session_id_state, request: gr.Request = None):
121
  """Update API keys for THIS SESSION ONLY"""
122
  session_id = get_session_id(request) if not session_id_state else session_id_state
123
  session = get_or_create_session_state(session_id)
 
190
  # Store session keys for API calls
191
  self.session_keys = {
192
  'mistral': mistral_key,
193
+ 'sambanova': sambanova_key
 
194
  }
195
 
196
  # Role definitions
 
987
  # Connect the save button
988
  save_keys_btn.click(
989
  update_session_api_keys,
990
+ inputs=[mistral_key_input, sambanova_key_input, session_state],
991
  outputs=[keys_status, session_state]
992
  )
993