from crewai import Crew
from llms.gpt import llm
from agents.learning_profiler import learning_profiler
from agents.learning_curator import learning_curator
from agents.article_evaluator import article_evaluator
from agents.curiosity_catalyst import curiosity_catalyst
from tasks.create_learning_profile import learning_profile_task
from tasks.new_article_suggestion import article_suggestion_task
from tasks.evaluate_articles import evaluation_task
from tasks.create_article_pitch import article_pitch_task

article_recommendation_crew = Crew(
    agents=[learning_profiler, learning_curator,
            article_evaluator, curiosity_catalyst],
    tasks=[learning_profile_task, article_suggestion_task,
           evaluation_task, article_pitch_task],
    verbose=True,
    # memory=True,
    manager_llm=llm,
    embedder={
        "provider": "openai",
        "config": {
                    "model": 'text-embedding-3-small'
        }
    }
)