Cleanup as per Bruce
Signed-off-by: Matt Williams <m@technovangelist.com>
This commit is contained in:
parent
aec742b6d2
commit
76bc4d0458
2 changed files with 4 additions and 4 deletions
|
@ -15,6 +15,8 @@ def chat(messages):
|
|||
|
||||
for line in r.iter_lines():
|
||||
body = json.loads(line)
|
||||
if "error" in body:
|
||||
raise Exception(body["error"])
|
||||
if body.get("done") is False:
|
||||
message = body.get("message", "")
|
||||
content = message.get("content", "")
|
||||
|
@ -22,8 +24,6 @@ def chat(messages):
|
|||
# the response streams one token at a time, print that as we receive it
|
||||
print(content, end="", flush=True)
|
||||
|
||||
if "error" in body:
|
||||
raise Exception(body["error"])
|
||||
|
||||
if body.get("done", False):
|
||||
message["content"] = output
|
||||
|
@ -32,7 +32,7 @@ def chat(messages):
|
|||
|
||||
def main():
|
||||
messages = []
|
||||
) # the context stores a conversation history, you can use this to make the model more context aware
|
||||
|
||||
while True:
|
||||
user_input = input("Enter a prompt: ")
|
||||
print()
|
||||
|
|
Loading…
Reference in a new issue