Add logfire integration
All checks were successful
CI / ci (push) Successful in 12s

This commit is contained in:
Joakim Hellsén 2026-06-08 19:09:20 +02:00
commit 72ad01ff2f
Signed by: Joakim Hellsén
SSH key fingerprint: SHA256:/9h/CsExpFp+PRhsfA0xznFx2CGfTT5R/kpuFfUgEQk
3 changed files with 10 additions and 5 deletions

View file

@ -6,6 +6,7 @@
"automerge", "automerge",
"buildx", "buildx",
"CLAHE", "CLAHE",
"deepseek",
"Denoise", "Denoise",
"denoising", "denoising",
"docstrings", "docstrings",
@ -26,6 +27,7 @@
"killyoy", "killyoy",
"levelname", "levelname",
"Licka", "Licka",
"logfire",
"Lördagsgodis", "Lördagsgodis",
"lovibot", "lovibot",
"Lovinator", "Lovinator",

10
main.py
View file

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

View file

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