fix mlx omni provider

This commit is contained in:
geoffsee
2025-06-18 14:33:07 -04:00
parent f1d7f52dbd
commit b7f02eb4fb
9 changed files with 78 additions and 43 deletions

View File

@@ -1,5 +1,5 @@
import { types } from "mobx-state-tree";
import { renderPage } from "vike/server";
export default types
.model("StaticAssetStore", {})
@@ -17,7 +17,7 @@ export default types
async handleSsr(
url: string,
headers: Headers,
env: Vike.PageContext["env"],
env: Vike.PageContext.env,
) {
console.log("handleSsr");
const pageContextInit = {

View File

@@ -13,7 +13,7 @@ import {XaiChatSdk} from "../providers/xai";
import {CerebrasSdk} from "../providers/cerebras";
import {CloudflareAISdk} from "../providers/cloudflareAi";
import {OllamaChatSdk} from "../providers/ollama";
import {MlxOmniChatSdk} from "../providers/mlx-omni";
import {MlxOmniChatProvider, MlxOmniChatSdk} from "../providers/mlx-omni";
import {ProviderRepository} from "../providers/_ProviderRepository";
export interface StreamParams {
@@ -126,7 +126,7 @@ const ChatService = types
// ----- Helpers ----------------------------------------------------------
const logger = console;
const useCache = false;
const useCache = true;
if(useCache) {
// ----- 1. Try cached value ---------------------------------------------
@@ -139,9 +139,10 @@ const ChatService = types
return new Response(JSON.stringify(parsed), { status: 200 });
}
logger.warn('Cache entry malformed refreshing');
throw new Error('Malformed cache entry');
}
} catch (err) {
logger.error('Error reading/parsing supportedModels cache', err);
logger.warn('Error reading/parsing supportedModels cache', err);
}
}
@@ -260,11 +261,8 @@ const ChatService = types
}) {
const {streamConfig, streamParams, controller, encoder, streamId} = params;
const useModelFamily = () => {
return ProviderRepository.getModelFamily(streamConfig.model, self.env)
}
const modelFamily = await useModelFamily();
const modelFamily = await ProviderRepository.getModelFamily(streamConfig.model, self.env);
const useModelHandler = () => {
return modelHandlers[modelFamily]