This commit is contained in:
parent
8ea1fd9014
commit
72ad01ff2f
3 changed files with 10 additions and 5 deletions
2
.vscode/settings.json
vendored
2
.vscode/settings.json
vendored
|
|
@ -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
10
main.py
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -8,6 +8,7 @@ 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