changes
Browse files
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(
|
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 |
|