refactor: update default gemini model and replace deprecated OpenAIModel class usage (#144)

Author: sdn4z

lgtm.toml

@@ -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

src/lightman_ai/ai/base/agent.py

@@ -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:

src/lightman_ai/ai/gemini/agent.py

@@ -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:

src/lightman_ai/ai/openai/agent.py

@@ -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]: