Upload useapi.py

useapi.py

@@ -13,6 +13,7 @@ character_url = "https://" + os.getenv("character_url")
 avatar_url = "https://" + os.getenv("avatar_url")
 image_url = "https://" + os.getenv("image_url")
 auth = os.getenv("auth")
+moment_url = os.getenv("moment_url")
 #headers
 def create_headers(language):
     # 映射
@@ -60,10 +61,10 @@ def get_models():
             for model_info in models:
                 # 过滤掉 'gpt-4o' 和 'gpt-4o-mini'
                 if model_info['model'] not in ['mythomax-13b']:
-                    if model_info['model'] in ['qwen-2-7b', 'gemma-2-9b', 'llama-3.1-8b', 'internLM-2.5-7b']:
-                        weight = 13
+                    if model_info['model'] in ['gemma-2-9b', 'llama-3.1-8b']:
+                        weight = 12  # Assign a low weight to reduce their frequency
                     else:
-                        weight = int(math.ceil(24 / model_info['price'] + 0.5))
+                        weight = int(math.ceil(25 / model_info['price'] + 0.5))
                     self.models.extend([model_info['model']] * weight)
 
     model_storage = ModelStorage()
@@ -78,26 +79,37 @@ def get_models():
         return model_storage.models
 
 #解析推荐json
-def extract_recommand(data):
-    return [
-        {
+async def extract_recommand(data, language):
+    result = []
+    for item in data["data"]["moments"]:
+        opening = await get_moment_opening(item["_id"], language)
+        result.append({
             "character_id": item["character_id"],
-            "avatar_url" : str(avatar_url+item["character_id"]+"_avatar.webp"),
+            "avatar_url": str(avatar_url + item["character_id"] + "_avatar.webp"),
             "_id": item["_id"],
-            "image_url" : str(image_url+item["_id"]+"_large.webp"),
+            "image_url": str(image_url + item["_id"] + "_large.webp"),
             "description": item["description"],
             "name": item["title"],
-            "opening": item["opening"]
-        }
-        for item in data["data"]["moments"]
-    ]
+            "opening": opening
+        })
+    return result
+
+async def get_moment_opening(moment_id, language):
+    url = f"{moment_url}{moment_id}"
+    async with httpx.AsyncClient() as client:
+        response = await client.get(url, headers=create_headers(language))
+        if response.status_code == 200:
+            data = response.json()
+            return data['data']['opening']
+    return None
 
 #请求推荐API
 async def recommand(char_id, language):
     recommand_url = str(recommand_base_url + char_id)
-    response = await client.get(recommand_url, headers=create_headers(language))
-    json_data = response.json()
-    return extract_recommand(json_data)
+    async with httpx.AsyncClient() as client:
+        response = await client.get(recommand_url, headers=create_headers(language))
+        json_data = response.json()
+    return await extract_recommand(json_data, language)
 
 async def fetch_stream(query, model, moment_id, session_id, bio, request_name, queue, language):
     payload = {"query": query, "model": model, "bio": bio, "moment_id": moment_id}
@@ -170,4 +182,4 @@ async def combine_streams(
             active_streams -= 1
 
     session_id_a = await task_a
-    session_id_b = await task_b
+    session_id_b = await task_b