Correct model associations and move to .env file configuration.

Author: qingyanbaby

.env.example

@@ -1,2 +1,7 @@
 CUSTOM_TOKEN=your_custom_token
-SYSTEM_TOKEN=your_poe_api_key
+SYSTEM_TOKEN=your_poe_api_key
+MODEL_MAPPING='{
+    "gpt-3.5-turbo": "GPT-3.5-Turbo",
+    "gpt-4o": "GPT-4o",
+    "gpt-4-turbo": "GPT-4-Turbo"
+}'

README.md

@@ -50,6 +50,7 @@ cd poe_2_openai
 
 3. Run the application
 ```shell
+docker compose build
 docker compose up -d
 ```
 
@@ -59,13 +60,11 @@ docker compose up -d
 ```
 
 ## Model Conversion Explanation
-```python
-model_mapping = {
-    "gpt-3.5-turbo-16k": "ChatGPT-16k",
-    "gpt-3.5-turbo": "ChatGPT",  
-    "gpt-4": "GPT-4",
-    "gpt-4-turbo": "Claude-3-Opus",
-    "gpt-4-vision-preview": "GPT-4-128k",
-    "gpt-4-turbo-preview": "Claude-3-Opus-200k"
-}
+```shell
+# edit the .env file
+MODEL_MAPPING='{
+    "gpt-3.5-turbo": "GPT-3.5-Turbo",
+    "gpt-4o": "GPT-4o",
+    "gpt-4-turbo": "GPT-4-Turbo"
+}'
 ```

README_ZH.md

@@ -54,17 +54,16 @@ cd poe_2_openai
 
 3.运行
 ```shell
+docker compose build
 docker compose up -d
 ```
 
 ## 模型转换说明
-```python
-model_mapping = {
-    "gpt-3.5-turbo-16k": "ChatGPT-16k",
-    "gpt-3.5-turbo": "ChatGPT",
-    "gpt-4": "GPT-4",
-    "gpt-4-turbo": "Claude-3-Opus",
-    "gpt-4-vision-preview": "GPT-4-128k",
-    "gpt-4-turbo-preview": "Claude-3-Opus-200k"
-}
+```shell
+# 请在 .env 文件自定义编辑
+MODEL_MAPPING='{
+    "gpt-3.5-turbo": "GPT-3.5-Turbo",
+    "gpt-4o": "GPT-4o",
+    "gpt-4-turbo": "GPT-4-Turbo"
+}'
 ```

docker-compose.yml

@@ -7,4 +7,9 @@ services:
       - "39527:39527"
     environment:
       CUSTOM_TOKEN: sk-your-own-custom-token
-      SYSTEM_TOKEN: your-poe-api-key
+      SYSTEM_TOKEN: your-poe-api-key
+      MODEL_MAPPING: '{
+        "gpt-3.5-turbo": "GPT-3.5-Turbo",
+        "gpt-4o": "GPT-4o",
+        "gpt-4-turbo": "GPT-4-Turbo"
+      }'

poe_api.py

@@ -1,4 +1,6 @@
+import json
 import logging
+import os
 
 from fastapi import WebSocket, Form
 from fastapi.responses import JSONResponse
@@ -11,16 +13,6 @@
 
 client_dict = {}
 
-model_mapping = {
-    "gpt-3.5-turbo-16k": "ChatGPT-16k",
-    "gpt-3.5-turbo": "ChatGPT",
-    "gpt-4": "GPT-4",
-    "gpt-4-turbo": "Claude-3-Opus",
-    "gpt-4-vision-preview": "GPT-4-128k",
-    "gpt-4-turbo-preview": "Claude-3-Opus-200k"
-}
-
-
 async def get_responses(api_key, prompt=[], bot="gpt-4"):
     bot_name = get_bot(bot)
     # "system", "user", "bot"
@@ -45,7 +37,7 @@ async def stream_get_responses(api_key, prompt, bot):
     bot_name = get_bot(bot)
     messages = openai_message_to_poe_message(prompt)
     async for partial in get_bot_response(messages=messages, bot_name=bot_name, api_key=api_key,
-                                          skip_system_prompt=False):
+                                          skip_system_prompt=True):
         yield partial.text
 
 
@@ -96,7 +88,8 @@ async def websocket_endpoint(websocket: WebSocket):
 
 
 def get_bot(model):
-    return model_mapping.get(model, "ChatGPT-16k")
+    model_mapping = json.loads(os.environ.get("MODEL_MAPPING", "{}"))
+    return model_mapping.get(model, "GPT-4o")
 
 
 def openai_message_to_poe_message(messages=[]):