1 line
2.3 KiB
JSON
1 line
2.3 KiB
JSON
{"repo": "awszyai/himcm-2025", "files": {"src/deepseek_api.py": {"179eb1c37db9339e41dae0d2a22e3ebf031185ee": {"t": 1785768192.0381854, "n": 2003, "h": "f5628d110ac2ba12", "c": "import os\nfrom dotenv import load_dotenv\nfrom openai import OpenAI\nload_dotenv()\n\n# (0) Official\nDeepseek_KEY = os.getenv('Deepseek_KEY')\nDeepseek_URL = \"https://api.deepseek.com\"\nclient = OpenAI(api_key=Deepseek_KEY, base_url=Deepseek_URL)\nMODEL_NAME = \"deepseek-chat\"\n\n# (1)\n# LUCHENTECH_KEY = os.getenv('LUCHENTECH_KEY')\n# LUCHENTECH_URL = 'https://cloud.luchentech.com/api/maas/'\n# client = OpenAI(api_key=LUCHENTECH_KEY, base_url=LUCHENTECH_URL)\n# MODEL_NAME = \"deepseek-ai/DeepSeek-R1\"\n\n# import requests\n\n# url = 'https://cloud.luchentech.com/api/maas/chat/completions'\n\n# headers = {\n# 'Content-Type': 'application/json',\n# 'Authorization': 'Bearer <your_api_key>'\n# }\n# payload = {\n# \"model\": \"deepseek-ai/DeepSeek-R1\",\n# \"messages\": [\n# {\n# \"role\": \"user\",\n# \"content\": \"写一首描绘春天的诗？\"\n# }\n# ],\n# \"stream\": False, \n# \"max_tokens\": 512\n# }\n\n# response = requests.request(\"POST\", url, json=payload, headers=headers)\n\n# print(response.json())\n\n# (2) Tencent\n# tencent_KEY = os.getenv('tencent_KEY')\n# tencent_KEY_wzj = os.getenv('tencent_KEY_wzj')\n# tencent_URL = \"https://api.lkeap.cloud.tencent.com/v1\"\n# # tencent_URL = \"https://api.lkeap.cloud.tencent.com/\"\n# client = OpenAI(api_key=tencent_KEY_wzj, base_url=tencent_URL)\n# MODEL_NAME = \"deepseek-r1\"\n\n\n# (3) 星河社区\n\n\n# Utils\ndef send_messages(messages, model=MODEL_NAME, tools=None):\n completion = client.chat.completions.create(\n model=MODEL_NAME,\n messages = messages,\n # temperature=1.3, #deepseek\n temperature=1, #kimi\n response_format={\"type\": \"json_object\"}, # 确保返回 JSON 格式\n n=1 # 请求返回1个结果\n )\n response = completion.choices[0].message.content.strip()\n return response\n\n# Usage example\nif __name__ == \"__main__\":\n messages = [\n {\"role\": \"system\", \"content\": \"You are a helpful assistant.\"},\n {\"role\": \"user\", \"content\": \"Tell me a joke.\"}\n ]\n response = send_messages(messages)\n print(response)\n \n\n"}}}} |