Compare commits

..

No commits in common. "72ad01ff2feb299c7e66b2383d781e4203304edb" and "a6d35ef73ff493505011ca1b3a69f1caca92de4b" have entirely different histories.

4 changed files with 6 additions and 11 deletions

View file

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

View file

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

10
main.py
View file

@ -8,7 +8,6 @@ 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
@ -18,7 +17,6 @@ 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
@ -60,10 +58,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", "")
@ -523,11 +521,11 @@ async def chat( # noqa: PLR0913, PLR0917
): ):
if author_name != bot_name: if author_name != bot_name:
message_history.append( message_history.append(
ModelRequest([UserPromptPart(content=message_content)]), ModelRequest(parts=[UserPromptPart(content=message_content)]),
) )
else: else:
message_history.append( message_history.append(
ModelResponse([TextPart(content=message_content)]), ModelResponse(parts=[TextPart(content=message_content)]),
) )
# Compact history to avoid exceeding model context limits # Compact history to avoid exceeding model context limits

View file

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