Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion lgtm.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
technologies = ["Python"]
categories = ["Correctness", "Quality", "Testing", "Security"]
exclude = ["*.md", "uv.lock", "eval/classified_articles.py"]
model = "gemini-2.5-pro-preview-*"
model = "gemini-2.5-pro"
silent = false
publish = true
4 changes: 2 additions & 2 deletions src/lightman_ai/ai/base/agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,11 +5,11 @@
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]
_class: type[OpenAIChatModel] | type[GoogleModel]
_default_model_name: str

def __init__(self, system_prompt: str, model: str | None = None, logger: logging.Logger | None = None) -> None:
Expand Down
2 changes: 1 addition & 1 deletion src/lightman_ai/ai/gemini/agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ 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"
_default_model_name = "gemini-2.5-pro"

@override
def _run_prompt(self, prompt: str) -> SelectedArticlesList:
Expand Down
4 changes: 2 additions & 2 deletions src/lightman_ai/ai/openai/agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,13 +5,13 @@
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
_class = OpenAIChatModel
_default_model_name = "gpt-4.1"

def _execute_agent(self, prompt: str) -> AgentRunResult[SelectedArticlesList]:
Expand Down
Loading