18
18
TokenizationRequestResult ,
19
19
)
20
20
from helm .proxy .clients .client import Client
21
- from helm .proxy .clients .critique_client import CritiqueClient
21
+ from helm .proxy .critique .critique_client import CritiqueClient
22
22
from helm .proxy .clients .huggingface_model_registry import get_huggingface_model_config
23
23
from helm .proxy .clients .toxicity_classifier_client import ToxicityClassifierClient
24
24
from helm .proxy .retry import NonRetriableException , retry_request
@@ -363,17 +363,17 @@ def get_critique_client(self) -> CritiqueClient:
363
363
return self ._critique_client
364
364
critique_type = self .credentials .get ("critiqueType" )
365
365
if critique_type == "random" :
366
- from helm .proxy .clients .critique_client import RandomCritiqueClient
366
+ from helm .proxy .critique .critique_client import RandomCritiqueClient
367
367
368
368
self ._critique_client = RandomCritiqueClient ()
369
369
elif critique_type == "mturk" :
370
- from helm .proxy .clients .mechanical_turk_critique_client import (
370
+ from helm .proxy .critique .mechanical_turk_critique_client import (
371
371
MechanicalTurkCritiqueClient ,
372
372
)
373
373
374
374
self ._critique_client = MechanicalTurkCritiqueClient ()
375
375
elif critique_type == "surgeai" :
376
- from helm .proxy .clients .surge_ai_critique_client import (
376
+ from helm .proxy .critique .surge_ai_critique_client import (
377
377
SurgeAICritiqueClient ,
378
378
)
379
379
@@ -382,15 +382,15 @@ def get_critique_client(self) -> CritiqueClient:
382
382
raise ValueError ("surgeaiApiKey credentials are required for SurgeAICritiqueClient" )
383
383
self ._critique_client = SurgeAICritiqueClient (surgeai_credentials , self ._build_cache_config ("surgeai" ))
384
384
elif critique_type == "model" :
385
- from helm .proxy .clients .model_critique_client import ModelCritiqueClient
385
+ from helm .proxy .critique .model_critique_client import ModelCritiqueClient
386
386
387
387
model_name : Optional [str ] = self .credentials .get ("critiqueModelName" )
388
388
if model_name is None :
389
389
raise ValueError ("critiqueModelName is required for ModelCritiqueClient" )
390
390
client : Client = self ._get_client (model_name )
391
391
self ._critique_client = ModelCritiqueClient (client , model_name )
392
392
elif critique_type == "scale" :
393
- from helm .proxy .clients .scale_critique_client import ScaleCritiqueClient
393
+ from helm .proxy .critique .scale_critique_client import ScaleCritiqueClient
394
394
395
395
scale_credentials = self .credentials .get ("scaleApiKey" )
396
396
scale_project = self .credentials .get ("scaleProject" , None )
0 commit comments