17 lines
470 B
Python
17 lines
470 B
Python
import requests
|
|
|
|
|
|
class ChatClient:
|
|
messages = []
|
|
|
|
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'
|