@@ -2,17 +2,29 @@ import { describe, it, expect } from "bun:test";
22import { getPreferredNameModel } from "./workspaceTitleGenerator" ;
33import type { AIService } from "./aiService" ;
44import { getKnownModel } from "@/common/constants/knownModels" ;
5+ import type { LanguageModel } from "ai" ;
6+ import type { Result } from "@/common/types/result" ;
7+ import type { SendMessageError } from "@/common/types/errors" ;
8+
9+ type CreateModelResult = Result < LanguageModel , SendMessageError > ;
510
611// Helper to create a mock AIService that succeeds for specific models
712function createMockAIService ( availableModels : string [ ] ) : AIService {
8- return {
9- createModel : async ( modelString : string ) => {
13+ const service : Partial < AIService > = {
14+ createModel : ( modelString : string ) : Promise < CreateModelResult > => {
1015 if ( availableModels . includes ( modelString ) ) {
11- return { success : true , data : { } as never } ;
16+ // Return a minimal success result - data is not used by getPreferredNameModel
17+ const result : CreateModelResult = { success : true , data : null as never } ;
18+ return Promise . resolve ( result ) ;
1219 }
13- return { success : false , error : { type : "api_key_not_found" , provider : "test" } } ;
20+ const err : CreateModelResult = {
21+ success : false ,
22+ error : { type : "api_key_not_found" , provider : "test" } ,
23+ } ;
24+ return Promise . resolve ( err ) ;
1425 } ,
15- } as unknown as AIService ;
26+ } ;
27+ return service as AIService ;
1628}
1729
1830describe ( "workspaceTitleGenerator" , ( ) => {
0 commit comments