Skip to content

Commit da9b321

Browse files
Refactor Ollama integration by cleaning comments
1 parent a2ece30 commit da9b321

1 file changed

Lines changed: 10 additions & 15 deletions

File tree

KowihanBackend/api_generator_platform_using_IA/ollama_integration.py

Lines changed: 10 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,4 @@
1-
"""
2-
Module d'intégration Ollama pour utiliser Llama localement
3-
Pas de limites de requêtes, 100% gratuit et privé
4-
"""
1+
52

63
import requests
74
import json
@@ -21,7 +18,7 @@ class OllamaConfig:
2118

2219

2320
class OllamaClient:
24-
"""Client pour communiquer avec Ollama (Llama local)"""
21+
2522

2623
def __init__(self, config: OllamaConfig = None):
2724
self.config = config or OllamaConfig()
@@ -34,15 +31,13 @@ def _check_ollama_available(self):
3431
if response.status_code == 200:
3532
models = response.json().get('models', [])
3633
available_models = [m['name'] for m in models]
37-
logger.info(f"✅ Ollama disponible. Modèles: {available_models}")
3834

39-
# Vérifier si le modèle configuré est disponible
35+
4036
if not any(self.config.model in m for m in available_models):
41-
logger.warning(f"⚠️ Modèle {self.config.model} non trouvé. Téléchargez-le avec: ollama pull {self.config.model}")
4237
else:
43-
logger.warning("⚠️ Ollama n'est pas accessible. Installez-le: https://ollama.ai")
38+
logger.warning(" Ollama n'est pas accessible. Installez-le: https://ollama.ai")
4439
except requests.exceptions.RequestException as e:
45-
logger.warning(f"⚠️ Ollama non disponible: {e}")
40+
logger.warning(f" Ollama non disponible: {e}")
4641
logger.info("Installation: curl -fsSL https://ollama.ai/install.sh | sh")
4742

4843
def generate(self, prompt: str, system_prompt: str = None) -> str:
@@ -393,9 +388,9 @@ def _security_audit_rules(self, specs: Dict) -> Dict:
393388
score = "B" if has_auth else "C"
394389

395390
warnings = [
396-
"⚠️ Définissez des SECRET_KEY robustes",
397-
"⚠️ Validez toutes les entrées utilisateur",
398-
"⚠️ Activez HTTPS en production"
391+
" Définissez des SECRET_KEY robustes",
392+
" Validez toutes les entrées utilisateur",
393+
" Activez HTTPS en production"
399394
]
400395

401396
recommendations = [
@@ -417,11 +412,11 @@ def _security_audit_rules(self, specs: Dict) -> Dict:
417412
}
418413

419414

420-
# Fonction utilitaire pour installer Ollama automatiquement
415+
421416
def install_ollama_instructions():
422417
"""Retourne les instructions d'installation d'Ollama"""
423418
return """
424-
# 🦙 Installation d'Ollama (Llama local)
419+
# Installation d'Ollama (Llama local)
425420
426421
## Linux / macOS:
427422
```bash

0 commit comments

Comments
 (0)