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

10
main.py
View file

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

View file

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