atypique-api / agents /cognitive_engine.py
RDS777's picture
Backend v3.0 complet
8e3a425
Raw
History Blame Contribute Delete
465 Bytes
# agents/cognitive_engine.py
import os
import logging
from core.multi_backend_engine import MultiBackendLLMEngine
logger = logging.getLogger("vortex.cognitive")
class MultiLLMEngine(MultiBackendLLMEngine):
def __init__(self, memory):
super().__init__(memory)
self.hf_token = os.environ.get("HF_TOKEN")
self.use_local = os.environ.get("USE_LOCAL_LLM", "0") == "1"
self.model_name = os.environ.get("LLM_MODEL", "microsoft/phi-2")