Files
llm/chat.py
2024-04-18 20:19:44 +02:00

17 lines
500 B
Python

import requests
class ChatClient:
messages = []
def input(self, user_name, message):
self.messages.append({'role': 'user', 'name': user_name, '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'