move summarizer modules to internals

Author: ks6088ts

.env.template

@@ -66,3 +66,6 @@ OTEL_COLLECTOR_ENDPOINT="http://localhost:4317"
 
 ## Scraper Settings
 SCRAPER_TYPE="mock" # Options: "mock", "httpx", "youtube_transcript"
+
+## Summarizer Settings
+SUMMARIZER_TYPE="mock" # Options: "mock", "llm"

scripts/agent_operator.py

@@ -18,11 +18,11 @@
     AgentState,
     Article,
 )
-from template_langgraph.agents.news_summarizer_agent.summarizers import (
-    LlmSummarizer,
-)
 from template_langgraph.agents.task_decomposer_agent.agent import graph as task_decomposer_agent_graph
 from template_langgraph.internals.scrapers import get_scraper
+from template_langgraph.internals.summarizers import (
+    get_summarizer,
+)
 from template_langgraph.loggers import get_logger
 
 # Initialize the Typer application
@@ -157,7 +157,7 @@ def news_summarizer_agent(
     graph = NewsSummarizerAgent(
         notifier=MockNotifier(),
         scraper=get_scraper(),
-        summarizer=LlmSummarizer(),
+        summarizer=get_summarizer(),
     ).create_graph()
     for event in graph.stream(
         input=AgentState(

template_langgraph/agents/news_summarizer_agent/agent.py

@@ -8,16 +8,16 @@
     StructuredArticle,
     SummarizeWebContentState,
 )
-from template_langgraph.agents.news_summarizer_agent.summarizers import (
-    BaseSummarizer,
-    LlmSummarizer,
-    MockSummarizer,
-)
 from template_langgraph.internals.scrapers import (
     BaseScraper,
     HttpxScraper,
     MockScraper,
 )
+from template_langgraph.internals.summarizers import (
+    BaseSummarizer,
+    LlmSummarizer,
+    MockSummarizer,
+)
 from template_langgraph.llms.azure_openais import AzureOpenAiWrapper
 from template_langgraph.loggers import get_logger
 

template_langgraph/agents/news_summarizer_agent/summarizers.py renamed to template_langgraph/internals/summarizers.py

@@ -3,9 +3,12 @@
 from __future__ import annotations
 
 from abc import ABC, abstractmethod
+from enum import Enum
+from functools import lru_cache
 from typing import Any
 
 from langchain_core.language_models.chat_models import BaseChatModel
+from pydantic_settings import BaseSettings, SettingsConfigDict
 
 from template_langgraph.agents.news_summarizer_agent.models import StructuredArticle
 from template_langgraph.llms.azure_openais import AzureOpenAiWrapper
@@ -14,6 +17,29 @@
 logger = get_logger(__name__)
 
 
+class SummarizerType(str, Enum):
+    """Enumeration of available summarizer types."""
+
+    MOCK = "mock"
+    LLM = "llm"
+
+
+class Settings(BaseSettings):
+    summarizer_type: SummarizerType = SummarizerType.MOCK
+
+    model_config = SettingsConfigDict(
+        env_file=".env",
+        env_ignore_empty=True,
+        extra="ignore",
+    )
+
+
+@lru_cache
+def get_summarizer_settings() -> Settings:
+    """Get summarizer settings."""
+    return Settings()
+
+
 class BaseSummarizer(ABC):
     """Abstract base summarizer returning a StructuredArticle."""
 
@@ -52,8 +78,13 @@ def summarize(self, prompt: str, content: str) -> StructuredArticle:  # noqa: D4
         )
 
 
-__all__ = [
-    "BaseSummarizer",
-    "MockSummarizer",
-    "LlmSummarizer",
-]
+def get_summarizer(settings: Settings = None) -> BaseSummarizer:
+    if settings is None:
+        settings = get_summarizer_settings()
+
+    if settings.summarizer_type == SummarizerType.MOCK:
+        return MockSummarizer()
+    elif settings.summarizer_type == SummarizerType.LLM:
+        return LlmSummarizer()
+    else:
+        raise ValueError(f"Unknown summarizer type: {settings.summarizer_type}")