proto怎么定义调用chatglm-6b的cli_demo.py

要使用chatglm-6b的cli_demo.py，你可以使用以下步骤：

1. 在proto文件中定义你的输入和输出格式，例如：

syntax = "proto3";

package chat;

message Input {
  string text = 1;
}

message Output {
  string text = 1;
}

2. 在cli_demo.py中导入定义的proto文件和grpc相关的模块，例如：

import grpc
import chat_pb2
import chat_pb2_grpc

3. 定义一个函数，用于向chatglm-6b发送请求并接收响应，例如：

def chat_with_gpt(input_text):
    channel = grpc.insecure_channel('localhost:50051')
    stub = chat_pb2_grpc.ChatStub(channel)
    response = stub.Generate(chat_pb2.Input(text=input_text))
    return response.text

4. 调用该函数并传入你想要输入的文本，例如：

response_text = chat_with_gpt("Hello, how are you?")
print(response_text)

请注意，这只是一个简单的示例，你需要根据你的实际情况进行调整。此外，确保chatglm-6b已启动并正在侦听指定的端口（默认为50051）。