Гефестыч: наш опыт автоматизации Code Review через LLM. «Грабли», решения, код

from pydantic_ai import Agent
from pydantic_ai.models.openai import OpenAIChatModel
from pydantic_ai.providers.ollama import OllamaProvider
ollama_provider = OllamaProvider(
    base_url="http://localhost:11434/",
    api_key="your_amazing_api_key",
)
model = OpenAIChatModel(
    model_name="qwen3-coder:latest",
    provider=ollama_provider,
)
agent = Agent(model=model)
result = agent.run_sync("Hello world!")
print(result.output)

# docs: Содержание,тайтл и коментарии к задаче
def _get_issue_context(issue_code:str) -> str:
    pass

# docs: Документация для сервиса/репозитория
def _get_service_docs(docs_ids: list[str]) -> list[str]:
    pass

• Github: https://patch-diff.githubusercontent.com/raw/{proje...}/pull/123.patch
  
• Bitbucket: https://your_host/api/latest/projects/{projectKey}/...RequestId}.patch

# docs: Запрос diff
async def get_diff(project: str, repository_slug: str, pr_id: str) -> str:
      pass

from pathlib import Path
# docs: Запрос файлов для внедрения в контекст
async def get_code_context(project: str, repository_slug: str, pr_id: str) -> list[Path]:
    pass

• Класть всю кодовую базу, формировать запрос на то, чтобы вытащить из каждого файла вектора и положить их в базу данных – в дальнейшем это поможет ориентироваться по их близости с вектором запроса.

• Написать парсер импортов. Если у вас небольшой проект, язык программирования один или два, то какие проблемы. Самостоятельно прошлись по древу, выбрали важное и подложили контекст.

• Сформировать source tree и попросить LLM выбрать необходимые для ревью файлы самостоятельно.

...
agent = Agent(model=model, deps_type=PRDeps)
# docs: Содержание, тайтл и коментарии к задаче
@agent.tool
def get_issue_context(ctx: RunContext[PRDeps]) -> str:
    issue_code = ctx.deps.issue_code
    issue_context = _get_issue_context(issue_code)
    return issue_context
# docs: Документация для сервиса/репозитория
@agent.tool
def get_service_docs(ctx: RunContext[PRDeps]) -> str:
    docs_ids = ctx.deps.docs_ids
    documentation = _get_service_docs(docs_ids)
    return documentation
# docs: Запрос diff
@agent.tool
def get_diff(ctx: RunContext[PRDeps]) -> str:
    diff = _get_diff(
        project=ctx.deps.project,
        repository_slug=ctx.deps.repository_slug,
        pr_id=ctx.deps.pr_id,
    )
    return diff
...

• Embedding: https://ai.pydantic.dev/embeddings/#quick-start
  
• RAG: https://ai.pydantic.dev/examples/rag/