From bdac281c425bd3bd27dbebf13df39d10ca398da2 Mon Sep 17 00:00:00 2001 From: FayazRahman Date: Thu, 11 May 2023 02:02:59 +0530 Subject: [PATCH] fix --- loopgpt/agent.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/loopgpt/agent.py b/loopgpt/agent.py index 4c73e30..944960e 100644 --- a/loopgpt/agent.py +++ b/loopgpt/agent.py @@ -71,6 +71,8 @@ def __init__( if model is None: model = OpenAIModel("gpt-3.5-turbo") + elif isinstance(model, str): + model = OpenAIModel(model) if embedding_provider is None: embedding_provider = OpenAIEmbeddingProvider()