Skip to content

Airflow ai sdk with gemini#268

Open
luisssguilherme wants to merge 6 commits intomainfrom
airflow-ai-sdk
Open

Airflow ai sdk with gemini#268
luisssguilherme wants to merge 6 commits intomainfrom
airflow-ai-sdk

Conversation

@luisssguilherme
Copy link
Contributor

Summaries on general and individuals Ro-DOU's reports.

@Heitorh3
Copy link
Contributor

Sugiro que as responsabilidades sejam divididas. Toda a lógica de geração de resumos com IA está implementada na classe dou_dag_generator.py, incluindo:

  • Preparação de contexto para IA (prepare_ai_context)
  • Construção de prompts (_build_ai_prompt)
  • Chamadas às APIs de LLM (_call_ai_api, _call_gemini_api, _call_openai_api)
  • Geração de resumos individuais (generate_individual_summaries)
  • Geração do resumo geral (generate_ai_summary)

Essa classe já acumula responsabilidades de:

  1. Geração de DAGs
  2. Execução de buscas
  3. Notificações de falha
  4. Integração com IA/LLM

As classes devem ter responsabilidades claras e bem definidas. Sugiro extrair a lógica de IA para uma nova classe, por exemplo AIReportGenerator ou LLMSummarizer, que encapsule toda a interação com provedores de LLM (Gemini, OpenAI) e a construção de prompts.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants