fix: cubic feedback

Author: RomneyDa

packages/llm-info/src/index.ts

@@ -10,7 +10,7 @@ import { OpenAi } from "./providers/openai.js";
 import { Vllm } from "./providers/vllm.js";
 import { Voyage } from "./providers/voyage.js";
 import { xAI } from "./providers/xAI.js";
-import { LlmInfo, ModelProvider, UseCase } from "./types.js";
+import { LlmInfoWithProvider, ModelProvider, UseCase } from "./types.js";
 
 export const allModelProviders: ModelProvider[] = [
   OpenAi,
@@ -27,28 +27,32 @@ export const allModelProviders: ModelProvider[] = [
   xAI,
 ];
 
-export const allLlms: LlmInfo[] = allModelProviders.flatMap((provider) =>
-  provider.models.map((model) => ({ ...model, provider: provider.id })),
+export const allLlms: LlmInfoWithProvider[] = allModelProviders.flatMap(
+  (provider) =>
+    provider.models.map((model) => ({ ...model, provider: provider.id })),
 );
 
 export function findLlmInfo(
   model: string,
   preferProviderId?: string,
-): LlmInfo | undefined {
+): LlmInfoWithProvider | undefined {
   if (preferProviderId) {
     const provider = allModelProviders.find((p) => p.id === preferProviderId);
     const info = provider?.models.find((llm) =>
       llm.regex ? llm.regex.test(model) : llm.model === model,
     );
     if (info) {
-      return info;
+      return {
+        ...info,
+        provider: preferProviderId,
+      };
     }
   }
   return allLlms.find((llm) =>
     llm.regex ? llm.regex.test(model) : llm.model === model,
   );
 }
 
-export function getAllRecommendedFor(useCase: UseCase): LlmInfo[] {
+export function getAllRecommendedFor(useCase: UseCase): LlmInfoWithProvider[] {
   return allLlms.filter((llm) => llm.recommendedFor?.includes(useCase));
 }