fix pydantic versioning

lambdaprompt/backends.py

@@ -1,6 +1,6 @@
 import os
 import aiohttp
-from pydantic import BaseModel, Extra
+from pydantic import BaseModel
 from typing import Optional, Union, List
 import tenacity
 
@@ -40,7 +40,7 @@ def get_backend(method):
 class Backend:
     class Parameters(BaseModel):
         class Config:
-            extra = Extra.forbid
+            extra = 'forbid'
 
     def __init__(self, **param_override):
         self.param_override = self.Parameters(**param_override)
@@ -92,7 +92,7 @@ class Parameters(RequestBackend.Parameters):
         model: str = 'gpt-3.5-turbo-instruct'
         presence_penalty: float = 0.2
         frequency_penalty: float = 0.2
-        stop: Optional[Union[str, List[str]]]
+        stop: Optional[Union[str, List[str]]] = None
 
     def __init__(self, openai_api_key=None, **param_override):
         self.openai_api_key = openai_api_key or os.environ.get("OPENAI_API_KEY")

lambdaprompt/server/main.py

@@ -12,7 +12,7 @@
 
 import aiosqlite
 from fastapi import BackgroundTasks, FastAPI, HTTPException
-from pydantic import BaseSettings
+from pydantic_settings import BaseSettings
 
 import lambdaprompt
 

pyproject.toml

@@ -13,7 +13,7 @@ classifiers = [
     "Programming Language :: Python :: 3",
 ]
 dependencies = [
-    "requests", "aiohttp", "python-dotenv", "jinja2", "nest_asyncio", "pyyaml", "tenacity", "pydantic"
+    "requests", "aiohttp", "python-dotenv", "jinja2", "nest_asyncio", "pyyaml", "tenacity", "pydantic", "pydantic-settings"
 ]
 urls = {homepage = "https://github.com/approximatelabs/lambdaprompt"}
 dynamic = ["version"]

tests/test_server.py

@@ -10,7 +10,7 @@ def client(mocker):
     from lambdaprompt.server import main
 
     class newSettings(main.Settings):
-        sqlite_path = ":memory:"
+        sqlite_path: str = ":memory:"
         prompt_library_paths: List[str] = [
             str(os.path.join(os.path.split(__file__)[0], "library"))
         ]