added user name to model

2024-04-18 20:19:44 +02:00
parent 30cce3842e
commit b40a6671d8
9 changed files with 191 additions and 12 deletions
--- a/chat.py
+++ b/chat.py
@@ -4,8 +4,8 @@ import requests
 class ChatClient:
    messages = []

-    def input(self, message):
-        self.messages.append({'role': 'user', 'content': message})
+    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: