gabykim commited on
Commit
3a9d0c3
·
1 Parent(s): ea31570

add evalution round config setting

Browse files
Files changed (1) hide show
  1. src/know_lang_bot/config.py +7 -1
src/know_lang_bot/config.py CHANGED
@@ -196,6 +196,12 @@ class ChatConfig(BaseSettings):
196
  description="Maximum number of characters per chunk"
197
  )
198
 
 
 
 
 
 
 
199
  class AppConfig(BaseSettings):
200
  model_config = SettingsConfigDict(
201
  env_file='.env',
@@ -204,7 +210,7 @@ class AppConfig(BaseSettings):
204
  )
205
 
206
  llm: LLMConfig = Field(default_factory=LLMConfig)
207
- evaluator: LLMConfig = Field(default_factory=LLMConfig)
208
  reranker: RerankerConfig = Field(default_factory=RerankerConfig)
209
  db: DBConfig = Field(default_factory=DBConfig)
210
  parser: ParserConfig = Field(default_factory=ParserConfig)
 
196
  description="Maximum number of characters per chunk"
197
  )
198
 
199
+ class EvaluatorConfig(LLMConfig):
200
+ evaluation_rounds: int = Field(
201
+ default=1,
202
+ description="Number of evaluation rounds per test case"
203
+ )
204
+
205
  class AppConfig(BaseSettings):
206
  model_config = SettingsConfigDict(
207
  env_file='.env',
 
210
  )
211
 
212
  llm: LLMConfig = Field(default_factory=LLMConfig)
213
+ evaluator: EvaluatorConfig = Field(default_factory=EvaluatorConfig)
214
  reranker: RerankerConfig = Field(default_factory=RerankerConfig)
215
  db: DBConfig = Field(default_factory=DBConfig)
216
  parser: ParserConfig = Field(default_factory=ParserConfig)