added discord bot

This commit is contained in:
wea_ondara
2024-04-17 21:06:42 +02:00
parent 41d5b4f591
commit 71b25ef337
4 changed files with 47 additions and 12 deletions

24
chat.py
View File

@@ -1,16 +1,16 @@
import requests
from utils.prompt import prompt
messages = []
while True:
user_prompt = prompt('>> User: ')
messages.append({'role': 'user', 'content': user_prompt})
class ChatClient:
messages = []
response = requests.post('http://localhost:8900/', json=messages)
if response.status_code == 200:
messages = response.json()
print('>> Bot : ' + messages[-1]['content'])
else:
messages = messages[0:-1]
print('### Error from backend')
def input(self, message):
self.messages.append({'role': 'user', 'content': message})
response = requests.post('http://localhost:8900/', json=self.messages)
if response.status_code == 200:
self.messages = response.json()
return self.messages[-1]['content']
else:
self.messages = self.messages[0:-1]
return '### Error from backend'