lightman_ai 0.22.0__tar.gz → 1.1.0__tar.gz

lightman_ai-0.22.0/README.md → lightman_ai-1.1.0/PKG-INFO

@@ -1,4 +1,29 @@
+Metadata-Version: 2.4
+Name: lightman_ai
+Version: 1.1.0
+Summary: LLM-powered cybersecurity news aggregator.
+Author-email: sdn4z <imsdn4z@gmail.com>, scastlara <s.cast.lara@gmail.com>
+License-File: LICENSE
+Requires-Python: <4,>=3.13
+Requires-Dist: click<9.0.0,>=8.1.7
+Requires-Dist: httpx<1.0.0,>=0.28.0
+Requires-Dist: pydantic-ai-slim[google,openai]<2.0.0,>=1.0.0
+Requires-Dist: pydantic-settings<3.0.0,>=2.9.1
+Requires-Dist: python-dotenv<2.0.0,>=1.1.1
+Requires-Dist: stamina<26.0.0,>=25.1.0
+Requires-Dist: tomlkit<1.0.0,>=0.13.3
+Provides-Extra: sentry
+Requires-Dist: sentry-sdk<3.0.0,>=2.21.0; extra == 'sentry'
+Description-Content-Type: text/markdown
+
 # 🔍 Lightman AI
