update how to get workai model

Signed-off-by: oilbeater <[email protected]>
oilbeater · Oct 11, 2024 · 8bb13a4 · 8bb13a4
1 parent d2ee7db
commit 8bb13a4
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 18 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/src/providers/workersAI.ts b/src/providers/workersAI.ts
@@ -15,10 +15,8 @@ export const workersAIProvider: AIProvider = {
     getModelName: getModelName,
     getTokenCount: getTokenCount,
     handleRequest: async (c: Context<{ Bindings: Env }>) => {
-        const provider = c.req.param('provider');
-        const repo = c.req.param('repo');
-        const model = c.req.param('model');
-        const response = await c.env.AI.run(`${provider}/${repo}/${model}`,
+        const model = getModelName(c);
+        const response = await c.env.AI.run(`${model}`,
             await c.req.json());
 
         if (response instanceof ReadableStream) {