fix

farizrahman4u · May 10, 2023 · bdac281 · bdac281
1 parent 9f317c8
commit bdac281
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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()