ldooks/dragon-bot
ldooks/dragon-bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

I'm dumb, dont make 3 rquests for the same json file

Browse Source
This commit is contained in:
Luke Robles 2024-04-19 10:31:13 -07:00
parent b91e6a7ca5
commit bd0d576c7e
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
app/bot.py
View File

@ -157,8 +157,9 @@ async def on_message(ctx):
llm_rule_endpoint = ( llm_rule_endpoint = (
"http://dragon-bot-json.dragon-bot.svc.cluster.local/rules.json" "http://dragon-bot-json.dragon-bot.svc.cluster.local/rules.json"
) )
llm_rules = requests.get(llm_rule_endpoint).json()
if ctx.author.id in requests.get(llm_rule_endpoint).json()["disallowed_users"]: if ctx.author.id in llm_rules["disallowed_users"]:
responses = [ responses = [
"You cant do that right now", "You cant do that right now",
"You cant use this feature right now", "You cant use this feature right now",
@ -168,7 +169,7 @@ async def on_message(ctx):
return return
url = "http://192.168.1.137:1337/v1/chat/completions" url = "http://192.168.1.137:1337/v1/chat/completions"
instructions = requests.get(llm_rule_endpoint).json()["prompt"] instructions = llm_rules["prompt"]
payload = { payload = {
"messages": [ "messages": [
@ -183,7 +184,7 @@ async def on_message(ctx):
"role": "user", "role": "user",
}, },
], ],
"model": requests.get(llm_rule_endpoint).json()["model"], "model": llm_rules["model"],
"stream": False, "stream": False,
"max_tokens": 4096, "max_tokens": 4096,
"stop": ["goon"], "stop": ["goon"],