提交 1e2124ff 作者: glide-the

参数校验

上级 c389f1a3
...@@ -184,7 +184,7 @@ class FastChatOpenAILLMChain(RemoteRpcModel, Chain, ABC): ...@@ -184,7 +184,7 @@ class FastChatOpenAILLMChain(RemoteRpcModel, Chain, ABC):
history = inputs[self.history_key] history = inputs[self.history_key]
streaming = inputs[self.streaming_key] streaming = inputs[self.streaming_key]
prompt = inputs[self.prompt_key] prompt = inputs[self.prompt_key]
stop = inputs['stop'] stop = inputs.get("stop", None)
print(f"__call:{prompt}") print(f"__call:{prompt}")
try: try:
...@@ -243,7 +243,6 @@ if __name__ == "__main__": ...@@ -243,7 +243,6 @@ if __name__ == "__main__":
chain.call_model_name("gpt-3.5-turbo") chain.call_model_name("gpt-3.5-turbo")
answer_result_stream_result = chain({"streaming": False, answer_result_stream_result = chain({"streaming": False,
"stop": "",
"prompt": "你好", "prompt": "你好",
"history": [] "history": []
}) })
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论