feat: adapt pullfrog for gitea + ollama
This commit is contained in:
+18
-44
@@ -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: [],
|
||||
};
|
||||
}),
|
||||
});
|
||||
|
||||
Reference in New Issue
Block a user