diff --git a/GramAddict/core/navigation/brain.py b/GramAddict/core/navigation/brain.py index 27f16d7..b83b57b 100644 --- a/GramAddict/core/navigation/brain.py +++ b/GramAddict/core/navigation/brain.py @@ -15,12 +15,8 @@ def ask_brain_for_action( return None cfg = Config() - url = ( - getattr(cfg.args, "ai_embedding_url", "http://localhost:11434/api/chat") - if hasattr(cfg, "args") - else "http://localhost:11434/api/chat" - ) - model = getattr(cfg.args, "ai_embedding_model", "llama3") if hasattr(cfg, "args") else "llama3" + url = getattr(cfg.args, "ai_model_url", "http://localhost:11434/api/generate") if hasattr(cfg, "args") else "http://localhost:11434/api/generate" + model = getattr(cfg.args, "ai_model", "llama3") if hasattr(cfg, "args") else "llama3" prompt = ( f"You are an autonomous Instagram agent. Your ultimate goal is: '{goal}'.\n"