mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-02-07 02:55:08 +08:00
Feat: support verify to set llm key and boost bigrams. (#12980)
#12863 ### Type of change - [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
@ -48,8 +48,8 @@ class API4ConversationService(CommonService):
|
||||
@DB.connection_context()
|
||||
def get_list(cls, dialog_id, tenant_id,
|
||||
page_number, items_per_page,
|
||||
orderby, desc, id, user_id=None, include_dsl=True, keywords="",
|
||||
from_date=None, to_date=None
|
||||
orderby, desc, id=None, user_id=None, include_dsl=True, keywords="",
|
||||
from_date=None, to_date=None, exp_user_id=None
|
||||
):
|
||||
if include_dsl:
|
||||
sessions = cls.model.select().where(cls.model.dialog_id == dialog_id)
|
||||
@ -66,6 +66,8 @@ class API4ConversationService(CommonService):
|
||||
sessions = sessions.where(cls.model.create_date >= from_date)
|
||||
if to_date:
|
||||
sessions = sessions.where(cls.model.create_date <= to_date)
|
||||
if exp_user_id:
|
||||
sessions = sessions.where(cls.model.exp_user_id == exp_user_id)
|
||||
if desc:
|
||||
sessions = sessions.order_by(cls.model.getter_by(orderby).desc())
|
||||
else:
|
||||
@ -74,6 +76,17 @@ class API4ConversationService(CommonService):
|
||||
sessions = sessions.paginate(page_number, items_per_page)
|
||||
|
||||
return count, list(sessions.dicts())
|
||||
|
||||
@classmethod
|
||||
@DB.connection_context()
|
||||
def get_names(cls, dialog_id, exp_user_id):
|
||||
fields = [cls.model.id, cls.model.name,]
|
||||
sessions = cls.model.select(*fields).where(
|
||||
cls.model.dialog_id == dialog_id,
|
||||
cls.model.exp_user_id == exp_user_id
|
||||
).order_by(cls.model.getter_by("create_date").desc())
|
||||
|
||||
return list(sessions.dicts())
|
||||
|
||||
@classmethod
|
||||
@DB.connection_context()
|
||||
|
||||
Reference in New Issue
Block a user