@astrive-ai/agent 1.0.0

package/src/reflection.ts

@@ -0,0 +1,70 @@
+import { IProvider, Message } from "@astrive-ai/types";
+
+export interface ReflectionContext {
+  provider: IProvider;
+  originalInput: string;
+}
+
+export interface ReviewResult {
+  hasIssues: boolean;
+  issues: string[];
+  suggestions: string[];
+}
+
+export class ReflectionEngine {
+  public async reflect(response: string, context: ReflectionContext): Promise<string> {
+    const review = await this.reviewResponse(response, context.provider);
+    
+    if (review.hasIssues) {
+      const prompt = `The previous response to "${context.originalInput}" had issues. Please improve it based on this feedback:
+Issues: ${review.issues.join(", ")}
+Suggestions: ${review.suggestions.join(", ")}
+Original response: ${response}
+
+Please provide the corrected response.`;
+
+      const improvedResponse = await context.provider.chat({
+        messages: [
+          { role: "system", content: "You are an expert editor who improves responses based on feedback. Provide ONLY the improved response." },
+          { role: "user", content: prompt }
+        ],
+        model: "llama-3.3-70b-versatile"
+      });
+      
+      return improvedResponse.content || response;
+    }
+    
+    return response;
+  }
+
+  private async reviewResponse(response: string, provider: IProvider): Promise<ReviewResult> {
+    const prompt = `Review the following response for accuracy, clarity, and completeness.
+Response: "${response}"
+
+Provide your review in the following JSON format ONLY:
+{
+  "hasIssues": boolean,
+  "issues": ["list of specific issues"],
+  "suggestions": ["list of actionable suggestions to fix issues"]
+}`;
+
+    try {
+      const reviewResponse = await provider.chat({
+        messages: [
+          { role: "system", content: "You are a critical reviewer. Always respond with valid JSON matching the requested schema." },
+          { role: "user", content: prompt }
+        ],
+        model: "llama-3.3-70b-versatile"
+      });
+
+      const content = (reviewResponse.content || "").trim();
+      const jsonStart = content.indexOf('{');
+      const jsonEnd = content.lastIndexOf('}') + 1;
+      const jsonString = content.slice(jsonStart, jsonEnd);
+      
+      return JSON.parse(jsonString) as ReviewResult;
+    } catch (e) {
+      return { hasIssues: false, issues: [], suggestions: [] };
+    }
+  }
+}

package/tsconfig.json

@@ -0,0 +1,8 @@
+{
+  "extends": "../../tsconfig.json",
+  "compilerOptions": {
+    "outDir": "./dist",
+    "rootDir": "./src"
+  },
+  "include": ["src/**/*"]
+}