mirror of
https://github.com/langgenius/dify.git
synced 2026-01-14 06:07:33 +08:00
5bbf685035
Signed-off-by: -LAN- <laipz8200@outlook.com> Signed-off-by: kenwoodjw <blackxin55+@gmail.com> Signed-off-by: Yongtao Huang <yongtaoh2022@gmail.com> Signed-off-by: yihong0618 <zouzou0208@gmail.com> Signed-off-by: zhanluxianshen <zhanluxianshen@163.com> Co-authored-by: -LAN- <laipz8200@outlook.com> Co-authored-by: GuanMu <ballmanjq@gmail.com> Co-authored-by: Davide Delbianco <davide.delbianco@outlook.com> Co-authored-by: NeatGuyCoding <15627489+NeatGuyCoding@users.noreply.github.com> Co-authored-by: kenwoodjw <blackxin55+@gmail.com> Co-authored-by: Yongtao Huang <yongtaoh2022@gmail.com> Co-authored-by: Yongtao Huang <99629139+hyongtao-db@users.noreply.github.com> Co-authored-by: Qiang Lee <18018968632@163.com> Co-authored-by: 李强04 <liqiang04@gaotu.cn> Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: Asuka Minato <i@asukaminato.eu.org> Co-authored-by: Matri Qi <matrixdom@126.com> Co-authored-by: huayaoyue6 <huayaoyue@163.com> Co-authored-by: Bowen Liang <liangbowen@gf.com.cn> Co-authored-by: znn <jubinkumarsoni@gmail.com> Co-authored-by: crazywoola <427733928@qq.com> Co-authored-by: crazywoola <100913391+crazywoola@users.noreply.github.com> Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> Co-authored-by: yihong <zouzou0208@gmail.com> Co-authored-by: Muke Wang <shaodwaaron@gmail.com> Co-authored-by: wangmuke <wangmuke@kingsware.cn> Co-authored-by: Wu Tianwei <30284043+WTW0313@users.noreply.github.com> Co-authored-by: quicksand <quicksandzn@gmail.com> Co-authored-by: 非法操作 <hjlarry@163.com> Co-authored-by: zxhlyh <jasonapring2015@outlook.com> Co-authored-by: Eric Guo <eric.guocz@gmail.com> Co-authored-by: Zhedong Cen <cenzhedong2@126.com> Co-authored-by: jiangbo721 <jiangbo721@163.com> Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com> Co-authored-by: hjlarry <25834719+hjlarry@users.noreply.github.com> Co-authored-by: lxsummer <35754229+lxjustdoit@users.noreply.github.com> Co-authored-by: 湛露先生 <zhanluxianshen@163.com> Co-authored-by: Guangdong Liu <liugddx@gmail.com> Co-authored-by: QuantumGhost <obelisk.reg+git@gmail.com> Co-authored-by: Claude <noreply@anthropic.com> Co-authored-by: Yessenia-d <yessenia.contact@gmail.com> Co-authored-by: huangzhuo1949 <167434202+huangzhuo1949@users.noreply.github.com> Co-authored-by: huangzhuo <huangzhuo1@xiaomi.com> Co-authored-by: 17hz <0x149527@gmail.com> Co-authored-by: Amy <1530140574@qq.com> Co-authored-by: Joel <iamjoel007@gmail.com> Co-authored-by: Nite Knite <nkCoding@gmail.com> Co-authored-by: Yeuoly <45712896+Yeuoly@users.noreply.github.com> Co-authored-by: Petrus Han <petrus.hanks@gmail.com> Co-authored-by: iamjoel <2120155+iamjoel@users.noreply.github.com> Co-authored-by: Kalo Chin <frog.beepers.0n@icloud.com> Co-authored-by: Ujjwal Maurya <ujjwalsbx@gmail.com> Co-authored-by: Maries <xh001x@hotmail.com>
78 lines
2.3 KiB
Python
78 lines
2.3 KiB
Python
from flask_restx import Api, Namespace, fields
|
|
|
|
from fields.conversation_fields import message_file_fields
|
|
from libs.helper import TimestampField
|
|
|
|
from .raws import FilesContainedField
|
|
|
|
feedback_fields = {
|
|
"rating": fields.String,
|
|
}
|
|
|
|
|
|
def build_feedback_model(api_or_ns: Api | Namespace):
|
|
"""Build the feedback model for the API or Namespace."""
|
|
return api_or_ns.model("Feedback", feedback_fields)
|
|
|
|
|
|
agent_thought_fields = {
|
|
"id": fields.String,
|
|
"chain_id": fields.String,
|
|
"message_id": fields.String,
|
|
"position": fields.Integer,
|
|
"thought": fields.String,
|
|
"tool": fields.String,
|
|
"tool_labels": fields.Raw,
|
|
"tool_input": fields.String,
|
|
"created_at": TimestampField,
|
|
"observation": fields.String,
|
|
"files": fields.List(fields.String),
|
|
}
|
|
|
|
|
|
def build_agent_thought_model(api_or_ns: Api | Namespace):
|
|
"""Build the agent thought model for the API or Namespace."""
|
|
return api_or_ns.model("AgentThought", agent_thought_fields)
|
|
|
|
|
|
retriever_resource_fields = {
|
|
"id": fields.String,
|
|
"message_id": fields.String,
|
|
"position": fields.Integer,
|
|
"dataset_id": fields.String,
|
|
"dataset_name": fields.String,
|
|
"document_id": fields.String,
|
|
"document_name": fields.String,
|
|
"data_source_type": fields.String,
|
|
"segment_id": fields.String,
|
|
"score": fields.Float,
|
|
"hit_count": fields.Integer,
|
|
"word_count": fields.Integer,
|
|
"segment_position": fields.Integer,
|
|
"index_node_hash": fields.String,
|
|
"content": fields.String,
|
|
"created_at": TimestampField,
|
|
}
|
|
|
|
message_fields = {
|
|
"id": fields.String,
|
|
"conversation_id": fields.String,
|
|
"parent_message_id": fields.String,
|
|
"inputs": FilesContainedField,
|
|
"query": fields.String,
|
|
"answer": fields.String(attribute="re_sign_file_url_answer"),
|
|
"feedback": fields.Nested(feedback_fields, attribute="user_feedback", allow_null=True),
|
|
"retriever_resources": fields.List(fields.Nested(retriever_resource_fields)),
|
|
"created_at": TimestampField,
|
|
"agent_thoughts": fields.List(fields.Nested(agent_thought_fields)),
|
|
"message_files": fields.List(fields.Nested(message_file_fields)),
|
|
"status": fields.String,
|
|
"error": fields.String,
|
|
}
|
|
|
|
message_infinite_scroll_pagination_fields = {
|
|
"limit": fields.Integer,
|
|
"has_more": fields.Boolean,
|
|
"data": fields.List(fields.Nested(message_fields)),
|
|
}
|