azettl commited on
Commit
34b1f00
Β·
1 Parent(s): b3f93dc
Files changed (1) hide show
  1. app.py +1 -16
app.py CHANGED
@@ -21,7 +21,6 @@ load_dotenv()
21
  # API Configuration - These will be updated by UI if needed
22
  MISTRAL_API_KEY = os.getenv("MISTRAL_API_KEY")
23
  SAMBANOVA_API_KEY = os.getenv("SAMBANOVA_API_KEY")
24
- HUGGINGFACE_API_TOKEN = os.getenv("HUGGINGFACE_API_TOKEN")
25
  MODERATOR_MODEL = os.getenv("MODERATOR_MODEL", "mistral")
26
 
27
  # Session-based storage for isolated discussions
@@ -60,7 +59,7 @@ class WebSearchAgent:
60
  WikipediaTool(),
61
  FinalAnswerTool()
62
  ],
63
- model=InferenceClientModel(model_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0"),
64
  max_steps=3,
65
  verbosity_level=0
66
  )
@@ -145,17 +144,6 @@ def update_session_api_keys(mistral_key, sambanova_key, huggingface_key, session
145
  else:
146
  status_messages.append("❌ No SambaNova API key available")
147
 
148
- if huggingface_key.strip():
149
- session["api_keys"]["huggingface"] = huggingface_key.strip()
150
- status_messages.append("βœ… Hugging Face token saved for this session")
151
- # Update environment for search agent
152
- os.environ["HUGGINGFACE_API_TOKEN"] = huggingface_key.strip()
153
- elif HUGGINGFACE_API_TOKEN:
154
- session["api_keys"]["huggingface"] = HUGGINGFACE_API_TOKEN
155
- status_messages.append("βœ… Using Hugging Face token from environment")
156
- else:
157
- status_messages.append("❌ No Hugging Face token available")
158
-
159
  return " | ".join(status_messages), session_id
160
 
161
  class VisualConsensusEngine:
@@ -171,7 +159,6 @@ class VisualConsensusEngine:
171
 
172
  mistral_key = session_keys.get("mistral") or MISTRAL_API_KEY
173
  sambanova_key = session_keys.get("sambanova") or SAMBANOVA_API_KEY
174
- hf_token = session_keys.get("huggingface") or HUGGINGFACE_API_TOKEN
175
 
176
  self.models = {
177
  'mistral': {
@@ -770,7 +757,6 @@ def check_model_status_session(session_id_state: str = None, request: gr.Request
770
  # Get session-specific keys or fall back to env vars
771
  mistral_key = session_keys.get("mistral") or MISTRAL_API_KEY
772
  sambanova_key = session_keys.get("sambanova") or SAMBANOVA_API_KEY
773
- hf_token = session_keys.get("huggingface") or HUGGINGFACE_API_TOKEN
774
 
775
  status_info = "## πŸ” Model Availability Status\n\n"
776
 
@@ -1040,7 +1026,6 @@ with gr.Blocks(title="🎭 Consilium: Visual AI Consensus Platform", theme=gr.th
1040
  ```bash
1041
  export MISTRAL_API_KEY=your_key_here
1042
  export SAMBANOVA_API_KEY=your_key_here
1043
- export HUGGINGFACE_API_TOKEN=your_token_here
1044
  export MODERATOR_MODEL=mistral
1045
  ```
1046
 
 
21
  # API Configuration - These will be updated by UI if needed
22
  MISTRAL_API_KEY = os.getenv("MISTRAL_API_KEY")
23
  SAMBANOVA_API_KEY = os.getenv("SAMBANOVA_API_KEY")
 
24
  MODERATOR_MODEL = os.getenv("MODERATOR_MODEL", "mistral")
25
 
26
  # Session-based storage for isolated discussions
 
59
  WikipediaTool(),
60
  FinalAnswerTool()
61
  ],
62
+ model=InferenceClientModel(),
63
  max_steps=3,
64
  verbosity_level=0
65
  )
 
144
  else:
145
  status_messages.append("❌ No SambaNova API key available")
146
 
 
 
 
 
 
 
 
 
 
 
 
147
  return " | ".join(status_messages), session_id
148
 
149
  class VisualConsensusEngine:
 
159
 
160
  mistral_key = session_keys.get("mistral") or MISTRAL_API_KEY
161
  sambanova_key = session_keys.get("sambanova") or SAMBANOVA_API_KEY
 
162
 
163
  self.models = {
164
  'mistral': {
 
757
  # Get session-specific keys or fall back to env vars
758
  mistral_key = session_keys.get("mistral") or MISTRAL_API_KEY
759
  sambanova_key = session_keys.get("sambanova") or SAMBANOVA_API_KEY
 
760
 
761
  status_info = "## πŸ” Model Availability Status\n\n"
762
 
 
1026
  ```bash
1027
  export MISTRAL_API_KEY=your_key_here
1028
  export SAMBANOVA_API_KEY=your_key_here
 
1029
  export MODERATOR_MODEL=mistral
1030
  ```
1031