Mirrors/ProxMenux
1
0
Fork 0
mirror of https://github.com/MacRimi/ProxMenux.git synced 2026-04-30 19:36:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update ollama_provider.py

Browse Source
This commit is contained in:
MacRimi
2026-03-19 20:45:35 +01:00
parent f2e804783b
commit c867c4ef51
1 changed files with 16 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
AppImage/scripts/ai_providers/ollama_provider.py
View File

@@ -62,8 +62,13 @@ class OllamaProvider(AIProvider):
'Content-Type': 'application/json', 'Content-Type': 'application/json',
} }
# Cloud models (e.g., kimi-k2.5:cloud, minimax-m2.7:cloud) need longer timeout
# because requests go through: ProxMenux -> Ollama -> Cloud Provider -> back
is_cloud_model = ':cloud' in self.model.lower()
timeout = 120 if is_cloud_model else 30 # 2 minutes for cloud, 30s for local
try: try:
result = self._make_request(url, payload, headers, timeout=30) result = self._make_request(url, payload, headers, timeout=timeout)
except AIProviderError as e: except AIProviderError as e:
if "Connection" in str(e) or "refused" in str(e).lower(): if "Connection" in str(e) or "refused" in str(e).lower():
raise AIProviderError( raise AIProviderError(
@@ -130,4 +135,14 @@ class OllamaProvider(AIProvider):
} }
# If server is up and model exists, do the actual test # If server is up and model exists, do the actual test
# For cloud models, we skip the full test (which sends a message)
# because it would take too long. The model availability check above is sufficient.
is_cloud_model = ':cloud' in self.model.lower()
if is_cloud_model:
return {
'success': True,
'message': f"Cloud model '{self.model}' is available via Ollama",
'model': self.model
}
return super().test_connection() return super().test_connection()