mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-08-07 12:07:43 +00:00
XianBW
46d8247697
* initial version * test requirements * fix bugs * fix bugs * add annotation * fix ruff error * fix CI * fix CI * fix CI * fix CI * change random usage * move cache_seed_gen to core/utils.py * fix CI * change cache_seed_gen name --------- Co-authored-by: Young <afe.young@gmail.com>
54 lines
1.9 KiB
Python
54 lines
1.9 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
from pydantic_settings import BaseSettings
|
|
|
|
# TODO: use pydantic for other modules in Qlib
|
|
# from pydantic_settings import BaseSettings
|
|
|
|
|
|
class RDAgentSettings(BaseSettings):
|
|
# TODO: (xiao) I think LLMSetting may be a better name.
|
|
# TODO: (xiao) I think most of the config should be in oai.config
|
|
# Log configs
|
|
# TODO: (xiao) think it can be a separate config.
|
|
log_trace_path: str | None = None
|
|
|
|
# Behavior of returning answers to the same question when caching is enabled
|
|
use_auto_chat_cache_seed_gen: bool = False
|
|
"""
|
|
`_create_chat_completion_inner_function` provdies a feature to pass in a seed to affect the cache hash key
|
|
We want to enable a auto seed generator to get different default seed for `_create_chat_completion_inner_function`
|
|
if seed is not given.
|
|
So the cache will only not miss you ask the same question on same round.
|
|
"""
|
|
init_chat_cache_seed: int = 42
|
|
|
|
# azure document intelligence configs
|
|
azure_document_intelligence_key: str = ""
|
|
azure_document_intelligence_endpoint: str = ""
|
|
# factor extraction conf
|
|
max_input_duplicate_factor_group: int = 300
|
|
max_output_duplicate_factor_group: int = 20
|
|
max_kmeans_group_number: int = 40
|
|
|
|
# workspace conf
|
|
workspace_path: Path = Path.cwd() / "git_ignore_folder" / "RD-Agent_workspace"
|
|
|
|
# multi processing conf
|
|
multi_proc_n: int = 1
|
|
|
|
# pickle cache conf
|
|
cache_with_pickle: bool = True # whether to use pickle cache
|
|
pickle_cache_folder_path_str: str = str(
|
|
Path.cwd() / "pickle_cache/",
|
|
) # the path of the folder to store the pickle cache
|
|
use_file_lock: bool = (
|
|
True # when calling the function with same parameters, whether to use file lock to avoid
|
|
# executing the function multiple times
|
|
)
|
|
|
|
|
|
RD_AGENT_SETTINGS = RDAgentSettings()
|