highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-09-19 20:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

var initialization

Browse Source
This commit is contained in:
joachim-danswer
2025-01-29 14:18:31 -08:00
committed by Evan Lohn
parent 8fea571f6e
commit 3d99ad7bc4
1 changed files with 53 additions and 113 deletions
Download Patch File Download Diff File Expand all files Collapse all files

166
backend/onyx/configs/agent_configs.py
View File

@@ -1,12 +1,12 @@
import os import os
AGENT_DEFAULT_RETRIEVAL_HITS = 25 AGENT_DEFAULT_RETRIEVAL_HITS = 15
AGENT_DEFAULT_RERANKING_HITS = 15 AGENT_DEFAULT_RERANKING_HITS = 10
AGENT_DEFAULT_SUB_QUESTION_MAX_CONTEXT_HITS = 8 AGENT_DEFAULT_SUB_QUESTION_MAX_CONTEXT_HITS = 8
AGENT_DEFAULT_NUM_DOCS_FOR_INITIAL_DECOMPOSITION = 3 AGENT_DEFAULT_NUM_DOCS_FOR_INITIAL_DECOMPOSITION = 3
AGENT_DEFAULT_NUM_DOCS_FOR_REFINED_DECOMPOSITION = 5 AGENT_DEFAULT_NUM_DOCS_FOR_REFINED_DECOMPOSITION = 5
AGENT_DEFAULT_EXPLORATORY_SEARCH_RESULTS = 3 AGENT_DEFAULT_EXPLORATORY_SEARCH_RESULTS = 3
AGENT_DEFAULT_MIN_ORIG_QUESTION_DOCS = 5 AGENT_DEFAULT_MIN_ORIG_QUESTION_DOCS = 3
AGENT_DEFAULT_MAX_ANSWER_CONTEXT_DOCS = 10 AGENT_DEFAULT_MAX_ANSWER_CONTEXT_DOCS = 10
AGENT_DEFAULT_MAX_STATIC_HISTORY_CHAR_LENGTH = 10000 AGENT_DEFAULT_MAX_STATIC_HISTORY_CHAR_LENGTH = 10000
@@ -14,124 +14,64 @@ AGENT_DEFAULT_MAX_STATIC_HISTORY_CHAR_LENGTH = 10000
# Agent Configs # Agent Configs
##### #####
agent_retrieval_stats_os: bool | str | None = os.environ.get(
"AGENT_RETRIEVAL_STATS", False
)
AGENT_RETRIEVAL_STATS: bool = False AGENT_RETRIEVAL_STATS = (
if isinstance(agent_retrieval_stats_os, str) and agent_retrieval_stats_os == "True": not os.environ.get("AGENT_RETRIEVAL_STATS") == "False"
AGENT_RETRIEVAL_STATS = True ) or True # default True
elif isinstance(agent_retrieval_stats_os, bool) and agent_retrieval_stats_os:
AGENT_RETRIEVAL_STATS = True
agent_max_query_retrieval_results_os: int | str = os.environ.get(
"AGENT_MAX_QUERY_RETRIEVAL_RESULTS", AGENT_DEFAULT_RETRIEVAL_HITS
)
AGENT_MAX_QUERY_RETRIEVAL_RESULTS: int = AGENT_DEFAULT_RETRIEVAL_HITS AGENT_MAX_QUERY_RETRIEVAL_RESULTS = int(
try: os.environ.get("AGENT_MAX_QUERY_RETRIEVAL_RESULTS") or AGENT_DEFAULT_RETRIEVAL_HITS
atmqrr = int(agent_max_query_retrieval_results_os) ) # 15
AGENT_MAX_QUERY_RETRIEVAL_RESULTS = atmqrr
except ValueError:
raise ValueError(
f"MAX_AGENT_QUERY_RETRIEVAL_RESULTS must be an integer, got {AGENT_MAX_QUERY_RETRIEVAL_RESULTS}"
)
AGENT_MAX_QUERY_RETRIEVAL_RESULTS = int(
os.environ.get("AGENT_MAX_QUERY_RETRIEVAL_RESULTS") or AGENT_DEFAULT_RETRIEVAL_HITS
) # 15
# Reranking agent configs # Reranking agent configs
# Reranking stats - no influence on flow outside of stats collection # Reranking stats - no influence on flow outside of stats collection
agent_reranking_stats_os: bool | str | None = os.environ.get( AGENT_RERANKING_STATS = (
"AGENT_RERANKING_TEST", False not os.environ.get("AGENT_RERANKING_STATS") == "True"
) ) or False # default False
AGENT_RERANKING_STATS: bool = False
if isinstance(agent_reranking_stats_os, str) and agent_reranking_stats_os == "True": AGENT_MAX_QUERY_RETRIEVAL_RESULTS = int(
AGENT_RERANKING_STATS = True os.environ.get("AGENT_MAX_QUERY_RETRIEVAL_RESULTS") or AGENT_DEFAULT_RETRIEVAL_HITS
elif isinstance(agent_reranking_stats_os, bool) and agent_reranking_stats_os: ) # 15
AGENT_RERANKING_STATS = True
AGENT_RERANKING_MAX_QUERY_RETRIEVAL_RESULTS = int(
os.environ.get("AGENT_RERANKING_MAX_QUERY_RETRIEVAL_RESULTS")
or AGENT_DEFAULT_RERANKING_HITS
) # 10
AGENT_NUM_DOCS_FOR_DECOMPOSITION = int(
os.environ.get("AGENT_NUM_DOCS_FOR_DECOMPOSITION")
or AGENT_DEFAULT_NUM_DOCS_FOR_INITIAL_DECOMPOSITION
) # 3
AGENT_NUM_DOCS_FOR_REFINED_DECOMPOSITION = int(
os.environ.get("AGENT_NUM_DOCS_FOR_REFINED_DECOMPOSITION")
or AGENT_DEFAULT_NUM_DOCS_FOR_REFINED_DECOMPOSITION
) # 5
AGENT_EXPLORATORY_SEARCH_RESULTS = int(
os.environ.get("AGENT_EXPLORATORY_SEARCH_RESULTS")
or AGENT_DEFAULT_EXPLORATORY_SEARCH_RESULTS
) # 3
AGENT_MIN_ORIG_QUESTION_DOCS = int(
os.environ.get("AGENT_MIN_ORIG_QUESTION_DOCS")
or AGENT_DEFAULT_MIN_ORIG_QUESTION_DOCS
) # 3
AGENT_MAX_ANSWER_CONTEXT_DOCS = int(
os.environ.get("AGENT_MAX_ANSWER_CONTEXT_DOCS")
or AGENT_DEFAULT_SUB_QUESTION_MAX_CONTEXT_HITS
) # 8
agent_reranking_max_query_retrieval_results_os: int | str = os.environ.get( AGENT_MAX_STATIC_HISTORY_CHAR_LENGTH = int(
"AGENT_RERANKING_MAX_QUERY_RETRIEVAL_RESULTS", AGENT_DEFAULT_RERANKING_HITS os.environ.get("AGENT_MAX_STATIC_HISTORY_CHAR_LENGTH")
) or AGENT_DEFAULT_MAX_STATIC_HISTORY_CHAR_LENGTH
) # 10000
AGENT_RERANKING_MAX_QUERY_RETRIEVAL_RESULTS: int = AGENT_DEFAULT_RERANKING_HITS
GRAPH_VERSION_NAME: str = "a" GRAPH_VERSION_NAME: str = "a"
try:
atmqrr = int(agent_reranking_max_query_retrieval_results_os)
AGENT_RERANKING_MAX_QUERY_RETRIEVAL_RESULTS = atmqrr
except ValueError:
raise ValueError(
f"AGENT_RERANKING_MAX_QUERY_RETRIEVAL_RESULTS must be an integer, got {AGENT_RERANKING_MAX_QUERY_RETRIEVAL_RESULTS}"
)
AGENT_NUM_DOCS_FOR_DECOMPOSITION_OS: int | str = os.environ.get(
"AGENT_NUM_DOCS_FOR_DECOMPOSITION", AGENT_DEFAULT_NUM_DOCS_FOR_INITIAL_DECOMPOSITION
)
try:
AGENT_NUM_DOCS_FOR_DECOMPOSITION = int(AGENT_NUM_DOCS_FOR_DECOMPOSITION_OS)
except ValueError:
raise ValueError(
f"AGENT_NUM_DOCS_FOR_DECOMPOSITION must be an integer, got {AGENT_NUM_DOCS_FOR_DECOMPOSITION_OS}"
)
AGENT_NUM_DOCS_FOR_REFINED_DECOMPOSITION_OS: int | str = os.environ.get(
"AGENT_NUM_DOCS_FOR_REFINED_DECOMPOSITION",
AGENT_DEFAULT_NUM_DOCS_FOR_REFINED_DECOMPOSITION,
)
try:
AGENT_NUM_DOCS_FOR_REFINED_DECOMPOSITION = int(
AGENT_NUM_DOCS_FOR_REFINED_DECOMPOSITION_OS
)
except ValueError:
raise ValueError(
f"AGENT_NUM_DOCS_FOR_REFINED_DECOMPOSITION must be an integer, got {AGENT_NUM_DOCS_FOR_REFINED_DECOMPOSITION_OS}"
)
AGENT_EXPLORATORY_SEARCH_RESULTS_OS: int | str = os.environ.get(
"AGENT_EXPLORATORY_SEARCH_RESULTS", AGENT_DEFAULT_EXPLORATORY_SEARCH_RESULTS
)
try:
AGENT_EXPLORATORY_SEARCH_RESULTS = int(AGENT_EXPLORATORY_SEARCH_RESULTS_OS)
except ValueError:
raise ValueError(
f"AGENT_EXPLORATORY_SEARCH_RESULTS must be an integer, got {AGENT_EXPLORATORY_SEARCH_RESULTS_OS}"
)
AGENT_MIN_ORIG_QUESTION_DOCS_OS: int | str = os.environ.get(
"AGENT_MIN_ORIG_QUESTION_DOCS", AGENT_DEFAULT_MIN_ORIG_QUESTION_DOCS
)
try:
AGENT_MIN_ORIG_QUESTION_DOCS = int(AGENT_MIN_ORIG_QUESTION_DOCS_OS)
except ValueError:
raise ValueError(
f"AGENT_MIN_ORIG_QUESTION_DOCS must be an integer, got {AGENT_MIN_ORIG_QUESTION_DOCS_OS}"
)
AGENT_MAX_ANSWER_CONTEXT_DOCS_OS: int | str = os.environ.get(
"AGENT_MAX_ANSWER_CONTEXT_DOCS", AGENT_DEFAULT_SUB_QUESTION_MAX_CONTEXT_HITS
)
try:
AGENT_MAX_ANSWER_CONTEXT_DOCS = int(AGENT_MAX_ANSWER_CONTEXT_DOCS_OS)
except ValueError:
raise ValueError(
f"AGENT_MAX_ANSWER_CONTEXT_DOCS must be an integer, got {AGENT_MAX_ANSWER_CONTEXT_DOCS_OS}"
)
AGENT_MAX_STATIC_HISTORY_CHAR_LENGTH_OS: int | str = os.environ.get(
"AGENT_MAX_STATIC_HISTORY_CHAR_LENGTH_OS",
AGENT_DEFAULT_MAX_STATIC_HISTORY_CHAR_LENGTH,
)
try:
AGENT_MAX_STATIC_HISTORY_CHAR_LENGTH = int(AGENT_MAX_STATIC_HISTORY_CHAR_LENGTH_OS)
except ValueError:
raise ValueError(
f"AGENT_MAX_STATIC_HISTORY_CHAR_LENGTH must be an integer, got {AGENT_MAX_STATIC_HISTORY_CHAR_LENGTH_OS}"
)