apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-06-02 19:55:18 +02:00
Code Issues Projects Releases Packages Wiki Activity

revert changes

Browse source
This commit is contained in:
Swayam 2025-10-07 01:11:00 +05:30
parent 9f99b09e0d
commit fc76e268a0
2 changed files with 3 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

2
surfsense_backend/app/db.py
View file

@ -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
)

15
surfsense_backend/app/schemas/search_space.py
View file

@ -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)