logdetective 1.7.0__tar.gz → 1.9.0__tar.gz

{logdetective-1.7.0 → logdetective-1.9.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: logdetective
-Version: 1.7.0
+Version: 1.9.0
 Summary: Log using LLM AI to search for build/test failures and provide ideas for fixing these.
 License: Apache-2.0
 Author: Jiri Podivin
@@ -27,6 +27,7 @@ Requires-Dist: backoff (==2.2.1) ; extra == "server" or extra == "server-testing
 Requires-Dist: drain3 (>=0.9.11,<0.10.0)
 Requires-Dist: fastapi (>=0.111.1) ; extra == "server" or extra == "server-testing"
 Requires-Dist: huggingface-hub (>0.23.2)
+Requires-Dist: koji (>=1.35.0,<2.0.0) ; extra == "server" or extra == "server-testing"
 Requires-Dist: llama-cpp-python (>0.2.56,!=0.2.86)
 Requires-Dist: matplotlib (>=3.8.4,<4.0.0) ; extra == "server" or extra == "server-testing"
 Requires-Dist: numpy (>=1.26.0)

{logdetective-1.7.0 → logdetective-1.9.0}/logdetective/server/database/models/merge_request_jobs.py

@@ -157,7 +157,7 @@ class GitlabMergeRequestJobs(Base):
         project_id: int,
         mr_iid: int,
         job_id: int,
-    ) -> Optional["GitlabMergeRequestJobs"]:
+    ) -> Self:
         """Search for a detailed merge request
         or create a new one if not found.
 

{logdetective-1.7.0 → logdetective-1.9.0}/logdetective/server/database/models/metrics.py

@@ -1,7 +1,7 @@
 import io
 import enum
 import datetime
-from typing import Optional, List
+from typing import Optional, List, Self, Tuple
 
 import backoff
 
@@ -114,7 +114,7 @@ class AnalyzeRequestMetrics(Base):
     def update(  # pylint: disable=too-many-arguments disable=too-many-positional-arguments
         cls,
         id_: int,
-        response_sent_at: datetime,
+        response_sent_at: DateTime,
         response_length: int,
         response_certainty: float,
         compressed_response: bytes,
@@ -123,6 +123,8 @@ class AnalyzeRequestMetrics(Base):
         with data related to the given response"""
         with transaction(commit=True) as session:
             metrics = session.query(AnalyzeRequestMetrics).filter_by(id=id_).first()
+            if metrics is None:
+                raise ValueError("Returned `AnalyzeRequestMetrics` table is empty.")
             metrics.response_sent_at = response_sent_at
             metrics.response_length = response_length
             metrics.response_certainty = response_certainty
@@ -134,11 +136,13 @@ class AnalyzeRequestMetrics(Base):
     def get_metric_by_id(
         cls,
         id_: int,
-    ) -> "AnalyzeRequestMetrics":
+    ) -> Self:
         """Update a row
         with data related to the given response"""
         with transaction(commit=True) as session:
             metric = session.query(AnalyzeRequestMetrics).filter_by(id=id_).first()
+            if metric is None:
+                raise ValueError("Returned `AnalyzeRequestMetrics` table is empty.")
             return metric
 
     def add_mr_job(
@@ -169,7 +173,7 @@ class AnalyzeRequestMetrics(Base):
         project_id: int,
         mr_iid: int,
         job_id: int,
-    ) -> List["AnalyzeRequestMetrics"]:
+    ) -> List[Self]:
         """Search for all requests triggered by the specified merge request job.
 
         Args:
@@ -204,11 +208,12 @@ class AnalyzeRequestMetrics(Base):
 
     @classmethod
     def get_dictionary_with_datetime_keys(
-        cls, time_format: str, values_dict: dict[str, any]
-    ) -> dict[datetime.datetime, any]:
-        """Convert from a dictionary with str keys to a dictionary with datetime keys"""
+        cls, time_format: str, query_results: List[Tuple[str, int]]
+    ) -> dict[datetime.datetime, int]:
+        """Convert from a list of tuples with str as first values
+        to a dictionary with datetime keys"""
         new_dict = {
-            datetime.datetime.strptime(r[0], time_format): r[1] for r in values_dict
+            datetime.datetime.strptime(e[0], time_format): e[1] for e in query_results
         }
         return new_dict
 

{logdetective-1.7.0 → logdetective-1.9.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "logdetective"
-version = "1.7.0"
+version = "1.9.0"
 description = "Log using LLM AI to search for build/test failures and provide ideas for fixing these."
 authors = ["Jiri Podivin <jpodivin@gmail.com>"]
 license = "Apache-2.0"
@@ -53,10 +53,11 @@ matplotlib = {version = "^3.8.4", optional = true }
 backoff = {version = "2.2.1", optional = true }
 sentry-sdk = {version = "^2.17.0", optional = true, extras = ["fastapi"]}
 openai = {version = "^1.82.1", optional = true}
+koji = {version = "^1.35.0", optional = true}
 
 [tool.poetry.extras]
-server = ["fastapi", "sqlalchemy", "psycopg2", "alembic", "matplotlib", "backoff", "aiolimiter", "sentry-sdk", "openai"]
-server-testing = ["fastapi", "sqlalchemy", "psycopg2-binary", "alembic", "matplotlib", "backoff", "pytest-asyncio", "sentry-sdk", "openai"]
+server = ["fastapi", "sqlalchemy", "psycopg2", "alembic", "matplotlib", "backoff", "aiolimiter", "sentry-sdk", "openai", "koji"]
+server-testing = ["fastapi", "sqlalchemy", "psycopg2-binary", "alembic", "matplotlib", "backoff", "pytest-asyncio", "sentry-sdk", "openai", "koji"]
 
 [build-system]
 requires = ["poetry-core"]