logdetective 0.2.9__tar.gz → 0.2.10__tar.gz

{logdetective-0.2.9 → logdetective-0.2.10}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: logdetective
-Version: 0.2.9
+Version: 0.2.10
 Summary: Log using LLM AI to search for build/test failures and provide ideas for fixing these.
 License: Apache-2.0
 Author: Jiri Podivin
@@ -22,7 +22,7 @@ Provides-Extra: server
 Requires-Dist: drain3 (>=0.9.11,<0.10.0)
 Requires-Dist: huggingface-hub (>0.23.2)
 Requires-Dist: llama-cpp-python (>0.2.56,!=0.2.86)
-Requires-Dist: numpy (>=1.26.0,<2.0.0)
+Requires-Dist: numpy (>=1.26.0)
 Requires-Dist: requests (>0.2.31)
 Project-URL: homepage, https://github.com/fedora-copr/logdetective
 Project-URL: issues, https://github.com/fedora-copr/logdetective/issues

{logdetective-0.2.9 → logdetective-0.2.10}/logdetective/extractors.py

@@ -16,7 +16,7 @@ class LLMExtractor:
     A class that extracts relevant information from logs using a language model.
     """
     def __init__(self, model: Llama, n_lines: int = 2):
-        self.model =  model
+        self.model = model
         self.n_lines = n_lines
         self.grammar = LlamaGrammar.from_string(
             "root ::= (\"Yes\" | \"No\")", verbose=False)

{logdetective-0.2.9 → logdetective-0.2.10}/logdetective/logdetective.py

@@ -9,6 +9,7 @@ from logdetective.extractors import LLMExtractor, DrainExtractor
 
 LOG = logging.getLogger("logdetective")
 
+
 def setup_args():
     """ Setup argument parser and return arguments. """
     parser = argparse.ArgumentParser("logdetective")

{logdetective-0.2.9 → logdetective-0.2.10}/logdetective/server.py

@@ -16,6 +16,7 @@ from logdetective.constants import (
 from logdetective.extractors import DrainExtractor
 from logdetective.utils import validate_url, compute_certainty
 
+
 class BuildLog(BaseModel):
     """Model of data submitted to API.
     """
@@ -47,7 +48,6 @@ class StagedResponse(Response):
 
 LOG = logging.getLogger("logdetective")
 
-
 LLM_CPP_HOST = os.environ.get("LLAMA_CPP_HOST", "localhost")
 LLM_CPP_SERVER_ADDRESS = f"http://{LLM_CPP_HOST}"
 LLM_CPP_SERVER_PORT = os.environ.get("LLAMA_CPP_SERVER_PORT", 8000)
@@ -55,6 +55,7 @@ LLM_CPP_SERVER_TIMEOUT = os.environ.get("LLAMA_CPP_SERVER_TIMEOUT", 600)
 LOG_SOURCE_REQUEST_TIMEOUT = os.environ.get("LOG_SOURCE_REQUEST_TIMEOUT", 60)
 API_TOKEN = os.environ.get("LOGDETECTIVE_TOKEN", None)
 
+
 def requires_token_when_set(authentication: Annotated[str | None, Header()] = None):
     """
     FastAPI Depend function that expects a header named Authentication
@@ -82,6 +83,7 @@ def requires_token_when_set(authentication: Annotated[str | None, Header()] = No
              API_TOKEN, token)
     raise HTTPException(status_code=401, detail=f"Token {token} not valid.")
 
+
 app = FastAPI(dependencies=[Depends(requires_token_when_set)])
 
 
@@ -99,7 +101,7 @@ def process_url(url: str) -> str:
         if not log_request.ok:
             raise HTTPException(status_code=400,
                                 detail="Something went wrong while getting the logs: "
-                                    f"[{log_request.status_code}] {log_request.text}")
+                                f"[{log_request.status_code}] {log_request.text}")
     else:
         LOG.error("Invalid URL received ")
         raise HTTPException(status_code=400,
@@ -120,9 +122,9 @@ def mine_logs(log: str) -> List[str]:
     LOG.debug("Log summary: \n %s", log_summary)
     LOG.info("Compression ratio: %s", ratio)
 
-
     return log_summary
 
+
 async def submit_text(text: str, max_tokens: int = 0, log_probs: int = 1, stream: bool = False,
                       model: str = "default-model"):
     """Submit prompt to LLM.
@@ -131,17 +133,17 @@ async def submit_text(text: str, max_tokens: int = 0, log_probs: int = 1, stream
     """
     LOG.info("Analyzing the text")
     data = {
-            "prompt": text,
-            "max_tokens": str(max_tokens),
-            "logprobs": str(log_probs),
-            "stream": stream,
-            "model": model}
+        "prompt": text,
+        "max_tokens": str(max_tokens),
+        "logprobs": str(log_probs),
+        "stream": stream,
+        "model": model}
 
     try:
         # Expects llama-cpp server to run on LLM_CPP_SERVER_ADDRESS:LLM_CPP_SERVER_PORT
         response = requests.post(
             f"{LLM_CPP_SERVER_ADDRESS}:{LLM_CPP_SERVER_PORT}/v1/completions",
-            headers={"Content-Type":"application/json"},
+            headers={"Content-Type": "application/json"},
             data=json.dumps(data),
             timeout=int(LLM_CPP_SERVER_TIMEOUT),
             stream=stream)
@@ -154,7 +156,7 @@ async def submit_text(text: str, max_tokens: int = 0, log_probs: int = 1, stream
             raise HTTPException(
                 status_code=400,
                 detail="Something went wrong while getting a response from the llama server: "
-                    f"[{response.status_code}] {response.text}")
+                       f"[{response.status_code}] {response.text}")
         try:
             response = json.loads(response.text)
         except UnicodeDecodeError as ex:

{logdetective-0.2.9 → logdetective-0.2.10}/logdetective/utils.py

@@ -80,6 +80,8 @@ def compute_certainty(probs: List[Dict[str, float] | None]) -> float:
     In this case it's just a matter of applying inverse operation exp.
     Of course that leaves you with a value in range <0, 1> so it needs to be multiplied by 100.
     Simply put, this is the most straightforward way to get the numbers out.
+
+    This function is used in the server codebase.
     """
 
     top_logprobs = [

{logdetective-0.2.9 → logdetective-0.2.10}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "logdetective"
-version = "0.2.9"
+version = "0.2.10"
 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"
@@ -31,7 +31,9 @@ requests = ">0.2.31"
 llama-cpp-python = ">0.2.56,!=0.2.86"
 drain3 = "^0.9.11"
 huggingface-hub = ">0.23.2"
-numpy = "^1.26.0"
+# rawhide has numpy 2, F40 and F41 are still on 1.26
+# we need to support both versions
+numpy = ">=1.26.0"
 
 [build-system]
 requires = ["poetry-core"]