[feat] better openrouter support

src/ts/model/openrouter.ts

@@ -13,10 +13,36 @@ export async function openRouterModels() {
             headers: headers
         })  
         const res = await (await aim).json()
-        return res.data.map((v:any) => {
-            return v.id
+        return res.data.map((model: any) => {
+            let name = model.name
+            let price = model.pricing.prompt
+            if(price > 0){
+                name += ` - $${(price*1000).toFixed(5)}/1k`
+            }
+            else{
+                name += " - Free"
+            }
+            return {
+                id: model.id,
+                name: name,
+                price: price,
+                context_length: model.context_length,
+            }
+        }).sort((a: any, b: any) => {
+            return a.price - b.price
+        }).filter((model: any) => {
+            return model.price >= 0
         })
     } catch (error) {
         return []
     }
+}
+
+export async function getFreeOpenRouterModel(){
+    const models = await openRouterModels()
+    return models.filter((model: any) => {
+        return model.name.endsWith("Free")
+    }).sort((a: any, b: any) => {
+        return b.context_length - a.context_length
+    })[0].id ?? ''
 }

src/ts/process/request.ts

@@ -20,6 +20,7 @@ import { cloneDeep } from "lodash";
 import { supportsInlayImage } from "../image";
 import { OaifixEmdash } from "../plugins/fixer";
 import { Capacitor } from "@capacitor/core";
+import { getFreeOpenRouterModel } from "../model/openrouter";
 
 
 
@@ -279,11 +280,15 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
 
             const oaiFunctionCall = oaiFunctions ? (arg.useEmotion ? {"name": "set_emotion"} : "auto") : undefined
             let requestModel = (aiModel === 'reverse_proxy' || aiModel === 'openrouter') ? db.proxyRequestModel : aiModel
-
+            let openrouterRequestModel = db.openrouterRequestModel
             if(aiModel === 'reverse_proxy' && db.proxyRequestModel === 'custom'){
                 requestModel = db.customProxyRequestModel
             }
 
+            if(aiModel === 'openrouter' && db.openrouterRequestModel === 'risu/free'){
+                openrouterRequestModel = await getFreeOpenRouterModel()
+            }
+
             if(aiModel.startsWith('mistral')){
                 requestModel = aiModel
 
@@ -384,7 +389,7 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
 
             db.cipherChat = false
             let body = ({
-                model: aiModel === 'openrouter' ? db.openrouterRequestModel :
+                model: aiModel === 'openrouter' ? openrouterRequestModel :
                     requestModel ===  'gpt35' ? 'gpt-3.5-turbo'
                     : requestModel ===  'gpt35_0613' ? 'gpt-3.5-turbo-0613'
                     : requestModel ===  'gpt35_16k' ? 'gpt-3.5-turbo-16k'

src/ts/storage/database.ts

@@ -370,6 +370,7 @@ export function setDatabase(data:Database){
         utilOverride: false
     }
     data.keiServerURL ??= ''
+    data.top_k ??= 0
 
     changeLanguage(data.language)
     DataBase.set(data)