39 lines
1.6 KiB
Scala
39 lines
1.6 KiB
Scala
package com.twitter.representationscorer
|
|
|
|
import com.google.inject.Module
|
|
import com.twitter.inject.thrift.modules.ThriftClientIdModule
|
|
import com.twitter.representationscorer.columns.ListScoreColumn
|
|
import com.twitter.representationscorer.columns.ScoreColumn
|
|
import com.twitter.representationscorer.columns.SimClustersRecentEngagementSimilarityColumn
|
|
import com.twitter.representationscorer.columns.SimClustersRecentEngagementSimilarityUserTweetEdgeColumn
|
|
import com.twitter.representationscorer.modules.CacheModule
|
|
import com.twitter.representationscorer.modules.EmbeddingStoreModule
|
|
import com.twitter.representationscorer.modules.RMSConfigModule
|
|
import com.twitter.representationscorer.modules.TimerModule
|
|
import com.twitter.representationscorer.twistlyfeatures.UserSignalServiceRecentEngagementsClientModule
|
|
import com.twitter.strato.fed._
|
|
import com.twitter.strato.fed.server._
|
|
|
|
object RepresentationScorerFedServerMain extends RepresentationScorerFedServer
|
|
|
|
trait RepresentationScorerFedServer extends StratoFedServer {
|
|
override def dest: String = "/s/representation-scorer/representation-scorer"
|
|
override val modules: Seq[Module] =
|
|
Seq(
|
|
CacheModule,
|
|
ThriftClientIdModule,
|
|
UserSignalServiceRecentEngagementsClientModule,
|
|
TimerModule,
|
|
RMSConfigModule,
|
|
EmbeddingStoreModule
|
|
)
|
|
|
|
override def columns: Seq[Class[_ <: StratoFed.Column]] =
|
|
Seq(
|
|
classOf[ListScoreColumn],
|
|
classOf[ScoreColumn],
|
|
classOf[SimClustersRecentEngagementSimilarityUserTweetEdgeColumn],
|
|
classOf[SimClustersRecentEngagementSimilarityColumn]
|
|
)
|
|
}
|