From a7238b96b742ae516fb10591667bfd8e21167210 Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Tue, 21 May 2024 20:11:04 -0700 Subject: [PATCH] fix --- pipelines/examples/azure_openai_pipeline.py | 2 +- pipelines/examples/openai_pipeline.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/pipelines/examples/azure_openai_pipeline.py b/pipelines/examples/azure_openai_pipeline.py index 2e1dcb6..0e8d77e 100644 --- a/pipelines/examples/azure_openai_pipeline.py +++ b/pipelines/examples/azure_openai_pipeline.py @@ -47,7 +47,7 @@ class Pipeline: r.raise_for_status() - if data["stream"]: + if body["stream"]: return r.iter_lines() else: return r.json() diff --git a/pipelines/examples/openai_pipeline.py b/pipelines/examples/openai_pipeline.py index 54f99c3..80b8439 100644 --- a/pipelines/examples/openai_pipeline.py +++ b/pipelines/examples/openai_pipeline.py @@ -44,7 +44,7 @@ class Pipeline: r.raise_for_status() - if data["stream"]: + if body["stream"]: return r.iter_lines() else: return r.json()