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()