Spaces:
Sleeping
Sleeping
add evalution round config setting
Browse files
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:
|
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)
|