NekoAI-Lab / app /config.py
nekoko
feat: Sticker DB
1c2b077
import os
# 应用配置
DATASET_ID = "Nekoko/StickerSet"
COZE_API_TOKEN = os.getenv('COZE_API_TOKEN')
HUGGING_FACE_TOKEN = os.getenv('HUGGING_FACE_TOKEN')
DEEPSEEK_API_KEY = os.getenv('DEEPSEEK_API_KEY')
MILVUS_DB_FILE = os.getenv('MILVUS_DB_FILE', "./sticker.db")
MILVUS_DB_URL = os.getenv('MILVUS_DB_URL', "./sticker.db")
MILVUS_DB_TOKEN = os.getenv('MILVUS_DB_TOKEN', "")
EMBEDDING_MODEL = 'shibing624/text2vec-base-chinese'
PUBLIC_URL = os.getenv('PUBLIC_URL')
STICKER_RERANKING_SYSTEM_PROMPT = """
## 角色
你是专业的表情包搜索助手,能找到最符合用户关键词的表情包,并为给定表情包列表中各表情包的关键词匹配度打分(0 - 1)并排序。
## 回复示例:
[{ "reason": "判断的理由", "sticker_id": "1", "score": 0.8,},
{ "reason": "判断的理由", "sticker_id": "2", "score": 0.3,}]
## 功能: 分析相关性
1. 评估表情包列表中表情包描述与关键词的匹配度,给出0到1的评分。
2. 给出判断的理由(中文)
## 限制
- 仅处理表情包搜索相关内容,不答无关问题,分清楚关键词和表情包列表。
- 请只输出纯JSON格式,不要包含任何Markdown标记如```json或```。
"""
# 服务器配置
HOST = "0.0.0.0"
PORT = 7860