Compare commits
No commits in common. "72ad01ff2feb299c7e66b2383d781e4203304edb" and "a6d35ef73ff493505011ca1b3a69f1caca92de4b" have entirely different histories.
72ad01ff2f
...
a6d35ef73f
4 changed files with 6 additions and 11 deletions
|
|
@ -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"]
|
||||||
|
|
|
||||||
4
.vscode/settings.json
vendored
4
.vscode/settings.json
vendored
|
|
@ -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",
|
||||||
|
|
@ -59,4 +57,4 @@
|
||||||
"Waifu",
|
"Waifu",
|
||||||
"Zenless"
|
"Zenless"
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
|
|
|
||||||
10
main.py
10
main.py
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -8,7 +8,6 @@ dependencies = [
|
||||||
"audioop-lts",
|
"audioop-lts",
|
||||||
"discord-py",
|
"discord-py",
|
||||||
"httpx",
|
"httpx",
|
||||||
"logfire[httpx]",
|
|
||||||
"numpy",
|
"numpy",
|
||||||
"ollama",
|
"ollama",
|
||||||
"openai",
|
"openai",
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue