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

feat: add interface params

Browse source
This commit is contained in:
Swayam 2025-10-05 14:30:03 +05:30
parent 38b25702bc
commit 1b77269da6
5 changed files with 131 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

10
surfsense_backend/app/db.py
View file

@ -27,6 +27,8 @@ from app.config import config
from app.retriver.chunks_hybrid_search import ChucksHybridSearchRetriever
from app.retriver.documents_hybrid_search import DocumentHybridSearchRetriever
from sqlalchemy.ext.mutable import MutableDict
if config.AUTH_TYPE == "GOOGLE":
from fastapi_users.db import SQLAlchemyBaseOAuthAccountTableUUID
@ -204,6 +206,14 @@ class SearchSpace(BaseModel, TimestampMixin):
name = Column(String(100), nullable=False, index=True)
description = Column(String(500), nullable=True)
inference_params = Column(
MutableDict.as_mutable(JSON),
nullable=True,
server_default=text(
'\'{"temperature":0.0,"max_tokens":0,"top_k":0,"top_p":0}\'::json'
)
)
user_id = Column(
UUID(as_uuid=True), ForeignKey("user.id", ondelete="CASCADE"), nullable=False
)