feat: adapt pullfrog for gitea + ollama

This commit is contained in:
2026-05-31 01:01:00 -05:00
parent 36ac64a5b6
commit 2aca1a3aa3
183 changed files with 1419 additions and 28292 deletions
+18 -44
View File
@@ -2,26 +2,6 @@ import { type } from "arktype";
import type { ToolContext } from "./server.ts";
import { execute, tool } from "./shared.ts";
const CLOSING_ISSUES_QUERY = `
query($owner: String!, $repo: String!, $number: Int!) {
repository(owner: $owner, name: $repo) {
pullRequest(number: $number) {
closingIssuesReferences(first: 10) {
nodes { number title }
}
}
}
}
`;
type ClosingIssuesResponse = {
repository: {
pullRequest: {
closingIssuesReferences: { nodes: Array<{ number: number; title: string }> };
};
};
};
export const PullRequestInfo = type({
pull_number: type.number.describe("The pull request number to fetch"),
});
@@ -30,28 +10,18 @@ export function PullRequestInfoTool(ctx: ToolContext) {
return tool({
name: "get_pull_request",
description:
"Retrieve PR metadata (title, body, state, branches, author, labels, linked issues). " +
"Retrieve PR metadata (title, body, state, branches, author, labels). " +
"Example: `get_pull_request({ pull_number: 1234 })`. " +
"To checkout a PR branch locally, use checkout_pr instead.",
parameters: PullRequestInfo,
execute: execute(async ({ pull_number }) => {
// fetch REST and GraphQL in parallel
const [restResponse, graphqlResponse] = await Promise.all([
ctx.octokit.rest.pulls.get({
owner: ctx.repo.owner,
repo: ctx.repo.name,
pull_number,
}),
ctx.octokit.graphql<ClosingIssuesResponse>(CLOSING_ISSUES_QUERY, {
owner: ctx.repo.owner,
repo: ctx.repo.name,
number: pull_number,
}),
]);
const data = restResponse.data;
const isFork = data.head.repo?.full_name !== data.base.repo.full_name;
const closingIssues = graphqlResponse.repository.pullRequest.closingIssuesReferences.nodes;
const result = await ctx.gitea.rest.repository.repoGetPullRequest({
owner: ctx.repo.owner,
repo: ctx.repo.name,
index: pull_number,
});
const data = result.data;
const isFork = data.head?.repo?.full_name !== data.base?.repo?.full_name;
return {
number: data.number,
@@ -61,14 +31,18 @@ export function PullRequestInfoTool(ctx: ToolContext) {
state: data.state,
draft: data.draft,
merged: data.merged,
maintainerCanModify: data.maintainer_can_modify,
base: data.base.ref,
head: data.head.ref,
maintainerCanModify: data.allow_maintainer_edit,
base: data.base?.label,
head: data.head?.label,
isFork,
author: data.user?.login,
assignees: data.assignees?.map((a) => a.login),
labels: data.labels.map((l) => l.name),
closingIssues: closingIssues.map((i) => ({ number: i.number, title: i.title })),
assignees: data.assignees
?.map((a) => a.login)
.filter((n): n is string => n !== undefined),
labels: data.labels
?.map((l) => (typeof l === "string" ? l : l.name))
.filter((n): n is string => n !== undefined),
closingIssues: [],
};
}),
});