wip

airtai · Jul 31, 2024 · ea945e9 · ea945e9
1 parent 7b1982e
commit ea945e9
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 4 deletions.
diff --git a/fastagency/app.py b/fastagency/app.py
@@ -12,7 +12,7 @@
 from pydantic import BaseModel, TypeAdapter, ValidationError
 
 from .auth_token.auth import DeploymentAuthToken, create_deployment_auth_token
-from .db.base import BackendDBProtocol, FrontendDBProtocol
+from .db.base import BackendDBProtocol, DefaultDB, FrontendDBProtocol
 from .db.prisma import PrismaBackendDB, PrismaFrontendDB
 from .helpers import (
     add_model_to_user,

diff --git a/fastagency/auth_token/auth.py b/fastagency/auth_token/auth.py
@@ -8,7 +8,7 @@
 from fastapi import HTTPException
 from pydantic import BaseModel
 
-from ..db.base import BackendDBProtocol, FrontendDBProtocol
+from ..db.base import BackendDBProtocol, DefaultDB, FrontendDBProtocol
 
 
 def generate_auth_token(length: int = 32) -> str:

diff --git a/fastagency/db/base.py b/fastagency/db/base.py
@@ -11,6 +11,11 @@
     runtime_checkable,
 )
 
+class KeyNotFound(ValueError):
+    pass
+
+class KeyExists(ValueError):
+    pass
 
 @runtime_checkable
 class BackendDBProtocol(Protocol):

diff --git a/fastagency/models/agents/base.py b/fastagency/models/agents/base.py
@@ -6,7 +6,7 @@
 
 from fastagency.openapi.client import Client
 
-from ...db.base import BackendDBProtocol
+from ...db.base import BackendDBProtocol, DefaultDB
 from ..base import Model
 from ..registry import Registry
 from ..toolboxes.toolbox import ToolboxRef

diff --git a/fastagency/models/base.py b/fastagency/models/base.py
@@ -5,7 +5,7 @@
 from pydantic import BaseModel, Field, create_model, model_validator
 from typing_extensions import TypeAlias
 
-from ..db.base import BackendDBProtocol
+from ..db.base import BackendDBProtocol, DefaultDB
 
 M = TypeVar("M", bound="Model")