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