import { z } from "zod";
import { config } from "./config.js";
// Helper function to get the graph ID
const getGraphId = (providedGraphId?: string): string | null => {
if (config.CODEGPT_REPO_URL || config.IS_MULTI_REPO) {
return null
}
if (config.CODEGPT_GRAPH_ID) {
return config.CODEGPT_GRAPH_ID;
}
if (!providedGraphId) {
throw new Error("Graph ID is required. Either set config.CODEGPT_GRAPH_ID environment variable or provide graphId parameter.");
}
return providedGraphId;
};
const createToolSchema = <T extends Record<string, any>>(baseSchema: T) => {
const hasRepoAccess = config.CODEGPT_GRAPH_ID || config.CODEGPT_REPO_URL || config.IS_MULTI_REPO;
let schema = hasRepoAccess ? baseSchema : {
...baseSchema,
graphId: z
.string()
.min(1, "Graph ID is required")
.describe("The ID of the graph to query")
};
if (config.IS_MULTI_REPO) {
const repoList = config.REPO_LIST.join(", ");
schema = {
...schema,
repository: z
.string()
.min(1, "Repository name is required")
.describe(
`The repository to query from the available repositories. Format is: org/repo. Available options are: ${repoList}`
),
};
}
return schema;
};
function extractRepoInfo(url: string): { repoName: string; repoOrg: string } {
const cleanUrl = url.endsWith('/') ? url.slice(0, -1) : url
const repoRegex = /([^\/]+)\/([^\/]+)$/
const match = cleanUrl.match(repoRegex)
if (!match) {
throw new Error('Invalid format. Expected format: name/repo')
}
const [, repoOrg, repoName] = match
if (!repoName || !repoOrg) {
throw new Error('Could not extract name and repo from the URL')
}
return { repoOrg, repoName }
}
export {
getGraphId,
createToolSchema,
extractRepoInfo
}
