kimi

docker-compose.yml

@@ -0,0 +1,13 @@
+version: '3'
+
+services:
+  reverse-kimi:
+    container_name: reverse-kimi
+    image: yunpengtai/reverse-kimi:latest
+    restart: always
+    ports:
+      - "6867:6867"
+    volumes:
+      - ./config.json:/app/config.json
+    environment:
+      - TZ=Asia/Shanghai

dockerfile

@@ -0,0 +1,18 @@
+# 使用官方 Python 运行时作为父镜像
+FROM python:3.9-slim
+
+# 设置工作目录为 /app
+WORKDIR /app
+
+# 将当前目录内容复制到位于 /app 的容器中
+COPY . /app
+
+# 设置环境变量
+ENV PYTHONUNBUFFERED=1
+
+RUN pip3 install --upgrade pip
+
+RUN pip3 install --no-cache-dir --upgrade -r requirements.txt
+
+# 在容器启动时先运行 main.py，然后运行 server.py
+CMD ["sh", "-c", "python3 main.py & python3 server.py"]

main.py

@@ -0,0 +1,150 @@
+import json
+import time
+import re
+
+import requests
+
+from utils import load_config, write_tokens
+
+
+def get_chat_url(
+    chat_id): return f'https://kimi.moonshot.cn/api/chat/{chat_id}/completion/stream'
+
+
+REFRESH_INTERVAL = 5 * 60
+REFRESH_URL = 'https://kimi.moonshot.cn/api/auth/token/refresh'
+LIST_URL = 'https://kimi.moonshot.cn/api/chat/list'
+NEW_CHAT_URL = 'https://kimi.moonshot.cn/api/chat'
+
+HEADERS = {
+    'user-agent':
+    'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/76.0.3809.132 Safari/537.36',
+    'Content-Type': 'application/json',
+    'Referer': 'https://kimi.moonshot.cn',
+    'Origin': 'https://kimi.moonshot.cn',
+}
+
+
+# list 和 create 头部都是 auth_token 来鉴权
+def list_conversations():
+    response = requests.post(LIST_URL, headers=HEADERS, json={}).json()
+    return response['items']
+
+
+def create_conversation(name):
+    response = requests.post(NEW_CHAT_URL,
+                             headers=HEADERS,
+                             json={'name': name}).json()
+    return response['id']
+
+
+def format_url(text):
+    url_pattern = r'http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\\(\\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+'
+    urls = re.findall(url_pattern, text)
+    for url in urls:
+        new_url = f'<url id="" type="url" status="" title="" wc="">{url}</url>'
+        text = text.replace(url, new_url)
+    return text
+
+
+def process_msg(messages):
+    concat_content = ''
+    for msg in messages:
+        role, content = msg['role'], msg['content']
+        single_msg = f'{role}: {content}\n\n'
+        concat_content += single_msg
+    concat_content = format_url(concat_content)
+    return [{'content': concat_content, 'role': 'user'}]
+
+
+async def get_reply(messages):
+    HEADERS['Authorization'] = load_config()['auth_token']
+    chat_id = create_conversation('新的聊天')
+    chat_url = get_chat_url(chat_id)
+    # 添加文件，需要在 Kimi 官方上传好
+    # messages.append({'content': 'https://prod-chat-kimi.tos-s3-cn-beijing.volces.com/prod-chat-kimi/ckiuir33aesg978thq90/2024-03-03/cnhtnlsudu6ec6vquo9g/3e4545519d5a8a56256980d9fda4f2de_720w_thumbnail.jpg?X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Credential=AKLTYTJlNjgwMjY2ZDBkNDFiYmI5YWNiZDBlZmFmYjIzZTA%2F20240303%2Fcn-beijing%2Fs3%2Faws4_request&X-Amz-Date=20240303T020928Z&X-Amz-Expires=518400&X-Amz-SignedHeaders=host&X-Amz-Signature=f9c5dc63e55256871a54f453bc23efde8a9a78078cd7e778cf4dad86987e0a4b', 'type': 'file'})
+    concat_msg = process_msg(messages)
+    response = requests.post(chat_url,
+                             headers=HEADERS,
+                             json={'messages': concat_msg})
+    status_code = response.status_code
+    if status_code != 200:
+        print(status_code)
+        print(response.content.decode('utf-8'))
+    else:
+        search_content = '\n\n参考：\n'
+        url = None
+        link_count = 1
+        for r in response.iter_lines():
+            if r:
+                decoded = r.decode('utf-8')
+                if decoded.startswith('data:'):
+                    json_data = json.loads(decoded[len('data: '):])
+                    if json_data['event'] == 'cmpl':
+                        answer = json_data['text']
+                        yield json.dumps(
+                            {
+                                'object': 'chat.completion.chunk',
+                                'model': 'moonshot-v1',
+                                'choices': [
+                                    {
+                                        'delta':  {'content': answer},
+                                        'index': 0,
+                                        'finish_reason': None,
+                                    }
+                                ],
+                            }
+                        )
+
+                    elif json_data['event'] == 'search_plus':
+                        msg = json_data['msg']
+                        if 'url' in msg:
+                            title, url = msg['title'], msg['url']
+                            search_content += f'{link_count}. [{title}]({url})\n'
+                            link_count += 1
+    if url is not None:
+        yield json.dumps(
+            {
+                'object': 'chat.completion.chunk',
+                'model': 'moonshot-v1',
+                'choices': [
+                    {
+                        'delta':  {'content': search_content},
+                        'index': 0,
+                        'finish_reason': 'search',
+                    }
+                ],
+            }
+        )
+
+    yield json.dumps(
+        {
+            'object': 'chat.completion.chunk',
+            'model': 'moonshot-v1',
+            'choices': [
+                {
+                    'delta':  {},
+                    'index': 0,
+                    'finish_reason': 'stop',
+                }
+            ],
+        }
+    )
+
+
+def refresh():
+    global refresh_token
+    refresh_token = load_config()['refresh_token']
+    HEADERS['Authorization'] = refresh_token
+    response = requests.get(REFRESH_URL, headers=HEADERS).json()
+    auth_token, refresh_token = list(response.values())
+    refresh_token = f'Bearer {refresh_token}'
+    auth_token = f'Bearer {auth_token}'
+    write_tokens(auth_token, refresh_token)
+
+
+if __name__ == '__main__':
+    while True:
+        refresh()
+        print('Refresh......')
+        time.sleep(REFRESH_INTERVAL)

