mirror of
https://github.com/qodo-ai/pr-agent.git
synced 2025-07-02 03:40:38 +08:00
14971c4f5f
* - Add support for documentation content exceeding token limits via two phase operation: 1. Ask LLM to rank headings which are most likely to contain an answer to a user question 2. Provide the corresponding files for the LLM to search for an answer. - Refactor of help_docs to make the code more readable - For the purpose of getting canonical path: git providers to use default branch and not the PR's source branch. - Refactor of token counting and making it clear on when an estimate factor will be used. * Code review changes: 1. Correctly handle exception during retry_with_fallback_models (to allow fallback model to run in case of failure) 2. Better naming for default_branch in bitbucket cloud provider
84 lines
2.7 KiB
Python
84 lines
2.7 KiB
Python
from os.path import abspath, dirname, join
|
|
from pathlib import Path
|
|
from typing import Optional
|
|
|
|
from dynaconf import Dynaconf
|
|
from starlette_context import context
|
|
|
|
PR_AGENT_TOML_KEY = 'pr-agent'
|
|
|
|
current_dir = dirname(abspath(__file__))
|
|
global_settings = Dynaconf(
|
|
envvar_prefix=False,
|
|
merge_enabled=True,
|
|
settings_files=[join(current_dir, f) for f in [
|
|
"settings/configuration.toml",
|
|
"settings/ignore.toml",
|
|
"settings/language_extensions.toml",
|
|
"settings/pr_reviewer_prompts.toml",
|
|
"settings/pr_questions_prompts.toml",
|
|
"settings/pr_line_questions_prompts.toml",
|
|
"settings/pr_description_prompts.toml",
|
|
"settings/code_suggestions/pr_code_suggestions_prompts.toml",
|
|
"settings/code_suggestions/pr_code_suggestions_prompts_not_decoupled.toml",
|
|
"settings/code_suggestions/pr_code_suggestions_reflect_prompts.toml",
|
|
"settings/pr_information_from_user_prompts.toml",
|
|
"settings/pr_update_changelog_prompts.toml",
|
|
"settings/pr_custom_labels.toml",
|
|
"settings/pr_add_docs.toml",
|
|
"settings/custom_labels.toml",
|
|
"settings/pr_help_prompts.toml",
|
|
"settings/pr_help_docs_prompts.toml",
|
|
"settings/pr_help_docs_headings_prompts.toml",
|
|
"settings/.secrets.toml",
|
|
"settings_prod/.secrets.toml",
|
|
]]
|
|
)
|
|
|
|
|
|
def get_settings(use_context=False):
|
|
"""
|
|
Retrieves the current settings.
|
|
|
|
This function attempts to fetch the settings from the starlette_context's context object. If it fails,
|
|
it defaults to the global settings defined outside of this function.
|
|
|
|
Returns:
|
|
Dynaconf: The current settings object, either from the context or the global default.
|
|
"""
|
|
try:
|
|
return context["settings"]
|
|
except Exception:
|
|
return global_settings
|
|
|
|
|
|
# Add local configuration from pyproject.toml of the project being reviewed
|
|
def _find_repository_root() -> Optional[Path]:
|
|
"""
|
|
Identify project root directory by recursively searching for the .git directory in the parent directories.
|
|
"""
|
|
cwd = Path.cwd().resolve()
|
|
no_way_up = False
|
|
while not no_way_up:
|
|
no_way_up = cwd == cwd.parent
|
|
if (cwd / ".git").is_dir():
|
|
return cwd
|
|
cwd = cwd.parent
|
|
return None
|
|
|
|
|
|
def _find_pyproject() -> Optional[Path]:
|
|
"""
|
|
Search for file pyproject.toml in the repository root.
|
|
"""
|
|
repo_root = _find_repository_root()
|
|
if repo_root:
|
|
pyproject = repo_root / "pyproject.toml"
|
|
return pyproject if pyproject.is_file() else None
|
|
return None
|
|
|
|
|
|
pyproject_path = _find_pyproject()
|
|
if pyproject_path is not None:
|
|
get_settings().load_file(pyproject_path, env=f'tool.{PR_AGENT_TOML_KEY}')
|