From 64d7b720a1ccbfa34d4fde74b352785cd16ece3b Mon Sep 17 00:00:00 2001
From: realTaki <shichuan@ualberta.ca>
Date: Wed, 4 Oct 2023 11:39:40 +0000
Subject: [PATCH] YamlModel is deprecated class

---
 embedbase/settings.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/embedbase/settings.py b/embedbase/settings.py
index d9bdd93..075bf01 100644
--- a/embedbase/settings.py
+++ b/embedbase/settings.py
@@ -2,7 +2,8 @@
 from functools import lru_cache
 import typing
 import os
-from pydantic_yaml import YamlModel
+from pydantic import BaseModel
+from pydantic_yaml import parse_yaml_file_as
 
 
 class VectorDatabaseEnum(str, Enum):
@@ -18,7 +19,7 @@ class EmbeddingProvider(str, Enum):
     COHERE = "cohere"
 
 
-class Settings(YamlModel):
+class Settings(BaseModel):
     openai_api_key: typing.Optional[str] = None
     openai_organization: typing.Optional[str] = None
     supabase_url: typing.Optional[str] = None
@@ -33,7 +34,7 @@ def get_settings_from_file(path: str = "config.yaml"):
     """
     Read settings from a file, only supports yaml for now
     """
-    settings = Settings.parse_file(path)
+    settings = parse_yaml_file_as(Settings, path)
 
     # TODO: move
     # if firebase, init firebase