From 650e1a1a6fe93e3953350918f892c165f7a5413f Mon Sep 17 00:00:00 2001 From: Vladimir Blagojevic Date: Wed, 26 Apr 2023 13:56:24 +0200 Subject: [PATCH] fix: gpt-3.5-turbo is an agent streaming model (#4673) * gpt-3.5 is also agent streaming model * Add more streaming capable models * Add end-of-file-fixer * List full model names for clarity --- docs/pydoc/config/retriever.yml | 1 - haystack/agents/utils.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/docs/pydoc/config/retriever.yml b/docs/pydoc/config/retriever.yml index 5bdc6b788..7819ce3ba 100644 --- a/docs/pydoc/config/retriever.yml +++ b/docs/pydoc/config/retriever.yml @@ -24,4 +24,3 @@ renderer: add_method_class_prefix: true add_member_class_prefix: false filename: retriever_api.md - diff --git a/haystack/agents/utils.py b/haystack/agents/utils.py index 792aa533c..4b59bd160 100644 --- a/haystack/agents/utils.py +++ b/haystack/agents/utils.py @@ -2,7 +2,7 @@ from typing import Optional from haystack.agents.types import Color -STREAMING_CAPABLE_MODELS = ["davinci"] +STREAMING_CAPABLE_MODELS = ["text-davinci-003", "gpt-3.5-turbo", "gpt-35-turbo", "gpt-4"] def print_text(text: str, end="", color: Optional[Color] = None) -> None: