🎨 chore: run black to format the code

This commit is contained in:
2023-10-31 12:57:06 +01:00
parent c0d6b3f2d5
commit fc370828df
4 changed files with 15 additions and 6 deletions

View File

@@ -108,7 +108,7 @@ class Chat:
messages: list[discord.Message] = await fetch_messages_history(
self.message.channel, 10, self.original_message
)
#if latst item id is not original message id, add original message to messages
# if latst item id is not original message id, add original message to messages
if messages[-1].id != self.original_message.id:
messages.append(self.original_message)
self.context = []

View File

@@ -34,7 +34,8 @@ def createPrompt(
sysprompt.replace("[server-name]", guildName)
.replace("[channel-name]", channelName)
.replace(
"[date-and-time]", datetime.datetime.utcnow().strftime("%d/%m/%Y %H:%M:%S")
"[date-and-time]",
datetime.datetime.utcnow().strftime("%d/%m/%Y %H:%M:%S"),
)
)
prompt[0]["content"] = sysprompt
@@ -44,7 +45,8 @@ def createPrompt(
.replace("[server-name]", guildName)
.replace("[channel-name]", channelName)
.replace(
"[date-and-time]", datetime.datetime.utcnow().strftime("%d/%m/%Y %H:%M:%S")
"[date-and-time]",
datetime.datetime.utcnow().strftime("%d/%m/%Y %H:%M:%S"),
)
)
else:

View File

@@ -7,8 +7,9 @@ anthropic = AsyncAnthropic(
api_key=os.getenv("ANTHROPIC_KEY"),
)
async def claude(messages):
#messages are a dict {} with content and roler.
# messages are a dict {} with content and roler.
prompt = ""
for message in messages:
if message["role"] == "system":
@@ -20,7 +21,12 @@ async def claude(messages):
elif message["role"] == "function":
...
prompt += AI_PROMPT
completion = await anthropic.completions.create(stop_sequences=["\n\nHuman (", "\n\nSYSTEM: "], model="claude-2", max_tokens_to_sample=512, prompt=prompt)
completion = await anthropic.completions.create(
stop_sequences=["\n\nHuman (", "\n\nSYSTEM: "],
model="claude-2",
max_tokens_to_sample=512,
prompt=prompt,
)
return {
"name": "send_message",
"arguments": {"message": completion.completion},

View File

@@ -5,6 +5,7 @@ from src.chatUtils.requesters.llama import llama
from src.chatUtils.requesters.llama2 import llama2
from src.chatUtils.requesters.claude import claude
class ModelNotFound(Exception):
pass