106 lines
3.9 KiB
TypeScript
106 lines
3.9 KiB
TypeScript
import type { AiDoc } from '../classes.aidoc.js';
|
|
import * as plugins from '../plugins.js';
|
|
import { ProjectContext } from './projectcontext.js';
|
|
import { logger } from '../logging.js';
|
|
|
|
interface IDescriptionInterface {
|
|
description: string;
|
|
keywords: string[];
|
|
}
|
|
|
|
export class Description {
|
|
// INSTANCE
|
|
private aiDocsRef: AiDoc;
|
|
private projectDir: string;
|
|
|
|
constructor(aiDocsRef: AiDoc, projectDirArg: string) {
|
|
this.aiDocsRef = aiDocsRef;
|
|
this.projectDir = projectDirArg;
|
|
}
|
|
|
|
public async build() {
|
|
// Use runAgent with filesystem tool for agent-driven exploration
|
|
const fsTools = plugins.smartagentTools.filesystemTool({ rootDir: this.projectDir });
|
|
|
|
const descriptionSystemPrompt = `
|
|
You create project descriptions and keywords for npm packages.
|
|
|
|
You have access to filesystem tools to explore the project.
|
|
|
|
IMPORTANT RULES:
|
|
- Only READ files (package.json, npmextra.json, source files in ts/)
|
|
- Do NOT write, delete, or modify any files
|
|
- Your final response must be valid JSON only
|
|
- Description must be a clear, concise one-sentence summary
|
|
- Keywords must be relevant to the project's use cases
|
|
- Both description and keywords fields must be present
|
|
- Do NOT wrap JSON in markdown code blocks
|
|
`;
|
|
|
|
const descriptionTaskPrompt = `
|
|
PROJECT DIRECTORY: ${this.projectDir}
|
|
|
|
Use the filesystem tools to explore the project and understand what it does:
|
|
1. First, use list_directory to see the project structure
|
|
2. Read package.json to understand the package name and current description
|
|
3. Read npmextra.json if it exists for additional metadata
|
|
4. Read key source files in ts/ directory to understand the implementation
|
|
|
|
Then generate a description and keywords based on your exploration.
|
|
|
|
Your FINAL response must be valid JSON adhering to this interface:
|
|
{
|
|
description: string; // a sensible short, one sentence description of the project
|
|
keywords: string[]; // an array of tags that describe the project based on use cases
|
|
}
|
|
|
|
Important: Answer only in valid JSON.
|
|
Your answer should be parseable with JSON.parse() without modifying anything.
|
|
Don't wrap the JSON in \`\`\`json\`\`\` - just return the raw JSON object.
|
|
`;
|
|
|
|
logger.log('info', 'Starting description generation with agent...');
|
|
|
|
const descriptionResult = await plugins.smartagent.runAgent({
|
|
model: this.aiDocsRef.model,
|
|
prompt: descriptionTaskPrompt,
|
|
system: descriptionSystemPrompt,
|
|
tools: fsTools,
|
|
maxSteps: 15,
|
|
onToolCall: (toolName) => logger.log('info', `[Description] Tool call: ${toolName}`),
|
|
});
|
|
|
|
console.log(descriptionResult.text);
|
|
const resultObject: IDescriptionInterface = JSON.parse(
|
|
descriptionResult.text.replace('```json', '').replace('```', ''),
|
|
);
|
|
|
|
// Use ProjectContext to get file handles for writing
|
|
const projectContext = new ProjectContext(this.projectDir);
|
|
const files = await projectContext.gatherFiles();
|
|
|
|
// Update npmextra.json
|
|
const npmextraJson = files.smartfilesNpmextraJSON;
|
|
const npmextraJsonContent = JSON.parse(npmextraJson.contents.toString());
|
|
|
|
npmextraJsonContent['@git.zone/cli'].module.description = resultObject.description;
|
|
npmextraJsonContent['@git.zone/cli'].module.keywords = resultObject.keywords;
|
|
|
|
npmextraJson.contents = Buffer.from(JSON.stringify(npmextraJsonContent, null, 2));
|
|
await npmextraJson.write();
|
|
|
|
// Update package.json
|
|
const packageJson = files.smartfilePackageJSON;
|
|
const packageJsonContent = JSON.parse(packageJson.contents.toString());
|
|
packageJsonContent.description = resultObject.description;
|
|
packageJsonContent.keywords = resultObject.keywords;
|
|
packageJson.contents = Buffer.from(JSON.stringify(packageJsonContent, null, 2));
|
|
await packageJson.write();
|
|
|
|
console.log(`\n======================\n`);
|
|
console.log(JSON.stringify(resultObject, null, 2));
|
|
console.log(`\n======================\n`);
|
|
return descriptionResult.text;
|
|
}
|
|
}
|