requirements.txt

@@ -0,0 +1,6 @@
+fastapi==0.110.1
+pydantic==2.7.0
+Requests==2.31.0
+retry==0.9.2
+sse_starlette==2.1.0
+uvicorn==0.29.0

server.py

@@ -0,0 +1,41 @@
+from fastapi import Depends, FastAPI, HTTPException, status
+from fastapi.security import OAuth2PasswordBearer
+from pydantic import BaseModel
+from sse_starlette.sse import EventSourceResponse
+
+from main import get_reply
+from utils import load_config
+
+app = FastAPI()
+TOKEN = load_config()['token']
+oauth2_scheme = OAuth2PasswordBearer(tokenUrl='token')
+
+
+class Request(BaseModel):
+    messages: list
+    model: str = None
+    stream: bool = None
+    temperature: float = None
+    presence_penalty: float = None
+    frequency_penalty: float = None
+    top_p: int = None
+    max_tokens: int = None
+
+
+@app.post('/v1/chat/completions')
+async def answer(request: Request, token: str = Depends(oauth2_scheme)):
+    if token != TOKEN:
+        print(f'请求的 token: {token}')
+        print(f'设置的 token: {TOKEN}')
+        raise HTTPException(
+            status_code=status.HTTP_401_UNAUTHORIZED,
+            detail='Invalid authentication credentials',
+            headers={'WWW-Authenticate': 'Bearer'},
+        )
+    # .model_dump() 相当于 .dict()
+    messages = request.model_dump()['messages']
+    return EventSourceResponse(get_reply(messages))
+
+if __name__ == '__main__':
+    import uvicorn
+    uvicorn.run(app, host='0.0.0.0', port=6867)

utils.py

@@ -0,0 +1,13 @@
+import json
+
+
+def load_config():
+    return json.load(open('config.json', encoding='utf-8'))
+
+
+def write_tokens(auth_token, refresh_token):
+    cfg = load_config()
+    cfg['auth_token'] = auth_token
+    cfg['refresh_token'] = refresh_token
+    with open('config.json', 'w', encoding='utf-8') as w:
+        w.write(json.dumps(cfg))