create-polos 0.1.0

package/dist/index.js

@@ -0,0 +1,445 @@
+#!/usr/bin/env node
+
+// src/index.ts
+import path2 from "path";
+import fs2 from "fs";
+import * as p from "@clack/prompts";
+
+// src/providers.ts
+var providers = [
+  {
+    label: "Anthropic",
+    value: "anthropic",
+    import: `import { anthropic } from '@ai-sdk/anthropic'`,
+    call: `anthropic('claude-sonnet-4-5')`,
+    package: "@ai-sdk/anthropic",
+    packageVersion: "^3.0.0",
+    envVar: "ANTHROPIC_API_KEY",
+    envPlaceholder: "sk-ant-..."
+  },
+  {
+    label: "OpenAI",
+    value: "openai",
+    import: `import { openai } from '@ai-sdk/openai'`,
+    call: `openai('gpt-4o-mini')`,
+    package: "@ai-sdk/openai",
+    packageVersion: "^3.0.0",
+    envVar: "OPENAI_API_KEY",
+    envPlaceholder: "sk-..."
+  },
+  {
+    label: "Google Gemini",
+    value: "google",
+    import: `import { google } from '@ai-sdk/google'`,
+    call: `google('gemini-2.0-flash')`,
+    package: "@ai-sdk/google",
+    packageVersion: "^3.0.0",
+    envVar: "GOOGLE_GENERATIVE_AI_API_KEY",
+    envPlaceholder: "AIza..."
+  }
+];
+
+// src/scaffold.ts
+import fs from "fs";
+import path from "path";
+import { execSync } from "child_process";
+
+// src/templates/package-json.ts
+function packageJsonTemplate(projectName, provider) {
+  return `{
+  "name": ${JSON.stringify(projectName)},
+  "version": "0.1.0",
+  "private": true,
+  "type": "module",
+  "scripts": {
+    "dev": "polos dev",
+    "start": "tsx src/main.ts"
+  },
+  "dependencies": {
+    "${provider.package}": "${provider.packageVersion}",
+    "@polos/sdk": "latest",
+    "dotenv": "^16.4.0"
+  },
+  "devDependencies": {
+    "@types/node": "^22.0.0",
+    "tsx": "^4.0.0",
+    "typescript": "^5.7.0"
+  }
+}
+`;
+}
+
+// src/templates/tsconfig-json.ts
+function tsconfigJsonTemplate() {
+  return `{
+  "compilerOptions": {
+    "target": "ES2022",
+    "module": "ESNext",
+    "moduleResolution": "bundler",
+    "esModuleInterop": true,
+    "strict": true,
+    "skipLibCheck": true,
+    "outDir": "dist",
+    "rootDir": "src"
+  },
+  "include": ["src"]
+}
+`;
+}
+
+// src/templates/env-example.ts
+function envExampleTemplate(provider) {
+  return `${provider.envVar}=${provider.envPlaceholder}
+TAVILY_API_KEY=tvly-...
+# POLOS_API_URL=http://localhost:8080
+# POLOS_API_KEY=
+`;
+}
+
+// src/templates/gitignore.ts
+function gitignoreTemplate() {
+  return `node_modules/
+dist/
+.env
+*.log
+`;
+}
+
+// src/templates/readme.ts
+function readmeTemplate(projectName, provider) {
+  return `# ${projectName}
+
+A Polos agent project using **${provider.label}**.
+
+## Getting Started
+
+1. Install dependencies:
+   \`\`\`bash
+   npm install
+   \`\`\`
+
+2. Copy \`.env.example\` to \`.env\` and add your API keys:
+   \`\`\`bash
+   cp .env.example .env
+   \`\`\`
+
+3. Start the development server:
+   \`\`\`bash
+   npx polos dev
+   \`\`\`
+
+## Project Structure
+
+\`\`\`
+src/
+  main.ts                    # Entry point \u2014 imports agents, starts Polos
+  agents/
+    coding-agent.ts          # Coding agent with local sandbox tools
+    assistant-agent.ts       # Assistant with sandbox tools + web search + ask user
+    text-review/
+      agents.ts              # Grammar, tone, correctness, and editor agents
+      workflow.ts            # Parallel review workflow
+\`\`\`
+
+## Agents
+
+- **coding_agent** \u2014 A coding agent with local sandbox tools (exec, read, write, edit, glob, grep)
+- **assistant_agent** \u2014 An assistant with sandbox tools, web search, and ask user
+- **text_review** \u2014 A workflow that runs 3 reviewers in parallel, then a final editor
+
+## Learn More
+
+- [Polos Documentation](https://docs.polos.dev)
+`;
+}
+
+// src/templates/main.ts.ts
+function mainTsTemplate() {
+  return `import 'dotenv/config';
+import { Polos } from '@polos/sdk';
+
+// Import agents and workflows for registration
+import './agents/coding-agent.js';
+import './agents/assistant-agent.js';
+import './agents/text-review/agents.js';
+import './agents/text-review/workflow.js';
+
+const polos = new Polos();
+await polos.serve();
+`;
+}
+
+// src/templates/coding-agent.ts.ts
+function codingAgentTemplate(provider) {
+  return `${provider.import};
+import { defineAgent, maxSteps, sandboxTools } from '@polos/sdk';
+import path from 'node:path';
+
+const workspaceDir = path.resolve(process.cwd(), 'workspace');
+
+const tools = sandboxTools({
+  env: 'local',
+  local: {
+    cwd: workspaceDir,
+    pathRestriction: workspaceDir,
+  },
+});
+
+export const codingAgent = defineAgent({
+  id: 'coding_agent',
+  model: ${provider.call},
+  systemPrompt: \`You are a coding agent with access to sandbox tools.
+Your workspace is at \${workspaceDir}.
+Use your tools to read, write, and execute code.\`,
+  tools,
+  stopConditions: [maxSteps({ count: 30 })],
+});
+`;
+}
+
+// src/templates/assistant-agent.ts.ts
+function assistantAgentTemplate(provider) {
+  return `${provider.import};
+import {
+  defineAgent,
+  maxSteps,
+  sandboxTools,
+  createWebSearchTool,
+  createAskUserTool,
+} from '@polos/sdk';
+import path from 'node:path';
+
+const workspaceDir = path.resolve(process.cwd(), 'workspace');
+
+const sandbox = sandboxTools({
+  env: 'local',
+  local: {
+    cwd: workspaceDir,
+    pathRestriction: workspaceDir,
+  },
+});
+
+const webSearch = createWebSearchTool({
+  maxResults: 5,
+  searchDepth: 'basic',
+  includeAnswer: true,
+  approval: 'always',
+});
+
+const askUser = createAskUserTool();
+
+export const assistantAgent = defineAgent({
+  id: 'assistant_agent',
+  model: ${provider.call},
+  systemPrompt: \`You are a helpful assistant with access to sandbox tools, web search, and the ability to ask the user for clarification.
+Your workspace is at \${workspaceDir}.
+Use your tools to help the user with their tasks.\`,
+  tools: [...sandbox, webSearch, askUser],
+  stopConditions: [maxSteps({ count: 30 })],
+});
+`;
+}
+
+// src/templates/text-review-agents.ts.ts
+function textReviewAgentsTemplate(provider) {
+  return `${provider.import};
+import { defineAgent, maxSteps } from '@polos/sdk';
+
+export const grammarReviewAgent = defineAgent({
+  id: 'grammar_reviewer',
+  model: ${provider.call},
+  systemPrompt: \`You are a grammar reviewer. Analyze the provided text for grammatical errors,
+punctuation issues, and sentence structure problems. Provide a concise review with specific
+suggestions for improvement. Return your review as a single string.\`,
+  stopConditions: [maxSteps({ count: 10 })],
+});
+
+export const toneConsistencyAgent = defineAgent({
+  id: 'tone_reviewer',
+  model: ${provider.call},
+  systemPrompt: \`You are a tone and consistency reviewer. Analyze the provided text for tone shifts,
+inconsistencies in voice, and style issues. Provide a concise review with specific suggestions
+for improvement. Return your review as a single string.\`,
+  stopConditions: [maxSteps({ count: 10 })],
+});
+
+export const correctnessAgent = defineAgent({
+  id: 'correctness_reviewer',
+  model: ${provider.call},
+  systemPrompt: \`You are a factual correctness reviewer. Analyze the provided text for factual accuracy,
+logical consistency, and unsupported claims. Provide a concise review with specific concerns.
+Return your review as a single string.\`,
+  stopConditions: [maxSteps({ count: 10 })],
+});
+
+export const finalEditorAgent = defineAgent({
+  id: 'final_editor',
+  model: ${provider.call},
+  systemPrompt: \`You are a final editor. You will receive the original text along with reviews from
+grammar, tone, and correctness reviewers. Synthesize all feedback and produce an improved
+version of the text. Return only the improved text.\`,
+  stopConditions: [maxSteps({ count: 10 })],
+});
+`;
+}
+
+// src/templates/text-review-workflow.ts.ts
+function textReviewWorkflowTemplate() {
+  return `import { defineWorkflow } from '@polos/sdk';
+import {
+  grammarReviewAgent,
+  toneConsistencyAgent,
+  correctnessAgent,
+  finalEditorAgent,
+} from './agents.js';
+
+interface TextReviewPayload {
+  text: string;
+}
+
+interface TextReviewResult {
+  originalText: string;
+  grammarReview: string;
+  toneReview: string;
+  correctnessReview: string;
+  finalText: string;
+}
+
+export const textReview = defineWorkflow<TextReviewPayload, unknown, TextReviewResult>(
+  { id: 'text_review' },
+  async (ctx, payload) => {
+    const { text } = payload;
+
+    // Run 3 reviewers in parallel
+    const reviewResults = await ctx.step.batchAgentInvokeAndWait<Record<string, unknown>>(
+      'parallel_reviews',
+      [
+        grammarReviewAgent.withInput(\`Review this text for grammar:\\n\\n\${text}\`),
+        toneConsistencyAgent.withInput(\`Review this text for tone:\\n\\n\${text}\`),
+        correctnessAgent.withInput(\`Review this text for correctness:\\n\\n\${text}\`),
+      ],
+    );
+
+    const grammarReview = (reviewResults[0]?.['result'] as string) ?? '';
+    const toneReview = (reviewResults[1]?.['result'] as string) ?? '';
+    const correctnessReview = (reviewResults[2]?.['result'] as string) ?? '';
+
+    // Send all reviews to the final editor
+    const editorPrompt = \`Original text:
+\${text}
+
+Grammar review:
+\${grammarReview}
+
+Tone review:
+\${toneReview}
+
+Correctness review:
+\${correctnessReview}
+
+Please produce an improved version of the original text incorporating the feedback above.\`;
+
+    const editorResult = (await ctx.step.agentInvokeAndWait(
+      'final_editor',
+      finalEditorAgent.withInput(editorPrompt),
+    )) as Record<string, unknown>;
+
+    return {
+      originalText: text,
+      grammarReview,
+      toneReview,
+      correctnessReview,
+      finalText: (editorResult['result'] as string) ?? '',
+    };
+  },
+);
+`;
+}
+
+// src/scaffold.ts
+function generateFiles(projectName, provider) {
+  return [
+    { path: "package.json", content: packageJsonTemplate(projectName, provider) },
+    { path: "tsconfig.json", content: tsconfigJsonTemplate() },
+    { path: ".env.example", content: envExampleTemplate(provider) },
+    { path: ".gitignore", content: gitignoreTemplate() },
+    { path: "README.md", content: readmeTemplate(projectName, provider) },
+    { path: "src/main.ts", content: mainTsTemplate() },
+    { path: "src/agents/coding-agent.ts", content: codingAgentTemplate(provider) },
+    { path: "src/agents/assistant-agent.ts", content: assistantAgentTemplate(provider) },
+    { path: "src/agents/text-review/agents.ts", content: textReviewAgentsTemplate(provider) },
+    { path: "src/agents/text-review/workflow.ts", content: textReviewWorkflowTemplate() }
+  ];
+}
+function scaffoldProject(projectDir, files) {
+  for (const file of files) {
+    const filePath = path.join(projectDir, file.path);
+    fs.mkdirSync(path.dirname(filePath), { recursive: true });
+    fs.writeFileSync(filePath, file.content);
+  }
+}
+function installDependencies(projectDir) {
+  try {
+    execSync("npm install", { cwd: projectDir, stdio: "pipe" });
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+// src/index.ts
+async function main() {
+  p.intro("Create a new Polos project");
+  const projectName = await p.text({
+    message: "What is your project name?",
+    placeholder: "my-polos-project",
+    defaultValue: "my-polos-project",
+    validate(value) {
+      if (!value) return "Project name is required";
+      if (!/^[a-z0-9@][a-z0-9._\-/]*$/.test(value)) {
+        return "Invalid project name \u2014 use lowercase letters, numbers, hyphens, and dots";
+      }
+    }
+  });
+  if (p.isCancel(projectName)) {
+    p.cancel("Operation cancelled.");
+    process.exit(0);
+  }
+  const providerValue = await p.select({
+    message: "Which LLM provider do you want to use?",
+    options: providers.map((prov) => ({
+      label: prov.label,
+      value: prov.value
+    })),
+    initialValue: "anthropic"
+  });
+  if (p.isCancel(providerValue)) {
+    p.cancel("Operation cancelled.");
+    process.exit(0);
+  }
+  const provider = providers.find((prov) => prov.value === providerValue);
+  const projectDir = path2.resolve(process.cwd(), projectName);
+  if (fs2.existsSync(projectDir)) {
+    p.cancel(`Directory "${projectName}" already exists.`);
+    process.exit(1);
+  }
+  const s = p.spinner();
+  s.start("Creating project files...");
+  const files = generateFiles(projectName, provider);
+  scaffoldProject(projectDir, files);
+  s.stop("Project files created.");
+  s.start("Installing dependencies...");
+  const installed = installDependencies(projectDir);
+  if (installed) {
+    s.stop("Dependencies installed.");
+  } else {
+    s.stop("Could not install dependencies. Run `npm install` manually.");
+  }
+  p.outro(`Your project is ready!
+
+  Next steps:
+    cd ${projectName}
+    cp .env.example .env     # add your ${provider.envVar}
+    npx polos dev`);
+}
+main().catch(console.error);

package/package.json

@@ -0,0 +1,31 @@
+{
+  "name": "create-polos",
+  "version": "0.1.0",
+  "description": "Scaffold a new Polos agent project",
+  "type": "module",
+  "bin": {
+    "create-polos": "./dist/index.js"
+  },
+  "scripts": {
+    "build": "tsup",
+    "dev": "tsup --watch"
+  },
+  "files": [
+    "dist"
+  ],
+  "keywords": [
+    "polos",
+    "agents",
+    "scaffolding",
+    "cli"
+  ],
+  "license": "Apache-2.0",
+  "dependencies": {
+    "@clack/prompts": "^0.9.1"
+  },
+  "devDependencies": {
+    "@types/node": "^22.0.0",
+    "tsup": "^8.0.0",
+    "typescript": "^5.7.0"
+  }
+}