+![Build Status](https://github.com/elementsinteractive/lightman-ai/actions/workflows/test.yml/badge.svg)
+[![PyPI version](https://img.shields.io/pypi/v/lightman-ai)](https://pypi.org/project/lightman-ai/)
+[![Docker version](https://img.shields.io/docker/v/elementsinteractive/lightman-ai?label=DockerHub&logo=docker&logoColor=f5f5f5)](https://hub.docker.com/r/elementsinteractive/lightman-ai)
+[![Python Version](https://img.shields.io/badge/python-3.9%20%7C%203.10%20%7C%203.11%20%7C%203.12%20%7C%203.13-blue?logo=python&logoColor=yellow)](https://pypi.org/project/lightman-ai/)
+[![Ruff](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json)](https://github.com/astral-sh/ruff)
+[![License](https://img.shields.io/github/license/elementsinteractive/lightman-ai)](LICENSE)
+
 
 > LLM-Powered Cybersecurity News Intelligence Platform
 
@@ -74,19 +99,37 @@ Lightman AI is an intelligent cybersecurity news aggregation and risk assessment
    agent = "openai"
    score_threshold = 8
    prompt = "development"
+   log_level = "INFO"
    
    [prompts]
    development = "Analyze cybersecurity news for relevance to our organization."' > lightman.toml
    ```
 
 3. **Run with Docker**:
+
    ```bash
    docker run --rm \
      -v $(pwd)/lightman.toml:/app/lightman.toml \
      -e OPENAI_API_KEY="your-api-key" \
      elementsinteractive/lightman-ai:latest \
-     lightman run --config-file /app/lightman.toml --score 7
+     run --config-file /app/lightman.toml --score 8 --agent openai
+   ```
+   
+   You use a .env file instead of setting the environment variables through the cli
+   
+   ```bash
+      cp .env.example .env
    ```
+   
+   Fill it with your values and run:
+   
+   ```bash
+   docker run --rm \
+     -v $(pwd)/lightman.toml:/app/lightman.toml \
+     --env-file .env \
+     elementsinteractive/lightman-ai:latest \
+     run --config-file /app/lightman.toml --score 8 --agent openai
+   ``` 
 
 ## 🔧 Usage
 
@@ -105,6 +148,8 @@ Lightman AI is an intelligent cybersecurity news aggregation and risk assessment
 | `--start-date` | Start date to retrieve articles | False |
 | `--today` | Retrieve articles from today | False |
 | `--yesterday` | Retrieve articles from yesterday | False |
+| `-v` | Be more verbose on output | False |
+
 
 ### Environment Variables:
 lightman-ai uses the following environment variables:
@@ -133,17 +178,33 @@ prompt = 'development'        # Prompt template to use
 service_desk_project_key = "SEC"
 service_desk_request_id_type = "incident"
 
+# alternative configuration
+[malware]
+agent = 'openai'              # AI agent to use (openai, gemini)
+score_threshold = 8           # Minimum relevance score (1-10)
+prompt = 'malware'            # Prompt template to use
+
+# Optional: Service desk integration
+service_desk_project_key = "SEC"
+service_desk_request_id_type = "incident"
+
 [prompts]
 development = """
 Analyze the following cybersecurity news articles and determine their relevance to our organization.
 Rate each article from 1-10 based on potential impact and urgency.
-Focus on: data breaches, malware, vulnerabilities, and threat intelligence.
-"""
+Focus on vulnerabilities."""
+
+malware = """
+Analyze the following cybersecurity news articles and determine their relevance to our organization.
+Rate each article from 1-10 based on potential impact and urgency.
+Focus on malware."""
 
 custom_prompt = """
 Your custom analysis prompt here...
 """
 ```
+Note how it supports different configurations and prompts.
+
 
 It also supports having separate files for your prompts and your configuration settings. Specify the path with `--prompt`.
 
@@ -200,7 +261,7 @@ lightman run --agent openai --score 8 --prompt security_critical --yesterday
 
 ### Development Installation
 In order to fully use the provided setup for local development and testing, this project requires the following dependencies:
-- Python 3.13
+- [Python 3.13](https://www.python.org/downloads/release/python-3130/)
 - [just](https://github.com/casey/just)
 - [uv](https://docs.astral.sh/uv/getting-started/installation/)
 

lightman_ai-0.22.0/PKG-INFO → lightman_ai-1.1.0/README.md

@@ -1,22 +1,11 @@
-Metadata-Version: 2.4
-Name: lightman_ai
-Version: 0.22.0
-Summary: LLM-powered cybersecurity news aggregator.
-Author-email: sdn4z <imsdn4z@gmail.com>
-License-File: LICENSE
-Requires-Python: <4,>=3.13
-Requires-Dist: click<9.0.0,>=8.1.7
-Requires-Dist: httpx<1.0.0,>=0.28.0
-Requires-Dist: pydantic-ai-slim[google,openai]>=0.4.4
-Requires-Dist: pydantic-settings<3.0.0,>=2.9.1
-Requires-Dist: python-dotenv<2.0.0,>=1.1.1
-Requires-Dist: stamina<26.0.0,>=25.1.0
-Requires-Dist: tomlkit<1.0.0,>=0.13.3
-Provides-Extra: sentry
-Requires-Dist: sentry-sdk<3.0.0,>=2.21.0; extra == 'sentry'
-Description-Content-Type: text/markdown
-
 # 🔍 Lightman AI
+![Build Status](https://github.com/elementsinteractive/lightman-ai/actions/workflows/test.yml/badge.svg)
+[![PyPI version](https://img.shields.io/pypi/v/lightman-ai)](https://pypi.org/project/lightman-ai/)
+[![Docker version](https://img.shields.io/docker/v/elementsinteractive/lightman-ai?label=DockerHub&logo=docker&logoColor=f5f5f5)](https://hub.docker.com/r/elementsinteractive/lightman-ai)
+[![Python Version](https://img.shields.io/badge/python-3.9%20%7C%203.10%20%7C%203.11%20%7C%203.12%20%7C%203.13-blue?logo=python&logoColor=yellow)](https://pypi.org/project/lightman-ai/)
+[![Ruff](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json)](https://github.com/astral-sh/ruff)
+[![License](https://img.shields.io/github/license/elementsinteractive/lightman-ai)](LICENSE)
+
 
 > LLM-Powered Cybersecurity News Intelligence Platform
 
@@ -92,19 +81,37 @@ Lightman AI is an intelligent cybersecurity news aggregation and risk assessment
    agent = "openai"
    score_threshold = 8
    prompt = "development"
+   log_level = "INFO"
    
    [prompts]
    development = "Analyze cybersecurity news for relevance to our organization."' > lightman.toml
    ```
 
 3. **Run with Docker**:
+
    ```bash
    docker run --rm \
      -v $(pwd)/lightman.toml:/app/lightman.toml \
      -e OPENAI_API_KEY="your-api-key" \
      elementsinteractive/lightman-ai:latest \
-     lightman run --config-file /app/lightman.toml --score 7
+     run --config-file /app/lightman.toml --score 8 --agent openai
+   ```
+   
+   You use a .env file instead of setting the environment variables through the cli
+   
+   ```bash
+      cp .env.example .env
    ```
+   
+   Fill it with your values and run:
+   
+   ```bash
+   docker run --rm \
+     -v $(pwd)/lightman.toml:/app/lightman.toml \
+     --env-file .env \
+     elementsinteractive/lightman-ai:latest \
+     run --config-file /app/lightman.toml --score 8 --agent openai
+   ``` 
 
 ## 🔧 Usage
 
@@ -123,6 +130,8 @@ Lightman AI is an intelligent cybersecurity news aggregation and risk assessment
 | `--start-date` | Start date to retrieve articles | False |
 | `--today` | Retrieve articles from today | False |
 | `--yesterday` | Retrieve articles from yesterday | False |
+| `-v` | Be more verbose on output | False |
+
 
 ### Environment Variables:
 lightman-ai uses the following environment variables:
@@ -151,17 +160,33 @@ prompt = 'development'        # Prompt template to use
 service_desk_project_key = "SEC"
 service_desk_request_id_type = "incident"
 
+# alternative configuration
+[malware]
+agent = 'openai'              # AI agent to use (openai, gemini)
+score_threshold = 8           # Minimum relevance score (1-10)
+prompt = 'malware'            # Prompt template to use
+
+# Optional: Service desk integration
+service_desk_project_key = "SEC"
+service_desk_request_id_type = "incident"
+
 [prompts]
 development = """
 Analyze the following cybersecurity news articles and determine their relevance to our organization.
 Rate each article from 1-10 based on potential impact and urgency.
-Focus on: data breaches, malware, vulnerabilities, and threat intelligence.
-"""
+Focus on vulnerabilities."""
+
+malware = """
+Analyze the following cybersecurity news articles and determine their relevance to our organization.
+Rate each article from 1-10 based on potential impact and urgency.
+Focus on malware."""
 
 custom_prompt = """
 Your custom analysis prompt here...
 """
 ```
+Note how it supports different configurations and prompts.
+
 
 It also supports having separate files for your prompts and your configuration settings. Specify the path with `--prompt`.
 
@@ -218,7 +243,7 @@ lightman run --agent openai --score 8 --prompt security_critical --yesterday
 
 ### Development Installation
 In order to fully use the provided setup for local development and testing, this project requires the following dependencies:
-- Python 3.13
+- [Python 3.13](https://www.python.org/downloads/release/python-3130/)
 - [just](https://github.com/casey/just)
 - [uv](https://docs.astral.sh/uv/getting-started/installation/)
 

lightman_ai-1.1.0/VERSION

@@ -0,0 +1 @@
+v1.1.0

{lightman_ai-0.22.0 → lightman_ai-1.1.0}/pyproject.toml

@@ -2,6 +2,7 @@
 [project]
 authors = [
     {name = "sdn4z", email = "imsdn4z@gmail.com"},
+    {name = "scastlara", email = "s.cast.lara@gmail.com"}
 ]
 requires-python = "<4,>=3.13"
 dependencies = [
@@ -11,7 +12,7 @@ dependencies = [
     "stamina<26.0.0,>=25.1.0",
     "pydantic-settings<3.0.0,>=2.9.1",
     "tomlkit<1.0.0,>=0.13.3",
-    "pydantic-ai-slim[google,openai]>=0.4.4",
+    "pydantic-ai-slim[google,openai]>=1.0.0,<2.0.0",
 ]
 name = "lightman_ai"
 description = "LLM-powered cybersecurity news aggregator."

{lightman_ai-0.22.0 → lightman_ai-1.1.0}/src/lightman_ai/ai/base/agent.py

@@ -1,26 +1,30 @@
 import logging
 from abc import ABC, abstractmethod
-from typing import Never
+from typing import ClassVar, Never, override
 
 from lightman_ai.article.models import SelectedArticlesList
 from pydantic_ai import Agent
 from pydantic_ai.models.google import GoogleModel
-from pydantic_ai.models.openai import OpenAIModel
+from pydantic_ai.models.openai import OpenAIChatModel
 
 
 class BaseAgent(ABC):
-    _class: type[OpenAIModel] | type[GoogleModel]
-    _default_model_name: str
+    _AGENT_CLASS: type[OpenAIChatModel] | type[GoogleModel]
+    _DEFAULT_MODEL_NAME: str
+    _AGENT_NAME: ClassVar[str]
 
     def __init__(self, system_prompt: str, model: str | None = None, logger: logging.Logger | None = None) -> None:
-        agent_model = self._class(model or self._default_model_name)
+        selected_model = model or self._DEFAULT_MODEL_NAME
+        agent_model = self._AGENT_CLASS(selected_model)
         self.agent: Agent[Never, SelectedArticlesList] = Agent(
             model=agent_model, output_type=SelectedArticlesList, system_prompt=system_prompt
         )
         self.logger = logger or logging.getLogger("lightman")
+        self.logger.info("Selected %s's %s model", self, selected_model)
 
-    def get_prompt_result(self, prompt: str) -> SelectedArticlesList:
-        return self._run_prompt(prompt)
+    @override
+    def __str__(self) -> str:
+        return self._AGENT_NAME
 
     @abstractmethod
-    def _run_prompt(self, prompt: str) -> SelectedArticlesList: ...
+    def run_prompt(self, prompt: str) -> SelectedArticlesList: ...

{lightman_ai-0.22.0 → lightman_ai-1.1.0}/src/lightman_ai/ai/gemini/agent.py

@@ -9,11 +9,12 @@ from pydantic_ai.models.google import GoogleModel
 class GeminiAgent(BaseAgent):
     """Class that provides an interface to operate with the Gemini model."""
 
-    _class = GoogleModel
-    _default_model_name = "gemini-2.5-pro-preview-05-06"
+    _AGENT_CLASS = GoogleModel
+    _DEFAULT_MODEL_NAME = "gemini-2.5-pro"
+    _AGENT_NAME = "Gemini"
 
     @override
-    def _run_prompt(self, prompt: str) -> SelectedArticlesList:
+    def run_prompt(self, prompt: str) -> SelectedArticlesList:
         with map_gemini_exceptions():
             result = self.agent.run_sync(prompt)
         return result.output

{lightman_ai-0.22.0 → lightman_ai-1.1.0}/src/lightman_ai/ai/openai/agent.py

@@ -5,21 +5,22 @@ from lightman_ai.ai.base.agent import BaseAgent
 from lightman_ai.ai.openai.exceptions import LimitTokensExceededError, map_openai_exceptions
 from lightman_ai.article.models import SelectedArticlesList
 from pydantic_ai.agent import AgentRunResult
-from pydantic_ai.models.openai import OpenAIModel
+from pydantic_ai.models.openai import OpenAIChatModel
 
 
 class OpenAIAgent(BaseAgent):
     """Class that provides an interface to operate with the OpenAI model."""
 
-    _class = OpenAIModel
-    _default_model_name = "gpt-4.1"
+    _AGENT_CLASS = OpenAIChatModel
+    _DEFAULT_MODEL_NAME = "gpt-4.1"
+    _AGENT_NAME = "OpenAI"
 
     def _execute_agent(self, prompt: str) -> AgentRunResult[SelectedArticlesList]:
         with map_openai_exceptions():
             return self.agent.run_sync(prompt)
 
     @override
-    def _run_prompt(self, prompt: str) -> SelectedArticlesList:
+    def run_prompt(self, prompt: str) -> SelectedArticlesList:
         try:
             result = self._execute_agent(prompt)
         except LimitTokensExceededError as err:

{lightman_ai-0.22.0 → lightman_ai-1.1.0}/src/lightman_ai/cli.py

@@ -1,20 +1,33 @@
 import logging
+import os
 from datetime import date
 from importlib import metadata
 
 import click
 from dotenv import load_dotenv
 from lightman_ai.ai.utils import AGENT_CHOICES
-from lightman_ai.constants import DEFAULT_CONFIG_FILE, DEFAULT_CONFIG_SECTION, DEFAULT_ENV_FILE
+from lightman_ai.constants import (
+    DEFAULT_AGENT,
+    DEFAULT_CONFIG_FILE,
+    DEFAULT_CONFIG_SECTION,
+    DEFAULT_ENV_FILE,
+    DEFAULT_LOG_LEVEL,
+    DEFAULT_SCORE,
+    DEFAULT_TIME_ZONE,
+    VERBOSE_LOG_LEVEL,
+)
 from lightman_ai.core.config import FileConfig, FinalConfig, PromptConfig
 from lightman_ai.core.exceptions import ConfigNotFoundError, InvalidConfigError, PromptNotFoundError
 from lightman_ai.core.sentry import configure_sentry
-from lightman_ai.core.settings import Settings
 from lightman_ai.exceptions import MultipleDateSourcesError
 from lightman_ai.main import lightman
 from lightman_ai.utils import get_start_date
 
 logger = logging.getLogger("lightman")
+logging.basicConfig(
+    format="%(asctime)s [%(levelname)s] %(name)s: %(message)s",
+    datefmt="%Y-%m-%d %H:%M:%S",
+)
 
 
 def get_version() -> str:
@@ -72,6 +85,7 @@ def entry_point() -> None:
 @click.option("--start-date", type=click.DateTime(formats=["%Y-%m-%d"]), help="Start date to retrieve articles")
 @click.option("--today", is_flag=True, help="Retrieve articles from today.")
 @click.option("--yesterday", is_flag=True, help="Retrieve articles from yesterday.")
+@click.option("-v", is_flag=True, help="Be more verbose on output.")
 def run(
     agent: str,
     prompt: str,
@@ -85,18 +99,30 @@ def run(
     start_date: date | None,
     today: bool,
     yesterday: bool,
+    v: bool,
 ) -> int:
     """
     Entrypoint of the application.
 
     Holds no logic. It loads the configuration, calls the main method and returns 0 when succesful .
     """
-    load_dotenv(env_file or DEFAULT_ENV_FILE)  # TODO refs: #112
-    configure_sentry()
+    load_dotenv(env_file or DEFAULT_ENV_FILE)
+
+    if v:
+        logger.setLevel(VERBOSE_LOG_LEVEL)
+    else:
+        try:
+            env_log_level = os.getenv("LOG_LEVEL")
+            log_level = env_log_level.upper() if env_log_level else DEFAULT_LOG_LEVEL
+            logger.setLevel(log_level)
+        except ValueError:
+            logger.setLevel(DEFAULT_LOG_LEVEL)
+            logger.warning("Invalid logging level. Using default value.")
+
+    configure_sentry(logger.level)
 
-    settings = Settings.try_load_from_file(env_file)
     try:
-        start_datetime = get_start_date(settings, yesterday, today, start_date)
+        start_datetime = get_start_date(os.getenv("TIME_ZONE", DEFAULT_TIME_ZONE), yesterday, today, start_date)
     except MultipleDateSourcesError as e:
         raise click.UsageError(e.args[0]) from e
 
@@ -105,9 +131,9 @@ def run(
         config_from_file = FileConfig.get_config_from_file(config_section=config, path=config_file)
         final_config = FinalConfig.init_from_dict(
             data={
-                "agent": agent or config_from_file.agent or settings.AGENT,
+                "agent": agent or config_from_file.agent or DEFAULT_AGENT,
                 "prompt": prompt or config_from_file.prompt,
-                "score_threshold": score or config_from_file.score_threshold or settings.SCORE,
+                "score_threshold": score or config_from_file.score_threshold or DEFAULT_SCORE,
                 "model": model or config_from_file.model,
             }
         )
@@ -126,5 +152,11 @@ def run(
         start_date=start_datetime,
     )
     relevant_articles_metadata = [f"{article.title} ({article.link})" for article in relevant_articles]
-    logger.warning("Found these articles: \n- %s", "\n- ".join(relevant_articles_metadata))
+
+    if relevant_articles_metadata:
+        articles = f"Found these articles:\n* {'\n* '.join(relevant_articles_metadata)} "
+        click.echo(click.style(articles))
+    else:
+        click.echo(click.style("No relevant articles found."))
+
     return 0

lightman_ai-1.1.0/src/lightman_ai/constants.py

@@ -0,0 +1,12 @@
+DEFAULT_CONFIG_SECTION = "default"
+
+DEFAULT_CONFIG_FILE = "lightman.toml"
+
+DEFAULT_ENV_FILE = ".env"
+
+DEFAULT_LOG_LEVEL = "WARNING"
+VERBOSE_LOG_LEVEL = "INFO"
+
+DEFAULT_AGENT = "openai"
+DEFAULT_SCORE = 8
+DEFAULT_TIME_ZONE = "UTC"

lightman_ai-1.1.0/src/lightman_ai/core/sentry.py

@@ -0,0 +1,32 @@
+import logging
+import os
+from importlib import metadata
+
+logger = logging.getLogger("lightman")
+
+
+def configure_sentry(log_level: int) -> None:
+    """Configure Sentry for error tracking."""
+    try:
+        import sentry_sdk  # noqa: PLC0415
+        from sentry_sdk.integrations.logging import LoggingIntegration  # noqa: PLC0415
+    except ImportError:
+        if os.getenv("SENTRY_DSN"):
+            logger.warning(
+                "Could not initialize sentry, it is not installed! Add it by installing the project with `lightman-ai[sentry]`."
+            )
+        return
+
+    if not os.getenv("SENTRY_DSN"):
+        logger.warning("SENTRY_DSN not configured, skipping Sentry initialization")
+        return
+
+    try:
+        sentry_logging = LoggingIntegration(level=logging.INFO, event_level=log_level)
+
+        sentry_sdk.init(
+            release=metadata.version("lightman-ai"),
+            integrations=[sentry_logging],
+        )
+    except Exception as e:
+        logger.warning("Could not instantiate Sentry! %s.\nContinuing with the execution.", e)

{lightman_ai-0.22.0 → lightman_ai-1.1.0}/src/lightman_ai/main.py

@@ -11,6 +11,7 @@ from lightman_ai.integrations.service_desk.integration import (
 from lightman_ai.sources.the_hacker_news import TheHackerNewsSource
 
 logger = logging.getLogger("lightman")
+logger.addHandler(logging.NullHandler())
 
 
 def _get_articles_from_source(start_date: datetime | None = None) -> ArticlesList:
@@ -18,7 +19,7 @@ def _get_articles_from_source(start_date: datetime | None = None) -> ArticlesLis
 
 
 def _classify_articles(articles: ArticlesList, agent: BaseAgent) -> SelectedArticlesList:
-    return agent.get_prompt_result(prompt=str(articles))
+    return agent.run_prompt(prompt=str(articles))
 
 
 def _create_service_desk_issues(
@@ -67,7 +68,6 @@ def lightman(
 
     agent_class = get_agent_class_from_agent_name(agent)
     agent_instance = agent_class(prompt, model, logger=logger)
-    logger.info("Selected %s.", agent_instance)
 
     classified_articles = _classify_articles(
         articles=articles,
@@ -77,10 +77,6 @@ def lightman(
     selected_articles: list[SelectedArticle] = classified_articles.get_articles_with_score_gte_threshold(
         score_threshold
     )
-    if selected_articles:
-        logger.info("Found these articles: %s", selected_articles)
-    else:
-        logger.info("No articles found to be relevant. Total returned articles by AI %s", len(classified_articles))
 
     if not dry_run:
         if not service_desk_project_key or not service_desk_request_id_type:

{lightman_ai-0.22.0 → lightman_ai-1.1.0}/src/lightman_ai/sources/the_hacker_news.py

@@ -25,8 +25,10 @@ class TheHackerNewsSource(BaseSource):
     @override
     def get_articles(self, date: datetime | None = None) -> ArticlesList:
         """Return the articles that are present in THN feed."""
+        logger.info("Downloading articles from %s", THN_URL)
         feed = self.get_feed()
         articles = self._xml_to_list_of_articles(feed)
+        logger.info("Articles properly downloaded and parsed.")
         if date:
             return ArticlesList.get_articles_from_date_onwards(articles=articles, start_date=date)
         else:

{lightman_ai-0.22.0 → lightman_ai-1.1.0}/src/lightman_ai/utils.py

@@ -1,11 +1,10 @@
 from datetime import date, datetime, time, timedelta
 from zoneinfo import ZoneInfo
 
-from lightman_ai.core.settings import Settings
 from lightman_ai.exceptions import MultipleDateSourcesError
 
 
-def get_start_date(settings: Settings, yesterday: bool, today: bool, start_date: date | None) -> datetime | None:
+def get_start_date(time_zone: str, yesterday: bool, today: bool, start_date: date | None) -> datetime | None:
     mutually_exclusive_date_fields = [x for x in [start_date, today, yesterday] if x]
 
     if len(mutually_exclusive_date_fields) > 1:
@@ -14,12 +13,12 @@ def get_start_date(settings: Settings, yesterday: bool, today: bool, start_date:
         )
 
     if today:
-        now = datetime.now(ZoneInfo(settings.TIME_ZONE))
-        return datetime.combine(now, time(0, 0), tzinfo=ZoneInfo(settings.TIME_ZONE))
+        now = datetime.now(ZoneInfo(time_zone))
+        return datetime.combine(now, time(0, 0), tzinfo=ZoneInfo(time_zone))
     elif yesterday:
-        yesterday_date = datetime.now(ZoneInfo(settings.TIME_ZONE)) - timedelta(days=1)
-        return datetime.combine(yesterday_date, time(0, 0), tzinfo=ZoneInfo(settings.TIME_ZONE))
+        yesterday_date = datetime.now(ZoneInfo(time_zone)) - timedelta(days=1)
+        return datetime.combine(yesterday_date, time(0, 0), tzinfo=ZoneInfo(time_zone))
     elif isinstance(start_date, date):
-        return datetime.combine(start_date, time(0, 0), tzinfo=ZoneInfo(settings.TIME_ZONE))
+        return datetime.combine(start_date, time(0, 0), tzinfo=ZoneInfo(time_zone))
     else:
         return None

lightman_ai-0.22.0/VERSION

@@ -1 +0,0 @@
-v0.22.0

lightman_ai-0.22.0/src/lightman_ai/constants.py

@@ -1,5 +0,0 @@
-DEFAULT_CONFIG_SECTION = "default"
-
-DEFAULT_CONFIG_FILE = "lightman.toml"
-
-DEFAULT_ENV_FILE = ".env"

lightman_ai-0.22.0/src/lightman_ai/core/sentry.py

@@ -1,39 +0,0 @@
-import logging
-import os
-from importlib import metadata
-
-logger = logging.getLogger("lightman")
-
-
-def configure_sentry() -> None:
-    """Configure Sentry for error tracking."""
-    try:
-        import sentry_sdk
-        from sentry_sdk.integrations.logging import LoggingIntegration
-    except ImportError:
-        logger.warning(
-            "Could not initialize sentry, it is not installed! Add it by installing the project with `lightman-ai[sentry]`."
-        )
-        return
-
-    try:
-        if not os.getenv("SENTRY_DSN"):
-            logger.info("SENTRY_DSN not configured, skipping Sentry initialization")
-            return
-
-        logging_level_str = os.getenv("LOGGING_LEVEL", "ERROR").upper()
-        try:
-            logging_level = getattr(logging, logging_level_str, logging.ERROR)
-        except AttributeError:
-            logger.warning("The specified logging level `%s` does not exist. Defaulting to ERROR.", logging_level_str)
-            logging_level = logging.ERROR
-
-        # Set up logging integration
-        sentry_logging = LoggingIntegration(level=logging.INFO, event_level=logging_level)
-
-        sentry_sdk.init(
-            release=metadata.version("lightman-ai"),
-            integrations=[sentry_logging],
-        )
-    except Exception as e:
-        logger.warning("Could not instantiate Sentry! %s.\nContinuing with the execution.", e)

lightman_ai-0.22.0/src/lightman_ai/core/settings.py

@@ -1,30 +0,0 @@
-import logging
-from pathlib import Path
-from typing import Any, Self
-
-from pydantic_settings import BaseSettings, SettingsConfigDict
-
-logger = logging.getLogger("lightman")
-
-
-class Settings(BaseSettings):
-    def __init__(self, *args: Any, **kwargs: Any) -> None:
-        super().__init__(*args, **kwargs)
-
-    AGENT: str = "openai"
-    SCORE: int = 8
-    TIME_ZONE: str = "UTC"
-    model_config = SettingsConfigDict(extra="ignore")
-
-    @classmethod
-    def try_load_from_file(cls, env_file: str | None = None) -> Self:
-        """
-        Initialize Settings class and returns an instance.
-
-        It tries to load env variables from the env file. Variables set in the environment take precendence.
-
-        If the env file is not present, it continues execution, following pydantic-settings' behaviour.
-        """
-        if env_file and not Path(env_file).exists():
-            logger.warning("env file `%s` not found.", env_file)
-        return cls(_env_file=env_file)