Add logfire integration

.pre-commit-config.yaml

@@ -21,7 +21,7 @@ repos:
       - id: trailing-whitespace
 
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.15.15
+    rev: v0.15.16
     hooks:
       - id: ruff-check
         args: ["--fix", "--exit-non-zero-on-fix"]

.vscode/settings.json

@@ -6,6 +6,7 @@
         "automerge",
         "buildx",
         "CLAHE",
+        "deepseek",
         "Denoise",
         "denoising",
         "docstrings",
@@ -26,6 +27,7 @@
         "killyoy",
         "levelname",
         "Licka",
+        "logfire",
         "Lördagsgodis",
         "lovibot",
         "Lovinator",
@@ -57,4 +59,4 @@
         "Waifu",
         "Zenless"
     ]
-}
+}

main.py

@@ -8,6 +8,7 @@ import os
 import re
 from collections import deque
 from dataclasses import dataclass
+from logging import basicConfig
 from typing import TYPE_CHECKING
 from typing import Any
 from typing import Literal
@@ -17,6 +18,7 @@ from typing import TypeVar
 import cv2
 import discord
 import httpx
+import logfire
 import numpy as np
 import ollama
 import openai
@@ -58,10 +60,10 @@ sentry_sdk.init(
     send_default_pii=True,
 )
 
+logfire.configure()
+basicConfig(handlers=[logfire.LogfireLoggingHandler()], level=logging.DEBUG)
 
 logger: logging.Logger = logging.getLogger(__name__)
-logger.setLevel(logging.DEBUG)
-
 
 discord_token: str = os.getenv("DISCORD_TOKEN", "")
 os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_TOKEN", "")
@@ -521,11 +523,11 @@ async def chat(  # noqa: PLR0913, PLR0917
     ):
         if author_name != bot_name:
             message_history.append(
-                ModelRequest(parts=[UserPromptPart(content=message_content)]),
+                ModelRequest([UserPromptPart(content=message_content)]),
             )
         else:
             message_history.append(
-                ModelResponse(parts=[TextPart(content=message_content)]),
+                ModelResponse([TextPart(content=message_content)]),
             )
 
     # Compact history to avoid exceeding model context limits

pyproject.toml

@@ -8,6 +8,7 @@ dependencies = [
     "audioop-lts",
     "discord-py",
     "httpx",
+    "logfire[httpx]",
     "numpy",
     "ollama",
     "openai",