(self, folder, session)
| 19 | self.log = {} |
| 20 | |
| 21 | def start(self, folder, session): |
| 22 | log_file = [] |
| 23 | client_socket, client_address = self.socket.accept() |
| 24 | while True: |
| 25 | data = client_socket.recv(10000).decode() |
| 26 | if data == "": |
| 27 | self.stop(client_socket) |
| 28 | break |
| 29 | else: |
| 30 | log_file.append({"role": "user", "content": data}) |
| 31 | try: |
| 32 | ret = session.model_inference(json.loads(data)) |
| 33 | log_file.append({"role": "agent", "content": ret}) |
| 34 | self.send_message(client_socket, ret) |
| 35 | except json.decoder.JSONDecodeError: |
| 36 | self.send_message(client_socket, "") |
| 37 | self.log[folder] = log_file |
| 38 | |
| 39 | def send_message(self, client_socket, message): |
| 40 | client_socket.sendall(message.encode()) |
no test coverage detected