fix: 🐛 utilize generic model object

iwpnd · Nov 21, 2021 · 1f95d2c · 1f95d2c
1 parent 04cad31
commit 1f95d2c
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/pyle38/responses.py b/pyle38/responses.py
@@ -3,7 +3,7 @@
 from pydantic import BaseModel as PydanticBaseModel
 from pydantic.generics import GenericModel as PydanticGenericModel
 
-T = TypeVar("T", bound=Dict)
+T = TypeVar("T", bound=Union[Dict, str])
 S = TypeVar("S", bound=str)
 
 
@@ -56,11 +56,7 @@ class ObjectsResponse(JSONResponse, GenericModel, Generic[T]):
     fields: Optional[List[str]] = None
 
 
-class StringObject(BaseModel):
-    object: str
-    id: Union[str, int]
-    distance: Optional[float] = None
-    fields: Optional[List[int]] = None
+StringObject = Object[str]
 
 
 class StringObjectResponse(JSONResponse, GenericModel):