From fc76e268a078cb39cb1d7fdd09b4483bbaf60e18 Mon Sep 17 00:00:00 2001 From: Swayam Date: Tue, 7 Oct 2025 01:11:00 +0530 Subject: [PATCH] revert changes --- surfsense_backend/app/db.py | 2 -- surfsense_backend/app/schemas/search_space.py | 15 +++------------ 2 files changed, 3 insertions(+), 14 deletions(-) diff --git a/surfsense_backend/app/db.py b/surfsense_backend/app/db.py index 2fe79f8d1..99be9f277 100644 --- a/surfsense_backend/app/db.py +++ b/surfsense_backend/app/db.py @@ -205,8 +205,6 @@ class SearchSpace(BaseModel, TimestampMixin): name = Column(String(100), nullable=False, index=True) description = Column(String(500), nullable=True) - inference_params = Column(JSON, nullable=True) - user_id = Column( UUID(as_uuid=True), ForeignKey("user.id", ondelete="CASCADE"), nullable=False ) diff --git a/surfsense_backend/app/schemas/search_space.py b/surfsense_backend/app/schemas/search_space.py index 3dd8da4e8..49b5cd094 100644 --- a/surfsense_backend/app/schemas/search_space.py +++ b/surfsense_backend/app/schemas/search_space.py @@ -1,21 +1,14 @@ import uuid from datetime import datetime -from pydantic import BaseModel, ConfigDict, Field +from pydantic import BaseModel, ConfigDict from .base import IDModel, TimestampModel -class InferenceParams(BaseModel): - temperature: float | None = Field(None, ge=0.0, le=2.0) - max_tokens: int | None = Field(None, ge=0) - top_k: int | None = Field(None, ge=0) - top_p: float | None = Field(None, ge=0.0, le=1.0) - class SearchSpaceBase(BaseModel): name: str description: str | None = None - inference_params: InferenceParams | None = None class SearchSpaceCreate(SearchSpaceBase): @@ -23,9 +16,7 @@ class SearchSpaceCreate(SearchSpaceBase): class SearchSpaceUpdate(SearchSpaceBase): - name: str | None = None - description: str | None = None - inference_params: InferenceParams | None = None + pass class SearchSpaceRead(SearchSpaceBase, IDModel, TimestampModel): @@ -33,4 +24,4 @@ class SearchSpaceRead(SearchSpaceBase, IDModel, TimestampModel): created_at: datetime user_id: uuid.UUID - model_config = ConfigDict(from_attributes=True) + model_config = ConfigDict(from_attributes=True) \ No newline at end of file